Add support for passing name to enum_value

Author: patrick91

strawberry/types/enum.py

@@ -50,6 +50,7 @@ def is_graphql_generic(self) -> bool:
 @dataclasses.dataclass
 class EnumValueDefinition:
     value: Any
+    graphql_name: Optional[str] = None
     deprecation_reason: Optional[str] = None
     directives: Iterable[object] = ()
     description: Optional[str] = None
@@ -60,6 +61,7 @@ def __int__(self) -> int:
 
 def enum_value(
     value: Any,
+    name: Optional[str] = None,
     deprecation_reason: Optional[str] = None,
     directives: Iterable[object] = (),
     description: Optional[str] = None,
@@ -68,6 +70,7 @@ def enum_value(
 
     Args:
         value: The value of the enum member.
+        name: The GraphQL name of the enum member.
         deprecation_reason: The deprecation reason of the enum member,
             setting this will mark the enum member as deprecated.
         directives: The directives to attach to the enum member.
@@ -90,6 +93,7 @@ class MyEnum(Enum):
     """
     return EnumValueDefinition(
         value=value,
+        graphql_name=name,
         deprecation_reason=deprecation_reason,
         directives=directives,
         description=description,
@@ -123,12 +127,16 @@ def _process_enum(
             item_directives = item_value.directives
             enum_value_description = item_value.description
             deprecation_reason = item_value.deprecation_reason
-            item_value = item_value.value
 
             # update _value2member_map_ so that doing `MyEnum.MY_VALUE` and
             # `MyEnum['MY_VALUE']` both work
-            cls._value2member_map_[item_value] = item
-            cls._member_map_[item_name]._value_ = item_value
+            cls._value2member_map_[item_value.value] = item
+            cls._member_map_[item_name]._value_ = item_value.value
+
+            if item_value.graphql_name:
+                item_name = item_value.graphql_name
+
+            item_value = item_value.value
 
         value = EnumValue(
             item_name,

tests/schema/test_enum.py

@@ -431,3 +431,46 @@ def receive_enum(self, test: TestEnum) -> str:
     assert result.data["a"] == "TestEnum.A"
     assert result.data["b"] == "TestEnum.B"
     assert result.data["c"] == "TestEnum.C"
+
+
+def test_can_give_custom_name_to_enum_value():
+    @strawberry.enum
+    class IceCreamFlavour(Enum):
+        VANILLA = "vanilla"
+        CHOCOLATE_COOKIE = strawberry.enum_value("chocolate", name="chocolateCookie")
+
+    @strawberry.type
+    class Query:
+        @strawberry.field
+        def best_flavour(self) -> IceCreamFlavour:
+            return IceCreamFlavour.CHOCOLATE_COOKIE
+
+    schema = strawberry.Schema(query=Query)
+
+    assert (
+        str(schema)
+        == dedent(
+            """
+        enum IceCreamFlavour {
+          VANILLA
+          chocolateCookie
+        }
+
+        type Query {
+          bestFlavour: IceCreamFlavour!
+        }
+        """
+        ).strip()
+    )
+
+    query = """
+    {
+        bestFlavour
+    }
+    """
+
+    result = schema.execute_sync(query)
+
+    assert not result.errors
+    assert result.data
+    assert result.data["bestFlavour"] == "chocolateCookie"