NeoAnd/netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2025-07-24 17:38:37 -06:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Extend filterset/model tests to cover PowerOutlet.status

Browse Source
This commit is contained in:
Jason Novinger 2025-02-26 17:09:56 -06:00
parent 8efcbddb37
commit 2dcf2d203c
2 changed files with 22 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -3684,6 +3684,7 @@ class PowerOutletTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedF
feed_leg=PowerOutletFeedLegChoices.FEED_LEG_A, feed_leg=PowerOutletFeedLegChoices.FEED_LEG_A,
description='First', description='First',
color='ff0000', color='ff0000',
status=PowerOutletStatusChoices.STATUS_ENABLED,
), ),
PowerOutlet( PowerOutlet(
device=devices[1], device=devices[1],
@ -3693,6 +3694,7 @@ class PowerOutletTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedF
feed_leg=PowerOutletFeedLegChoices.FEED_LEG_B, feed_leg=PowerOutletFeedLegChoices.FEED_LEG_B,
description='Second', description='Second',
color='00ff00', color='00ff00',
status=PowerOutletStatusChoices.STATUS_DISABLED,
), ),
PowerOutlet( PowerOutlet(
device=devices[2], device=devices[2],
@ -3702,6 +3704,7 @@ class PowerOutletTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedF
feed_leg=PowerOutletFeedLegChoices.FEED_LEG_C, feed_leg=PowerOutletFeedLegChoices.FEED_LEG_C,
description='Third', description='Third',
color='0000ff', color='0000ff',
status=PowerOutletStatusChoices.STATUS_FAULTY,
), ),
) )
PowerOutlet.objects.bulk_create(power_outlets) PowerOutlet.objects.bulk_create(power_outlets)
@ -3796,6 +3799,23 @@ class PowerOutletTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedF
params = {'connected': False} params = {'connected': False}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_status(self):
params = {'status': [PowerOutletStatusChoices.STATUS_ENABLED]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
params = {'status': [PowerOutletStatusChoices.STATUS_DISABLED]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
params = {'status': [PowerOutletStatusChoices.STATUS_FAULTY]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
params = {'status': [
PowerOutletStatusChoices.STATUS_ENABLED,
PowerOutletStatusChoices.STATUS_DISABLED,
PowerOutletStatusChoices.STATUS_FAULTY,
]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
class InterfaceTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests): class InterfaceTestCase(TestCase, DeviceComponentFilterSetTests, ChangeLoggedFilterSetTests):
queryset = Interface.objects.all() queryset = Interface.objects.all()

3
netbox/dcim/tests/test_models.py
View File

@ -465,7 +465,8 @@ class DeviceTestCase(TestCase):
device=device, device=device,
name='Power Outlet 1', name='Power Outlet 1',
power_port=powerport, power_port=powerport,
feed_leg=PowerOutletFeedLegChoices.FEED_LEG_A feed_leg=PowerOutletFeedLegChoices.FEED_LEG_A,
status=PowerOutletStatusChoices.STATUS_ENABLED,
) )
self.assertEqual(poweroutlet.cf['cf1'], 'foo') self.assertEqual(poweroutlet.cf['cf1'], 'foo')