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

Re-implemented connected device API view & test

Browse Source
This commit is contained in:
Jeremy Stretch
2018-11-06 11:44:32 -05:00
parent ce00226bc6
commit bb114ea66b
2 changed files with 60 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -3300,44 +3300,46 @@ class ConnectionTest(APITestCase):
self.assertEqual(circuittermination1.connected_endpoint, interface1)
# class ConnectedDeviceTest(APITestCase):
#
# def setUp(self):
#
# super(ConnectedDeviceTest, self).setUp()
#
# 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(
# manufacturer=manufacturer, model='Test Device Type 1', slug='test-device-type-1'
# )
# self.devicetype2 = DeviceType.objects.create(
# manufacturer=manufacturer, model='Test Device Type 2', slug='test-device-type-2'
# )
# self.devicerole1 = DeviceRole.objects.create(
# name='Test Device Role 1', slug='test-device-role-1', color='ff0000'
# )
# self.devicerole2 = DeviceRole.objects.create(
# name='Test Device Role 2', slug='test-device-role-2', color='00ff00'
# )
# self.device1 = Device.objects.create(
# device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice1', site=self.site1
# )
# self.device2 = Device.objects.create(
# device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice2', site=self.site1
# )
# self.interface1 = Interface.objects.create(device=self.device1, name='eth0')
# self.interface2 = Interface.objects.create(device=self.device2, name='eth0')
# InterfaceConnection.objects.create(interface_a=self.interface1, interface_b=self.interface2)
#
# def test_get_connected_device(self):
#
# url = reverse('dcim-api:connected-device-list')
# response = self.client.get(url + '?peer-device=TestDevice2&peer-interface=eth0', **self.header)
#
# self.assertHttpStatus(response, status.HTTP_200_OK)
# self.assertEqual(response.data['name'], self.device1.name)
class ConnectedDeviceTest(APITestCase):
def setUp(self):
super(ConnectedDeviceTest, self).setUp()
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(
manufacturer=manufacturer, model='Test Device Type 1', slug='test-device-type-1'
)
self.devicetype2 = DeviceType.objects.create(
manufacturer=manufacturer, model='Test Device Type 2', slug='test-device-type-2'
)
self.devicerole1 = DeviceRole.objects.create(
name='Test Device Role 1', slug='test-device-role-1', color='ff0000'
)
self.devicerole2 = DeviceRole.objects.create(
name='Test Device Role 2', slug='test-device-role-2', color='00ff00'
)
self.device1 = Device.objects.create(
device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice1', site=self.site1
)
self.device2 = Device.objects.create(
device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice2', site=self.site1
)
self.interface1 = Interface.objects.create(device=self.device1, name='eth0')
self.interface2 = Interface.objects.create(device=self.device2, name='eth0')
cable = Cable(termination_a=self.interface1, termination_b=self.interface2)
cable.save()
def test_get_connected_device(self):
url = reverse('dcim-api:connected-device-list')
response = self.client.get(url + '?peer_device=TestDevice2&peer_interface=eth0', **self.header)
self.assertHttpStatus(response, status.HTTP_200_OK)
self.assertEqual(response.data['name'], self.device1.name)
class VirtualChassisTest(APITestCase):