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

diff --git a/src/lib.rs b/src/lib.rs
index 820f43e..0e2007d 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,3 +1,37 @@
+#[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> {
+    vec![RowInput {
+        label: String::from("foo"),
+        entries: vec![],
+    }]
+    .into_iter()
+}
+
 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![]
+            }]
+        );
+    }
+}