Read return statements without value

1 files changed, 6 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 35756f5..d455301 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
@@ -43,6 +43,12 @@ import static org.zwobble.couscous.util.ExtraSets.set;
 
 public class JavaReaderTests {
     @Test
+    public void canReadReturnWithoutValue() {
+        StatementNode statement = readStatement("void", "return;");
+        assertEquals(statement, returns(LiteralNode.UNIT));
+    }
+
+    @Test
     public void canDeclareAndReferenceLocalVariables() {
         List<StatementNode> statements = readStatements("int", "int x = 4; return x;");