Working factories

tests/test_cities.py

@@ -0,0 +1,28 @@
+import pytest
+from litestar.status_codes import HTTP_200_OK
+from litestar.testing import AsyncTestClient
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from main import app
+from tests.factories.city_factory import CityFactory
+
+
+class TestCitiesEndpoints:
+    @pytest.fixture(scope="function", autouse=True)
+    def setup_class(self, db_session: AsyncSession):
+        self.db_session = db_session
+
+    @pytest.mark.asyncio
+    async def test_get(self):
+        city_1 = await CityFactory.create()
+        city_2 = await CityFactory.create()
+        city_3 = await CityFactory.create()
+        city_4 = await CityFactory.create()
+        city_5 = await CityFactory.create()
+
+        async with AsyncTestClient(app=app) as client:
+            response = await client.get("/v1/cities")
+            print("#" * 100)
+            print(response.content)
+            print("#" * 100)
+            assert response.status_code == HTTP_200_OK