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

Move TreeNodeMultipleChoiceFilter tests to utilities (follow-up to #3616)

Browse Source
This commit is contained in:
Jeremy Stretch
2019-11-01 15:01:24 -04:00
parent ee854eff2c
commit 875e09013c
3 changed files with 87 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -127,9 +127,6 @@ class SiteTest(APITestCase):
self.region2 = Region.objects.create(name='Test Region 2', slug='test-region-2')
self.site1 = Site.objects.create(region=self.region1, name='Test Site 1', slug='test-site-1')
self.site2 = Site.objects.create(region=self.region1, name='Test Site 2', slug='test-site-2')
self.site3 = Site.objects.create(region=self.region2, name='Test Site 3', slug='test-site-3')
self.site_non_region1 = Site.objects.create(name='Test Site Null Region1', slug='test-site-no-region1')
self.site_non_region2 = Site.objects.create(name='Test Site Null Region2', slug='test-site-no-region2')
def test_get_site(self):
@ -164,7 +161,7 @@ class SiteTest(APITestCase):
url = reverse('dcim-api:site-list')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 5)
self.assertEqual(response.data['count'], 3)
def test_list_sites_brief(self):
@ -176,20 +173,6 @@ class SiteTest(APITestCase):
['id', 'name', 'slug', 'url']
)
def test_list_sites_null_region(self):
url = reverse('dcim-api:site-list')
response = self.client.get('{}?region=null'.format(url), **self.header)
self.assertEqual(response.data['count'], 2)
def test_list_sites_multiple_regions(self):
url = reverse('dcim-api:site-list')
response = self.client.get('{}?region=null&region=test-region-1'.format(url), **self.header)
self.assertEqual(response.data['count'], 4)
def test_create_site(self):
data = {
@ -203,7 +186,7 @@ class SiteTest(APITestCase):
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Site.objects.count(), 6)
self.assertEqual(Site.objects.count(), 4)
site4 = Site.objects.get(pk=response.data['id'])
self.assertEqual(site4.name, data['name'])
self.assertEqual(site4.slug, data['slug'])
@ -236,7 +219,7 @@ class SiteTest(APITestCase):
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Site.objects.count(), 8)
self.assertEqual(Site.objects.count(), 6)
self.assertEqual(response.data[0]['name'], data[0]['name'])
self.assertEqual(response.data[1]['name'], data[1]['name'])
self.assertEqual(response.data[2]['name'], data[2]['name'])
@ -253,7 +236,7 @@ class SiteTest(APITestCase):
response = self.client.put(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_200_OK)
self.assertEqual(Site.objects.count(), 5)
self.assertEqual(Site.objects.count(), 3)
site1 = Site.objects.get(pk=response.data['id'])
self.assertEqual(site1.name, data['name'])
self.assertEqual(site1.slug, data['slug'])
@ -265,7 +248,7 @@ class SiteTest(APITestCase):
response = self.client.delete(url, **self.header)
self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
self.assertEqual(Site.objects.count(), 4)
self.assertEqual(Site.objects.count(), 2)
class RackGroupTest(APITestCase):
@ -1769,8 +1752,7 @@ class DeviceTest(APITestCase):
super().setUp()
region = Region.objects.create(name='Test Region', slug='test-region-1')
self.site1 = Site.objects.create(region=region, name='Test Site 1', slug='test-site-1')
self.site1 = Site.objects.create(name='Test Site 1', slug='test-site-1')
self.site2 = Site.objects.create(name='Test Site 2', slug='test-site-2')
manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1')
self.devicetype1 = DeviceType.objects.create(
@ -1818,20 +1800,6 @@ class DeviceTest(APITestCase):
'B': 2
}
)
self.device_non_region1 = Device.objects.create(
device_type=self.devicetype1,
device_role=self.devicerole1,
name='Test Device Null Region1',
site=self.site2,
cluster=self.cluster1
)
self.device_non_region2 = Device.objects.create(
device_type=self.devicetype1,
device_role=self.devicerole1,
name='Test Device Null Region2',
site=self.site2,
cluster=self.cluster1
)
def test_get_device(self):
@ -1847,7 +1815,7 @@ class DeviceTest(APITestCase):
url = reverse('dcim-api:device-list')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 6)
self.assertEqual(response.data['count'], 4)
def test_list_devices_brief(self):
@ -1859,20 +1827,6 @@ class DeviceTest(APITestCase):
['display_name', 'id', 'name', 'url']
)
def test_list_devices_null_region(self):
url = reverse('dcim-api:device-list')
response = self.client.get('{}?region=null'.format(url), **self.header)
self.assertEqual(response.data['count'], 2)
def test_list_devices_multiple_regions(self):
url = reverse('dcim-api:device-list')
response = self.client.get('{}?region=null&region=test-region-1'.format(url), **self.header)
self.assertEqual(response.data['count'], 6)
def test_create_device(self):
data = {
@ -1887,7 +1841,7 @@ class DeviceTest(APITestCase):
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Device.objects.count(), 7)
self.assertEqual(Device.objects.count(), 5)
device4 = Device.objects.get(pk=response.data['id'])
self.assertEqual(device4.device_type_id, data['device_type'])
self.assertEqual(device4.device_role_id, data['device_role'])
@ -1922,7 +1876,7 @@ class DeviceTest(APITestCase):
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Device.objects.count(), 9)
self.assertEqual(Device.objects.count(), 7)
self.assertEqual(response.data[0]['name'], data[0]['name'])
self.assertEqual(response.data[1]['name'], data[1]['name'])
self.assertEqual(response.data[2]['name'], data[2]['name'])
@ -1946,7 +1900,7 @@ class DeviceTest(APITestCase):
response = self.client.put(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_200_OK)
self.assertEqual(Device.objects.count(), 6)
self.assertEqual(Device.objects.count(), 4)
device1 = Device.objects.get(pk=response.data['id'])
self.assertEqual(device1.device_type_id, data['device_type'])
self.assertEqual(device1.device_role_id, data['device_role'])
@ -1961,7 +1915,7 @@ class DeviceTest(APITestCase):
response = self.client.delete(url, **self.header)
self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
self.assertEqual(Device.objects.count(), 5)
self.assertEqual(Device.objects.count(), 3)
def test_config_context_included_by_default_in_list_view(self):