attribute tests

1 files changed, 13 insertions, 0 deletions

diff --git a/tests/support/mod.rs b/tests/support/mod.rs
new file mode 100644
index 0000000..51ad600
--- /dev/null
+++ b/tests/support/mod.rs
@@ -0,0 +1,13 @@
+use vt100;
+
+pub trait TestHelpers {
+    fn assert_process(&mut self, s: &[u8]) -> u64;
+}
+
+impl TestHelpers for vt100::Screen {
+    fn assert_process(&mut self, s: &[u8]) -> u64 {
+        let ret = self.process(s);
+        assert_eq!(ret, s.len() as u64);
+        ret
+    }
+}