|
|
@@ -12,6 +12,23 @@ from virtualization.models import Cluster, ClusterType
|
|
|
from wireless.choices import WirelessChannelChoices, WirelessRoleChoices
|
|
|
|
|
|
|
|
|
+class DeviceComponentFilterSetTests:
|
|
|
+
|
|
|
+ def test_device_type(self):
|
|
|
+ device_types = DeviceType.objects.all()[:2]
|
|
|
+ params = {'device_type_id': [device_types[0].pk, device_types[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'device_type': [device_types[0].model, device_types[1].model]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
+ def test_device_role(self):
|
|
|
+ device_role = DeviceRole.objects.all()[:2]
|
|
|
+ params = {'device_role_id': [device_role[0].pk, device_role[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'device_role': [device_role[0].slug, device_role[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
+
|
|
|
class RegionTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
queryset = Region.objects.all()
|
|
|
filterset = RegionFilterSet
|
|
|
@@ -1994,7 +2011,7 @@ class ModuleTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
-class ConsolePortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class ConsolePortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = ConsolePort.objects.all()
|
|
|
filterset = ConsolePortFilterSet
|
|
|
|
|
|
@@ -2023,10 +2040,23 @@ class ConsolePortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -2044,10 +2074,10 @@ class ConsolePortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[0], device_role=device_roles[0], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -2161,7 +2191,7 @@ class ConsolePortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
|
|
|
-class ConsoleServerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class ConsoleServerPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = ConsoleServerPort.objects.all()
|
|
|
filterset = ConsoleServerPortFilterSet
|
|
|
|
|
|
@@ -2190,10 +2220,23 @@ class ConsoleServerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -2211,10 +2254,10 @@ class ConsoleServerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -2328,7 +2371,7 @@ class ConsoleServerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
|
|
|
-class PowerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class PowerPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = PowerPort.objects.all()
|
|
|
filterset = PowerPortFilterSet
|
|
|
|
|
|
@@ -2357,10 +2400,23 @@ class PowerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -2378,10 +2434,10 @@ class PowerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -2503,7 +2559,7 @@ class PowerPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
|
|
|
-class PowerOutletTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class PowerOutletTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = PowerOutlet.objects.all()
|
|
|
filterset = PowerOutletFilterSet
|
|
|
|
|
|
@@ -2532,10 +2588,23 @@ class PowerOutletTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -2553,10 +2622,10 @@ class PowerOutletTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -2674,7 +2743,7 @@ class PowerOutletTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
|
|
|
|
-class InterfaceTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class InterfaceTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = Interface.objects.all()
|
|
|
filterset = InterfaceFilterSet
|
|
|
|
|
|
@@ -2703,10 +2772,23 @@ class InterfaceTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -2724,10 +2806,10 @@ class InterfaceTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3097,7 +3179,7 @@ class InterfaceTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
|
|
|
|
-class FrontPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = FrontPort.objects.all()
|
|
|
filterset = FrontPortFilterSet
|
|
|
|
|
|
@@ -3126,10 +3208,23 @@ class FrontPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -3147,10 +3242,10 @@ class FrontPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3273,7 +3368,7 @@ class FrontPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
-class RearPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class RearPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = RearPort.objects.all()
|
|
|
filterset = RearPortFilterSet
|
|
|
|
|
|
@@ -3302,10 +3397,23 @@ class RearPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
module_type = ModuleType.objects.create(manufacturer=manufacturer, model='Module Type 1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -3323,10 +3431,10 @@ class RearPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
- Device(name=None, device_type=device_type, device_role=device_role, site=sites[3]), # For cable connections
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name=None, device_type=device_types[2], device_role=device_roles[2], site=sites[3]), # For cable connections
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3443,7 +3551,7 @@ class RearPortTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
-class ModuleBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class ModuleBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = ModuleBay.objects.all()
|
|
|
filterset = ModuleBayFilterSet
|
|
|
|
|
|
@@ -3472,9 +3580,21 @@ class ModuleBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Model 1', slug='model-1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -3492,9 +3612,9 @@ class ModuleBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3560,7 +3680,7 @@ class ModuleBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
-class DeviceBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
+class DeviceBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = DeviceBay.objects.all()
|
|
|
filterset = DeviceBayFilterSet
|
|
|
|
|
|
@@ -3589,9 +3709,21 @@ class DeviceBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Site(name='Site 3', slug='site-3', region=regions[2], group=groups[2]),
|
|
|
Site(name='Site X', slug='site-x'),
|
|
|
))
|
|
|
+
|
|
|
manufacturer = Manufacturer.objects.create(name='Manufacturer 1', slug='manufacturer-1')
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturer, model='Model 1', slug='model-1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturer, model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
locations = (
|
|
|
Location(name='Location 1', slug='location-1', site=sites[0]),
|
|
|
@@ -3609,9 +3741,9 @@ class DeviceBayTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3690,8 +3822,19 @@ class InventoryItemTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
)
|
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
|
|
|
|
- device_type = DeviceType.objects.create(manufacturer=manufacturers[0], model='Model 1', slug='model-1')
|
|
|
- device_role = DeviceRole.objects.create(name='Device Role 1', slug='device-role-1')
|
|
|
+ device_types = (
|
|
|
+ DeviceType(manufacturer=manufacturers[0], model='Device Type 1', slug='device-type-1'),
|
|
|
+ DeviceType(manufacturer=manufacturers[0], model='Device Type 2', slug='device-type-2'),
|
|
|
+ DeviceType(manufacturer=manufacturers[0], model='Device Type 3', slug='device-type-3'),
|
|
|
+ )
|
|
|
+ DeviceType.objects.bulk_create(device_types)
|
|
|
+
|
|
|
+ device_roles = (
|
|
|
+ DeviceRole(name='Device Role 1', slug='device-role-1'),
|
|
|
+ DeviceRole(name='Device Role 2', slug='device-role-2'),
|
|
|
+ DeviceRole(name='Device Role 3', slug='device-role-3'),
|
|
|
+ )
|
|
|
+ DeviceRole.objects.bulk_create(device_roles)
|
|
|
|
|
|
regions = (
|
|
|
Region(name='Region 1', slug='region-1'),
|
|
|
@@ -3732,9 +3875,9 @@ class InventoryItemTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Rack.objects.bulk_create(racks)
|
|
|
|
|
|
devices = (
|
|
|
- Device(name='Device 1', device_type=device_type, device_role=device_role, site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
- Device(name='Device 2', device_type=device_type, device_role=device_role, site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
- Device(name='Device 3', device_type=device_type, device_role=device_role, site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
+ Device(name='Device 1', device_type=device_types[0], device_role=device_roles[0], site=sites[0], location=locations[0], rack=racks[0]),
|
|
|
+ Device(name='Device 2', device_type=device_types[1], device_role=device_roles[1], site=sites[1], location=locations[1], rack=racks[1]),
|
|
|
+ Device(name='Device 3', device_type=device_types[2], device_role=device_roles[2], site=sites[2], location=locations[2], rack=racks[2]),
|
|
|
)
|
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
@@ -3825,6 +3968,20 @@ class InventoryItemTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
params = {'rack': [racks[0].name, racks[1].name]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
+ def test_device_type(self):
|
|
|
+ device_types = DeviceType.objects.all()[:2]
|
|
|
+ params = {'device_type_id': [device_types[0].pk, device_types[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+ params = {'device_type': [device_types[0].model, device_types[1].model]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+
|
|
|
+ def test_device_role(self):
|
|
|
+ device_role = DeviceRole.objects.all()[:2]
|
|
|
+ params = {'device_role_id': [device_role[0].pk, device_role[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+ params = {'device_role': [device_role[0].slug, device_role[1].slug]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
+
|
|
|
def test_device(self):
|
|
|
devices = Device.objects.all()[:2]
|
|
|
params = {'device_id': [devices[0].pk, devices[1].pk]}
|