Handle lambdas that return void properly

1 files changed, 13 insertions, 1 deletions

diff --git a/src/test/java/org/zwobble/couscous/tests/frontends/java/InnerTypeReadingTests.java b/src/test/java/org/zwobble/couscous/tests/frontends/java/InnerTypeReadingTests.java
index 8f5c305..0c6245c 100644
--- a/src/test/java/org/zwobble/couscous/tests/frontends/java/InnerTypeReadingTests.java
+++ b/src/test/java/org/zwobble/couscous/tests/frontends/java/InnerTypeReadingTests.java
@@ -8,7 +8,7 @@ import static org.zwobble.couscous.tests.frontends.java.JavaReading.readTypes;
 
 public class InnerTypeReadingTests {
     @Test
-    public void lambdaWithNoArgumentsAndExpressionBodyIsReadAsAnonymousClassCreation() {
+    public void lambdaWithNoArgumentsAndReturnValueAndExpressionBodyIsReadAsAnonymousClassCreation() {
         assertEquals(
             readTypes(generateMethodSource("void",
                 "java.util.function.IntSupplier supplier = new java.util.function.IntSupplier() {\n" +
@@ -37,6 +37,18 @@ public class InnerTypeReadingTests {
             readTypes(generateMethodSource("void",
                 "java.util.function.Function<Integer, Integer> function = (Integer x) -> x;")));
     }
+    @Test
+    public void lambdaWithNoReturnValueAndExpressionBodyIsReadAsAnonymousClassCreation() {
+        assertEquals(
+            readTypes(generateMethodSource("void",
+                "java.lang.Runnable runnable = new java.lang.Runnable() {\n" +
+                    "    public void run() {\n" +
+                    "        System.out.println();\n" +
+                    "    }\n" +
+                    "};")),
+            readTypes(generateMethodSource("void",
+                "java.lang.Runnable runnable = () -> System.out.println();")));
+    }
 
     @Test
     public void expressionStaticMethodReferenceWithStaticReceiverIsReadAsLambda() {