Browse Source

Fix tests in api

pull/1689/head
Hironsan 3 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 importlib import reload
from os import environ from os import environ
from config.settings import base as settings
from django.test import TestCase from django.test import TestCase
from config.settings import base as settings
class TestDatabaseUrl(TestCase): class TestDatabaseUrl(TestCase):
def test_sslmode_defaults_to_required(self): 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.contrib.auth.models import User
from django.test import TestCase, override_settings 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_USER_GROUPS="X-AuthProxy-Groups")
@override_settings(HEADER_AUTH_ADMIN_GROUP_NAME="Admin") @override_settings(HEADER_AUTH_ADMIN_GROUP_NAME="Admin")
@ -11,7 +14,9 @@ class HeaderAuthMiddlewareTest(TestCase):
user = User.objects.create_user(username="TestUser") user = User.objects.create_user(username="TestUser")
user.is_superuser = False 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"}) middleware.process_user_groups(user, {"HTTP_X_AUTHPROXY_GROUPS": "Admin;Reader"})
self.assertTrue(user.is_superuser) self.assertTrue(user.is_superuser)
@ -20,7 +25,9 @@ class HeaderAuthMiddlewareTest(TestCase):
user = User.objects.create_user(username="TestUser") user = User.objects.create_user(username="TestUser")
user.is_superuser = True 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"}) middleware.process_user_groups(user, {"HTTP_X_AUTHPROXY_GROUPS": "Guest;Reader"})
self.assertFalse(user.is_superuser) self.assertFalse(user.is_superuser)
Loading…
Cancel
Save