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

9856 circuits graphql tests working

Browse Source
This commit is contained in:
Arthur
2024-03-06 10:21:23 -08:00
parent 2f719269e8
commit aa7c00ec32
1 changed files with 11 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -21,6 +21,7 @@ from .utils import disable_warnings
from ipam.graphql.types import IPAddressFamilyType from ipam.graphql.types import IPAddressFamilyType
from strawberry.lazy_type import LazyType from strawberry.lazy_type import LazyType
from strawberry.type import StrawberryList, StrawberryOptional from strawberry.type import StrawberryList, StrawberryOptional
from strawberry.union import StrawberryUnion
__all__ = ( __all__ = (
'APITestCase', 'APITestCase',
@ -450,14 +451,18 @@ class APIViewTestCases:
fields_string = '' fields_string = ''
for field in type_class.__strawberry_definition__.fields: for field in type_class.__strawberry_definition__.fields:
# for field_name, field in type_class._meta.fields.items(): """
# print(f"field_name: {field.name} type: {field.type}") print(f"field_name: {field.name} type: {field.type}")
if field.name == 'site': if field.name == 'provider':
# breakpoint() breakpoint()
pass pass
"""
if type(field.type) is StrawberryList: if type(field.type) is StrawberryList:
if type(field.type.of_type) is StrawberryUnion:
# this would require a fragment query
continue
fields_string += f'{field.name} {{ id }}\n' fields_string += f'{field.name} {{ id }}\n'
elif field.type is strawberry_django.fields.types.DjangoModelType: elif field.type is strawberry_django.fields.types.DjangoModelType:
# Dynamic fields must specify a subselection # Dynamic fields must specify a subselection
@ -467,6 +472,8 @@ class APIViewTestCases:
fields_string += f'{field.name} {{ id }}\n' fields_string += f'{field.name} {{ id }}\n'
elif field.type.of_type == strawberry_django.fields.types.DjangoModelType: elif field.type.of_type == strawberry_django.fields.types.DjangoModelType:
fields_string += f'{field.name} {{ pk }}\n' fields_string += f'{field.name} {{ pk }}\n'
elif field.is_relation:
fields_string += f'{field.name} {{ id }}\n'
# TODO: Improve field detection logic to avoid nested ArrayFields # TODO: Improve field detection logic to avoid nested ArrayFields
elif field.name == 'extra_choices': elif field.name == 'extra_choices':
continue continue
@ -487,15 +494,6 @@ class APIViewTestCases:
}} }}
""" """
if "_list" not in name:
query = f"""
{{
{name}_list {{
{fields_string}
}}
}}
"""
return query return query
@override_settings(LOGIN_REQUIRED=True) @override_settings(LOGIN_REQUIRED=True)