def setUp(self):
self.user = accounts.models.User.objects.create_user(
username="******",
email="*****@*****.**",
password="******",
email_verified=True,
twofa_enabled=True)
self.user._test_agree_all_tos()
self.dead_totp_device = models.TOTPDevice(owner=self.user,
last_t=0,
activated_at=timezone.now(),
deleted_at=timezone.now())
self.dead_totp_device.save()
self.totp_device = models.TOTPDevice(owner=self.user,
last_t=0,
activated_at=timezone.now())
self.totp_device.save()
self.backup_device = models.PaperDevice(owner=self.user,
activated_at=timezone.now())
self.backup_device.save()
self.client = django.test.Client()
self.login(self.client)
def test_can_render_totp(self):
self.device = models.TOTPDevice(
owner=self.user, activated_at=timezone.now(),
last_t=0, base32_secret='GEZDGNBVGY3TQOJQGEZDGNBVGY3TQOJQ')
self.device.save()
self.client.get(self.path())
def test_allows_totp_if_previous_disabled(self):
existing_device = models.TOTPDevice(
owner=self.user, last_t=0, activated_at=timezone.now(), deleted_at=timezone.now()
)
existing_device.save()
resp = self.client.get(self.path())
assert resp.status_code == 200
def test_disallows_multiple_totp(self):
existing_device = models.TOTPDevice(owner=self.user,
last_t=0,
activated_at=timezone.now())
existing_device.save()
resp = self.client.get(self.path())
assert resp.status_code == 302
def test_returns_active_device(self):
device = models.TOTPDevice(
owner=self.user, activated_at=timezone.now(),
last_t=0)
device.save()
got_device, got_other_devices = views._get_verify_device(self.user, None)
assert got_device == device
assert set(got_other_devices) == set()
def setUp(self):
self.user = accounts.models.User.objects.create_user(
username='******',
email='*****@*****.**',
password='******',
email_verified=True,
twofa_enabled=True)
self.other_user = accounts.models.User.objects.create_user(
username='******',
email='*****@*****.**',
password='******',
email_verified=True,
twofa_enabled=True)
self.dead_backup_device = models.PaperDevice(
owner=self.user,
activated_at=timezone.now(),
deleted_at=timezone.now())
self.dead_backup_device.save()
self.backup_device = models.PaperDevice(owner=self.user,
activated_at=timezone.now())
self.backup_device.save()
self.totp_device = models.TOTPDevice(owner=self.user,
activated_at=timezone.now(),
last_t=0)
self.totp_device.save()
self.bobs_totp_device = models.TOTPDevice(owner=self.other_user,
activated_at=timezone.now(),
last_t=0)
self.bobs_totp_device.save()
self.client = django.test.Client()
self.login(self.client)
