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

+ 4 - 1
netbox/utilities/testing/api.py

@@ -454,7 +454,7 @@ class APIViewTestCases:
                 """
                 print(f"field_name: {field.name} type: {field.type}")
 
-                if field.name == 'front_image':
+                if field.name == 'assigned_object':
                     breakpoint()
                     pass
                 """
@@ -467,6 +467,9 @@ class APIViewTestCases:
                 elif type(field.type) is StrawberryList and type(field.type.of_type) is StrawberryUnion:
                     # this would require a fragment query
                     continue
+                elif type(field.type) is StrawberryUnion:
+                    # this would require a fragment query
+                    continue
                 elif field.type is strawberry_django.fields.types.DjangoModelType:
                     # Dynamic fields must specify a subselection
                     fields_string += f'{field.name} {{ pk }}\n'

+ 2 - 2
netbox/vpn/graphql/types.py

@@ -103,11 +103,11 @@ class IPSecProposalType(OrganizationalObjectType):
 class IPSecPolicyType(OrganizationalObjectType):
 
     @strawberry_django.field
-    def proposals(self) -> List[Annotated["IKEProposalType", strawberry.lazy('vpn.graphql.types')]]:
+    def proposals(self) -> List[Annotated["IPSecProposalType", strawberry.lazy('vpn.graphql.types')]]:
         return self.proposals.all()
 
     @strawberry_django.field
-    def ipsec_profiles(self) -> List[Annotated["IPSecProposalType", strawberry.lazy('vpn.graphql.types')]]:
+    def ipsec_profiles(self) -> List[Annotated["IPSecProfileType", strawberry.lazy('vpn.graphql.types')]]:
         return self.ipsec_profiles.all()