Add endpoint tests

backend/endpoints/tests/test_heartbeat.py

@@ -0,0 +1,9 @@
+from fastapi.testclient import TestClient
+
+from main import app
+
+def test_heartbeat():
+    with TestClient(app) as client:
+        response = client.get("/heartbeat")
+        assert response.status_code == 200
+        assert response.json() == {"status": "ok"}

backend/endpoints/tests/test_platform.py

@@ -0,0 +1,9 @@
+from fastapi.testclient import TestClient
+
+from main import app
+
+def test_platforms():
+    with TestClient(app) as client:
+        response = client.get("/platforms")
+        assert response.status_code == 200
+        assert response.json() == {"status": "ok"}

backend/main.py

@@ -11,7 +11,11 @@
 from config import DEV_PORT, DEV_HOST, ROMM_AUTH_SECRET_KEY
 from endpoints import search, platform, rom, identity, oauth, scan  # noqa
 from utils.socket import socket_app
-from utils.auth import HybridAuthBackend, CustomCSRFMiddleware, create_default_admin_user
+from utils.auth import (
+    HybridAuthBackend,
+    CustomCSRFMiddleware,
+    create_default_admin_user,
+)
 
 app = FastAPI()
 

backend/utils/tests/test_oauth.py

@@ -40,7 +40,7 @@ async def test_get_current_active_user_from_token_invalid_token():
 async def test_get_current_active_user_from_token_invalid_user():
     token = create_oauth_token({"sub": "invalid_user"})
 
-    with pytest.raises(HTTPException) as e:
+    with pytest.raises(HTTPException):
         await get_current_active_user_from_token(token)
 
 

pyproject.toml

@@ -39,6 +39,7 @@ itsdangerous = "^2.1.2"
 stream-zip = "^0.0.67"
 starlette-csrf = "^3.0.0"
 pytest-asyncio = "^0.21.1"
+httpx = "^0.24.1"
 
 [build-system]
 requires = ["poetry-core"]