test: add TestConvertDicts

Author: NightMarcher

fastapi_esql/__init__.py

@@ -21,7 +21,7 @@
     timing,
 )
 
-__version__ = "0.0.9"
+__version__ = "0.0.10"
 
 __all__ = [
     "QsParsingError",

fastapi_esql/utils/converter.py

@@ -8,9 +8,13 @@ def convert_dicts(dicts, converters: Dict[str, Callable]):
     if not converters:
         return
     for d in dicts:
-        for f, cvt in converters.items():
-            v = d[f]
+        for field, converter in converters.items():
+            if field not in d:
+                logger.warning(f"Item `{field}` does not exist in dict `{d}`")
+                continue
+
+            value = d[field]
             try:
-                d[f] = cvt(v)
+                d[field] = converter(value)
             except Exception as e:
-                logger.warning(f"Converting value `{v}` by `{cvt.__name__}` failed => {e}")
+                logger.warning(f"Converting value `{value}` by `{converter.__name__}` failed => {e}")

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "fastapi-efficient-sql"
-version = "0.0.9"
+version = "0.0.10"
 description = "Generate bulk DML SQL and execute them based on Tortoise ORM and mysql8.0+, and integrated with FastAPI."
 authors = ["BryanLee <[email protected]>"]
 keywords = ["sql", "fastapi", "tortoise-orm", "mysql8", "bulk-operation"]

tests/test_converter.py

@@ -0,0 +1,25 @@
+from copy import deepcopy
+from json import loads
+from unittest import TestCase
+
+from fastapi_esql import convert_dicts
+
+
+class TestConvertDicts(TestCase):
+
+    dicts = [{"id": 1, "value": "1"}, {"id": 2, "value": '{"k": [true, null]}'}]
+
+    def test_normal(self):
+        dicts = deepcopy(self.dicts)
+        convert_dicts(dicts, {"value": loads})
+        assert dicts == [{"id": 1, "value": 1}, {"id": 2, "value": {"k": [True, None]}}]
+
+    def test_wrong_field(self):
+        dicts = deepcopy(self.dicts)
+        convert_dicts(dicts, {"wrong_value": loads})
+        assert dicts == self.dicts
+
+    def test_wrong_converter(self):
+        dicts = deepcopy(self.dicts)
+        convert_dicts(dicts, {"value": int})
+        assert dicts == [{"id": 1, "value": 1}, {"id": 2, "value": '{"k": [true, null]}'}]

tests/test_orm.py

@@ -12,7 +12,8 @@ def get_ro_conn(self):
         return None
 
 
-class TestOrm(TestCase):
+class TestORM(TestCase):
+
     def test_no_model(self):
         with self.assertRaises(NotImplementedError):
             class AccountMgr(BaseManager, metaclass=DemoMetaclass):