Fix AuthTokens (#176)

DIAGNijmegen · Nov 20, 2024 · b606be8 · b606be8
1 parent ad9f6ff
commit b606be8
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/tests/scripts/create_test_fixtures.py b/tests/scripts/create_test_fixtures.py
@@ -34,8 +34,7 @@
 )
 from grandchallenge.verifications.models import Verification
 from grandchallenge.workstations.models import Workstation
-from knox import crypto
-from knox.models import AuthToken
+from knox.models import AuthToken, hash_token
 from knox.settings import CONSTANTS
 
 from .constants import USER_TOKENS
@@ -227,11 +226,9 @@ def _create_archive(users):
 def _create_user_tokens(users):
     out = f"{'*' * 80}\n"
     for user, token in USER_TOKENS.items():
-        digest = crypto.hash_token(token)
-
         AuthToken(
             token_key=token[: CONSTANTS.TOKEN_KEY_LENGTH],
-            digest=digest,
+            key=hash_token(token),
             user=users[user],
             expiry=None,
         ).save()