NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-08-12 10:38:16 -06:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Fix tests for #2528

Browse Source
This commit is contained in:
Arnoud Vermeer 2018-11-01 13:53:39 +01:00
parent c776770e7f
commit 71a5581d95
1 changed files with 37 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
netbox/circuits/tests/test_api.py
View File

@ -5,12 +5,11 @@ from rest_framework import status
from circuits.constants import CIRCUIT_STATUS_ACTIVE, TERM_SIDE_A, TERM_SIDE_Z from circuits.constants import CIRCUIT_STATUS_ACTIVE, TERM_SIDE_A, TERM_SIDE_Z
from circuits.models import Circuit, CircuitTermination, CircuitType, Provider from circuits.models import Circuit, CircuitTermination, CircuitType, Provider
from dcim.models import Site from dcim.models import Site, Manufacturer, DeviceRole, DeviceType, Interface, Device
from extras.constants import GRAPH_TYPE_PROVIDER from extras.constants import GRAPH_TYPE_PROVIDER
from extras.models import Graph from extras.models import Graph
from utilities.testing import APITestCase from utilities.testing import APITestCase
class ProviderTest(APITestCase): class ProviderTest(APITestCase):
def setUp(self): def setUp(self):
@ -337,14 +336,43 @@ class CircuitTerminationTest(APITestCase):
self.circuit3 = Circuit.objects.create(cid='TEST0003', provider=provider, type=circuittype) self.circuit3 = Circuit.objects.create(cid='TEST0003', provider=provider, type=circuittype)
self.site1 = Site.objects.create(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') 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='Test Device 1',
site=self.site1
)
self.device2 = Device.objects.create(
device_type=self.devicetype1,
device_role=self.devicerole1,
name='Test Device 2',
site=self.site1,
)
self.interface1 = Interface.objects.create(device=self.device1, name='Test Interface 1')
self.interface2 = Interface.objects.create(device=self.device2, name='Test Interface 2')
self.interface3 = Interface.objects.create(device=self.device2, name='Test Interface 3')
self.interface4 = Interface.objects.create(device=self.device2, name='Test Interface 4')
self.circuittermination1 = CircuitTermination.objects.create( self.circuittermination1 = CircuitTermination.objects.create(
circuit=self.circuit1, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000 circuit=self.circuit1, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000, interface=self.interface1
) )
self.circuittermination2 = CircuitTermination.objects.create( self.circuittermination2 = CircuitTermination.objects.create(
circuit=self.circuit2, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000 circuit=self.circuit2, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000, interface=self.interface2
) )
self.circuittermination3 = CircuitTermination.objects.create( self.circuittermination3 = CircuitTermination.objects.create(
circuit=self.circuit3, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000 circuit=self.circuit3, term_side=TERM_SIDE_A, site=self.site1, port_speed=1000000, interface=self.interface3
) )
def test_get_circuittermination(self): def test_get_circuittermination(self):
@ -368,6 +396,7 @@ class CircuitTerminationTest(APITestCase):
'term_side': TERM_SIDE_Z, 'term_side': TERM_SIDE_Z,
'site': self.site2.pk, 'site': self.site2.pk,
'port_speed': 1000000, 'port_speed': 1000000,
'interface': self.interface4.pk,
} }
url = reverse('circuits-api:circuittermination-list') url = reverse('circuits-api:circuittermination-list')
@ -380,6 +409,7 @@ class CircuitTerminationTest(APITestCase):
self.assertEqual(circuittermination4.term_side, data['term_side']) self.assertEqual(circuittermination4.term_side, data['term_side'])
self.assertEqual(circuittermination4.site_id, data['site']) self.assertEqual(circuittermination4.site_id, data['site'])
self.assertEqual(circuittermination4.port_speed, data['port_speed']) self.assertEqual(circuittermination4.port_speed, data['port_speed'])
self.assertEqual(circuittermination4.interface, self.interface4)
def test_update_circuittermination(self): def test_update_circuittermination(self):
@ -388,6 +418,7 @@ class CircuitTerminationTest(APITestCase):
'term_side': TERM_SIDE_Z, 'term_side': TERM_SIDE_Z,
'site': self.site2.pk, 'site': self.site2.pk,
'port_speed': 1000000, 'port_speed': 1000000,
'interface': self.interface4.pk,
} }
url = reverse('circuits-api:circuittermination-detail', kwargs={'pk': self.circuittermination1.pk}) url = reverse('circuits-api:circuittermination-detail', kwargs={'pk': self.circuittermination1.pk})
@ -400,6 +431,7 @@ class CircuitTerminationTest(APITestCase):
self.assertEqual(circuittermination1.term_side, data['term_side']) self.assertEqual(circuittermination1.term_side, data['term_side'])
self.assertEqual(circuittermination1.site_id, data['site']) self.assertEqual(circuittermination1.site_id, data['site'])
self.assertEqual(circuittermination1.port_speed, data['port_speed']) self.assertEqual(circuittermination1.port_speed, data['port_speed'])
self.assertEqual(circuittermination1.interface, self.interface4)
def test_delete_circuittermination(self): def test_delete_circuittermination(self):