mirror/netbox-community-netbox
1
0
Fork 0
mirror of https://github.com/netbox-community/netbox.git synced 2024-05-10 07:54:54 +00:00
Code Activity

Apply filterset & test changes for #14631 & #14629

Browse Source
This commit is contained in:
Jeremy Stretch
2023-12-28 14:43:08 -05:00
parent d5c1cb0ef6
commit d930c4e36e
5 changed files with 119 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
netbox/virtualization/tests/test_filtersets.py
View File

@@ -669,12 +669,16 @@ class VirtualDiskTestCase(TestCase, ChangeLoggedFilterSetTests):
VirtualMachine.objects.bulk_create(vms)
disks = (
VirtualDisk(virtual_machine=vms[0], name='Disk 1', size=1, description='A'),
VirtualDisk(virtual_machine=vms[1], name='Disk 2', size=2, description='B'),
VirtualDisk(virtual_machine=vms[2], name='Disk 3', size=3, description='C'),
VirtualDisk(virtual_machine=vms[0], name='Disk 1', size=1, description='foobar1'),
VirtualDisk(virtual_machine=vms[1], name='Disk 2', size=2, description='foobar2'),
VirtualDisk(virtual_machine=vms[2], name='Disk 3', size=3, description='foobar3'),
)
VirtualDisk.objects.bulk_create(disks)
def test_q(self):
params = {'q': 'foobar1'}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_virtual_machine(self):
vms = VirtualMachine.objects.all()[:2]
params = {'virtual_machine_id': [vms[0].pk, vms[1].pk]}
@@ -691,5 +695,5 @@ class VirtualDiskTestCase(TestCase, ChangeLoggedFilterSetTests):
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_description(self):
params = {'description': ['A', 'B']}
params = {'description': ['foobar1', 'foobar2']}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)