From 7b13fe3b90542a276355e8b8de1fe49afe68c7b3 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 22 Nov 2015 10:57:39 +0000
Subject: Combine code for reading methods and lambdas

---
 .../couscous/tests/frontends/java/JavaReaderTests.java        | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

(limited to 'src/test/java')

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() {
-- 
cgit v1.2.3