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

Inspect many-to-many fields

Browse Source
This commit is contained in:
Jeremy Stretch
2024-03-06 15:09:48 -05:00
parent 5cb7af88d4
commit 0a0dae3d35
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
netbox/utilities/testing/filtersets.py
View File

@ -80,6 +80,10 @@ class BaseFilterSetTests:
if model_field.name in chain(self.ignore_fields, IGNORE_FIELDS): if model_field.name in chain(self.ignore_fields, IGNORE_FIELDS):
continue continue
# Skip reverse ForeignKey relationships
if type(model_field) is ManyToOneRel:
continue
# One-to-one & one-to-many relationships # One-to-one & one-to-many relationships
if issubclass(model_field.__class__, ForeignKey) or type(model_field) is OneToOneRel: if issubclass(model_field.__class__, ForeignKey) or type(model_field) is OneToOneRel:
if model_field.related_model is ContentType: if model_field.related_model is ContentType:
@ -90,10 +94,14 @@ class BaseFilterSetTests:
filter_name = model_field.name filter_name = model_field.name
else: else:
filter_name = f'{model_field.name}_id' filter_name = f'{model_field.name}_id'
self.assertIn(filter_name, filterset_fields, f'No filter found for {model_field.name}!') self.assertIn(filter_name, filterset_fields, f'No filter found for {filter_name}!')
# TODO: Many-to-one & many-to-many relationships # TODO: Many-to-many relationships
elif type(model_field) in (ManyToOneRel, ManyToManyField, ManyToManyRel): elif type(model_field) is ManyToManyField:
related_model = model_field.related_model._meta.model_name
filter_name = f'{related_model}_id'
self.assertIn(filter_name, filterset_fields, f'M2M: No filter found for {filter_name}!')
elif type(model_field) is ManyToManyRel:
continue continue
# TODO: Generic relationships # TODO: Generic relationships