fix: auto_apis unit bugs

Author: freemindcore

tests/conftest.py

@@ -28,37 +28,38 @@ def user(db) -> User:
 def easy_api_client(user) -> Callable:
     orig_func = copy.deepcopy(JWTAuthAsync.__call__)
 
-    orig_has_perm_fuc = copy.deepcopy(user.has_perm)
-
-    def mock_has_perm_true(*args, **kwargs):
-        return True
-
-    def mock_has_perm_false(*args, **kwargs):
-        return False
-
-    async def mock_func(self, request):
-        setattr(request, "user", user)
-        return True
-
-    setattr(JWTAuthAsync, "__call__", mock_func)
-
     def create_client(
         api: Union[EasyAPI, Router, Type[ControllerBase]],
+        api_user=None,  # type: ignore
         is_staff: bool = False,
         is_superuser: bool = False,
         has_perm: bool = False,
     ) -> "EasyTestClient":
-        setattr(user, "is_staff", is_staff)
-        setattr(user, "is_superuser", is_superuser)
+        if api_user is None:
+            api_user = user
+        setattr(api_user, "is_staff", is_staff)
+        setattr(api_user, "is_superuser", is_superuser)
+
+        def mock_has_perm_true(*args, **kwargs):
+            return True
+
+        def mock_has_perm_false(*args, **kwargs):
+            return False
+
+        async def mock_func(self, request):
+            setattr(request, "user", api_user)
+            return True
+
+        setattr(JWTAuthAsync, "__call__", mock_func)
+
         if is_superuser:
-            setattr(user, "is_staff", True)
+            setattr(api_user, "is_staff", True)
         if has_perm:
-            setattr(user, "has_perm", mock_has_perm_true)
+            setattr(api_user, "has_perm", mock_has_perm_true)
         else:
-            setattr(user, "has_perm", mock_has_perm_false)
+            setattr(api_user, "has_perm", mock_has_perm_false)
         client = EasyTestClient(api, auth=jwt_auth_async)
         return client
 
     yield create_client
     setattr(JWTAuthAsync, "__call__", orig_func)
-    setattr(user, "has_perm", orig_has_perm_fuc)

tests/test_auto_api_creation.py

@@ -32,21 +32,33 @@ def test_auto_generate_admin_api():
     assert "TypeAdminAPIController" in controller_names
 
 
-async def test_auto_apis(transactional_db, easy_api_client):
+async def test_auto_apis(transactional_db, user, easy_api_client):
     for controller_class in controllers:
         if not str(controller_class).endswith("ClientAdminAPIController"):
             continue
-
-        client = easy_api_client(controller_class)
-        response = await client.get("/")
-        # TODO: figure out why user.is_authenticated is False in auto created API
-
+        client = easy_api_client(controller_class, api_user=user, has_perm=True)
+        response = await client.get("/", data={}, json={}, user=user)
         assert response.status_code == 403
-        # assert response.json()["data"] == []
 
+        client = easy_api_client(
+            controller_class, api_user=user, has_perm=True, is_staff=True
+        )
+        response = await client.get("/", data={}, json={}, user=user)
+        assert response.status_code == 200
+        assert response.json()["data"] == []
+
+        client = easy_api_client(
+            controller_class, api_user=user, has_perm=True, is_staff=True
+        )
         response = await client.delete("/20000")
         assert response.status_code == 403
-        # assert response.json()["code"] == 404
+
+        client = easy_api_client(
+            controller_class, api_user=user, has_perm=True, is_staff=True
+        )
+        response = await client.delete("/20000", data={}, json={}, user=user)
+        assert response.status_code == 200
+        assert response.json()["code"] == 404
 
 
 async def test_auto_generation_settings(settings):