fix: tests

strollby · Jan 13, 2024 · e322202 · e322202
1 parent 5f3aa58
commit e322202
Show file tree

Hide file tree

Showing 38 changed files with 781 additions and 969 deletions.
diff --git a/graphene_federation/appolo_versions/v1_0.py b/graphene_federation/appolo_versions/v1_0.py
@@ -2,6 +2,7 @@
 from graphql import GraphQLArgument, GraphQLDirective, GraphQLNonNull
 
 from ..scalars import _FieldSet
+from ..transform import field_set_case_transform
 from ..validators import validate_key, validate_requires
 
 key_directive = CustomDirective(
@@ -15,6 +16,7 @@
     is_repeatable=True,
     add_definition_to_schema=False,
     non_field_validator=validate_key,
+    input_transform=field_set_case_transform,
 )
 
 requires_directive = CustomDirective(
@@ -27,6 +29,7 @@
     is_repeatable=True,
     add_definition_to_schema=False,
     field_validator=validate_requires,
+    input_transform=field_set_case_transform,
 )
 
 

diff --git a/graphene_federation/appolo_versions/v2_0.py b/graphene_federation/appolo_versions/v2_0.py
@@ -9,6 +9,7 @@
 
 from .v1_0 import extends_directive
 from ..scalars import FieldSet
+from ..transform import field_set_case_transform
 from ..validators import validate_key, validate_requires
 
 key_directive = CustomDirective(
@@ -26,6 +27,7 @@
     is_repeatable=True,
     add_definition_to_schema=False,
     non_field_validator=validate_key,
+    input_transform=field_set_case_transform,
 )
 
 requires_directive = CustomDirective(
@@ -38,6 +40,7 @@
     is_repeatable=True,
     add_definition_to_schema=False,
     field_validator=validate_requires,
+    input_transform=field_set_case_transform,
 )
 
 

diff --git a/graphene_federation/directives/utils.py b/graphene_federation/directives/utils.py
@@ -1,6 +1,6 @@
 from typing import Any
 
-from graphene import Enum, InputObjectType, Interface, ObjectType, Scalar, Union
+from graphene import Enum, InputObjectType, Interface, ObjectType, Union
 
 
 def is_non_field(field: Any):
@@ -11,7 +11,6 @@ def is_non_field(field: Any):
             or issubclass(field, InputObjectType)
             or issubclass(field, Enum)
             or issubclass(field, Union)
-            or issubclass(field, Scalar)
         ):
             return True
         else: