From 1695c5f83e91e9fc293d92a3d4be231a3b025338 Mon Sep 17 00:00:00 2001 From: SKairinos Date: Thu, 25 Jan 2024 09:20:12 +0000 Subject: [PATCH] fix test --- .../user/tests/auth/backends/test_otp_bypass_token.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/codeforlife/user/tests/auth/backends/test_otp_bypass_token.py b/codeforlife/user/tests/auth/backends/test_otp_bypass_token.py index 4c944c67..973efc5d 100644 --- a/codeforlife/user/tests/auth/backends/test_otp_bypass_token.py +++ b/codeforlife/user/tests/auth/backends/test_otp_bypass_token.py @@ -2,7 +2,6 @@ from django.test import RequestFactory, TestCase from django.utils import timezone -from django.utils.crypto import get_random_string from ....auth.backends import OtpBypassTokenBackend from ....models import ( @@ -38,9 +37,7 @@ def setUp(self): auth_factor=self.auth_factor, ) - self.tokens = [ - get_random_string(8) for _ in range(OtpBypassToken.max_count) - ] + self.tokens = OtpBypassToken.generate_tokens() self.otp_bypass_tokens = OtpBypassToken.objects.bulk_create( [ OtpBypassToken(user=self.user, token=token) @@ -52,7 +49,7 @@ def test_authenticate(self): request = self.request_factory.post("/") request.user = self.user - user = self.backend.authenticate(request, token=self.tokens[0]) + user = self.backend.authenticate(request, token=next(iter(self.tokens))) assert user == self.user assert self.otp_bypass_tokens[0].id is None