1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Clean up tests

This commit is contained in:
jeremystretch
2022-11-17 10:24:03 -05:00
parent 977b79ecee
commit d4a231585a
9 changed files with 353 additions and 288 deletions

View File

@@ -4,7 +4,8 @@ from django.test import TestCase
class UserConfigTest(TestCase):
def setUp(self):
@classmethod
def setUpTestData(cls):
user = User.objects.create_user(username='testuser')
user.config.data = {
@@ -27,10 +28,8 @@ class UserConfigTest(TestCase):
}
user.config.save()
self.userconfig = user.config
def test_get(self):
userconfig = self.userconfig
userconfig = User.objects.get(username='testuser').config
# Retrieve root and nested values
self.assertEqual(userconfig.get('a'), True)
@@ -50,7 +49,7 @@ class UserConfigTest(TestCase):
self.assertEqual(userconfig.get('b.foo.x.invalid', 'DEFAULT'), 'DEFAULT')
def test_all(self):
userconfig = self.userconfig
userconfig = User.objects.get(username='testuser').config
flattened_data = {
'a': True,
'b.foo': 101,
@@ -64,7 +63,7 @@ class UserConfigTest(TestCase):
self.assertEqual(userconfig.all(), flattened_data)
def test_set(self):
userconfig = self.userconfig
userconfig = User.objects.get(username='testuser').config
# Overwrite existing values
userconfig.set('a', 'abc')
@@ -93,7 +92,7 @@ class UserConfigTest(TestCase):
userconfig.set('a.x', 1)
def test_clear(self):
userconfig = self.userconfig
userconfig = User.objects.get(username='testuser').config
# Clear existing values
userconfig.clear('a')