Fix filterset tests for Device and VirtualMachine

This commit is contained in:
Brian Tiemann 2024-10-31 14:36:59 -04:00
parent 5be20924e7
commit 8e27d2b7bf
2 changed files with 8 additions and 4 deletions

View File

@ -2502,6 +2502,10 @@ class DeviceTestCase(TestCase, ChangeLoggedFilterSetTests):
params = {'airflow': DeviceAirflowChoices.AIRFLOW_FRONT_TO_REAR}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
def test_mac_address(self):
params = {'mac_address': ['00-00-00-00-00-01', '00-00-00-00-00-02']}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
def test_serial(self):
params = {'serial': ['ABC', 'DEF']}
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)

View File

@ -225,10 +225,10 @@ class VirtualMachineFilterSet(
to_field_name='slug',
label=_('Platform (slug)'),
)
# mac_address = MultiValueMACAddressFilter(
# field_name='interfaces___mac_address',
# label=_('MAC address'),
# )
mac_address = MultiValueMACAddressFilter(
field_name='interfaces__mac_addresses__mac_address',
label=_('MAC address'),
)
has_primary_ip = django_filters.BooleanFilter(
method='_has_primary_ip',
label=_('Has a primary IP'),