NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-12-16 13:29:37 -06:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Standardize on the use of v2 tokens in tests

Browse Source
This commit is contained in:
Jeremy Stretch 2025-10-02 16:37:28 -04:00
parent f82f084c02
commit adce67a7cf
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
netbox/core/tests/test_api.py
View File

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

4
netbox/netbox/tests/test_authentication.py
View File

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

4
netbox/utilities/testing/api.py
View File

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