Refs #787 -- Activate 2FA after adding a device by default

This commit is contained in:
Raphael Michel
2018-04-09 18:48:00 +02:00
parent 5d73221b06
commit 9dace592c0
4 changed files with 42 additions and 7 deletions

View File

@@ -222,11 +222,14 @@ class UserSettings2FATest(SoupTest):
totp = TOTP(d.bin_key, d.step, d.t0, d.digits, d.drift)
totp.time = time.time()
r = self.client.post('/control/settings/2fa/totp/{}/confirm'.format(d.pk), {
'token': str(totp.token())
'token': str(totp.token()),
'activate': 'on'
}, follow=True)
d.refresh_from_db()
assert d.confirmed
assert 'alert-success' in r.rendered_content
self.user.refresh_from_db()
assert self.user.require_2fa
def test_confirm_totp_failed(self):
self.client.post('/control/settings/2fa/add', {
@@ -271,11 +274,14 @@ class UserSettings2FATest(SoupTest):
d = U2FDevice.objects.first()
r = self.client.post('/control/settings/2fa/u2f/{}/confirm'.format(d.pk), {
'token': 'FOO'
'token': 'FOO',
'activate': 'on'
}, follow=True)
d.refresh_from_db()
assert d.confirmed
assert 'alert-success' in r.rendered_content
self.user.refresh_from_db()
assert self.user.require_2fa
m.undo()