Mascali
/
netbox
mirror of https://github.com/netbox-community/netbox.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

9856 fix tests / remove debug code

This commit is contained in:
Arthur 2024-03-07 12:48:57 -08:00
parent 133d6bfcb3
commit 7b6a603111
1 changed files with 9 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -452,15 +452,13 @@ class APIViewTestCases:
fields_string = ''
for field in type_class.__strawberry_definition__.fields:
"""
print(f"field_name: {field.name} type: {field.type}")
if field.name == 'assigned_object':
breakpoint()
pass
"""
if field.type in (strawberry_django.fields.types.DjangoFileType, strawberry_django.fields.types.DjangoImageType):
if (
type(field.type) in (
strawberry_django.fields.types.DjangoFileType, strawberry_django.fields.types.DjangoImageType) or
type(field.type) is StrawberryOptional and field.type.of_type in (
strawberry_django.fields.types.DjangoFileType, strawberry_django.fields.types.DjangoImageType)
):
# image / file fields nullable or not...
fields_string += f'{field.name} {{ name }}\n'
elif type(field.type) is StrawberryList and type(field.type.of_type) is LazyType:
# List of related objects (queryset)
@ -471,22 +469,8 @@ class APIViewTestCases:
elif type(field.type) is StrawberryUnion:
# this would require a fragment query
continue
elif field.type is strawberry_django.fields.types.DjangoModelType:
print("DjangoModelType")
print("--------------------------")
print(f"{self.model} -> {field.name}")
print("")
# Dynamic fields must specify a subselection
fields_string += f'{field.name} {{ pk }}\n'
elif type(field.type) is StrawberryOptional:
if type(field.type.of_type) is LazyType:
fields_string += f'{field.name} {{ id }}\n'
elif field.type.of_type == strawberry_django.fields.types.DjangoModelType:
print("DjangoModelType")
print("--------------------------")
print(f"{self.model} -> {field.name}")
print("")
fields_string += f'{field.name} {{ pk }}\n'
elif type(field.type) is StrawberryOptional and type(field.type.of_type) is LazyType:
fields_string += f'{field.name} {{ id }}\n'
elif hasattr(field, 'is_relation') and field.is_relation:
# Note: StrawberryField types do not have is_relation
fields_string += f'{field.name} {{ id }}\n'