Fix: enum values should be renamed using field renaming

1 files changed, 4 insertions, 1 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/EnumNode.java b/src/main/java/org/zwobble/couscous/ast/EnumNode.java
index 162d2a0..6ecec65 100644
--- a/src/main/java/org/zwobble/couscous/ast/EnumNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/EnumNode.java
@@ -8,6 +8,7 @@ import org.zwobble.couscous.types.Type;
 import java.util.List;
 import java.util.Set;
 
+import static org.zwobble.couscous.util.ExtraLists.eagerMap;
 import static org.zwobble.couscous.util.ExtraLists.list;
 import static org.zwobble.couscous.util.ExtraSets.set;
 
@@ -50,7 +51,9 @@ public class EnumNode implements TypeNode {
 
     @Override
     public TypeNode transform(NodeTransformer transformer) {
-        return new EnumNode(transformer.transform(name), values);
+        return new EnumNode(
+            transformer.transform(name),
+            eagerMap(values, transformer::transformFieldName));
     }
 
     @Override