codegen: Fix enum value generated name to not be lower cased

Fixes the generated enum value names to not be lower cased when capitalized. Related to aws/aws-sdk-go-v2#1013
jasdel · Jan 6, 2021 · 51e377a · 51e377a
1 parent 12a5ddb
commit 51e377a
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/EnumGenerator.java b/codegen/smithy-go-codegen/src/main/java/software/amazon/smithy/go/codegen/EnumGenerator.java
@@ -59,8 +59,14 @@ public void run() {
                 for (EnumDefinition definition : enumTrait.getValues()) {
                     StringBuilder labelBuilder = new StringBuilder(symbol.getName());
                     String name = definition.getName().get();
-                    for (String part : name.split("(?U)[\\W_]")) {
-                        labelBuilder.append(StringUtils.capitalize(part.toLowerCase(Locale.US)));
+
+                    // Only split, capital case, and join the elements if there are non-word characters.
+                    if (name.matches("(?U)[\\W_]")) {
+                        for (String part : name.split("(?U)[\\W_]")) {
+                            labelBuilder.append(StringUtils.capitalize(part.toLowerCase(Locale.US)));
+                        }
+                    } else {
+                        labelBuilder.append(StringUtils.capitalize(name));
                     }
                     String label = labelBuilder.toString();