|
@@ -2247,13 +2247,21 @@ class ModuleBayTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
|
|
|
ModuleBayTemplate.objects.bulk_create(
|
|
ModuleBayTemplate.objects.bulk_create(
|
|
|
(
|
|
(
|
|
|
ModuleBayTemplate(
|
|
ModuleBayTemplate(
|
|
|
- device_type=device_types[0], name='Module Bay 1', description='foobar1'
|
|
|
|
|
|
|
+ device_type=device_types[0], name='Module Bay 1', enabled=True, description='foobar1'
|
|
|
),
|
|
),
|
|
|
ModuleBayTemplate(
|
|
ModuleBayTemplate(
|
|
|
- device_type=device_types[1], name='Module Bay 2', description='foobar2', module_type=module_types[0]
|
|
|
|
|
|
|
+ device_type=device_types[1],
|
|
|
|
|
+ name='Module Bay 2',
|
|
|
|
|
+ enabled=False,
|
|
|
|
|
+ description='foobar2',
|
|
|
|
|
+ module_type=module_types[0],
|
|
|
),
|
|
),
|
|
|
ModuleBayTemplate(
|
|
ModuleBayTemplate(
|
|
|
- device_type=device_types[2], name='Module Bay 3', description='foobar3', module_type=module_types[1]
|
|
|
|
|
|
|
+ device_type=device_types[2],
|
|
|
|
|
+ name='Module Bay 3',
|
|
|
|
|
+ enabled=True,
|
|
|
|
|
+ description='foobar3',
|
|
|
|
|
+ module_type=module_types[1],
|
|
|
),
|
|
),
|
|
|
)
|
|
)
|
|
|
)
|
|
)
|
|
@@ -2262,6 +2270,12 @@ class ModuleBayTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
|
|
|
params = {'name': ['Module Bay 1', 'Module Bay 2']}
|
|
params = {'name': ['Module Bay 1', 'Module Bay 2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_enabled(self):
|
|
|
|
|
+ params = {'enabled': True}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+ params = {'enabled': False}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
+
|
|
|
def test_module_type(self):
|
|
def test_module_type(self):
|
|
|
module_types = ModuleType.objects.all()[:2]
|
|
module_types = ModuleType.objects.all()[:2]
|
|
|
params = {'module_type_id': [module_types[0].pk, module_types[1].pk]}
|
|
params = {'module_type_id': [module_types[0].pk, module_types[1].pk]}
|
|
@@ -2284,16 +2298,30 @@ class DeviceBayTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
|
|
|
)
|
|
)
|
|
|
DeviceType.objects.bulk_create(device_types)
|
|
DeviceType.objects.bulk_create(device_types)
|
|
|
|
|
|
|
|
- DeviceBayTemplate.objects.bulk_create((
|
|
|
|
|
- DeviceBayTemplate(device_type=device_types[0], name='Device Bay 1', description='foobar1'),
|
|
|
|
|
- DeviceBayTemplate(device_type=device_types[1], name='Device Bay 2', description='foobar2'),
|
|
|
|
|
- DeviceBayTemplate(device_type=device_types[2], name='Device Bay 3', description='foobar3'),
|
|
|
|
|
- ))
|
|
|
|
|
|
|
+ DeviceBayTemplate.objects.bulk_create(
|
|
|
|
|
+ (
|
|
|
|
|
+ DeviceBayTemplate(
|
|
|
|
|
+ device_type=device_types[0], name='Device Bay 1', enabled=True, description='foobar1'
|
|
|
|
|
+ ),
|
|
|
|
|
+ DeviceBayTemplate(
|
|
|
|
|
+ device_type=device_types[1], name='Device Bay 2', enabled=False, description='foobar2'
|
|
|
|
|
+ ),
|
|
|
|
|
+ DeviceBayTemplate(
|
|
|
|
|
+ device_type=device_types[2], name='Device Bay 3', enabled=True, description='foobar3'
|
|
|
|
|
+ ),
|
|
|
|
|
+ )
|
|
|
|
|
+ )
|
|
|
|
|
|
|
|
def test_name(self):
|
|
def test_name(self):
|
|
|
params = {'name': ['Device Bay 1', 'Device Bay 2']}
|
|
params = {'name': ['Device Bay 1', 'Device Bay 2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_enabled(self):
|
|
|
|
|
+ params = {'enabled': True}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+ params = {'enabled': False}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
+
|
|
|
|
|
|
|
|
class InventoryItemTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, ChangeLoggedFilterSetTests):
|
|
class InventoryItemTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, ChangeLoggedFilterSetTests):
|
|
|
queryset = InventoryItemTemplate.objects.all()
|
|
queryset = InventoryItemTemplate.objects.all()
|
|
@@ -5778,11 +5806,11 @@ class ModuleBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
Device.objects.bulk_create(devices)
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
|
|
module_bays = (
|
|
module_bays = (
|
|
|
- ModuleBay(device=devices[0], name='Module Bay 1', label='A', description='First'),
|
|
|
|
|
- ModuleBay(device=devices[1], name='Module Bay 2', label='B', description='Second'),
|
|
|
|
|
- ModuleBay(device=devices[2], name='Module Bay 3', label='C', description='Third'),
|
|
|
|
|
- ModuleBay(device=devices[2], name='Module Bay 4', label='D', description='Fourth'),
|
|
|
|
|
- ModuleBay(device=devices[2], name='Module Bay 5', label='E', description='Fifth'),
|
|
|
|
|
|
|
+ ModuleBay(device=devices[0], name='Module Bay 1', label='A', enabled=True, description='First'),
|
|
|
|
|
+ ModuleBay(device=devices[1], name='Module Bay 2', label='B', enabled=False, description='Second'),
|
|
|
|
|
+ ModuleBay(device=devices[2], name='Module Bay 3', label='C', enabled=True, description='Third'),
|
|
|
|
|
+ ModuleBay(device=devices[2], name='Module Bay 4', label='D', enabled=False, description='Fourth'),
|
|
|
|
|
+ ModuleBay(device=devices[2], name='Module Bay 5', label='E', enabled=True, description='Fifth'),
|
|
|
)
|
|
)
|
|
|
for module_bay in module_bays:
|
|
for module_bay in module_bays:
|
|
|
module_bay.save()
|
|
module_bay.save()
|
|
@@ -5806,6 +5834,12 @@ class ModuleBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
params = {'label': ['A', 'B']}
|
|
params = {'label': ['A', 'B']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_enabled(self):
|
|
|
|
|
+ params = {'enabled': True}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+ params = {'enabled': False}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+
|
|
|
def test_description(self):
|
|
def test_description(self):
|
|
|
params = {'description': ['First', 'Second']}
|
|
params = {'description': ['First', 'Second']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
@@ -5965,6 +5999,7 @@ class DeviceBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
device=devices[0],
|
|
device=devices[0],
|
|
|
name='Device Bay 1',
|
|
name='Device Bay 1',
|
|
|
label='A',
|
|
label='A',
|
|
|
|
|
+ enabled=True,
|
|
|
description='First',
|
|
description='First',
|
|
|
_site=devices[0].site,
|
|
_site=devices[0].site,
|
|
|
_location=devices[0].location,
|
|
_location=devices[0].location,
|
|
@@ -5974,6 +6009,7 @@ class DeviceBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
device=devices[1],
|
|
device=devices[1],
|
|
|
name='Device Bay 2',
|
|
name='Device Bay 2',
|
|
|
label='B',
|
|
label='B',
|
|
|
|
|
+ enabled=False,
|
|
|
description='Second',
|
|
description='Second',
|
|
|
_site=devices[1].site,
|
|
_site=devices[1].site,
|
|
|
_location=devices[1].location,
|
|
_location=devices[1].location,
|
|
@@ -5983,6 +6019,7 @@ class DeviceBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
device=devices[2],
|
|
device=devices[2],
|
|
|
name='Device Bay 3',
|
|
name='Device Bay 3',
|
|
|
label='C',
|
|
label='C',
|
|
|
|
|
+ enabled=True,
|
|
|
description='Third',
|
|
description='Third',
|
|
|
_site=devices[2].site,
|
|
_site=devices[2].site,
|
|
|
_location=devices[2].location,
|
|
_location=devices[2].location,
|
|
@@ -5999,6 +6036,12 @@ class DeviceBayTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
|
|
|
params = {'label': ['A', 'B']}
|
|
params = {'label': ['A', 'B']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
|
|
|
|
+ def test_enabled(self):
|
|
|
|
|
+ params = {'enabled': True}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
+ params = {'enabled': False}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
|
|
+
|
|
|
def test_description(self):
|
|
def test_description(self):
|
|
|
params = {'description': ['First', 'Second']}
|
|
params = {'description': ['First', 'Second']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|