Misago/misago/acl/tests/test_user_acl_middleware.py

from unittest.mock import Mock

from django.contrib.auth import get_user_model
from django.test import TestCase

from misago.acl.middleware import user_acl_middleware
from misago.conftest import get_cache_versions

User = get_user_model()

cache_versions = get_cache_versions()


class MiddlewareTests(TestCase):
    def test_middleware_sets_attr_on_request(self):
        user = User.objects.create_user("User", "user@example.com")

        get_response = Mock()
        request = Mock(user=user, cache_versions=cache_versions)
        middleware = user_acl_middleware(get_response)
        middleware(request)
        assert request.user_acl

    def test_middleware_calls_get_response(self):
        user = User.objects.create_user("User", "user@example.com")
        get_response = Mock()
        request = Mock(user=user, cache_versions=cache_versions)
        middleware = user_acl_middleware(get_response)
        middleware(request)
        get_response.assert_called_once()