diff --git a/netbox/ipam/api/serializers.py b/netbox/ipam/api/serializers.py index c33694d88..8189a8fb3 100644 --- a/netbox/ipam/api/serializers.py +++ b/netbox/ipam/api/serializers.py @@ -1,4 +1,5 @@ from rest_framework import serializers +from rest_framework.validators import UniqueTogetherValidator from dcim.api.serializers import NestedDeviceSerializer, InterfaceSerializer, NestedSiteSerializer from extras.api.customfields import CustomFieldModelSerializer @@ -99,7 +100,7 @@ class WritableAggregateSerializer(serializers.ModelSerializer): class Meta: model = Aggregate - fields = ['id', 'family', 'prefix', 'rir', 'date_added', 'description'] + fields = ['id', 'prefix', 'rir', 'date_added', 'description'] # @@ -127,6 +128,18 @@ class WritableVLANGroupSerializer(serializers.ModelSerializer): class Meta: model = VLANGroup fields = ['id', 'name', 'slug', 'site'] + validators = [] + + def validate(self, data): + + # Validate uniqueness of name and slug if a site has been assigned. + if data.get('site', None): + for field in ['name', 'slug']: + validator = UniqueTogetherValidator(queryset=VLAN.objects.all(), fields=('site', field)) + validator.set_context(self) + validator(data) + + return data # @@ -163,6 +176,18 @@ class WritableVLANSerializer(serializers.ModelSerializer): fields = [ 'id', 'site', 'group', 'vid', 'name', 'tenant', 'status', 'role', 'description', ] + validators = [] + + def validate(self, data): + + # Validate uniqueness of vid and name if a group has been assigned. + if data.get('group', None): + for field in ['vid', 'name']: + validator = UniqueTogetherValidator(queryset=VLAN.objects.all(), fields=('group', field)) + validator.set_context(self) + validator(data) + + return data # @@ -198,7 +223,7 @@ class WritablePrefixSerializer(serializers.ModelSerializer): class Meta: model = Prefix fields = [ - 'id', 'family', 'prefix', 'site', 'vrf', 'tenant', 'vlan', 'status', 'role', 'is_pool', 'description', + 'id', 'prefix', 'site', 'vrf', 'tenant', 'vlan', 'status', 'role', 'is_pool', 'description', ] @@ -235,7 +260,7 @@ class WritableIPAddressSerializer(serializers.ModelSerializer): class Meta: model = IPAddress - fields = ['id', 'family', 'address', 'vrf', 'tenant', 'status', 'interface', 'description', 'nat_inside'] + fields = ['id', 'address', 'vrf', 'tenant', 'status', 'interface', 'description', 'nat_inside'] # diff --git a/netbox/ipam/tests/test_api.py b/netbox/ipam/tests/test_api.py new file mode 100644 index 000000000..64d8c98a1 --- /dev/null +++ b/netbox/ipam/tests/test_api.py @@ -0,0 +1,659 @@ +from netaddr import IPNetwork + +from rest_framework import status +from rest_framework.test import APITestCase + +from django.contrib.auth.models import User +from django.urls import reverse + +from dcim.models import Device, DeviceRole, DeviceType, Manufacturer, Site +from ipam.models import ( + Aggregate, IPAddress, IP_PROTOCOL_TCP, IP_PROTOCOL_UDP, Prefix, RIR, Role, Service, VLAN, VLANGroup, VRF, +) +from users.models import Token + + +class VRFTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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', rd='65000:3') + + 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_create_vrf(self): + + data = { + 'name': 'Test VRF 4', + 'rd': '65000:4', + } + + url = reverse('ipam-api:vrf-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, status.HTTP_201_CREATED) + self.assertEqual(VRF.objects.count(), 4) + vrf4 = VRF.objects.get(pk=response.data['id']) + self.assertEqual(vrf4.name, data['name']) + self.assertEqual(vrf4.rd, data['rd']) + + 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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(VRF.objects.count(), 2) + + +class RIRTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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_create_rir(self): + + data = { + 'name': 'Test RIR 4', + 'slug': 'test-rir-4', + } + + url = reverse('ipam-api:rir-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(RIR.objects.count(), 2) + + +class AggregateTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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['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_create_aggregate(self): + + data = { + 'prefix': '192.0.2.0/24', + 'rir': self.rir1.pk, + } + + url = reverse('ipam-api:aggregate-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_update_aggregate(self): + + 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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(Aggregate.objects.count(), 2) + + +class RoleTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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_create_role(self): + + data = { + 'name': 'Test Role 4', + 'slug': 'test-role-4', + } + + url = reverse('ipam-api:role-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(Role.objects.count(), 2) + + +class PrefixTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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['prefix'], str(self.prefix1.prefix)) + + def test_list_prefixs(self): + + url = reverse('ipam-api:prefix-list') + response = self.client.get(url, **self.header) + + self.assertEqual(response.data['count'], 3) + + def test_create_prefix(self): + + data = { + 'prefix': '192.168.4.0/24', + 'site': self.site1.pk, + 'vrf': self.vrf1.pk, + 'vlan': self.vlan1.pk, + 'role': self.role1.pk, + } + + url = reverse('ipam-api:prefix-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(Prefix.objects.count(), 2) + + +class IPAddressTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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['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_create_ipaddress(self): + + data = { + 'address': '192.168.0.4/24', + 'vrf': self.vrf1.pk, + } + + url = reverse('ipam-api:ipaddress-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(IPAddress.objects.count(), 2) + + +class VLANGroupTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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_create_vlangroup(self): + + data = { + 'name': 'Test VLAN Group 4', + 'slug': 'test-vlan-group-4', + } + + url = reverse('ipam-api:vlangroup-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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_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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(VLANGroup.objects.count(), 2) + + +class VLANTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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') + + 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_create_vlan(self): + + data = { + 'vid': 4, + 'name': 'Test VLAN 4', + } + + url = reverse('ipam-api:vlan-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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']) + + 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, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(VLAN.objects.count(), 2) + + +class ServiceTest(APITestCase): + + def setUp(self): + + user = User.objects.create(username='testuser', is_superuser=True) + token = Token.objects.create(user=user) + self.header = {'HTTP_AUTHORIZATION': 'Token {}'.format(token.key)} + + 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 + ) + 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=IP_PROTOCOL_TCP, port=1 + ) + self.service1 = Service.objects.create( + device=self.device1, name='Test Service 2', protocol=IP_PROTOCOL_TCP, port=2 + ) + self.service1 = Service.objects.create( + device=self.device1, name='Test Service 3', protocol=IP_PROTOCOL_TCP, port=3 + ) + + 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': IP_PROTOCOL_TCP, + 'port': 4, + } + + url = reverse('ipam-api:service-list') + response = self.client.post(url, data, **self.header) + + self.assertEqual(response.status_code, 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']) + + def test_update_service(self): + + data = { + 'device': self.device2.pk, + 'name': 'Test Service X', + 'protocol': IP_PROTOCOL_UDP, + 'port': 99, + } + + url = reverse('ipam-api:service-detail', kwargs={'pk': self.service1.pk}) + response = self.client.put(url, data, **self.header) + + self.assertEqual(response.status_code, 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.assertEqual(response.status_code, status.HTTP_204_NO_CONTENT) + self.assertEqual(Service.objects.count(), 2)