From d310a8cf9487f253fbc4152d182dfdef460aeb6d Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Wed, 27 Apr 2016 21:56:02 +0100
Subject: Read return statements without value

---
 .../org/zwobble/couscous/frontends/java/JavaStatementReader.java     | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org')

diff --git a/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java b/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
index e75585f..6b71684 100644
--- a/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
+++ b/src/main/java/org/zwobble/couscous/frontends/java/JavaStatementReader.java
@@ -93,7 +93,10 @@ class JavaStatementReader {
     }
 
     private StatementNode readReturnStatement(ReturnStatement statement) {
-        return ReturnNode.returns(readExpression(returnType.get(), statement.getExpression()));
+        ExpressionNode value = statement.getExpression() == null
+            ? LiteralNode.UNIT
+            : readExpression(returnType.get(), statement.getExpression());
+        return ReturnNode.returns(value);
     }
 
     private StatementNode readThrowStatement(ThrowStatement statement) {
-- 
cgit v1.2.3