Răsfoiți Sursa

9856 test fixes

Arthur 1 an în urmă
părinte
comite
7c66a6aba8
1 a modificat fișierele cu 20 adăugiri și 13 ștergeri
  1. 20 13
      netbox/netbox/tests/dummy_plugin/graphql.py

+ 20 - 13
netbox/netbox/tests/dummy_plugin/graphql.py

@@ -1,21 +1,28 @@
-import graphene
-from graphene_django import DjangoObjectType
-
-from netbox.graphql.fields import ObjectField, ObjectListField
+from typing import List
+import strawberry
+import strawberry_django
+from strawberry.schema.config import StrawberryConfig
 
 
 from . import models
 from . import models
 
 
 
 
-class DummyModelType(DjangoObjectType):
-
-    class Meta:
-        model = models.DummyModel
-        fields = '__all__'
+@strawberry_django.type(
+    models.DummyModel,
+    fields='__all__',
+)
+class DummyModelType:
+    pass
 
 
 
 
-class DummyQuery(graphene.ObjectType):
-    dummymodel = ObjectField(DummyModelType)
-    dummymodel_list = ObjectListField(DummyModelType)
+@strawberry.type
+class DummyQuery:
+    @strawberry.field
+    def dummymodel(self, id: int) -> DummyModelType:
+        return None
+    dummymodel_list: List[DummyModelType] = strawberry_django.field()
 
 
 
 
-schema = DummyQuery
+schema = strawberry.Schema(
+    query=DummyQuery,
+    config=StrawberryConfig(auto_camel_case=False),
+)