diff --git a/netbox/netbox/tests/dummy_plugin/graphql.py b/netbox/netbox/tests/dummy_plugin/graphql.py index 27ecd9ce0..bc6643ba9 100644 --- a/netbox/netbox/tests/dummy_plugin/graphql.py +++ b/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 -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), +)