diff --git a/backend/api/tests/test_config.py b/backend/api/tests/test_config.py index aa73dece..0f2e685c 100644 --- a/backend/api/tests/test_config.py +++ b/backend/api/tests/test_config.py @@ -2,9 +2,10 @@ from contextlib import contextmanager from importlib import reload from os import environ -from config.settings import base as settings from django.test import TestCase +from config.settings import base as settings + class TestDatabaseUrl(TestCase): def test_sslmode_defaults_to_required(self): diff --git a/backend/api/tests/test_middleware.py b/backend/api/tests/test_middleware.py index 5d42c198..7ea1fe46 100644 --- a/backend/api/tests/test_middleware.py +++ b/backend/api/tests/test_middleware.py @@ -1,7 +1,10 @@ -from api.middleware import HeaderAuthMiddleware +from unittest import mock + from django.contrib.auth.models import User from django.test import TestCase, override_settings +from api.middleware import HeaderAuthMiddleware + @override_settings(HEADER_AUTH_USER_GROUPS="X-AuthProxy-Groups") @override_settings(HEADER_AUTH_ADMIN_GROUP_NAME="Admin") @@ -11,7 +14,9 @@ class HeaderAuthMiddlewareTest(TestCase): user = User.objects.create_user(username="TestUser") user.is_superuser = False - middleware = HeaderAuthMiddleware() + get_response = mock.MagicMock() + + middleware = HeaderAuthMiddleware(get_response) middleware.process_user_groups(user, {"HTTP_X_AUTHPROXY_GROUPS": "Admin;Reader"}) self.assertTrue(user.is_superuser) @@ -20,7 +25,9 @@ class HeaderAuthMiddlewareTest(TestCase): user = User.objects.create_user(username="TestUser") user.is_superuser = True - middleware = HeaderAuthMiddleware() + get_response = mock.MagicMock() + + middleware = HeaderAuthMiddleware(get_response) middleware.process_user_groups(user, {"HTTP_X_AUTHPROXY_GROUPS": "Guest;Reader"}) self.assertFalse(user.is_superuser)