Fix unit tests

Flagsmith · Jul 30, 2024 · 59b45b1 · 59b45b1
1 parent 231bcc3
commit 59b45b1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/api/e2etests/middleware.py b/api/e2etests/middleware.py
@@ -7,7 +7,10 @@ def __init__(self, get_response):
 
     def __call__(self, request):
         request.is_e2e = False
-        if request.META.get("E2E_TEST_AUTH_TOKEN") == settings.E2E_TEST_AUTH_TOKEN:
+        if (
+            request.META.get("HTTP_X_E2E_TEST_AUTH_TOKEN")
+            == settings.E2E_TEST_AUTH_TOKEN
+        ):
             request.is_e2e = True
 
         return self.get_response(request)
diff --git a/api/tests/integration/e2etests/end_to_end/test_integration_e2e_tests.py b/api/tests/integration/e2etests/end_to_end/test_integration_e2e_tests.py
@@ -15,8 +15,8 @@ def test_e2e_teardown(settings, db) -> None:
     token = "test-token"
     register_url = "/api/v1/auth/users/"
     settings.ENABLE_FE_E2E = True
-
-    os.environ["E2E_TEST_AUTH_TOKEN"] = token
+    settings.E2E_TEST_AUTH_TOKEN = token
+    settings.MIDDLEWARE.append("e2etests.middleware.E2ETestMiddleware")
 
     client = APIClient(HTTP_X_E2E_TEST_AUTH_TOKEN=token)