This repository has been archived on 2024-05-31. You can view files and clone it, but cannot push or open issues or pull requests.
authentik/passbook/totp/tests/test_middleware.py

26 lines
787 B
Python
Raw Normal View History

2018-12-14 09:09:57 +00:00
"""passbook TOTP Middleware Test"""
2018-11-16 08:10:35 +00:00
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
2018-12-14 09:09:57 +00:00
from passbook.totp.middleware import totp_force_verify
2018-11-16 08:10:35 +00:00
class TestMiddleware(TestCase):
2018-12-14 09:09:57 +00:00
"""passbook TOTP Middleware Test"""
2018-11-16 08:10:35 +00:00
def setUp(self):
os.environ['RECAPTCHA_TESTING'] = 'True'
self.factory = RequestFactory()
2018-12-14 09:09:57 +00:00
def test_totp_force_verify_anon(self):
2018-11-16 09:08:15 +00:00
"""Test Anonymous TFA Force"""
request = self.factory.get(reverse('passbook_core:overview'))
2018-11-16 08:10:35 +00:00
request.user = AnonymousUser()
2018-12-14 09:09:57 +00:00
response = totp_force_verify(overview.OverviewView.as_view())(request)
2018-11-16 08:10:35 +00:00
self.assertEqual(response.status_code, 302)