Bug Fix: Custom Enum (#15)

graphene_federation/tests/test_custom_enum.py

@@ -0,0 +1,52 @@
+import graphene
+from graphene import ObjectType
+from graphql import graphql_sync
+
+from graphene_federation import build_schema, shareable, inaccessible
+
+
+def test_custom_enum():
+    class Episode(graphene.Enum):
+        NEWHOPE = 4
+        EMPIRE = 5
+        JEDI = 6
+
+    @shareable
+    class TestCustomEnum(graphene.ObjectType):
+        test_shareable_scalar = shareable(Episode())
+        test_inaccessible_scalar = inaccessible(Episode())
+
+    class Query(ObjectType):
+        test = Episode()
+        test2 = graphene.List(TestCustomEnum, required=True)
+
+    schema = build_schema(
+        query=Query, enable_federation_2=True, types=(TestCustomEnum,)
+    )
+    query = """
+        query {
+            _service {
+                sdl
+            }
+        }
+        """
+    result = graphql_sync(schema.graphql_schema, query)
+    assert (
+        result.data["_service"]["sdl"].strip()
+        == """extend schema @link(url: "https://specs.apollo.dev/federation/v2.0", import: ["@inaccessible", "@shareable"])
+type TestCustomEnum  @shareable {
+  testShareableScalar: Episode @shareable
+  testInaccessibleScalar: Episode @inaccessible
+}
+
+enum Episode {
+  NEWHOPE
+  EMPIRE
+  JEDI
+}
+
+type Query {
+  test: Episode
+  test2: [TestCustomEnum]!
+}"""
+    )

graphene_federation/utils.py

@@ -3,6 +3,7 @@
 import graphene
 from graphene import Schema, ObjectType
 from graphene.types.definitions import GrapheneObjectType
+from graphene.types.enum import EnumOptions
 from graphene.types.scalars import ScalarOptions
 from graphene.types.union import UnionOptions
 from graphene.utils.str_converters import to_camel_case
@@ -93,6 +94,7 @@ def get_attributed_fields(attribute: str, schema: Schema):
             not hasattr(type_, "graphene_type")
             or isinstance(type_.graphene_type._meta, UnionOptions)
             or isinstance(type_.graphene_type._meta, ScalarOptions)
+            or isinstance(type_.graphene_type._meta, EnumOptions)
         ):
             continue
         for field in list(type_.graphene_type._meta.fields):