26 lines
787 B
Python
26 lines
787 B
Python
|
"""passbook TOTP 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.totp.middleware import totp_force_verify
|
||
|
|
|
||
|
|
|
||
|
|
class TestMiddleware(TestCase):
|
||
|
|
"""passbook TOTP Middleware Test"""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
os.environ['RECAPTCHA_TESTING'] = 'True'
|
||
|
|
self.factory = RequestFactory()
|
||
|
|
|
||
|
|
def test_totp_force_verify_anon(self):
|
||
|
|
"""Test Anonymous TFA Force"""
|
||
|
|
request = self.factory.get(reverse('passbook_core:overview'))
|
||
|
|
request.user = AnonymousUser()
|
||
|
|
response = totp_force_verify(overview.OverviewView.as_view())(request)
|
||
|
|
self.assertEqual(response.status_code, 302)
|