diff --git a/netbox/ipam/tests/test_api.py b/netbox/ipam/tests/test_api.py index c20ef6158..06c1f41df 100644 --- a/netbox/ipam/tests/test_api.py +++ b/netbox/ipam/tests/test_api.py @@ -7,7 +7,7 @@ from rest_framework import status from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site from ipam.choices import * from ipam.models import Aggregate, IPAddress, Prefix, RIR, Role, Service, VLAN, VLANGroup, VRF -from utilities.testing import APITestCase, disable_warnings +from utilities.testing import APITestCase, APIViewTestCases, disable_warnings class AppTest(APITestCase): @@ -20,536 +20,158 @@ class AppTest(APITestCase): self.assertEqual(response.status_code, 200) -class VRFTest(APITestCase): +class VRFTest(APIViewTestCases.APIViewTestCase): + model = VRF + brief_fields = ['id', 'name', 'prefix_count', 'rd', 'url'] + create_data = [ + { + 'name': 'VRF 4', + 'rd': '65000:4', + }, + { + 'name': 'VRF 5', + 'rd': '65000:5', + }, + { + 'name': 'VRF 6', + 'rd': '65000:6', + }, + ] - def setUp(self): + @classmethod + def setUpTestData(cls): - super().setUp() - - self.vrf1 = VRF.objects.create(name='Test VRF 1', rd='65000:1') - self.vrf2 = VRF.objects.create(name='Test VRF 2', rd='65000:2') - self.vrf3 = VRF.objects.create(name='Test VRF 3') # No RD - - def test_get_vrf(self): - - url = reverse('ipam-api:vrf-detail', kwargs={'pk': self.vrf1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.vrf1.name) - - def test_list_vrfs(self): - - url = reverse('ipam-api:vrf-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_vrfs_brief(self): - - url = reverse('ipam-api:vrf-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['id', 'name', 'prefix_count', 'rd', 'url'] + vrfs = ( + VRF(name='VRF 1', rd='65000:1'), + VRF(name='VRF 2', rd='65000:2'), + VRF(name='VRF 3'), # No RD ) - - def test_create_vrf(self): - - data_list = [ - # VRF with RD - { - 'name': 'Test VRF 4', - 'rd': '65000:4', - }, - # VRF without RD - { - 'name': 'Test VRF 5', - } - ] - - url = reverse('ipam-api:vrf-list') - - for data in data_list: - response = self.client.post(url, data, format='json', **self.header) - self.assertHttpStatus(response, status.HTTP_201_CREATED) - vrf = VRF.objects.get(pk=response.data['id']) - self.assertEqual(vrf.name, data['name']) - self.assertEqual(vrf.rd, data['rd'] if 'rd' in data else None) - - def test_create_vrf_bulk(self): - - data = [ - { - 'name': 'Test VRF 4', - 'rd': '65000:4', - }, - { - 'name': 'Test VRF 5', - 'rd': '65000:5', - }, - { - 'name': 'Test VRF 6', - 'rd': '65000:6', - }, - ] - - url = reverse('ipam-api:vrf-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(VRF.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']) - - def test_update_vrf(self): - - data = { - 'name': 'Test VRF X', - 'rd': '65000:99', - } - - url = reverse('ipam-api:vrf-detail', kwargs={'pk': self.vrf1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(VRF.objects.count(), 3) - vrf1 = VRF.objects.get(pk=response.data['id']) - self.assertEqual(vrf1.name, data['name']) - self.assertEqual(vrf1.rd, data['rd']) - - def test_delete_vrf(self): - - url = reverse('ipam-api:vrf-detail', kwargs={'pk': self.vrf1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(VRF.objects.count(), 2) + VRF.objects.bulk_create(vrfs) -class RIRTest(APITestCase): +class RIRTest(APIViewTestCases.APIViewTestCase): + model = RIR + brief_fields = ['aggregate_count', 'id', 'name', 'slug', 'url'] + create_data = [ + { + 'name': 'RIR 4', + 'slug': 'rir-4', + }, + { + 'name': 'RIR 5', + 'slug': 'rir-5', + }, + { + 'name': 'RIR 6', + 'slug': 'rir-6', + }, + ] - def setUp(self): + @classmethod + def setUpTestData(cls): - super().setUp() - - self.rir1 = RIR.objects.create(name='Test RIR 1', slug='test-rir-1') - self.rir2 = RIR.objects.create(name='Test RIR 2', slug='test-rir-2') - self.rir3 = RIR.objects.create(name='Test RIR 3', slug='test-rir-3') - - def test_get_rir(self): - - url = reverse('ipam-api:rir-detail', kwargs={'pk': self.rir1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.rir1.name) - - def test_list_rirs(self): - - url = reverse('ipam-api:rir-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_rirs_brief(self): - - url = reverse('ipam-api:rir-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['aggregate_count', 'id', 'name', 'slug', 'url'] + rirs = ( + RIR(name='RIR 1', slug='rir-1'), + RIR(name='RIR 2', slug='rir-2'), + RIR(name='RIR 3', slug='rir-3'), ) - - def test_create_rir(self): - - data = { - 'name': 'Test RIR 4', - 'slug': 'test-rir-4', - } - - url = reverse('ipam-api:rir-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(RIR.objects.count(), 4) - rir4 = RIR.objects.get(pk=response.data['id']) - self.assertEqual(rir4.name, data['name']) - self.assertEqual(rir4.slug, data['slug']) - - def test_create_rir_bulk(self): - - data = [ - { - 'name': 'Test RIR 4', - 'slug': 'test-rir-4', - }, - { - 'name': 'Test RIR 5', - 'slug': 'test-rir-5', - }, - { - 'name': 'Test RIR 6', - 'slug': 'test-rir-6', - }, - ] - - url = reverse('ipam-api:rir-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(RIR.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']) - - def test_update_rir(self): - - data = { - 'name': 'Test RIR X', - 'slug': 'test-rir-x', - } - - url = reverse('ipam-api:rir-detail', kwargs={'pk': self.rir1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(RIR.objects.count(), 3) - rir1 = RIR.objects.get(pk=response.data['id']) - self.assertEqual(rir1.name, data['name']) - self.assertEqual(rir1.slug, data['slug']) - - def test_delete_rir(self): - - url = reverse('ipam-api:rir-detail', kwargs={'pk': self.rir1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(RIR.objects.count(), 2) + RIR.objects.bulk_create(rirs) -class AggregateTest(APITestCase): +class AggregateTest(APIViewTestCases.APIViewTestCase): + model = Aggregate + brief_fields = ['family', 'id', 'prefix', 'url'] - def setUp(self): + @classmethod + def setUpTestData(cls): - super().setUp() - - self.rir1 = RIR.objects.create(name='Test RIR 1', slug='test-rir-1') - self.rir2 = RIR.objects.create(name='Test RIR 2', slug='test-rir-2') - self.aggregate1 = Aggregate.objects.create(prefix=IPNetwork('10.0.0.0/8'), rir=self.rir1) - self.aggregate2 = Aggregate.objects.create(prefix=IPNetwork('172.16.0.0/12'), rir=self.rir1) - self.aggregate3 = Aggregate.objects.create(prefix=IPNetwork('192.168.0.0/16'), rir=self.rir1) - - def test_get_aggregate(self): - - url = reverse('ipam-api:aggregate-detail', kwargs={'pk': self.aggregate1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['family']['value'], 4) - self.assertEqual(response.data['prefix'], str(self.aggregate1.prefix)) - - def test_list_aggregates(self): - - url = reverse('ipam-api:aggregate-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_aggregates_brief(self): - - url = reverse('ipam-api:aggregate-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['family', 'id', 'prefix', 'url'] + rirs = ( + RIR(name='RIR 1', slug='rir-1'), + RIR(name='RIR 2', slug='rir-2'), ) + RIR.objects.bulk_create(rirs) - def test_create_aggregate(self): + aggregates = ( + Aggregate(prefix=IPNetwork('10.0.0.0/8'), rir=rirs[0]), + Aggregate(prefix=IPNetwork('172.16.0.0/12'), rir=rirs[0]), + Aggregate(prefix=IPNetwork('192.168.0.0/16'), rir=rirs[0]), + ) + Aggregate.objects.bulk_create(aggregates) - data = { - 'prefix': '192.0.2.0/24', - 'rir': self.rir1.pk, - } - - url = reverse('ipam-api:aggregate-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Aggregate.objects.count(), 4) - aggregate4 = Aggregate.objects.get(pk=response.data['id']) - self.assertEqual(str(aggregate4.prefix), data['prefix']) - self.assertEqual(aggregate4.rir_id, data['rir']) - - def test_create_aggregate_bulk(self): - - data = [ + cls.create_data = [ { 'prefix': '100.0.0.0/8', - 'rir': self.rir1.pk, + 'rir': rirs[1].pk, }, { 'prefix': '101.0.0.0/8', - 'rir': self.rir1.pk, + 'rir': rirs[1].pk, }, { 'prefix': '102.0.0.0/8', - 'rir': self.rir1.pk, + 'rir': rirs[1].pk, }, ] - url = reverse('ipam-api:aggregate-list') - response = self.client.post(url, data, format='json', **self.header) - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Aggregate.objects.count(), 6) - self.assertEqual(response.data[0]['prefix'], data[0]['prefix']) - self.assertEqual(response.data[1]['prefix'], data[1]['prefix']) - self.assertEqual(response.data[2]['prefix'], data[2]['prefix']) +class RoleTest(APIViewTestCases.APIViewTestCase): + model = Role + brief_fields = ['id', 'name', 'prefix_count', 'slug', 'url', 'vlan_count'] + create_data = [ + { + 'name': 'Role 4', + 'slug': 'role-4', + }, + { + 'name': 'Role 5', + 'slug': 'role-5', + }, + { + 'name': 'Role 6', + 'slug': 'role-6', + }, + ] - def test_update_aggregate(self): + @classmethod + def setUpTestData(cls): - data = { - 'prefix': '11.0.0.0/8', - 'rir': self.rir2.pk, - } - - url = reverse('ipam-api:aggregate-detail', kwargs={'pk': self.aggregate1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(Aggregate.objects.count(), 3) - aggregate1 = Aggregate.objects.get(pk=response.data['id']) - self.assertEqual(str(aggregate1.prefix), data['prefix']) - self.assertEqual(aggregate1.rir_id, data['rir']) - - def test_delete_aggregate(self): - - url = reverse('ipam-api:aggregate-detail', kwargs={'pk': self.aggregate1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(Aggregate.objects.count(), 2) - - -class RoleTest(APITestCase): - - def setUp(self): - - super().setUp() - - self.role1 = Role.objects.create(name='Test Role 1', slug='test-role-1') - self.role2 = Role.objects.create(name='Test Role 2', slug='test-role-2') - self.role3 = Role.objects.create(name='Test Role 3', slug='test-role-3') - - def test_get_role(self): - - url = reverse('ipam-api:role-detail', kwargs={'pk': self.role1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.role1.name) - - def test_list_roles(self): - - url = reverse('ipam-api:role-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_roles_brief(self): - - url = reverse('ipam-api:role-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['id', 'name', 'prefix_count', 'slug', 'url', 'vlan_count'] + roles = ( + Role(name='Role 1', slug='role-1'), + Role(name='Role 2', slug='role-2'), + Role(name='Role 3', slug='role-3'), ) - - def test_create_role(self): - - data = { - 'name': 'Test Role 4', - 'slug': 'test-role-4', - } - - url = reverse('ipam-api:role-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Role.objects.count(), 4) - role4 = Role.objects.get(pk=response.data['id']) - self.assertEqual(role4.name, data['name']) - self.assertEqual(role4.slug, data['slug']) - - def test_create_role_bulk(self): - - data = [ - { - 'name': 'Test Role 4', - 'slug': 'test-role-4', - }, - { - 'name': 'Test Role 5', - 'slug': 'test-role-5', - }, - { - 'name': 'Test Role 6', - 'slug': 'test-role-6', - }, - ] - - url = reverse('ipam-api:role-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Role.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']) - - def test_update_role(self): - - data = { - 'name': 'Test Role X', - 'slug': 'test-role-x', - } - - url = reverse('ipam-api:role-detail', kwargs={'pk': self.role1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(Role.objects.count(), 3) - role1 = Role.objects.get(pk=response.data['id']) - self.assertEqual(role1.name, data['name']) - self.assertEqual(role1.slug, data['slug']) - - def test_delete_role(self): - - url = reverse('ipam-api:role-detail', kwargs={'pk': self.role1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(Role.objects.count(), 2) + Role.objects.bulk_create(roles) -class PrefixTest(APITestCase): - - def setUp(self): - - super().setUp() - - self.site1 = Site.objects.create(name='Test Site 1', slug='test-site-1') - self.vrf1 = VRF.objects.create(name='Test VRF 1', rd='65000:1') - self.vlan1 = VLAN.objects.create(vid=1, name='Test VLAN 1') - self.role1 = Role.objects.create(name='Test Role 1', slug='test-role-1') - self.prefix1 = Prefix.objects.create(prefix=IPNetwork('192.168.1.0/24')) - self.prefix2 = Prefix.objects.create(prefix=IPNetwork('192.168.2.0/24')) - self.prefix3 = Prefix.objects.create(prefix=IPNetwork('192.168.3.0/24')) - - def test_get_prefix(self): - - url = reverse('ipam-api:prefix-detail', kwargs={'pk': self.prefix1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['family']['value'], 4) - self.assertEqual(response.data['prefix'], str(self.prefix1.prefix)) - - def test_list_prefixes(self): - - url = reverse('ipam-api:prefix-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_prefixes_brief(self): - - url = reverse('ipam-api:prefix-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['family', 'id', 'prefix', 'url'] - ) - - def test_create_prefix(self): - - data = { +class PrefixTest(APIViewTestCases.APIViewTestCase): + model = Prefix + brief_fields = ['family', 'id', 'prefix', 'url'] + create_data = [ + { 'prefix': '192.168.4.0/24', - 'site': self.site1.pk, - 'vrf': self.vrf1.pk, - 'vlan': self.vlan1.pk, - 'role': self.role1.pk, - } + }, + { + 'prefix': '192.168.5.0/24', + }, + { + 'prefix': '192.168.6.0/24', + }, + ] - url = reverse('ipam-api:prefix-list') - response = self.client.post(url, data, format='json', **self.header) + @classmethod + def setUpTestData(cls): - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Prefix.objects.count(), 4) - prefix4 = Prefix.objects.get(pk=response.data['id']) - self.assertEqual(str(prefix4.prefix), data['prefix']) - self.assertEqual(prefix4.site_id, data['site']) - self.assertEqual(prefix4.vrf_id, data['vrf']) - self.assertEqual(prefix4.vlan_id, data['vlan']) - self.assertEqual(prefix4.role_id, data['role']) - - def test_create_prefix_bulk(self): - - data = [ - { - 'prefix': '10.0.1.0/24', - }, - { - 'prefix': '10.0.2.0/24', - }, - { - 'prefix': '10.0.3.0/24', - }, - ] - - url = reverse('ipam-api:prefix-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Prefix.objects.count(), 6) - self.assertEqual(response.data[0]['prefix'], data[0]['prefix']) - self.assertEqual(response.data[1]['prefix'], data[1]['prefix']) - self.assertEqual(response.data[2]['prefix'], data[2]['prefix']) - - def test_update_prefix(self): - - data = { - 'prefix': '192.168.99.0/24', - 'site': self.site1.pk, - 'vrf': self.vrf1.pk, - 'vlan': self.vlan1.pk, - 'role': self.role1.pk, - } - - url = reverse('ipam-api:prefix-detail', kwargs={'pk': self.prefix1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(Prefix.objects.count(), 3) - prefix1 = Prefix.objects.get(pk=response.data['id']) - self.assertEqual(str(prefix1.prefix), data['prefix']) - self.assertEqual(prefix1.site_id, data['site']) - self.assertEqual(prefix1.vrf_id, data['vrf']) - self.assertEqual(prefix1.vlan_id, data['vlan']) - self.assertEqual(prefix1.role_id, data['role']) - - def test_delete_prefix(self): - - url = reverse('ipam-api:prefix-detail', kwargs={'pk': self.prefix1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(Prefix.objects.count(), 2) + prefixes = ( + Prefix(prefix=IPNetwork('192.168.1.0/24')), + Prefix(prefix=IPNetwork('192.168.2.0/24')), + Prefix(prefix=IPNetwork('192.168.3.0/24')), + ) + Prefix.objects.bulk_create(prefixes) def test_list_available_prefixes(self): - + """ + Test retrieval of all available prefixes within a parent prefix. + """ prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/24')) Prefix.objects.create(prefix=IPNetwork('192.0.2.64/26')) Prefix.objects.create(prefix=IPNetwork('192.0.2.192/27')) @@ -562,7 +184,9 @@ class PrefixTest(APITestCase): self.assertEqual(p['prefix'], available_prefixes[i]) def test_create_single_available_prefix(self): - + """ + Test retrieval of the first available prefix within a parent prefix. + """ vrf = VRF.objects.create(name='Test VRF 1', rd='1234') prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/28'), vrf=vrf, is_pool=True) url = reverse('ipam-api:prefix-available-prefixes', kwargs={'pk': prefix.pk}) @@ -596,7 +220,9 @@ class PrefixTest(APITestCase): self.assertIn('prefix_length', response.data[0]) def test_create_multiple_available_prefixes(self): - + """ + Test the creation of available prefixes within a parent prefix. + """ prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/28'), is_pool=True) url = reverse('ipam-api:prefix-available-prefixes', kwargs={'pk': prefix.pk}) @@ -622,7 +248,9 @@ class PrefixTest(APITestCase): self.assertEqual(len(response.data), 4) def test_list_available_ips(self): - + """ + Test retrieval of all available IP addresses within a parent prefix. + """ prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/29'), is_pool=True) url = reverse('ipam-api:prefix-available-ips', kwargs={'pk': prefix.pk}) @@ -637,7 +265,9 @@ class PrefixTest(APITestCase): self.assertEqual(len(response.data), 6) # 8 - 2 because prefix.is_pool = False def test_create_single_available_ip(self): - + """ + Test retrieval of the first available IP address within a parent prefix. + """ vrf = VRF.objects.create(name='Test VRF 1', rd='1234') prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/30'), vrf=vrf, is_pool=True) url = reverse('ipam-api:prefix-available-ips', kwargs={'pk': prefix.pk}) @@ -658,7 +288,9 @@ class PrefixTest(APITestCase): self.assertIn('detail', response.data) def test_create_multiple_available_ips(self): - + """ + Test the creation of available IP addresses within a parent prefix. + """ prefix = Prefix.objects.create(prefix=IPNetwork('192.0.2.0/29'), is_pool=True) url = reverse('ipam-api:prefix-available-ips', kwargs={'pk': prefix.pk}) @@ -679,317 +311,107 @@ class PrefixTest(APITestCase): self.assertEqual(len(response.data), 8) -class IPAddressTest(APITestCase): - - def setUp(self): - - super().setUp() - - self.vrf1 = VRF.objects.create(name='Test VRF 1', rd='65000:1') - self.ipaddress1 = IPAddress.objects.create(address=IPNetwork('192.168.0.1/24')) - self.ipaddress2 = IPAddress.objects.create(address=IPNetwork('192.168.0.2/24')) - self.ipaddress3 = IPAddress.objects.create(address=IPNetwork('192.168.0.3/24')) - - def test_get_ipaddress(self): - - url = reverse('ipam-api:ipaddress-detail', kwargs={'pk': self.ipaddress1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['family']['value'], 4) - self.assertEqual(response.data['address'], str(self.ipaddress1.address)) - - def test_list_ipaddresss(self): - - url = reverse('ipam-api:ipaddress-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_ipaddresses_brief(self): - - url = reverse('ipam-api:ipaddress-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['address', 'family', 'id', 'url'] - ) - - def test_create_ipaddress(self): - - data = { +class IPAddressTest(APIViewTestCases.APIViewTestCase): + model = IPAddress + brief_fields = ['address', 'family', 'id', 'url'] + create_data = [ + { 'address': '192.168.0.4/24', - 'vrf': self.vrf1.pk, - } + }, + { + 'address': '192.168.0.5/24', + }, + { + 'address': '192.168.0.6/24', + }, + ] - url = reverse('ipam-api:ipaddress-list') - response = self.client.post(url, data, format='json', **self.header) + @classmethod + def setUpTestData(cls): - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(IPAddress.objects.count(), 4) - ipaddress4 = IPAddress.objects.get(pk=response.data['id']) - self.assertEqual(str(ipaddress4.address), data['address']) - self.assertEqual(ipaddress4.vrf_id, data['vrf']) - - def test_create_ipaddress_bulk(self): - - data = [ - { - 'address': '192.168.0.4/24', - }, - { - 'address': '192.168.0.5/24', - }, - { - 'address': '192.168.0.6/24', - }, - ] - - url = reverse('ipam-api:ipaddress-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(IPAddress.objects.count(), 6) - self.assertEqual(response.data[0]['address'], data[0]['address']) - self.assertEqual(response.data[1]['address'], data[1]['address']) - self.assertEqual(response.data[2]['address'], data[2]['address']) - - def test_update_ipaddress(self): - - data = { - 'address': '192.168.0.99/24', - 'vrf': self.vrf1.pk, - } - - url = reverse('ipam-api:ipaddress-detail', kwargs={'pk': self.ipaddress1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(IPAddress.objects.count(), 3) - ipaddress1 = IPAddress.objects.get(pk=response.data['id']) - self.assertEqual(str(ipaddress1.address), data['address']) - self.assertEqual(ipaddress1.vrf_id, data['vrf']) - - def test_delete_ipaddress(self): - - url = reverse('ipam-api:ipaddress-detail', kwargs={'pk': self.ipaddress1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(IPAddress.objects.count(), 2) - - -class VLANGroupTest(APITestCase): - - def setUp(self): - - super().setUp() - - self.site1 = Site.objects.create(name='Test Site 1', slug='test-site-1') - self.vlangroup1 = VLANGroup.objects.create(name='Test VLAN Group 1', slug='test-vlan-group-1') - self.vlangroup2 = VLANGroup.objects.create(name='Test VLAN Group 2', slug='test-vlan-group-2') - self.vlangroup3 = VLANGroup.objects.create(name='Test VLAN Group 3', slug='test-vlan-group-3') - - def test_get_vlangroup(self): - - url = reverse('ipam-api:vlangroup-detail', kwargs={'pk': self.vlangroup1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.vlangroup1.name) - - def test_list_vlangroups(self): - - url = reverse('ipam-api:vlangroup-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_vlangroups_brief(self): - - url = reverse('ipam-api:vlangroup-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['id', 'name', 'slug', 'url', 'vlan_count'] + ip_addresses = ( + IPAddress(address=IPNetwork('192.168.0.1/24')), + IPAddress(address=IPNetwork('192.168.0.2/24')), + IPAddress(address=IPNetwork('192.168.0.3/24')), ) - - def test_create_vlangroup(self): - - data = { - 'name': 'Test VLAN Group 4', - 'slug': 'test-vlan-group-4', - 'site': self.site1.pk, - } - - url = reverse('ipam-api:vlangroup-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(VLANGroup.objects.count(), 4) - vlangroup4 = VLANGroup.objects.get(pk=response.data['id']) - self.assertEqual(vlangroup4.name, data['name']) - self.assertEqual(vlangroup4.slug, data['slug']) - - def test_create_vlangroup_bulk(self): - - data = [ - { - 'name': 'Test VLAN Group 4', - 'slug': 'test-vlan-group-4', - }, - { - 'name': 'Test VLAN Group 5', - 'slug': 'test-vlan-group-5', - }, - { - 'name': 'Test VLAN Group 6', - 'slug': 'test-vlan-group-6', - }, - ] - - url = reverse('ipam-api:vlangroup-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(VLANGroup.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']) - - def test_update_vlangroup(self): - - data = { - 'name': 'Test VLAN Group X', - 'slug': 'test-vlan-group-x', - } - - url = reverse('ipam-api:vlangroup-detail', kwargs={'pk': self.vlangroup1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(VLANGroup.objects.count(), 3) - vlangroup1 = VLANGroup.objects.get(pk=response.data['id']) - self.assertEqual(vlangroup1.name, data['name']) - self.assertEqual(vlangroup1.slug, data['slug']) - - def test_delete_vlangroup(self): - - url = reverse('ipam-api:vlangroup-detail', kwargs={'pk': self.vlangroup1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(VLANGroup.objects.count(), 2) + IPAddress.objects.bulk_create(ip_addresses) -class VLANTest(APITestCase): +class VLANGroupTest(APIViewTestCases.APIViewTestCase): + model = VLANGroup + brief_fields = ['id', 'name', 'slug', 'url', 'vlan_count'] + create_data = [ + { + 'name': 'VLAN Group 4', + 'slug': 'vlan-group-4', + }, + { + 'name': 'VLAN Group 5', + 'slug': 'vlan-group-5', + }, + { + 'name': 'VLAN Group 6', + 'slug': 'vlan-group-6', + }, + ] - def setUp(self): + @classmethod + def setUpTestData(cls): - super().setUp() - - self.group1 = VLANGroup.objects.create(name='Test VLAN Group 1', slug='test-vlan-group-1') - self.vlan1 = VLAN.objects.create(vid=1, name='Test VLAN 1') - self.vlan2 = VLAN.objects.create(vid=2, name='Test VLAN 2') - self.vlan3 = VLAN.objects.create(vid=3, name='Test VLAN 3') - self.prefix1 = Prefix.objects.create(prefix=IPNetwork('192.168.1.0/24')) - - def test_get_vlan(self): - - url = reverse('ipam-api:vlan-detail', kwargs={'pk': self.vlan1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.vlan1.name) - - def test_list_vlans(self): - - url = reverse('ipam-api:vlan-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_list_vlans_brief(self): - - url = reverse('ipam-api:vlan-list') - response = self.client.get('{}?brief=1'.format(url), **self.header) - - self.assertEqual( - sorted(response.data['results'][0]), - ['display_name', 'id', 'name', 'url', 'vid'] + vlan_groups = ( + VLANGroup(name='VLAN Group 1', slug='vlan-group-1'), + VLANGroup(name='VLAN Group 2', slug='vlan-group-2'), + VLANGroup(name='VLAN Group 3', slug='vlan-group-3'), ) + VLANGroup.objects.bulk_create(vlan_groups) - def test_create_vlan(self): - data = { - 'vid': 4, - 'name': 'Test VLAN 4', - 'group': self.group1.pk, - } +class VLANTest(APIViewTestCases.APIViewTestCase): + model = VLAN + brief_fields = ['display_name', 'id', 'name', 'url', 'vid'] - url = reverse('ipam-api:vlan-list') - response = self.client.post(url, data, format='json', **self.header) + @classmethod + def setUpTestData(cls): - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(VLAN.objects.count(), 4) - vlan4 = VLAN.objects.get(pk=response.data['id']) - self.assertEqual(vlan4.vid, data['vid']) - self.assertEqual(vlan4.name, data['name']) + vlan_groups = ( + VLANGroup(name='VLAN Group 1', slug='vlan-group-1'), + VLANGroup(name='VLAN Group 2', slug='vlan-group-2'), + ) + VLANGroup.objects.bulk_create(vlan_groups) - def test_create_vlan_bulk(self): + vlans = ( + VLAN(name='VLAN 1', vid=1, group=vlan_groups[0]), + VLAN(name='VLAN 2', vid=2, group=vlan_groups[0]), + VLAN(name='VLAN 3', vid=3, group=vlan_groups[0]), + ) + VLAN.objects.bulk_create(vlans) - data = [ + cls.create_data = [ { 'vid': 4, - 'name': 'Test VLAN 4', + 'name': 'VLAN 4', + 'group': vlan_groups[1].pk, }, { 'vid': 5, - 'name': 'Test VLAN 5', + 'name': 'VLAN 5', + 'group': vlan_groups[1].pk, }, { 'vid': 6, - 'name': 'Test VLAN 6', + 'name': 'VLAN 6', + 'group': vlan_groups[1].pk, }, ] - url = reverse('ipam-api:vlan-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(VLAN.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']) - - def test_update_vlan(self): - - data = { - 'vid': 99, - 'name': 'Test VLAN X', - } - - url = reverse('ipam-api:vlan-detail', kwargs={'pk': self.vlan1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(VLAN.objects.count(), 3) - vlan1 = VLAN.objects.get(pk=response.data['id']) - self.assertEqual(vlan1.vid, data['vid']) - self.assertEqual(vlan1.name, data['name']) - - def test_delete_vlan(self): - - url = reverse('ipam-api:vlan-detail', kwargs={'pk': self.vlan1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(VLAN.objects.count(), 2) - def test_delete_vlan_with_prefix(self): - self.prefix1.vlan = self.vlan1 - self.prefix1.save() + """ + Attempt and fail to delete a VLAN with a Prefix assigned to it. + """ + vlan = VLAN.objects.first() + Prefix.objects.create(prefix=IPNetwork('192.0.2.0/24'), vlan=vlan) - url = reverse('ipam-api:vlan-detail', kwargs={'pk': self.vlan1.pk}) + url = reverse('ipam-api:vlan-detail', kwargs={'pk': vlan.pk}) with disable_warnings('django.request'): response = self.client.delete(url, **self.header) @@ -1000,125 +422,46 @@ class VLANTest(APITestCase): self.assertTrue(content['detail'].startswith('Unable to delete object.')) -class ServiceTest(APITestCase): +class ServiceTest(APIViewTestCases.APIViewTestCase): + model = Service - def setUp(self): + @classmethod + def setUpTestData(cls): + site = Site.objects.create(name='Site 1', slug='site-1') + manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') + devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1') + devicerole = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1') - super().setUp() - - site = Site.objects.create(name='Test Site 1', slug='test-site-1') - manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') - devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Test Device Type 1') - devicerole = DeviceRole.objects.create(name='Test Device Role 1', slug='test-device-role-1') - self.device1 = Device.objects.create( - name='Test Device 1', site=site, device_type=devicetype, device_role=devicerole + devices = ( + Device(name='Device 1', site=site, device_type=devicetype, device_role=devicerole), + Device(name='Device 2', site=site, device_type=devicetype, device_role=devicerole), ) - self.device2 = Device.objects.create( - name='Test Device 2', site=site, device_type=devicetype, device_role=devicerole - ) - self.service1 = Service.objects.create( - device=self.device1, name='Test Service 1', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=1 - ) - self.service1 = Service.objects.create( - device=self.device1, name='Test Service 2', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=2 - ) - self.service1 = Service.objects.create( - device=self.device1, name='Test Service 3', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=3 + Device.objects.bulk_create(devices) + + services = ( + Service(device=devices[0], name='Service 1', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=1), + Service(device=devices[0], name='Service 2', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=2), + Service(device=devices[0], name='Service 3', protocol=ServiceProtocolChoices.PROTOCOL_TCP, port=3), ) + Service.objects.bulk_create(services) - def test_get_service(self): - - url = reverse('ipam-api:service-detail', kwargs={'pk': self.service1.pk}) - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['name'], self.service1.name) - - def test_list_services(self): - - url = reverse('ipam-api:service-list') - response = self.client.get(url, **self.header) - - self.assertEqual(response.data['count'], 3) - - def test_create_service(self): - - data = { - 'device': self.device1.pk, - 'name': 'Test Service 4', - 'protocol': ServiceProtocolChoices.PROTOCOL_TCP, - 'port': 4, - 'tags': ['Foo', 'Bar'], - } - - url = reverse('ipam-api:service-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Service.objects.count(), 4) - service4 = Service.objects.get(pk=response.data['id']) - self.assertEqual(service4.device_id, data['device']) - self.assertEqual(service4.name, data['name']) - self.assertEqual(service4.protocol, data['protocol']) - self.assertEqual(service4.port, data['port']) - tags = [tag.name for tag in service4.tags.all()] - self.assertEqual(sorted(tags), sorted(data['tags'])) - - def test_create_service_bulk(self): - - data = [ + cls.create_data = [ { - 'device': self.device1.pk, - 'name': 'Test Service 4', + 'device': devices[1].pk, + 'name': 'Service 4', 'protocol': ServiceProtocolChoices.PROTOCOL_TCP, 'port': 4, }, { - 'device': self.device1.pk, - 'name': 'Test Service 5', + 'device': devices[1].pk, + 'name': 'Service 5', 'protocol': ServiceProtocolChoices.PROTOCOL_TCP, 'port': 5, }, { - 'device': self.device1.pk, - 'name': 'Test Service 6', + 'device': devices[1].pk, + 'name': 'Service 6', 'protocol': ServiceProtocolChoices.PROTOCOL_TCP, 'port': 6, }, ] - - url = reverse('ipam-api:service-list') - response = self.client.post(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_201_CREATED) - self.assertEqual(Service.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']) - - def test_update_service(self): - - data = { - 'device': self.device2.pk, - 'name': 'Test Service X', - 'protocol': ServiceProtocolChoices.PROTOCOL_UDP, - 'port': 99, - } - - url = reverse('ipam-api:service-detail', kwargs={'pk': self.service1.pk}) - response = self.client.put(url, data, format='json', **self.header) - - self.assertHttpStatus(response, status.HTTP_200_OK) - self.assertEqual(Service.objects.count(), 3) - service1 = Service.objects.get(pk=response.data['id']) - self.assertEqual(service1.device_id, data['device']) - self.assertEqual(service1.name, data['name']) - self.assertEqual(service1.protocol, data['protocol']) - self.assertEqual(service1.port, data['port']) - - def test_delete_service(self): - - url = reverse('ipam-api:service-detail', kwargs={'pk': self.service1.pk}) - response = self.client.delete(url, **self.header) - - self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT) - self.assertEqual(Service.objects.count(), 2)