<Test>: Add test on main.py

app/constants.py

@@ -166,3 +166,5 @@ classNames = [
     "hair drier",
     "toothbrush",
 ]
+
+test_uid = "VAbtgG1IrWSOqoJgxe7u2yEqpIg2"

app/dependencies.py

@@ -13,7 +13,7 @@ async def get_current_user(
     credentials_exception = HTTPException(
         status_code=status.HTTP_401_UNAUTHORIZED,
         detail="Could not validate credentials",
-        headers={"WWW-Authenticate": "Bearer"},
+        headers={"WWW-Authenticate": "Bearer "},
     )
     try:
         payload = auth.verify_id_token(credentials.credentials)

app/main.py

@@ -12,7 +12,7 @@ app.include_router(image.router)
 app.include_router(video.router)
 
 
-@app.get("/login")
+@app.get("/me")
 def getProfile(current_user: Annotated[any, Depends(get_current_user)]):
     return current_user
 

app/requirements.txt

@@ -10,3 +10,4 @@ supabase==1.0.4
 uvicorn==0.23.2
 python-multipart==0.0.6
 firebase_admin==6.2.0
+pytest

app/routers/test_image.py

@@ -5,12 +5,19 @@ from app.routers.image import inferenceImage
 import pytest
 import pytest
 from fastapi.testclient import TestClient
+from fastapi.routing import APIRoute
 from app.main import app
-
+def endpoints():
+    endpoints = []
+    for route in app.routes:
+        if isinstance(route, APIRoute):
+            endpoints.append(route.path)
+    return endpoints
 @pytest.fixture
 def client():
     client = TestClient(app, "http://0.0.0.0:3000")
     yield client
+@pytest.mark.skipif("/image" not in endpoints(), reason="Route not defined")
 class TestImageRoute():
     img = mmcv.imread('demo.jpg')
     url = "http://0.0.0.0:3000/image"

app/test_main.py

@@ -0,0 +1,82 @@
+from fastapi.testclient import TestClient
+import mmcv
+from app.main import app
+import pytest
+import json
+import os
+import site
+import shutil
+from fastapi.routing import APIRoute
+from app import firebase_app
+from firebase_admin import firestore
+def get_site_packages():
+    # Get the list of directories 
+    site_packages_dirs = site.getsitepackages()
+
+    # Find the "site-packages" directory in the list
+    for dir in site_packages_dirs:
+        if dir.endswith("site-packages"):
+            target_dir = dir
+            break
+        else:
+            target_dir=None
+    return target_dir
+
+def endpoints():
+    endpoints = []
+    for route in app.routes:
+        if isinstance(route, APIRoute):
+            endpoints.append(route.path)
+    return endpoints
+@pytest.fixture(autouse=True)
+def modify_mmcv_image():
+    site_packages_path = get_site_packages()
+    dirList = os.listdir(site_packages_path)
+    if "mmcv" in dirList:
+        shutil.copyfile("libs/image.py", os.path.join(site_packages_path, "mmcv/visualization/image.py"))
+    else:
+        pytest.exit('Error: Cannot modified mmcv.Image')
+        
+@pytest.fixture
+def client():
+    client = TestClient(app, "http://0.0.0.0:3000")
+    yield client
+@pytest.fixture
+def token():
+    token = ""
+    yield token
+class TestFireBaseAPI():
+    def test_get_me(self, client, token):
+        if "/me" not in endpoints():
+            pytest.skip("This route isn't defined")
+        else:
+            if token != "":
+                payload = ""
+                headers = {
+                    'accept': 'application/json',
+                    "Authorization":"Bearer " + token
+                }
+                response = client.request("get", 'me', headers=headers, data=payload)
+                assert response.status_code == 200
+            payload = ""
+            headers = {
+                'accept': 'application/json',
+                'Content-Type': 'application/json',
+            }
+            response = client.request("get", 'me', headers=headers, data=payload)
+            assert response.status_code == 403
+    def test_invitation(self,client, token):
+        if"/invitation" not in endpoints():
+            pytest.skip("This route isn't defined")
+        else:
+            payload = ''
+            headers = {
+                'accept': 'application/json',
+                'Content-Type': 'application/json',
+                "Authorization": "Bearer " + token
+            }
+            response = client.request("post", 'invitation', headers=headers, data=payload)
+            assert response.status_code == 200
+            result = mmcv.imfrombytes(response.read())
+            assert result.shape[0] == 3
+