Clean up GraphQL tests

netbox/circuits/graphql/types.py

@@ -1,33 +1,21 @@
 from circuits import filtersets, models
-from netbox.graphql.types import *
+from netbox.graphql.types import BaseObjectType, ObjectType, TaggedObjectType
 
 __all__ = (
-    'CircuitType',
     'CircuitTerminationType',
+    'CircuitType',
     'CircuitTypeType',
     'ProviderType',
     'ProviderNetworkType',
 )
 
 
-#
+class CircuitTerminationType(BaseObjectType):
-# Object types
-#
-
-class ProviderType(TaggedObjectType):
 
     class Meta:
-        model = models.Provider
+        model = models.CircuitTermination
         fields = '__all__'
-        filterset_class = filtersets.ProviderFilterSet
+        filterset_class = filtersets.CircuitTerminationFilterSet
-
-
-class ProviderNetworkType(TaggedObjectType):
-
-    class Meta:
-        model = models.ProviderNetwork
-        fields = '__all__'
-        filterset_class = filtersets.ProviderNetworkFilterSet
 
 
 class CircuitType(TaggedObjectType):
@@ -46,9 +34,17 @@ class CircuitTypeType(ObjectType):
         filterset_class = filtersets.CircuitTypeFilterSet
 
 
-class CircuitTerminationType(BaseObjectType):
+class ProviderType(TaggedObjectType):
 
     class Meta:
-        model = models.CircuitTermination
+        model = models.Provider
         fields = '__all__'
-        filterset_class = filtersets.CircuitTerminationFilterSet
+        filterset_class = filtersets.ProviderFilterSet
+
+
+class ProviderNetworkType(TaggedObjectType):
+
+    class Meta:
+        model = models.ProviderNetwork
+        fields = '__all__'
+        filterset_class = filtersets.ProviderNetworkFilterSet

netbox/netbox/graphql/schema.py

@@ -1,10 +1,12 @@
 import graphene
 
 from circuits.graphql.schema import CircuitsQuery
+from ipam.graphql.schema import IPAMQuery
 
 
 class Query(
     CircuitsQuery,
+    IPAMQuery,
     graphene.ObjectType
 ):
     pass

netbox/utilities/testing/api.py

@@ -428,7 +428,7 @@ class APIViewTestCases:
         @override_settings(LOGIN_REQUIRED=True)
         def test_graphql_get_object(self):
             url = reverse('graphql')
-            object_type = self.model._meta.verbose_name.replace(' ', '_')
+            object_type = self.model._meta.verbose_name.lower().replace(' ', '_')
             object_id = self._get_queryset().first().pk
             query = f"""
             {{
@@ -459,7 +459,7 @@ class APIViewTestCases:
         @override_settings(LOGIN_REQUIRED=True)
         def test_graphql_list_objects(self):
             url = reverse('graphql')
-            object_type = self.model._meta.verbose_name_plural.replace(' ', '_')
+            object_type = self.model._meta.verbose_name_plural.lower().replace(' ', '_')
             query = f"""
             {{
                 {object_type} {{