Tests: check in-memory store #20

Author: dan-obx

tests/common.py

@@ -21,17 +21,14 @@ def autocleanup():
     remove_test_dir()
 
 
-def load_empty_test_objectbox(name: str = "") -> objectbox.ObjectBox:
+def load_empty_test_objectbox(db_name: str = test_dir) -> objectbox.ObjectBox:
     model = objectbox.Model()
     from objectbox.model import IdUid
     model.entity(TestEntity, last_property_id=IdUid(27, 1027))
     model.last_entity_id = IdUid(2, 2)
 
-    db_name = test_dir if len(name) == 0 else test_dir + "/" + name
-
     return objectbox.Builder().model(model).directory(db_name).build()
 
-
 def load_empty_test_datetime(name: str = "") -> objectbox.ObjectBox:
     model = objectbox.Model()
     from objectbox.model import IdUid

tests/test_inmemory.py

@@ -0,0 +1,29 @@
+import objectbox
+from tests.common import load_empty_test_objectbox
+from tests.model import TestEntity
+import os.path
+import shutil
+
+def test_inmemory():
+    # Expect path for persistent store
+    db_name = "testdata_persistent"
+    ob = load_empty_test_objectbox(db_name) 
+    box = objectbox.Box(ob, TestEntity)
+    object = TestEntity()
+    id = box.put(object)
+    assert id == 1
+    assert id == object.id
+    assert os.path.exists(db_name)
+    del box
+    ob.close()
+    shutil.rmtree(db_name)
+
+    # Expect no path for in-memory store
+    db_name = "memory:testdata"
+    ob = load_empty_test_objectbox(db_name) 
+    box = objectbox.Box(ob, TestEntity)
+    object = TestEntity()
+    id = box.put(object)
+    assert id == 1
+    assert id == object.id
+    assert not os.path.exists(db_name)