trent/trentpalmerdotorg
1
0
Fork 0
mirror of https://github.com/TrentSPalmer/trentpalmerdotorg.git synced 2024-11-21 17:11:30 -08:00
Code Issues Projects Releases Wiki Activity

update accounts/tests/test_totp_enabled_login_view.py add function

Browse Source 
test_TOTP_enabled_log_in_view_already_logged_in
This commit is contained in:
Trent Palmer 2021-04-20 20:28:20 -07:00
parent 9d48b5b0ed
commit d30af17e60
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
accounts/tests/test_totp_enabled_login_view.py
View File

@ -16,6 +16,16 @@ class TestTOTPEnabledLogInViewTestCase(TestCase):
user=user_a, user=user_a,
use_totp=True, totp_key=pyotp.random_base32()) use_totp=True, totp_key=pyotp.random_base32())
def test_TOTP_enabled_log_in_view_already_logged_in(self):
self.client.login(username='user_a', password='password_user_a')
user_a = User.objects.get(username='user_a')
totp_code = pyotp.TOTP(user_a.account.totp_key).now()
response = self.client.post(reverse('accounts:two_factor_input'), {
'totp_code': totp_code}, follow=True)
self.assertEquals(response.status_code, 200)
self.assertTemplateUsed(response, 'audio/index.html')
self.assertEquals(response.request['PATH_INFO'], '/')
def test_TOTP_enabled_log_in_view_wrong_code_with_retries(self): def test_TOTP_enabled_log_in_view_wrong_code_with_retries(self):
login_response = self.client.post(reverse('accounts:login'), { login_response = self.client.post(reverse('accounts:login'), {
'username': 'user_a', 'username': 'user_a',