tests: add "khprintln!" macro to check stdout in tests

1 files changed, 32 insertions, 0 deletions

diff --git a/src/macros.rs b/src/macros.rs
new file mode 100644
index 0000000..a393399
--- /dev/null
+++ b/src/macros.rs
@@ -0,0 +1,32 @@
+#[macro_export]
+macro_rules! khprint {
+  () => ();
+  ($($arg:tt)*) => ({
+    let line = format!($($arg)*);
+    #[cfg(test)] {
+      use testutils;
+      testutils::test_stdout_write(&line);
+    }
+    print!("{}", line);
+  })
+}
+
+#[macro_export]
+macro_rules! khprintln {
+  () => ({
+    #[cfg(test)] {
+      use testutils;
+      testutils::test_stdout_write("\n");
+    }
+    println!();
+  });
+  ($($arg:tt)*) => ({
+    let line = format!($($arg)*);
+    #[cfg(test)] {
+      use testutils;
+      testutils::test_stdout_write(&line);
+      testutils::test_stdout_write("\n");
+    }
+    println!("{}", line);
+  })
+}