|
1 | 1 | from django.contrib.auth.models import User |
| 2 | +from django.test import override_settings |
2 | 3 | from django.urls import reverse |
3 | 4 | from rest_framework import status |
4 | 5 |
|
@@ -1490,40 +1491,35 @@ def setUp(self): |
1490 | 1491 |
|
1491 | 1492 | super().setUp() |
1492 | 1493 |
|
1493 | | - self.site1 = Site.objects.create(name='Test Site 1', slug='test-site-1') |
1494 | | - self.site2 = Site.objects.create(name='Test Site 2', slug='test-site-2') |
1495 | | - manufacturer = Manufacturer.objects.create(name='Test Manufacturer 1', slug='test-manufacturer-1') |
1496 | | - self.devicetype1 = DeviceType.objects.create( |
1497 | | - manufacturer=manufacturer, model='Test Device Type 1', slug='test-device-type-1' |
1498 | | - ) |
1499 | | - self.devicetype2 = DeviceType.objects.create( |
1500 | | - manufacturer=manufacturer, model='Test Device Type 2', slug='test-device-type-2' |
1501 | | - ) |
1502 | | - self.devicerole1 = DeviceRole.objects.create( |
1503 | | - name='Test Device Role 1', slug='test-device-role-1', color='ff0000' |
1504 | | - ) |
1505 | | - self.devicerole2 = DeviceRole.objects.create( |
1506 | | - name='Test Device Role 2', slug='test-device-role-2', color='00ff00' |
1507 | | - ) |
| 1494 | + site = Site.objects.create(name='Site 1', slug='site-1') |
| 1495 | + manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1') |
| 1496 | + devicetype = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1') |
| 1497 | + devicerole = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1', color='ff0000') |
1508 | 1498 | self.device1 = Device.objects.create( |
1509 | | - device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice1', site=self.site1 |
| 1499 | + device_type=devicetype, device_role=devicerole, name='TestDevice1', site=site |
1510 | 1500 | ) |
1511 | 1501 | self.device2 = Device.objects.create( |
1512 | | - device_type=self.devicetype1, device_role=self.devicerole1, name='TestDevice2', site=self.site1 |
| 1502 | + device_type=devicetype, device_role=devicerole, name='TestDevice2', site=site |
1513 | 1503 | ) |
1514 | 1504 | self.interface1 = Interface.objects.create(device=self.device1, name='eth0') |
1515 | 1505 | self.interface2 = Interface.objects.create(device=self.device2, name='eth0') |
| 1506 | + self.interface3 = Interface.objects.create(device=self.device1, name='eth1') # Not connected |
1516 | 1507 |
|
1517 | 1508 | cable = Cable(termination_a=self.interface1, termination_b=self.interface2) |
1518 | 1509 | cable.save() |
1519 | 1510 |
|
| 1511 | + @override_settings(EXEMPT_VIEW_PERMISSIONS=['*']) |
1520 | 1512 | def test_get_connected_device(self): |
1521 | | - |
1522 | 1513 | url = reverse('dcim-api:connected-device-list') |
1523 | | - response = self.client.get(url + '?peer_device=TestDevice2&peer_interface=eth0', **self.header) |
1524 | 1514 |
|
| 1515 | + url_params = f'?peer_device={self.device1.name}&peer_interface={self.interface1.name}' |
| 1516 | + response = self.client.get(url + url_params, **self.header) |
1525 | 1517 | self.assertHttpStatus(response, status.HTTP_200_OK) |
1526 | | - self.assertEqual(response.data['name'], self.device1.name) |
| 1518 | + self.assertEqual(response.data['name'], self.device2.name) |
| 1519 | + |
| 1520 | + url_params = f'?peer_device={self.device1.name}&peer_interface={self.interface3.name}' |
| 1521 | + response = self.client.get(url + url_params, **self.header) |
| 1522 | + self.assertHttpStatus(response, status.HTTP_404_NOT_FOUND) |
1527 | 1523 |
|
1528 | 1524 |
|
1529 | 1525 | class VirtualChassisTest(APIViewTestCases.APIViewTestCase): |
|
0 commit comments