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:
@@ -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')
|
||||
|
Reference in New Issue
Block a user