Răsfoiți Sursa

Standardize on the use of v2 tokens in tests

Jeremy Stretch 4 luni în urmă
părinte
comite
adce67a7cf

+ 2 - 2
netbox/core/tests/test_api.py

@@ -135,8 +135,8 @@ class BackgroundTaskTestCase(TestCase):
         """
         # Create the test user and assign permissions
         self.user = User.objects.create_user(username='testuser', is_active=True)
-        self.token = Token.objects.create(version=1, user=self.user)
-        self.header = {'HTTP_AUTHORIZATION': f'Token {self.token.token}'}
+        self.token = Token.objects.create(user=self.user)
+        self.header = {'HTTP_AUTHORIZATION': f'Bearer {self.token.key}.{self.token.token}'}
 
         # Clear all queues prior to running each test
         get_queue('default').connection.flushall()

+ 2 - 2
netbox/netbox/tests/test_authentication.py

@@ -502,8 +502,8 @@ class ObjectPermissionAPIViewTestCase(TestCase):
         Create a test user and token for API calls.
         """
         self.user = User.objects.create(username='testuser')
-        self.token = Token.objects.create(version=1, user=self.user)
-        self.header = {'HTTP_AUTHORIZATION': f'Token {self.token.token}'}
+        self.token = Token.objects.create(user=self.user)
+        self.header = {'HTTP_AUTHORIZATION': f'Bearer {self.token.key}.{self.token.token}'}
 
     @override_settings(EXEMPT_VIEW_PERMISSIONS=[])
     def test_get_object(self):

+ 2 - 2
netbox/utilities/testing/api.py

@@ -49,8 +49,8 @@ class APITestCase(ModelTestCase):
         # Create the test user and assign permissions
         self.user = User.objects.create_user(username='testuser')
         self.add_permissions(*self.user_permissions)
-        self.token = Token.objects.create(version=1, user=self.user)
-        self.header = {'HTTP_AUTHORIZATION': f'Token {self.token.plaintext}'}
+        self.token = Token.objects.create(user=self.user)
+        self.header = {'HTTP_AUTHORIZATION': f'Bearer {self.token.key}.{self.token.token}'}
 
     def _get_view_namespace(self):
         return f'{self.view_namespace or self.model._meta.app_label}-api'