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

Automatically create UserConfig for users

Browse Source
This commit is contained in:
Jeremy Stretch
2020-04-23 16:36:12 -04:00
parent afa0565a44
commit f3012ed839
3 changed files with 47 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
netbox/users/tests/test_models.py
View File

@@ -9,7 +9,7 @@ class UserConfigTest(TestCase):
def setUp(self):
user = User.objects.create_user(username='testuser')
initial_data = {
user.config.data = {
'a': True,
'b': {
'foo': 101,
@@ -27,8 +27,9 @@ class UserConfigTest(TestCase):
}
}
}
user.config.save()
self.userconfig = UserConfig(user=user, data=initial_data)
self.userconfig = user.config
def test_get(self):
userconfig = self.userconfig
@@ -58,12 +59,12 @@ class UserConfigTest(TestCase):
userconfig.set('b.baz', 'abc')
self.assertEqual(userconfig.data['d'], 'abc')
self.assertEqual(userconfig.data['b']['baz'], 'abc')
self.assertIsNone(userconfig.pk)
# Set a value and commit to the database
userconfig.set('a', 'def', commit=True)
userconfig.refresh_from_db()
self.assertEqual(userconfig.data['a'], 'def')
self.assertIsNotNone(userconfig.pk)
# Attempt to change a branch node to a leaf node
with self.assertRaises(TypeError):