Add testing
This commit is contained in:
parent
62b8be1a10
commit
a164c5d67c
@ -420,20 +420,30 @@ if AUTO_LOGIN_USERNAME:
|
|||||||
# regular login in case the provided user does not exist.
|
# regular login in case the provided user does not exist.
|
||||||
MIDDLEWARE.insert(_index + 1, "paperless.auth.AutoLoginMiddleware")
|
MIDDLEWARE.insert(_index + 1, "paperless.auth.AutoLoginMiddleware")
|
||||||
|
|
||||||
|
|
||||||
|
def _parse_remote_user_settings() -> str:
|
||||||
ENABLE_HTTP_REMOTE_USER = __get_boolean("PAPERLESS_ENABLE_HTTP_REMOTE_USER")
|
ENABLE_HTTP_REMOTE_USER = __get_boolean("PAPERLESS_ENABLE_HTTP_REMOTE_USER")
|
||||||
HTTP_REMOTE_USER_HEADER_NAME = os.getenv(
|
header_name = os.getenv(
|
||||||
"PAPERLESS_HTTP_REMOTE_USER_HEADER_NAME",
|
"PAPERLESS_HTTP_REMOTE_USER_HEADER_NAME",
|
||||||
"HTTP_REMOTE_USER",
|
"HTTP_REMOTE_USER",
|
||||||
)
|
)
|
||||||
|
|
||||||
if ENABLE_HTTP_REMOTE_USER:
|
if ENABLE_HTTP_REMOTE_USER:
|
||||||
MIDDLEWARE.append("paperless.auth.HttpRemoteUserMiddleware")
|
MIDDLEWARE.append("paperless.auth.HttpRemoteUserMiddleware")
|
||||||
AUTHENTICATION_BACKENDS.insert(0, "django.contrib.auth.backends.RemoteUserBackend")
|
AUTHENTICATION_BACKENDS.insert(
|
||||||
|
0,
|
||||||
|
"django.contrib.auth.backends.RemoteUserBackend",
|
||||||
|
)
|
||||||
REST_FRAMEWORK["DEFAULT_AUTHENTICATION_CLASSES"].insert(
|
REST_FRAMEWORK["DEFAULT_AUTHENTICATION_CLASSES"].insert(
|
||||||
0,
|
0,
|
||||||
"paperless.auth.PaperlessRemoteUserAuthentication",
|
"paperless.auth.PaperlessRemoteUserAuthentication",
|
||||||
)
|
)
|
||||||
|
|
||||||
|
return header_name
|
||||||
|
|
||||||
|
|
||||||
|
HTTP_REMOTE_USER_HEADER_NAME = _parse_remote_user_settings()
|
||||||
|
|
||||||
# X-Frame options for embedded PDF display:
|
# X-Frame options for embedded PDF display:
|
||||||
X_FRAME_OPTIONS = "ANY" if DEBUG else "SAMEORIGIN"
|
X_FRAME_OPTIONS = "ANY" if DEBUG else "SAMEORIGIN"
|
||||||
|
|
||||||
|
75
src/paperless/tests/test_remote_user.py
Normal file
75
src/paperless/tests/test_remote_user.py
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
import os
|
||||||
|
from unittest import mock
|
||||||
|
|
||||||
|
from django.contrib.auth.models import User
|
||||||
|
from rest_framework import status
|
||||||
|
from rest_framework.test import APITestCase
|
||||||
|
|
||||||
|
from documents.tests.utils import DirectoriesMixin
|
||||||
|
from paperless.settings import _parse_remote_user_settings
|
||||||
|
|
||||||
|
|
||||||
|
class TestRemoteUser(DirectoriesMixin, APITestCase):
|
||||||
|
def setUp(self):
|
||||||
|
super().setUp()
|
||||||
|
|
||||||
|
self.user = User.objects.create_superuser(
|
||||||
|
username="temp_admin",
|
||||||
|
)
|
||||||
|
|
||||||
|
def test_remote_user(self):
|
||||||
|
"""
|
||||||
|
GIVEN:
|
||||||
|
- Configured user
|
||||||
|
- Remote user auth is enabled
|
||||||
|
WHEN:
|
||||||
|
- API call is made to get documents
|
||||||
|
THEN:
|
||||||
|
- Call succeeds
|
||||||
|
"""
|
||||||
|
|
||||||
|
with mock.patch.dict(
|
||||||
|
os.environ,
|
||||||
|
{
|
||||||
|
"PAPERLESS_ENABLE_HTTP_REMOTE_USER": "True",
|
||||||
|
},
|
||||||
|
):
|
||||||
|
_parse_remote_user_settings()
|
||||||
|
|
||||||
|
response = self.client.get("/api/documents/")
|
||||||
|
|
||||||
|
# 403 testing locally, 401 on ci...
|
||||||
|
self.assertIn(
|
||||||
|
response.status_code,
|
||||||
|
[status.HTTP_401_UNAUTHORIZED, status.HTTP_403_FORBIDDEN],
|
||||||
|
)
|
||||||
|
|
||||||
|
response = self.client.get(
|
||||||
|
"/api/documents/",
|
||||||
|
headers={
|
||||||
|
"Remote-User": self.user.username,
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
self.assertEqual(response.status_code, status.HTTP_200_OK)
|
||||||
|
|
||||||
|
def test_remote_user_header_setting(self):
|
||||||
|
"""
|
||||||
|
GIVEN:
|
||||||
|
- Remote user header name is set
|
||||||
|
WHEN:
|
||||||
|
- Settings are parsed
|
||||||
|
THEN:
|
||||||
|
- Correct header name is returned
|
||||||
|
"""
|
||||||
|
|
||||||
|
with mock.patch.dict(
|
||||||
|
os.environ,
|
||||||
|
{
|
||||||
|
"PAPERLESS_ENABLE_HTTP_REMOTE_USER": "True",
|
||||||
|
"PAPERLESS_HTTP_REMOTE_USER_HEADER_NAME": "HTTP_FOO",
|
||||||
|
},
|
||||||
|
):
|
||||||
|
header_name = _parse_remote_user_settings()
|
||||||
|
|
||||||
|
self.assertEqual(header_name, "HTTP_FOO")
|
Loading…
x
Reference in New Issue
Block a user