test: add TestSQLizer.test_resolve_wheres

Author: NightMarcher

Pipfile

@@ -17,4 +17,4 @@ tortoise-orm = "*"
 uvicorn = "*"
 
 [requires]
-python_version = "3.7"
+python_version = ">=3.7"

examples/service/__init__.py

@@ -86,7 +86,7 @@ def init_db(app: FastAPI):
         },
         "apps": {
             "demo": {
-                "models": ["examples.service.models.demo.__init__"],
+                "models": ["examples.service.models.demo"],
                 "default_connection": "demo_ro"
             }
         }

requirements-dev.txt

@@ -0,0 +1,8 @@
+aiomysql
+autopep8
+cryptography
+faker
+fastapi
+pytest
+tortoise-orm
+uvicorn

tests/__init__.py

@@ -0,0 +1,34 @@
+import asyncio
+
+from tortoise import Tortoise
+
+
+def init_test_orm():
+    """
+    CREATE USER 'demo_test'@'localhost' IDENTIFIED BY 'demo_TEST#0';
+    GRANT ALL ON demo.* TO 'demo_test'@'localhost';
+    """
+    config = {
+        "timezone": "Asia/Shanghai",
+        "connections": {
+            "test": {
+                "engine": "tortoise.backends.mysql",
+                "credentials": {
+                    "host": "localhost",
+                    "port": 3306,
+                    "user": "demo_test",
+                    "password": "demo_TEST#0",
+                    "database": "demo",
+                    "pool_recycle": 3600,
+                    "maxsize": 10,
+                },
+            },
+        },
+        "apps": {
+            "demo": {
+                "models": ["examples.service.models.demo"],
+                "default_connection": "test"
+            }
+        }
+    }
+    asyncio.run(Tortoise.init(config=config))

tests/test_sqlizer.py

@@ -1,7 +1,13 @@
 from datetime import datetime
 from unittest import TestCase
 
-from fastapi_esql import Cases, RawSQL, SQLizer
+from fastapi_esql import (
+    Cases, RawSQL, SQLizer,
+    Q, QsParsingError, WrongParamsError,
+)
+
+from . import init_test_orm
+from examples.service.models.demo import Account
 
 
 class TestRawSQL(TestCase):
@@ -27,6 +33,31 @@ def test_no_default(self):
 
 class TestSQLizer(TestCase):
 
+    def test_resolve_wheres(self):
+        with self.assertRaises(WrongParamsError):
+            SQLizer.resolve_wheres(object())
+
+        aids = (1, 2, 3)
+        assert SQLizer.resolve_wheres(
+            f"id IN ({','.join(map(str, aids))}) AND gender=1"
+        ) == "id IN (1,2,3) AND gender=1"
+        init_test_orm()
+        assert SQLizer.resolve_wheres(
+            Q(Q(id__in=aids), Q(gender=1), join_type="AND"), Account
+        ) == "`id` IN (1,2,3) AND `gender`=1"
+        assert SQLizer.resolve_wheres(
+            {"id__in": aids, "gender": 1}, Account
+        ) == "`id` IN (1,2,3) AND `gender`=1"
+        assert SQLizer.resolve_wheres(
+            [Q(id__in=aids), Q(gender=1)], Account
+        ) == "`id` IN (1,2,3) AND `gender`=1"
+
+        with self.assertRaises(WrongParamsError):
+            SQLizer.resolve_wheres(set())
+
+        with self.assertRaises(QsParsingError):
+            SQLizer.resolve_wheres({}, Account)
+
     def test_resolve_orders(self):
         orders = SQLizer.resolve_orders(["-created_at", "name"])
         assert orders == "created_at DESC, name ASC"