From 078228deac471521f2034728ec1f4f4fe99d337e Mon Sep 17 00:00:00 2001
From: sharkdp <davidpeter@web.de>
Date: Sun, 7 Oct 2018 12:29:38 +0200
Subject: [PATCH] Add unit test for InputFileReader

---
 src/inputfile.rs | 27 +++++++++++++++++++++++++++
 1 file changed, 27 insertions(+)

diff --git a/src/inputfile.rs b/src/inputfile.rs
index f8dbb8ff..d4e2e52e 100644
--- a/src/inputfile.rs
+++ b/src/inputfile.rs
@@ -45,3 +45,30 @@ impl<'a> InputFile<'a> {
         }
     }
 }
+
+#[test]
+fn basic() {
+    let content = b"hello\nworld";
+    let mut reader = InputFileReader::new(&content[..]);
+
+    let mut buffer = vec![];
+
+    let res = reader.read_line(&mut buffer);
+    assert!(res.is_ok());
+    assert_eq!(true, res.unwrap());
+    assert_eq!(b"hello\n", &buffer[..]);
+
+    buffer.clear();
+
+    let res = reader.read_line(&mut buffer);
+    assert!(res.is_ok());
+    assert_eq!(true, res.unwrap());
+    assert_eq!(b"world", &buffer[..]);
+
+    buffer.clear();
+
+    let res = reader.read_line(&mut buffer);
+    assert!(res.is_ok());
+    assert_eq!(false, res.unwrap());
+    assert!(buffer.is_empty());
+}