lib: Move `insert()` to a new module

Split this code out into its own module like we've done with `select()`.
Feels better organised this way.

1 files changed, 2 insertions, 193 deletions

diff --git a/src/lib.rs b/src/lib.rs
index 6dcbb56..f434ef2 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,8 +1,10 @@
+pub mod insert;
 pub mod select;
 pub mod sqlite;
 pub mod yaml;
 
 
+pub use insert::*;
 pub use select::*;
 
 
@@ -11,196 +13,3 @@ pub enum Error {
     #[error("SQL error: {0}")]
     Sqlite(#[from] rusqlite::Error),
 }
-
-
-// TODO: Move to an `insert` module
-pub fn insert(
-    dbconn: &mut rusqlite::Connection,
-    table_name: &str,
-    data: &mut [yaml_rust::Yaml],
-) -> Result<(), Error> {
-    let table_columns = crate::sqlite::get_column_names(&dbconn, table_name)?;
-
-    for mut doc in data {
-        let tx = dbconn.transaction()?;
-
-        crate::yaml::extract(&mut doc, &tx, &table_name, &table_columns)?;
-
-        tx.commit()?;
-    }
-
-    Ok(())
-}
-
-
-#[cfg(test)]
-mod tests {
-    use super::*;
-
-    #[derive(Debug, PartialEq)]
-    struct TestRecord {
-        id: i8,
-        count: i16,
-        weight: f32,
-        description: String,
-    }
-
-    fn test_yaml_insert(yaml_str: &str, expected: &[TestRecord]) {
-        let mut conn = rusqlite::Connection::open_in_memory().unwrap();
-
-        conn.execute(
-            r#"
-                CREATE TABLE "test" (
-                    id INTEGER PRIMARY KEY,
-                    count INTEGER,
-                    weight REAL,
-                    description TEXT
-                );
-            "#,
-            []
-        ).unwrap();
-
-        let mut data = yaml_rust::YamlLoader::load_from_str(&yaml_str).unwrap();
-
-        insert(&mut conn, "test", &mut data).unwrap();
-
-        {
-            let mut stmt = conn.prepare(r#"
-                SELECT
-                    id, count, weight, description
-                FROM "test";
-            "#).unwrap();
-
-            let rows = stmt.query_map(
-                [],
-                |row| {
-                    Ok(
-                        TestRecord {
-                            id: row.get(0).unwrap(),
-                            count: row.get(1).unwrap(),
-                            weight: row.get(2).unwrap(),
-                            description: row.get(3).unwrap(),
-                        }
-                    )
-                }
-            ).unwrap();
-
-            let got: Vec<TestRecord> = rows.map(|r| r.unwrap()).collect();
-
-            assert_eq!(expected, got);
-        }
-
-        conn.close().unwrap();
-    }
-
-    #[test]
-    fn inserts_yaml_in_database() {
-        let expected = TestRecord {
-            id: 1,
-            count: 99,
-            weight: 3.14,
-            description: r#"This is a test.
-Another paragraph with a flowed line."#.to_owned(),
-        };
-
-        let description = r#"This is a test.
-
-    Another paragraph
-    with a flowed line."#;
-
-        let yaml_str = format!(
-r#"- description: >-
-    {}
-  count: {}
-  weight: {}
-"#,
-            description,
-            expected.count,
-            expected.weight,
-        );
-
-        test_yaml_insert(&yaml_str, &vec![expected]);
-    }
-
-    #[test]
-    fn ignores_yaml_fields_that_are_not_column_names() {
-        let expected = TestRecord {
-            id: 1,
-            count: 55,
-            weight: 7.65,
-            description: "Some text content.".to_owned(),
-        };
-
-        let yaml_str = format!(
-r#"- description: >-
-    {}
-  count: {}
-  weight: {}
-  nonexistent_column: Must not be inserted.
-"#,
-            expected.description,
-            expected.count,
-            expected.weight,
-        );
-
-        test_yaml_insert(&yaml_str, &vec![expected]);
-    }
-
-    #[test]
-    fn inserts_multiple_records() {
-        let expected = vec![
-            TestRecord {
-                id: 1,
-                count: 10,
-                weight: 33.2,
-                description: "First".to_owned(),
-            },
-            TestRecord {
-                id: 2,
-                count: 12,
-                weight: 180.5,
-                description: "Second".to_owned(),
-            },
-        ];
-
-        let yaml_str = format!(
-r#"- description: {}
-  count: {}
-  weight: {}
-- description: {}
-  count: {}
-  weight: {}
-"#,
-            expected[0].description,
-            expected[0].count,
-            expected[0].weight,
-            expected[1].description,
-            expected[1].count,
-            expected[1].weight,
-        );
-
-        test_yaml_insert(&yaml_str, &expected);
-    }
-
-    #[test]
-    fn inserts_yaml_hash() {
-        let expected = TestRecord {
-            id: 1,
-            count: 255,
-            weight: 86.6,
-            description: "Some text content.".to_owned(),
-        };
-
-        let yaml_str = format!(
-r#"description: {}
-count: {}
-weight: {}
-"#,
-            expected.description,
-            expected.count,
-            expected.weight,
-        );
-
-        test_yaml_insert(&yaml_str, &vec![expected]);
-    }
-}