test: memory store basic functionality tests

yxonic · Jun 6, 2022 · 4a174c0 · 4a174c0
1 parent 98ca8cf
commit 4a174c0
Showing 1 changed file with 48 additions and 0 deletions.
diff --git a/rcommunity_core/src/store/memory.rs b/rcommunity_core/src/store/memory.rs
@@ -84,3 +84,51 @@ impl Transaction for MemoryTransaction {
         unimplemented!();
     }
 }
+
+impl Drop for MemoryTransaction {
+    fn drop(&mut self) {
+        // ensure transaction lock is released
+        self.release_txn_lock();
+    }
+}
+
+#[cfg(test)]
+mod test {
+    use crate::store::{Store, Transaction};
+
+    use super::MemoryStore;
+
+    #[tokio::test]
+    async fn test_memory_store() {
+        let mut store = MemoryStore::default();
+        {
+            let mut txn = store.txn_begin().await.unwrap();
+            assert!(txn.get("key".into()).await.unwrap().is_none());
+            txn.put("key".into(), "value".into()).await.unwrap();
+            assert_eq!(
+                txn.get("key".into()).await.unwrap().unwrap(),
+                String::from("value")
+            );
+        }
+        {
+            let mut txn = store.txn_begin().await.unwrap();
+            assert_eq!(
+                txn.get_for_update("key".into()).await.unwrap().unwrap(),
+                String::from("value")
+            );
+            txn.put("key".into(), "".into()).await.unwrap();
+            assert_eq!(
+                txn.get("key".into()).await.unwrap().unwrap(),
+                String::from("")
+            );
+            txn.commit().await.unwrap();
+        }
+        {
+            let txn = store.txn_begin().await.unwrap();
+            assert_eq!(
+                txn.get("key".into()).await.unwrap().unwrap(),
+                String::from("")
+            );
+        }
+    }
+}