Updated NameUtils.GetEnumValue to preserve leading underscore (#7272)

ChilliCream · Jul 19, 2024 · 0605e2a · 0605e2a
1 parent a7ce3ac
commit 0605e2a
Show file tree

Hide file tree

Showing 3 changed files with 507 additions and 0 deletions.
diff --git a/src/StrawberryShake/CodeGeneration/src/CodeGeneration/Utilities/NameUtils.cs b/src/StrawberryShake/CodeGeneration/src/CodeGeneration/Utilities/NameUtils.cs
@@ -71,6 +71,11 @@ public static string GetEnumValue(string enumValue)
             if (enumValue[i] == '_')
             {
                 upper = true;
+
+                if (i == 0)
+                {
+                    value.Append('_');
+                }
             }
             else if (upper)
             {

diff --git a/src/StrawberryShake/CodeGeneration/test/CodeGeneration.CSharp.Tests/SchemaGeneratorTests.cs b/src/StrawberryShake/CodeGeneration/test/CodeGeneration.CSharp.Tests/SchemaGeneratorTests.cs
@@ -529,4 +529,32 @@ query GetPeopleByPk($id: uuid!) {
             "extend schema @key(fields: \"id\")",
             FileResource.Open("HasuraSchema.graphql"));
     }
+
+    [Fact]
+    public void EnumWithUnderscorePrefixedValues()
+    {
+        AssertResult(
+            """
+            schema {
+                query: Query
+            }
+
+            type Query {
+                field1: Enum1
+            }
+
+            enum Enum1 {
+                _a   # -> "_A"
+                _a_b # -> "_AB"
+                _1   # -> "_1"
+                _1_2 # -> "_12"
+                __2  # -> "_2"
+            }
+            """,
+            """
+            query GetField1 {
+                field1
+            }
+            """);
+    }
 }