Browse Source

Fix tests in api

pull/1689/head
Hironsan 2 years ago
parent
commit
3e2e306ff4
2 changed files with 12 additions and 4 deletions
  1. 3
      backend/api/tests/test_config.py
  2. 13
      backend/api/tests/test_middleware.py

3
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):

13
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)
Loading…
Cancel
Save