diff --git a/netbox/dcim/tests/test_filtersets.py b/netbox/dcim/tests/test_filtersets.py index 296f6d433..bd44d4f5d 100644 --- a/netbox/dcim/tests/test_filtersets.py +++ b/netbox/dcim/tests/test_filtersets.py @@ -2048,6 +2048,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, device_type=device_types[0], name='Front Port 1', type=PortTypeChoices.TYPE_8P8C, + positions=1, color=ColorChoices.COLOR_RED, description='foobar1' ), @@ -2055,6 +2056,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, device_type=device_types[1], name='Front Port 2', type=PortTypeChoices.TYPE_110_PUNCH, + positions=2, color=ColorChoices.COLOR_GREEN, description='foobar2' ), @@ -2062,6 +2064,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, device_type=device_types[2], name='Front Port 3', type=PortTypeChoices.TYPE_BNC, + positions=3, color=ColorChoices.COLOR_BLUE, description='foobar3' ), @@ -2085,6 +2088,10 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]} self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) + def test_positions(self): + params = {'positions': [1, 2]} + self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) + class RearPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, ChangeLoggedFilterSetTests): queryset = RearPortTemplate.objects.all() @@ -5118,6 +5125,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil name='Front Port 4', label='D', type=PortTypeChoices.TYPE_FC, + positions=2, _site=devices[3].site, _location=devices[3].location, _rack=devices[3].rack, @@ -5127,6 +5135,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil name='Front Port 5', label='E', type=PortTypeChoices.TYPE_FC, + positions=3, _site=devices[3].site, _location=devices[3].location, _rack=devices[3].rack, @@ -5136,6 +5145,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil name='Front Port 6', label='F', type=PortTypeChoices.TYPE_FC, + positions=4, _site=devices[3].site, _location=devices[3].location, _rack=devices[3].rack, @@ -5172,6 +5182,10 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]} self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) + def test_positions(self): + params = {'positions': [2, 3]} + self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) + def test_description(self): params = {'description': ['First', 'Second']} self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)