Refactor terminal_size_wrapper to allow mocking in CI

1 files changed, 2 insertions, 10 deletions

diff --git a/CLI/rust/src/game_of_life/opts/validators.rs b/CLI/rust/src/game_of_life/opts/validators.rs
index 5ea37e2..4abcc19 100644
--- a/CLI/rust/src/game_of_life/opts/validators.rs
+++ b/CLI/rust/src/game_of_life/opts/validators.rs
@@ -3,16 +3,8 @@
 //! Used for validating different CLI args (like width, height and delay)
 //! Ref: https://docs.rs/clap/3.0.0-beta.2/clap/struct.Arg.html#method.validator
 
-use terminal_size::{terminal_size, Height, Width};
-
-#[cfg(test)]
-use mocktopus::macros::mockable;
-
-/// An internal method used to allow mocking terminal size in tests
-#[cfg_attr(test, mockable)]
-fn terminal_size_wrapper() -> (Width, Height) {
-    terminal_size().unwrap()
-}
+use terminal_size::{Height, Width};
+use super::terminal_size_wrapper;
 
 /// Validator to check if width configuration can be used
 ///