Add return type to MethodNode

1 files changed, 11 insertions, 0 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
index a39cb2a..cb1f3ac 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/JavaReaderTests.java
@@ -394,6 +394,17 @@ public class JavaReaderTests {
         LocalVariableDeclarationNode declaration = (LocalVariableDeclarationNode) statement;
         assertEquals(typeCoercion(literal(4), ObjectValues.OBJECT), declaration.getInitialValue());
     }
+
+    @Test
+    public void returnTypeOfMethodIsRead() {
+        ClassNode classNode = readClass(
+            "public int identity(int value) {" +
+            "    return value;" +
+            "}");
+
+        MethodNode method = classNode.getMethods().get(0);
+        assertEquals(IntegerValue.REF, method.getReturnType());
+    }
     
     @Test
     public void canDeclareAndReferenceArguments() {