Improve support for reading method references

1 files changed, 10 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 9901931..8f5c305 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
@@ -48,7 +48,7 @@ public class InnerTypeReadingTests {
     }
 
     @Test
-    public void expressionInstanceMethodReferenceWithStaticReceiverIsReadAsLambda() {
+    public void expressionInstanceMethodReferenceWithStaticReceiverAndNoArgsIsReadAsLambda() {
         assertEquals(
             readTypes(generateMethodSource("void",
                 "java.util.function.Function<String, Integer> function = arg0 -> arg0.length();")),
@@ -57,6 +57,15 @@ public class InnerTypeReadingTests {
     }
 
     @Test
+    public void expressionInstanceMethodReferenceWithStaticReceiverAndSomeArgsIsReadAsLambda() {
+        assertEquals(
+            readTypes(generateMethodSource("void",
+                "java.util.function.BiPredicate<Object, Object> function = (arg0, arg1) -> arg0.equals(arg1);")),
+            readTypes(generateMethodSource("void",
+                "java.util.function.BiPredicate<Object, Object> function = Object::equals;")));
+    }
+
+    @Test
     public void expressionConstructorReferenceWithStaticReceiverIsReadAsLambda() {
         assertEquals(
             readTypes(generateMethodSource("void",