Fix lint errors

Author: sinkozs

backend/.idea/.gitignore

@@ -2,7 +2,6 @@
 from typing import Any
 
 import jwt
-from fastapi import Response
 from fastapi.responses import JSONResponse
 from passlib.context import CryptContext
 
@@ -20,7 +19,7 @@ def create_access_token(subject: str | Any, expires_delta: timedelta) -> str:
     return encoded_jwt
 
 
-def set_auth_cookie(subject: str | Any, expires_delta: timedelta) -> Response:
+def set_auth_cookie(subject: str | Any, expires_delta: timedelta) -> JSONResponse:
     access_token = create_access_token(subject, expires_delta)
     response = JSONResponse(content={"message": "Login successful"})
     # Note: The secure flag on cookies ensures they're only sent over encrypted HTTPS connections. For local development (HTTP) set it to False
@@ -37,7 +36,7 @@ def set_auth_cookie(subject: str | Any, expires_delta: timedelta) -> Response:
     return response
 
 
-def delete_auth_cookie() -> Response:
+def delete_auth_cookie() -> JSONResponse:
     response = JSONResponse(content={"message": "Logout successful"})
 
     response.delete_cookie(

backend/.idea/backend.iml

@@ -13,6 +13,7 @@ def user_authentication_headers(
     data = {"username": email, "password": password}
 
     r = client.post(f"{settings.API_V1_STR}/login/access-token", data=data)
+    print(f"user_authentication_headers: {r}")
     return extract_cookies(r)
 
 

backend/.idea/inspectionProfiles/profiles_settings.xml

@@ -1,6 +1,8 @@
 import random
 import string
 
+import httpx
+from fastapi import Response
 from fastapi.responses import JSONResponse
 from fastapi.testclient import TestClient
 
@@ -24,13 +26,22 @@ def get_superuser_auth_cookies(client: TestClient) -> dict[str, str]:
     return extract_cookies(r)
 
 
-def extract_cookies(response: JSONResponse) -> dict[str, str]:
-    cookie_header = response.headers.get("Set-Cookie")
-
-    cookie_value = None
-    if cookie_header and "http_only_auth_cookie=" in cookie_header:
-        cookie_value = cookie_header.split("http_only_auth_cookie=")[1].split(";")[0]
-
-    assert cookie_value, "Cookie value not found"
-
-    return {"http_only_auth_cookie": cookie_value}
+def extract_cookies(
+    response: JSONResponse | httpx._models.Response | Response,
+) -> dict[str, str]:
+    if isinstance(response, httpx._models.Response):
+        # Handle httpx Response
+        cookie_value = response.cookies.get("http_only_auth_cookie")
+        if cookie_value:
+            return {"http_only_auth_cookie": cookie_value}
+    else:
+        # Handle Starlette Response
+        cookie_header = response.headers.get("Set-Cookie")
+        if cookie_header and "http_only_auth_cookie=" in cookie_header:
+            cookie_value = cookie_header.split("http_only_auth_cookie=")[1].split(";")[
+                0
+            ]
+            if cookie_value:
+                return {"http_only_auth_cookie": cookie_value}
+
+    raise AssertionError("Cookie value not found")