X-Git-Url: http://git.scottworley.com/tablify/blobdiff_plain/75bb888a36b5fdc005a0661c964779d6e0277298..9dfa98b71aca915f8daf3b5c8a4ad702a1f9e8da:/src/lib.rs?ds=sidebyside

diff --git a/src/lib.rs b/src/lib.rs
index 0e2007d..0584cd2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,4 +1,6 @@
 #[cfg(test)]
+use std::io::BufRead;
+#[cfg(test)]
 use std::iter::Iterator;
 
 #[derive(Debug, PartialEq, Eq)]
@@ -8,9 +10,13 @@ struct RowInput {
 }
 
 #[cfg(test)]
-fn read_rows(_input: &impl std::io::Read) -> impl Iterator<Item = RowInput> {
+fn read_rows(input: impl std::io::Read) -> impl Iterator<Item = RowInput> {
     vec![RowInput {
-        label: String::from("foo"),
+        label: std::io::BufReader::new(input)
+            .lines()
+            .nth(0)
+            .unwrap()
+            .unwrap(),
         entries: vec![],
     }]
     .into_iter()
@@ -27,11 +33,18 @@ mod tests {
     #[test]
     fn test_read_rows() {
         assert_eq!(
-            read_rows(&&b"foo"[..]).collect::<Vec<_>>(),
+            read_rows(&b"foo"[..]).collect::<Vec<_>>(),
             vec![RowInput {
                 label: String::from("foo"),
                 entries: vec![]
             }]
         );
+        assert_eq!(
+            read_rows(&b"bar"[..]).collect::<Vec<_>>(),
+            vec![RowInput {
+                label: String::from("bar"),
+                entries: vec![]
+            }]
+        );
     }
 }