introcue stdin buffer for tests

1 files changed, 14 insertions, 1 deletions

diff --git a/src/testutils.rs b/src/testutils.rs
index 308a793..7593f98 100644
--- a/src/testutils.rs
+++ b/src/testutils.rs
@@ -4,7 +4,8 @@ use std::path::PathBuf;
 
 use std::cell::RefCell;
 thread_local! {
-  pub static STDOUT_BUF: RefCell<String> = RefCell::new(String::new())
+  pub static STDOUT_BUF: RefCell<String> = RefCell::new(String::new());
+  pub static STDIN_BUF: RefCell<String> = RefCell::new(String::new());
 }
 
 use defaults;
@@ -36,3 +37,15 @@ pub fn test_stdout_clear() -> String {
     result
   })
 }
+
+pub fn test_stdin_write(text: &str) {
+  STDIN_BUF.with(|cell| cell.borrow_mut().push_str(&text));
+}
+
+pub fn test_stdin_clear() -> Vec<String> {
+  STDIN_BUF.with(|cell| {
+    let result = cell.borrow().lines().map(|line| line.to_owned()).collect();
+    *cell.borrow_mut() = String::new();
+    result
+  })
+}