NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-26 01:06:11 -06:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

12794 call get_user_model once in tests

Browse Source
This commit is contained in:
Arthur 2023-06-15 09:33:42 -07:00
parent ea184b66b6
commit cc5f5acc0b
4 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
netbox/dcim/tests/test_api.py
View File

@ -14,6 +14,9 @@ from wireless.choices import WirelessChannelChoices
from wireless.models import WirelessLAN from wireless.models import WirelessLAN
User = get_user_model()
class AppTest(APITestCase): class AppTest(APITestCase):
def test_root(self): def test_root(self):
@ -363,7 +366,7 @@ class RackReservationTest(APIViewTestCases.APIViewTestCase):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
user = get_user_model().objects.create(username='user1', is_active=True) user = User.objects.create(username='user1', is_active=True)
site = Site.objects.create(name='Test Site 1', slug='test-site-1') site = Site.objects.create(name='Test Site 1', slug='test-site-1')
racks = ( racks = (

5
netbox/extras/tests/test_api.py
View File

@ -15,6 +15,9 @@ from extras.scripts import BooleanVar, IntegerVar, Script, StringVar
from utilities.testing import APITestCase, APIViewTestCases from utilities.testing import APITestCase, APIViewTestCases
User = get_user_model()
class AppTest(APITestCase): class AppTest(APITestCase):
def test_root(self): def test_root(self):
@ -396,7 +399,7 @@ class JournalEntryTest(APIViewTestCases.APIViewTestCase):
@classmethod @classmethod
def setUpTestData(cls): def setUpTestData(cls):
user = get_user_model().objects.first() user = User.objects.first()
site = Site.objects.create(name='Site 1', slug='site-1') site = Site.objects.create(name='Site 1', slug='site-1')
journal_entries = ( journal_entries = (

5
netbox/users/tests/test_preferences.py
View File

@ -16,6 +16,9 @@ DEFAULT_USER_PREFERENCES = {
} }
User = get_user_model()
class UserPreferencesTest(TestCase): class UserPreferencesTest(TestCase):
user_permissions = ['dcim.view_site'] user_permissions = ['dcim.view_site']
@ -39,7 +42,7 @@ class UserPreferencesTest(TestCase):
@override_settings(DEFAULT_USER_PREFERENCES=DEFAULT_USER_PREFERENCES) @override_settings(DEFAULT_USER_PREFERENCES=DEFAULT_USER_PREFERENCES)
def test_default_preferences(self): def test_default_preferences(self):
user = get_user_model().objects.create(username='User 1') user = User.objects.create(username='User 1')
userconfig = user.config userconfig = user.config
self.assertEqual(userconfig.data, DEFAULT_USER_PREFERENCES) self.assertEqual(userconfig.data, DEFAULT_USER_PREFERENCES)

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

@ -26,6 +26,9 @@ __all__ = (
) )
User = get_user_model()
# #
# REST/GraphQL API Tests # REST/GraphQL API Tests
# #
@ -45,7 +48,7 @@ class APITestCase(ModelTestCase):
Create a user and token for API calls. Create a user and token for API calls.
""" """
# Create the test user and assign permissions # Create the test user and assign permissions
self.user = get_user_model().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(user=self.user) self.token = Token.objects.create(user=self.user)
self.header = {'HTTP_AUTHORIZATION': f'Token {self.token.key}'} self.header = {'HTTP_AUTHORIZATION': f'Token {self.token.key}'}