Add CsharpCompilerMethodTests

3 files changed, 17 insertions, 4 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java b/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
index 358b4a8..c284d1f 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodCallNode.java
@@ -90,7 +90,7 @@ public class MethodCallNode implements ExpressionNode {
     public ExpressionNode transform(NodeTransformer transformer) {
         return new MethodCallNode(
             transformer.transformReceiver(receiver),
-            methodName,
+            transformer.transformMethodName(signature()),
             transformer.transformExpressions(arguments),
             transformer.transform(type));
     }
diff --git a/src/main/java/org/zwobble/couscous/ast/MethodNode.java b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
index b5dfd39..e0b2432 100644
--- a/src/main/java/org/zwobble/couscous/ast/MethodNode.java
+++ b/src/main/java/org/zwobble/couscous/ast/MethodNode.java
@@ -156,7 +156,7 @@ public class MethodNode implements CallableNode {
         return new MethodNode(
             transformer.transformAnnotations(annotations),
             isStatic,
-            name,
+            transformer.transformMethodName(signature()),
             transformer.transformFormalArguments(arguments),
             transformer.transform(returnType),
             transformer.transformStatements(body));
diff --git a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
index 683b3b9..277813c 100644
--- a/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
+++ b/src/main/java/org/zwobble/couscous/ast/visitors/NodeTransformer.java
@@ -150,6 +150,7 @@ public class NodeTransformer {
     public static class Builder {
         private Function<ExpressionNode, Optional<ExpressionNode>> transformExpression = expression -> Optional.empty();
         private Function<TypeName, TypeName> transformType = type -> type;
+        private Function<MethodSignature, String> transformMethodName = MethodSignature::getName;
 
         public Builder transformExpression(
             Function<ExpressionNode, Optional<ExpressionNode>> transformExpression)
@@ -165,26 +166,38 @@ public class NodeTransformer {
             return this;
         }
 
+        public Builder transformMethodName(Function<MethodSignature, String> transformMethodName) {
+            this.transformMethodName = transformMethodName;
+            return this;
+        }
+
         public NodeTransformer build() {
-            return new NodeTransformer(transformExpression, transformType);
+            return new NodeTransformer(transformExpression, transformType, transformMethodName);
         }
     }
 
     private final Function<ExpressionNode, Optional<ExpressionNode>> transformExpression;
     private final Function<TypeName, TypeName> transformType;
+    private final Function<MethodSignature, String> transformMethodName;
 
     private NodeTransformer(
         Function<ExpressionNode, Optional<ExpressionNode>> transformExpression,
-        Function<TypeName, TypeName> transformType)
+        Function<TypeName, TypeName> transformType,
+        Function<MethodSignature, String> transformMethodName)
     {
         this.transformExpression = transformExpression;
         this.transformType = transformType;
+        this.transformMethodName = transformMethodName;
     }
 
     public TypeName transform(TypeName type) {
         return transformType.apply(type);
     }
 
+    public String transformMethodName(MethodSignature signature) {
+        return transformMethodName.apply(signature);
+    }
+
     public ExpressionNode transformExpression(ExpressionNode expression) {
         return transformExpression.apply(expression)
             .orElseGet(() -> expression.transform(this));