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

Improved create/update validation

Browse Source
This commit is contained in:
Jeremy Stretch
2017-03-16 14:17:14 -04:00
parent d58f9031d1
commit bbc355df07
2 changed files with 51 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
netbox/tenancy/tests/test_api.py
View File

@ -46,6 +46,9 @@ class TenantGroupTest(APITestCase):
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(TenantGroup.objects.count(), 4)
tenantgroup4 = TenantGroup.objects.get(pk=response.data['id'])
self.assertEqual(tenantgroup4.name, data['name'])
self.assertEqual(tenantgroup4.slug, data['slug'])
def test_update_tenantgroup(self):
@ -59,8 +62,9 @@ class TenantGroupTest(APITestCase):
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(TenantGroup.objects.count(), 3)
self.assertEqual(TenantGroup.objects.get(pk=self.tenantgroup1.pk).name, data['name'])
self.assertEqual(TenantGroup.objects.get(pk=self.tenantgroup1.pk).slug, data['slug'])
tenantgroup1 = TenantGroup.objects.get(pk=response.data['id'])
self.assertEqual(tenantgroup1.name, data['name'])
self.assertEqual(tenantgroup1.slug, data['slug'])
def test_delete_tenantgroup(self):
@ -79,9 +83,11 @@ class TenantTest(APITestCase):
token = Token.objects.create(user=user)
self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)}
self.tenant1 = Tenant.objects.create(name='Test Tenant 1', slug='test-tenant-1')
self.tenant2 = Tenant.objects.create(name='Test Tenant 2', slug='test-tenant-2')
self.tenant3 = Tenant.objects.create(name='Test Tenant 3', slug='test-tenant-3')
self.tenantgroup1 = TenantGroup.objects.create(name='Test Tenant Group 1', slug='test-tenant-group-1')
self.tenantgroup2 = TenantGroup.objects.create(name='Test Tenant Group 2', slug='test-tenant-group-2')
self.tenant1 = Tenant.objects.create(name='Test Tenant 1', slug='test-tenant-1', group=self.tenantgroup1)
self.tenant2 = Tenant.objects.create(name='Test Tenant 2', slug='test-tenant-2', group=self.tenantgroup1)
self.tenant3 = Tenant.objects.create(name='Test Tenant 3', slug='test-tenant-3', group=self.tenantgroup1)
def test_get_tenant(self):
@ -102,6 +108,7 @@ class TenantTest(APITestCase):
data = {
'name': 'Test Tenant 4',
'slug': 'test-tenant-4',
'group': self.tenantgroup1.pk,
}
url = reverse('tenancy-api:tenant-list')
@ -109,12 +116,17 @@ class TenantTest(APITestCase):
self.assertEqual(response.status_code, status.HTTP_201_CREATED)
self.assertEqual(Tenant.objects.count(), 4)
tenant4 = Tenant.objects.get(pk=response.data['id'])
self.assertEqual(tenant4.name, data['name'])
self.assertEqual(tenant4.slug, data['slug'])
self.assertEqual(tenant4.group_id, data['group'])
def test_update_tenant(self):
data = {
'name': 'Test Tenant X',
'slug': 'test-tenant-x',
'group': self.tenantgroup2.pk,
}
url = reverse('tenancy-api:tenant-detail', kwargs={'pk': self.tenant1.pk})
@ -122,8 +134,10 @@ class TenantTest(APITestCase):
self.assertEqual(response.status_code, status.HTTP_200_OK)
self.assertEqual(Tenant.objects.count(), 3)
self.assertEqual(Tenant.objects.get(pk=self.tenant1.pk).name, data['name'])
self.assertEqual(Tenant.objects.get(pk=self.tenant1.pk).slug, data['slug'])
tenant1 = Tenant.objects.get(pk=response.data['id'])
self.assertEqual(tenant1.name, data['name'])
self.assertEqual(tenant1.slug, data['slug'])
self.assertEqual(tenant1.group_id, data['group'])
def test_delete_tenant(self):