diff options
Diffstat (limited to 'src/lib.rs')
| -rw-r--r-- | src/lib.rs | 195 |
1 files changed, 2 insertions, 193 deletions
@@ -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]); - } -} |