|
|
@ -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) |