fix: tests

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

Hide file tree

Showing 38 changed files with 781 additions and 968 deletions.
diff --git a/graphene_federation/appolo_versions/__init__.py b/graphene_federation/appolo_versions/__init__.py
@@ -30,6 +30,6 @@ def get_directive_from_name(
     )
     if directive is None:
         raise ValueError(
-            f"@{directive_name} not found on federation version {federation_version}"
+            f"@{directive_name} not supported on federation version {federation_version}"
         )
     return directive
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,
 )