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

Add tests for new positions filters

Browse Source
This commit is contained in:
Jeremy Stretch
2025-12-01 12:24:46 -05:00
parent a3909d5c48
commit d5ce6d2166
1 changed files with 14 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -2048,6 +2048,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
device_type=device_types[0], device_type=device_types[0],
name='Front Port 1', name='Front Port 1',
type=PortTypeChoices.TYPE_8P8C, type=PortTypeChoices.TYPE_8P8C,
positions=1,
color=ColorChoices.COLOR_RED, color=ColorChoices.COLOR_RED,
description='foobar1' description='foobar1'
), ),
@@ -2055,6 +2056,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
device_type=device_types[1], device_type=device_types[1],
name='Front Port 2', name='Front Port 2',
type=PortTypeChoices.TYPE_110_PUNCH, type=PortTypeChoices.TYPE_110_PUNCH,
positions=2,
color=ColorChoices.COLOR_GREEN, color=ColorChoices.COLOR_GREEN,
description='foobar2' description='foobar2'
), ),
@@ -2062,6 +2064,7 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
device_type=device_types[2], device_type=device_types[2],
name='Front Port 3', name='Front Port 3',
type=PortTypeChoices.TYPE_BNC, type=PortTypeChoices.TYPE_BNC,
positions=3,
color=ColorChoices.COLOR_BLUE, color=ColorChoices.COLOR_BLUE,
description='foobar3' description='foobar3'
), ),
@@ -2085,6 +2088,10 @@ class FrontPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests,
params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]} params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) 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): class RearPortTemplateTestCase(TestCase, DeviceComponentTemplateFilterSetTests, ChangeLoggedFilterSetTests):
queryset = RearPortTemplate.objects.all() queryset = RearPortTemplate.objects.all()
@@ -5118,6 +5125,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
name='Front Port 4', name='Front Port 4',
label='D', label='D',
type=PortTypeChoices.TYPE_FC, type=PortTypeChoices.TYPE_FC,
positions=2,
_site=devices[3].site, _site=devices[3].site,
_location=devices[3].location, _location=devices[3].location,
_rack=devices[3].rack, _rack=devices[3].rack,
@@ -5127,6 +5135,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
name='Front Port 5', name='Front Port 5',
label='E', label='E',
type=PortTypeChoices.TYPE_FC, type=PortTypeChoices.TYPE_FC,
positions=3,
_site=devices[3].site, _site=devices[3].site,
_location=devices[3].location, _location=devices[3].location,
_rack=devices[3].rack, _rack=devices[3].rack,
@@ -5136,6 +5145,7 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
name='Front Port 6', name='Front Port 6',
label='F', label='F',
type=PortTypeChoices.TYPE_FC, type=PortTypeChoices.TYPE_FC,
positions=4,
_site=devices[3].site, _site=devices[3].site,
_location=devices[3].location, _location=devices[3].location,
_rack=devices[3].rack, _rack=devices[3].rack,
@@ -5172,6 +5182,10 @@ class FrontPortTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFil
params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]} params = {'color': [ColorChoices.COLOR_RED, ColorChoices.COLOR_GREEN]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) 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): 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)