Misago/misago/admin/tests/test_admin_views.py

from django.test import TestCase
from django.urls import reverse

from ...users.test import create_test_user
from ..test import AdminTestCase
from ..views import get_protected_namespace


class MockRequest:
    def __init__(self, path):
        self.path = path


class AdminProtectedNamespaceTests(TestCase):
    def test_valid_cases(self):
        """get_protected_namespace returns true for protected links"""
        TEST_CASES = ("", "somewhere/", "ejksajdlksajldjskajdlksajlkdas")

        links_prefix = reverse("misago:admin:index")

        for case in TEST_CASES:
            request = MockRequest(links_prefix + case)
            self.assertEqual(get_protected_namespace(request), "misago:admin")

    def test_invalid_cases(self):
        """get_protected_namespace returns none for other links"""
        TEST_CASES = ("/", "/somewhere/", "/ejksajdlksajldjskajdlksajlkdas")

        for case in TEST_CASES:
            request = MockRequest(case)
            self.assertEqual(get_protected_namespace(request), None)


class AdminViewAccessTests(AdminTestCase):
    def test_admin_denies_non_staff_non_superuser(self):
        """admin middleware rejects user thats non staff and non superuser"""
        self.user.is_staff = False
        self.user.is_superuser = False
        self.user.save()

        response = self.client.get(reverse("misago:admin:index"))
        self.assertContains(response, "Sign in")

    def test_admin_denies_non_staff_superuser(self):
        """admin middleware rejects user thats non staff and superuser"""
        self.user.is_staff = False
        self.user.is_superuser = True
        self.user.save()

        response = self.client.get(reverse("misago:admin:index"))
        self.assertContains(response, "Sign in")

    def test_admin_passess_in_staff_non_superuser(self):
        """admin middleware passess user thats staff and non superuser"""
        self.user.is_staff = True
        self.user.is_superuser = False
        self.user.save()

        response = self.client.get(reverse("misago:admin:index"))
        self.assertContains(response, self.user.username)

    def test_admin_passess_in_staff_superuser(self):
        """admin middleware passess user thats staff and superuser"""
        self.user.is_staff = True
        self.user.is_superuser = True
        self.user.save()

        response = self.client.get(reverse("misago:admin:index"))
        self.assertContains(response, self.user.username)


class Admin404ErrorTests(AdminTestCase):
    def test_list_search_unicode_handling(self):
        """querystring creation handles unicode strings"""
        test_link = "%stotally-errored/" % reverse("misago:admin:index")

        response = self.client.get(test_link)

        self.assertContains(
            response, "Requested page could not be found.", status_code=404
        )


class AdminGenericViewsTests(AdminTestCase):
    def test_view_redirected_queryvar(self):
        """querystring redirected value is handled"""
        test_link = reverse("misago:admin:users:accounts:index")

        # request resulted in redirect with redirected=1 bit
        response = self.client.get("%s?username=lorem" % test_link)
        self.assertEqual(response.status_code, 302)
        self.assertIn("redirected=1", response["location"])

        # request with flag muted redirect
        response = self.client.get("%s?redirected=1&username=lorem" % test_link)
        self.assertEqual(response.status_code, 200)

    def test_list_search_unicode_handling(self):
        """querystring creation handles unicode strings"""
        test_link = reverse("misago:admin:users:accounts:index")
        response = self.client.get("%s?redirected=1&username=%s" % (test_link, "łut"))
        self.assertEqual(response.status_code, 200)