Arthur 1 год назад
Родитель
Сommit
005a339745
2 измененных файлов с 3 добавлено и 6 удалено
  1. 1 0
      netbox/dcim/graphql/types.py
  2. 2 6
      netbox/utilities/testing/api.py

+ 1 - 0
netbox/dcim/graphql/types.py

@@ -129,6 +129,7 @@ class CableTerminationType(NetBoxObjectType):
 )
 class CableType(NetBoxObjectType):
     color: str
+    tenant: Annotated["TenantType", strawberry.lazy('tenancy.graphql.types')] | None
 
     @strawberry_django.field
     def terminations(self) -> List[CableTerminationType]:

+ 2 - 6
netbox/utilities/testing/api.py

@@ -472,25 +472,21 @@ class APIViewTestCases:
                     # this would require a fragment query
                     continue
                 elif field.type is strawberry_django.fields.types.DjangoModelType:
-                    print("")
-                    print("")
                     print("DjangoModelType")
                     print("--------------------------")
                     print(f"{self.model} -> {field.name}")
                     print("")
                     # Dynamic fields must specify a subselection
-                    fields_string += f'{field.name} {{ id }}\n'
+                    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("")
-                        print("")
                         print("DjangoModelType")
                         print("--------------------------")
                         print(f"{self.model} -> {field.name}")
                         print("")
-                        fields_string += f'{field.name} {{ id }}\n'
+                        fields_string += f'{field.name} {{ pk }}\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'