mirrors/bookwyrm
1
0
Fork 1
mirror of https://github.com/bookwyrm-social/bookwyrm.git synced 2024-06-02 06:22:21 +00:00
Code Issues Projects Releases Wiki Activity

add test_post_login_with_2fa

Browse source
This commit is contained in:
Hugh Rundle 2022-10-15 07:11:03 +11:00
parent a1c3f15d80
commit 905aa66f38
1 changed files with 20 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

40
bookwyrm/tests/views/preferences/test_two_factor_auth.py
View file

@ -110,29 +110,29 @@ class TwoFactorViews(TestCase):
self.assertIsInstance(result, TemplateResponse)
self.assertEqual(result.status_code, 200)
# def test_post_login_with_2fa(self, *_):
# """check 2FA login works"""
# NOTE this throws a redis error.
# Possibly because the function it tests wants to check the user database?
# can we mock that?
# view = views.LoginWith2FA.as_view()
# form = forms.Confirm2FAForm()
# totp = pyotp.TOTP('UEWMVJHO23G5XLMVSOCL6TNTSSACJH2X')
def test_post_login_with_2fa(self, *_):
"""check 2FA login works"""
view = views.LoginWith2FA.as_view()
form = forms.Confirm2FAForm()
totp = pyotp.TOTP("UEWMVJHO23G5XLMVSOCL6TNTSSACJH2X")
# form.data["otp"] = totp.now()
# request = self.factory.post("", form.data)
# request.user = self.local_user
form.data["otp"] = totp.now()
request = self.factory.post("", form.data)
request.user = self.local_user
# middleware = SessionMiddleware(request)
# middleware.process_request(request)
# request.session['2fa_auth_time'] = time.time()
# request.session['2fa_user'] = self.local_user.username
# request.session.save()
middleware = SessionMiddleware(request)
middleware.process_request(request)
request.session["2fa_auth_time"] = time.time()
request.session["2fa_user"] = self.local_user.username
request.session.save()
# with patch("bookwyrm.views.preferences.two_factor_auth.LoginWith2FA"):
# result = view(request)
# self.assertIsInstance(result, TemplateResponse)
# self.assertEqual(result.status_code, 200)
with patch("bookwyrm.views.preferences.two_factor_auth.LoginWith2FA"), patch(
"bookwyrm.views.preferences.two_factor_auth.login"
):
result = view(request)
self.assertEqual(result.url, "/")
self.assertEqual(result.status_code, 302)
self.assertTrue(request.user.is_authenticated)
def test_post_login_with_2fa_wrong_code(self, *_):
"""check 2FA login fails"""