add tests

Author: mkrd

testing_orm.py

@@ -1,5 +1,4 @@
-from dictdatabase.object_mapper import DictModel, FileDictItemModel, FileDictModel
-import dictdatabase as DDB
+
 
 
 class WorkTime(DictModel):
@@ -21,7 +20,6 @@ def full_name(self):
 
 class Users(FileDictModel[User]):
 	__file__ = "users"
-	__item_model__ = User
 
 
 
@@ -45,8 +43,8 @@ class Users(FileDictModel[User]):
 assert u.work_times[1].end == "17:00"
 assert len(u.work_times) == 2
 
-print("check")
-print(u)
+
+print("u type:", type(u))
 
 
 

tests/conftest.py

@@ -2,11 +2,13 @@
 from tests import TEST_DIR
 import pytest
 import shutil
+import os
 
 
 @pytest.fixture(scope="session")
 def use_test_dir(request):
 	DDB.config.storage_directory = TEST_DIR
+	os.makedirs(TEST_DIR, exist_ok=True)
 	request.addfinalizer(lambda: shutil.rmtree(TEST_DIR))
 
 

tests/test_object_mapper.py

@@ -0,0 +1,50 @@
+from dictdatabase.object_mapper import DictModel, FileDictItemModel, FileDictModel
+import dictdatabase as DDB
+import pytest
+
+
+def test_object_mapper_docs_example(use_test_dir):
+	name = "object_mapper_docs_example_users"
+	DDB.at(name).create({
+		"u1": {
+			"first_name": "John",
+			"last_name": "Doe",
+			"age": 21
+		},
+		"u2": {
+			"first_name": "Jane",
+			"last_name": "Smith",
+			"age": 30,
+			"phone": "0123456"
+		},
+	}, force_overwrite=True)
+
+	class User(FileDictItemModel):
+		first_name: str
+		last_name: str
+		age: int
+		phone: str | None
+
+		def full_name(self):
+			return f"{self.first_name} {self.last_name}"
+
+	class Users(FileDictModel[User]):
+		__file__ = name
+
+	u1: User = Users.get_at_key("u1")
+	assert u1.full_name() == "John Doe"
+	assert u1.age == 21
+	assert u1.phone is None
+
+	with pytest.raises(AttributeError):
+		u1.no
+
+	u2: User = Users.get_at_key("u2")
+	assert u2.full_name() == "Jane Smith"
+	assert u2.age == 30
+	assert u2.phone == "0123456"
+
+
+
+	for uid, user in Users.get_all().items():
+		assert user.age in [21, 30]

tests/test_read.py

@@ -28,7 +28,7 @@ def test_invalid_params(use_test_dir, use_compression, use_orjson, indent):
 
 
 
-def test_read_integrity():
+def test_read_integrity(use_test_dir, use_compression, use_orjson, indent):
 	cases = [
 		r'{"a": "\\", "b": 2}',
 		r'{"a": "\\\\", "b": 2}',
@@ -42,8 +42,7 @@ def test_read_integrity():
 		r'{"a": "\\\"\"", "b": 2}',
 	]
 
-
-	for i, case in enumerate(cases):
+	for case in cases:
 		with open(f"{DDB.config.storage_directory}/test_read_integrity.json", "w") as f:
 			f.write(case)
 		dd = DDB.at("test_read_integrity", key="a").read()