1
0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00

Added tests for Cable model

This commit is contained in:
Jeremy Stretch
2018-10-31 15:17:51 -04:00
parent d22c23290f
commit c2656c4ce1

View File

@ -2805,6 +2805,147 @@ class CableTest(APITestCase):
super(CableTest, self).setUp()
site = Site.objects.create(name='Test Site 1', slug='test-site-1')
self.manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1')
devicetype = DeviceType.objects.create(
manufacturer=self.manufacturer, model='Test Device Type 1', slug='test-device-type-1'
)
devicerole = DeviceRole.objects.create(
name='Test Device Role 1', slug='test-device-role-1', color='ff0000'
)
self.device1 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 1', site=site
)
self.device2 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 2', site=site
)
for device in [self.device1, self.device2]:
for i in range(0, 10):
Interface(device=device, form_factor=IFACE_FF_1GE_FIXED, name='eth{}'.format(i)).save()
self.cable1 = Cable(
termination_a=self.device1.interfaces.get(name='eth0'),
termination_b=self.device2.interfaces.get(name='eth0'),
label='Test Cable 1'
)
self.cable1.save()
self.cable2 = Cable(
termination_a=self.device1.interfaces.get(name='eth1'),
termination_b=self.device2.interfaces.get(name='eth1'),
label='Test Cable 2'
)
self.cable2.save()
self.cable3 = Cable(
termination_a=self.device1.interfaces.get(name='eth2'),
termination_b=self.device2.interfaces.get(name='eth2'),
label='Test Cable 3'
)
self.cable3.save()
def test_get_cable(self):
url = reverse('dcim-api:cable-detail', kwargs={'pk': self.cable1.pk})
response = self.client.get(url, **self.header)
self.assertEqual(response.data['id'], self.cable1.pk)
def test_list_cables(self):
url = reverse('dcim-api:cable-list')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 3)
def test_create_cable(self):
interface_a = self.device1.interfaces.get(name='eth3')
interface_b = self.device2.interfaces.get(name='eth3')
data = {
'termination_a_type': 'dcim.interface',
'termination_a_id': interface_a.pk,
'termination_b_type': 'dcim.interface',
'termination_b_id': interface_b.pk,
'status': CONNECTION_STATUS_PLANNED,
'label': 'Test Cable 4',
}
url = reverse('dcim-api:cable-list')
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Cable.objects.count(), 4)
cable4 = Cable.objects.get(pk=response.data['id'])
self.assertEqual(cable4.termination_a, interface_a)
self.assertEqual(cable4.termination_b, interface_b)
self.assertEqual(cable4.status, data['status'])
self.assertEqual(cable4.label, data['label'])
def test_create_cable_bulk(self):
data = [
{
'termination_a_type': 'dcim.interface',
'termination_a_id': self.device1.interfaces.get(name='eth3').pk,
'termination_b_type': 'dcim.interface',
'termination_b_id': self.device2.interfaces.get(name='eth3').pk,
'label': 'Test Cable 4',
},
{
'termination_a_type': 'dcim.interface',
'termination_a_id': self.device1.interfaces.get(name='eth4').pk,
'termination_b_type': 'dcim.interface',
'termination_b_id': self.device2.interfaces.get(name='eth4').pk,
'label': 'Test Cable 5',
},
{
'termination_a_type': 'dcim.interface',
'termination_a_id': self.device1.interfaces.get(name='eth5').pk,
'termination_b_type': 'dcim.interface',
'termination_b_id': self.device2.interfaces.get(name='eth5').pk,
'label': 'Test Cable 6',
},
]
url = reverse('dcim-api:cable-list')
response = self.client.post(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_201_CREATED)
self.assertEqual(Cable.objects.count(), 6)
self.assertEqual(response.data[0]['label'], data[0]['label'])
self.assertEqual(response.data[1]['label'], data[1]['label'])
self.assertEqual(response.data[2]['label'], data[2]['label'])
def test_update_cable(self):
data = {
'label': 'Test Cable X',
'status': CONNECTION_STATUS_CONNECTED,
}
url = reverse('dcim-api:cable-detail', kwargs={'pk': self.cable1.pk})
response = self.client.patch(url, data, format='json', **self.header)
self.assertHttpStatus(response, status.HTTP_200_OK)
self.assertEqual(Cable.objects.count(), 3)
cable1 = Cable.objects.get(pk=response.data['id'])
self.assertEqual(cable1.status, data['status'])
self.assertEqual(cable1.label, data['label'])
def test_delete_cable(self):
url = reverse('dcim-api:cable-detail', kwargs={'pk': self.cable1.pk})
response = self.client.delete(url, **self.header)
self.assertHttpStatus(response, status.HTTP_204_NO_CONTENT)
self.assertEqual(Cable.objects.count(), 2)
class ConnectionTest(APITestCase):
def setUp(self):
super(ConnectionTest, self).setUp()
self.site = Site.objects.create(
name='Test Site 1', slug='test-site-1'
)
@ -3163,88 +3304,6 @@ class CableTest(APITestCase):
self.assertEqual(circuittermination1.connected_endpoint, interface1)
class ConsoleConnectionTest(APITestCase):
def setUp(self):
super(ConsoleConnectionTest, self).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', slug='test-device-type-1'
)
devicerole = DeviceRole.objects.create(
name='Test Device Role 1', slug='test-device-role-1', color='ff0000'
)
device1 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 1', site=site
)
device2 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 2', site=site
)
consoleserverport1 = ConsoleServerPort.objects.create(device=device1, name='Test Console Server Port 1')
consoleserverport2 = ConsoleServerPort.objects.create(device=device1, name='Test Console Server Port 2')
consoleserverport3 = ConsoleServerPort.objects.create(device=device1, name='Test Console Server Port 3')
ConsolePort.objects.create(
device=device2, connected_endpoint=consoleserverport1, name='Test Console Port 1', connection_status=True
)
ConsolePort.objects.create(
device=device2, connected_endpoint=consoleserverport2, name='Test Console Port 2', connection_status=True
)
ConsolePort.objects.create(
device=device2, connected_endpoint=consoleserverport3, name='Test Console Port 3', connection_status=True
)
def test_list_consoleconnections(self):
url = reverse('dcim-api:consoleconnections-list')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 3)
class PowerConnectionTest(APITestCase):
def setUp(self):
super(PowerConnectionTest, self).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', slug='test-device-type-1'
)
devicerole = DeviceRole.objects.create(
name='Test Device Role 1', slug='test-device-role-1', color='ff0000'
)
device1 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 1', site=site
)
device2 = Device.objects.create(
device_type=devicetype, device_role=devicerole, name='Test Device 2', site=site
)
poweroutlet1 = PowerOutlet.objects.create(device=device1, name='Test Power Outlet 1')
poweroutlet2 = PowerOutlet.objects.create(device=device1, name='Test Power Outlet 2')
poweroutlet3 = PowerOutlet.objects.create(device=device1, name='Test Power Outlet 3')
PowerPort.objects.create(
device=device2, connected_endpoint=poweroutlet1, name='Test Power Port 1', connection_status=True
)
PowerPort.objects.create(
device=device2, connected_endpoint=poweroutlet2, name='Test Power Port 2', connection_status=True
)
PowerPort.objects.create(
device=device2, connected_endpoint=poweroutlet3, name='Test Power Port 3', connection_status=True
)
def test_list_powerconnections(self):
url = reverse('dcim-api:powerconnections-list')
response = self.client.get(url, **self.header)
self.assertEqual(response.data['count'], 3)
# class ConnectedDeviceTest(APITestCase):
#
# def setUp(self):