authentik/passbook/tfa/tests/test_middleware.py

"""passbook Mod 2FA Middleware Test"""

import os

from django.contrib.auth.models import AnonymousUser
from django.test import RequestFactory, TestCase
from django.urls import reverse

from passbook.core.views import overview
from passbook.tfa.middleware import tfa_force_verify


class TestMiddleware(TestCase):
    """passbook 2FA Middleware Test"""

    def setUp(self):
        os.environ['RECAPTCHA_TESTING'] = 'True'
        self.factory = RequestFactory()

    def test_tfa_force_verify_anon(self):
        """Test Anonymous TFA Force"""
        request = self.factory.get(reverse('passbook_core:overview'))
        request.user = AnonymousUser()
        response = tfa_force_verify(overview.OverviewView.as_view())(request)
        self.assertEqual(response.status_code, 302)