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

16078 add tests for graphql filtering

Browse Source
This commit is contained in:
Arthur 2024-05-13 14:10:18 -07:00
parent b3dd5576cd
commit 9265c8a4e4
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
netbox/utilities/testing/api.py
View File

@ -441,6 +441,10 @@ class APIViewTestCases:
return getattr(self, 'graphql_base_name', base_name)
def _build_query_with_filter(self, name, filter_string):
"""
Called by either _build_query or _build_filtered_query - construct the actual
query given a name and filter string
"""
type_class = get_graphql_type_for_model(self.model)
# Compile list of fields to include
@ -488,6 +492,9 @@ class APIViewTestCases:
return query
def _build_filtered_query(self, name, **filters):
"""
Create a filtered query: i.e. ip_address_list(filters: {address: "1.1.1.1/24"}){.
"""
if filters:
filter_string = ', '.join(f'{k}: "{v}"' for k, v in filters.items())
filter_string = f'(filters: {{{filter_string}}})'
@ -497,6 +504,9 @@ class APIViewTestCases:
return self._build_query_with_filter(name, filter_string)
def _build_query(self, name, **filters):
"""
Create a normal query - unfiltered or with a string query: i.e. site(name: "aaa"){.
"""
if filters:
filter_string = ', '.join(f'{k}:{v}' for k, v in filters.items())
filter_string = f'({filter_string})'