Combine code for reading methods and lambdas

1 files changed, 10 insertions, 1 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 5fd26c0..9dfe8b8 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
@@ -155,7 +155,7 @@ public class JavaReaderTests {
     }
 
     @Test
-    public void lambdaIsReadAsAnonymousClassCreation() {
+    public void lambdaWithNoArgumentsAndExpressionBodyIsReadAsAnonymousClassCreation() {
         assertEquals(
             readClasses(generateMethodSource("void",
                 "java.util.function.Supplier<Integer> supplier = new java.util.function.Supplier<Integer>() {\n" +
@@ -166,6 +166,15 @@ public class JavaReaderTests {
             readClasses(generateMethodSource("void",
                 "java.util.function.Supplier<Integer> supplier = () -> 42;")));
     }
+
+    @Test
+    public void lambdaWithNoArgumentsAndBlockBodyIsReadAsAnonymousClassCreation() {
+        assertEquals(
+            readClasses(generateMethodSource("void",
+                "java.util.function.Supplier<Integer> supplier = () -> 42;")),
+            readClasses(generateMethodSource("void",
+                "java.util.function.Supplier<Integer> supplier = () -> { return 42; };")));
+    }
     
     @Test
     public void argumentIsBoxedIfNecessary() {