From 7c66a6aba8457458bfbd781cb3fbec43e3d88c04 Mon Sep 17 00:00:00 2001
From: Arthur <worldnomad@gmail.com>
Date: Wed, 6 Mar 2024 15:25:06 -0800
Subject: [PATCH] 9856 test fixes

---
 netbox/netbox/tests/dummy_plugin/graphql.py | 33 +++++++++++++--------
 1 file changed, 20 insertions(+), 13 deletions(-)

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),
+)