"""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 common 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('common-index')) request.user = AnonymousUser() response = tfa_force_verify(common.IndexView.as_view())(request) self.assertEqual(response.status_code, 302)