From c09802a7da260d04809471fd1526a83e488fe3fb Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Fri, 15 Jan 2016 18:03:20 +0000 Subject: Add return type to MethodNode --- .../couscous/tests/frontends/java/JavaReaderTests.java | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'src/test/java/org') 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() { -- cgit v1.2.3