X-Git-Url: http://git.scottworley.com/tablify/blobdiff_plain/ece97615567c7bc537f960d1784d652b11c0e66c..2aa9ef947fa4104deef034305f1238275427a092:/src/lib.rs

diff --git a/src/lib.rs b/src/lib.rs
index 820f43e..1d098b0 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,3 +1,58 @@
+#[cfg(test)]
+use std::io::BufRead;
+#[cfg(test)]
+use std::iter::Iterator;
+
+#[derive(Debug, PartialEq, Eq)]
+struct RowInput {
+    label: String,
+    entries: Vec<String>,
+}
+
+#[cfg(test)]
+fn read_rows(input: impl std::io::Read) -> impl Iterator<Item = RowInput> {
+    std::io::BufReader::new(input).lines().map(|line| RowInput {
+        label: line.unwrap(),
+        entries: vec![],
+    })
+}
+
 pub fn tablify(_input: &impl std::io::Read) -> String {
     String::from("Hello, world!")
 }
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn test_read_rows() {
+        assert_eq!(
+            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![]
+            }]
+        );
+        assert_eq!(
+            read_rows(&b"foo\nbar\n"[..]).collect::<Vec<_>>(),
+            vec![
+                RowInput {
+                    label: String::from("foo"),
+                    entries: vec![]
+                },
+                RowInput {
+                    label: String::from("bar"),
+                    entries: vec![]
+                }
+            ]
+        );
+    }
+}