Skip to content

Commit 1695c5f

Browse files
committed
fix test
1 parent a7e8bc4 commit 1695c5f

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

codeforlife/user/tests/auth/backends/test_otp_bypass_token.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22

33
from django.test import RequestFactory, TestCase
44
from django.utils import timezone
5-
from django.utils.crypto import get_random_string
65

76
from ....auth.backends import OtpBypassTokenBackend
87
from ....models import (
@@ -38,9 +37,7 @@ def setUp(self):
3837
auth_factor=self.auth_factor,
3938
)
4039

41-
self.tokens = [
42-
get_random_string(8) for _ in range(OtpBypassToken.max_count)
43-
]
40+
self.tokens = OtpBypassToken.generate_tokens()
4441
self.otp_bypass_tokens = OtpBypassToken.objects.bulk_create(
4542
[
4643
OtpBypassToken(user=self.user, token=token)
@@ -52,7 +49,7 @@ def test_authenticate(self):
5249
request = self.request_factory.post("/")
5350
request.user = self.user
5451

55-
user = self.backend.authenticate(request, token=self.tokens[0])
52+
user = self.backend.authenticate(request, token=next(iter(self.tokens)))
5653

5754
assert user == self.user
5855
assert self.otp_bypass_tokens[0].id is None

0 commit comments

Comments
 (0)