move db tests; remove old imports

1 files changed, 57 insertions, 0 deletions

diff --git a/database/src/table.rs b/database/src/table.rs
index ace0fb8..4535b4c 100644
--- a/database/src/table.rs
+++ b/database/src/table.rs
@@ -103,3 +103,60 @@ impl Table {
         self.query(txn, query).next()
     }
 }
+
+#[cfg(test)]
+mod test {
+    use crate::{
+        backends::{Database, new_memory},
+        table::Table,
+        test_shared::{NAME, new_bob},
+    };
+    use anyhow::Result;
+
+    #[test]
+    pub fn insert_get() -> Result<()> {
+        let db = new_memory();
+        let table = Table::new(5);
+
+        let mut bob_row = 0;
+        db.write_transaction(&mut |txn| {
+            bob_row = table.insert(txn, new_bob())?;
+            Ok(())
+        })?;
+
+        let mut bob = None;
+        db.read_transaction(&mut |txn| {
+            bob = table.get(txn, bob_row)?;
+            Ok(())
+        })?;
+
+        assert_eq!(bob.unwrap().as_object().get(NAME).unwrap(), "Bob");
+        Ok(())
+    }
+
+    #[test]
+    pub fn update() -> Result<()> {
+        let db = new_memory();
+        let table = Table::new(5);
+
+        let mut bob_row = 0;
+        let mut bob = None;
+
+        db.write_transaction(&mut |txn| {
+            bob_row = table.insert(txn, new_bob())?;
+            Ok(())
+        })?;
+        db.write_transaction(&mut |txn| {
+            let better_bob = new_bob().as_object().insert(NAME, "Better Bob");
+            table.update(txn, bob_row, better_bob)?;
+            Ok(())
+        })?;
+        db.read_transaction(&mut |txn| {
+            bob = table.get(txn, bob_row)?;
+            Ok(())
+        })?;
+
+        assert_eq!(bob.unwrap().as_object().get(NAME).unwrap(), "Better Bob");
+        Ok(())
+    }
+}