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

Address PR feedback, add ServiceFilterSet test for fhrpgroup

Browse Source
This commit is contained in:
Jason Novinger 2025-04-07 15:35:39 -05:00
parent 37a2d902a5
commit be3d55a032
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
netbox/ipam/tests/test_filtersets.py
View File

@ -2329,6 +2329,11 @@ class ServiceTestCase(TestCase, ChangeLoggedFilterSetTests):
VirtualMachine(name='Virtual Machine 3', cluster=cluster), VirtualMachine(name='Virtual Machine 3', cluster=cluster),
) )
VirtualMachine.objects.bulk_create(virtual_machines) VirtualMachine.objects.bulk_create(virtual_machines)
fhrp_group = FHRPGroup.objects.create(
name='telnet',
protocol=FHRPGroupProtocolChoices.PROTOCOL_CARP,
group_id=101,
)
services = ( services = (
Service( Service(
@ -2369,6 +2374,12 @@ class ServiceTestCase(TestCase, ChangeLoggedFilterSetTests):
protocol=ServiceProtocolChoices.PROTOCOL_UDP, protocol=ServiceProtocolChoices.PROTOCOL_UDP,
ports=[2003], ports=[2003],
), ),
Service(
parent=fhrp_group,
name='Service 7',
protocol=ServiceProtocolChoices.PROTOCOL_UDP,
ports=[2004],
),
) )
Service.objects.bulk_create(services) Service.objects.bulk_create(services)
services[0].ipaddresses.add(ip_addresses[0]) services[0].ipaddresses.add(ip_addresses[0])
@ -2409,6 +2420,13 @@ class ServiceTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'virtual_machine': [vms[0].name, vms[1].name]} params = {'virtual_machine': [vms[0].name, vms[1].name]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2) self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_fhrp_group(self):
fhrp_group = FHRPGroup.objects.get()
params = {'fhrpgroup_id': [fhrp_group.pk]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
params = {'fhrpgroup': [fhrp_group.name]}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_ip_address(self): def test_ip_address(self):
ips = IPAddress.objects.all()[:2] ips = IPAddress.objects.all()[:2]
params = {'ip_address_id': [ips[0].pk, ips[1].pk]} params = {'ip_address_id': [ips[0].pk, ips[1].pk]}