NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2026-01-23 20:12:42 -06:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

#11969: Cleanup (#17308)

Browse Source 
* Shorten choice constant names

* Add filter tests for airflow
This commit is contained in:
Jeremy Stretch
2024-08-29 15:47:11 -04:00
committed by GitHub
parent db6246a437
commit 00d8eb40fc
2 changed files with 40 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
netbox/dcim/tests/test_filtersets.py
View File

@@ -500,7 +500,8 @@ class RackTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
weight=10,
max_weight=1000,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar1'
description='foobar1',
airflow=RackAirflowChoices.FRONT_TO_REAR
),
RackType(
manufacturer=manufacturers[1],
@@ -518,7 +519,8 @@ class RackTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
weight=20,
max_weight=2000,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar2'
description='foobar2',
airflow=RackAirflowChoices.REAR_TO_FRONT
),
RackType(
manufacturer=manufacturers[2],
@@ -615,6 +617,10 @@ class RackTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'weight_unit': WeightUnitChoices.UNIT_POUND}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_airflow(self):
params = {'airflow': RackAirflowChoices.REAR_TO_FRONT}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
class RackTestCase(TestCase, ChangeLoggedFilterSetTests):
queryset = Rack.objects.all()
@@ -745,7 +751,8 @@ class RackTestCase(TestCase, ChangeLoggedFilterSetTests):
weight=10,
max_weight=1000,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar1'
description='foobar1',
airflow=RackAirflowChoices.FRONT_TO_REAR
),
Rack(
name='Rack 2',
@@ -767,7 +774,8 @@ class RackTestCase(TestCase, ChangeLoggedFilterSetTests):
weight=20,
max_weight=2000,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar2'
description='foobar2',
airflow=RackAirflowChoices.REAR_TO_FRONT
),
Rack(
name='Rack 3',
@@ -958,6 +966,10 @@ class RackTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'rack_type': [rack_types[0].slug, rack_types[1].slug]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_airflow(self):
params = {'airflow': RackAirflowChoices.FRONT_TO_REAR}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
class RackReservationTestCase(TestCase, ChangeLoggedFilterSetTests):
queryset = RackReservation.objects.all()
@@ -1380,7 +1392,8 @@ class ModuleTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
part_number='Part Number 1',
weight=10,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar1'
description='foobar1',
airflow=ModuleAirflowChoices.FRONT_TO_REAR
),
ModuleType(
manufacturer=manufacturers[1],
@@ -1388,7 +1401,8 @@ class ModuleTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
part_number='Part Number 2',
weight=20,
weight_unit=WeightUnitChoices.UNIT_POUND,
description='foobar2'
description='foobar2',
airflow=ModuleAirflowChoices.REAR_TO_FRONT
),
ModuleType(
manufacturer=manufacturers[2],
@@ -1499,6 +1513,10 @@ class ModuleTypeTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'weight_unit': WeightUnitChoices.UNIT_POUND}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_airflow(self):
params = {'airflow': RackAirflowChoices.FRONT_TO_REAR}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
class ConsolePortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, ChangeLoggedFilterSetTests):
queryset = ConsolePortTemplate.objects.all()