Parcourir la source

secrets: imported SecretValidationHasher.iterations for SecretTestCase

Zach Moody il y a 8 ans
Parent
commit
f7e768d166
1 fichiers modifiés avec 2 ajouts et 1 suppressions
  1. 2 1
      netbox/secrets/tests/test_models.py

+ 2 - 1
netbox/secrets/tests/test_models.py

@@ -6,6 +6,7 @@ from django.core.exceptions import ValidationError
 from django.test import TestCase
 
 from secrets.models import UserKey, Secret, generate_master_key, encrypt_master_key, decrypt_master_key
+from secrets.hashers import SecretValidationHasher
 
 
 class UserKeyTestCase(TestCase):
@@ -100,7 +101,7 @@ class SecretTestCase(TestCase):
         # Ensure proper hashing algorithm is used
         hasher, iterations, salt, sha256 = s.hash.split('$')
         self.assertEqual(hasher, 'pbkdf2_sha256', "Hashing algorithm has been modified to: {}".format(hasher))
-        self.assertGreaterEqual(int(iterations), 24000, "Insufficient iteration count ({}) for hash".format(iterations))
+        self.assertGreaterEqual(int(iterations), SecretValidationHasher.iterations, "Insufficient iteration count ({}) for hash".format(iterations))
         self.assertGreaterEqual(len(salt), 12, "Hash salt is too short ({} chars)".format(len(salt)))
 
         # Test hash validation