yaqlite::insert(): Add test

Check that the function inserts a record into the database based on a
YAML input.

Adjust the `extract()` function to take the table name as an argument.

1 files changed, 5 insertions, 2 deletions

diff --git a/src/yaml.rs b/src/yaml.rs
index 532658a..2517d62 100644
--- a/src/yaml.rs
+++ b/src/yaml.rs
@@ -12,12 +12,13 @@ pub use sql::*;
 pub fn extract(
     doc: &mut yaml::Yaml,
     tx: &rusqlite::Transaction,
+    table_name: &str,
     table_columns: &HashMap<String, crate::sqlite::Zero>,
 ) {
     match doc {
         yaml::Yaml::Array(ref mut array) => {
             for yaml_value in array {
-                extract(yaml_value, tx, table_columns);
+                extract(yaml_value, tx, table_name, table_columns);
             }
         }
         yaml::Yaml::Hash(ref mut hash) => {
@@ -35,11 +36,13 @@ pub fn extract(
             let mut stmt = tx.prepare(
                 &format!(
                     r#"
-                        INSERT INTO "people"
+                        INSERT INTO "{}"
                             ({})
                         VALUES
                             ({});
                     "#,
+                    table_name,
+
                     // Wrap column names in quotes.
                     hash.keys()
                         .map(|k| format!(r#""{}""#, k.as_str().unwrap()))