1 /*
  2  * Copyright (c) 2024, 2026, Oracle and/or its affiliates. All rights reserved.
  3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  4  *
  5  * This code is free software; you can redistribute it and/or modify it
  6  * under the terms of the GNU General Public License version 2 only, as
  7  * published by the Free Software Foundation.
  8  *
  9  * This code is distributed in the hope that it will be useful, but WITHOUT
 10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 12  * version 2 for more details (a copy is included in the LICENSE file that
 13  * accompanied this code).
 14  *
 15  * You should have received a copy of the GNU General Public License version
 16  * 2 along with this work; if not, write to the Free Software Foundation,
 17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 18  *
 19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 20  * or visit www.oracle.com if you need additional information or have any
 21  * questions.
 22  */
 23 
 24 import jdk.incubator.code.Reflect;
 25 import java.util.function.BiConsumer;
 26 import java.util.function.Consumer;
 27 import java.util.function.Function;
 28 import java.util.function.IntFunction;
 29 import java.util.function.Supplier;
 30 
 31 /*
 32  * @test
 33  * @summary Smoke test for code reflection with method reference expressions.
 34  * @modules jdk.incubator.code
 35  * @build MethodReferenceTest
 36  * @build CodeReflectionTester
 37  * @run main CodeReflectionTester MethodReferenceTest
 38  */
 39 
 40 public class MethodReferenceTest {
 41 
 42     static void m_s(String s) {}
 43 
 44     void m(String s) {}
 45 
 46     @Reflect
 47     @IR("""
 48             func @"test1" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
 49                 %1 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda @lambda.isReflectable=true (%2 : java.type:"java.lang.String")java.type:"void" -> {
 50                     %3 : Var<java.type:"java.lang.String"> = var %2 @"x$0";
 51                     %4 : java.type:"java.lang.String" = var.load %3;
 52                     invoke %4 @java.ref:"MethodReferenceTest::m_s(java.lang.String):void";
 53                     return;
 54                 };
 55                 %5 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %1 @"c";
 56                 return;
 57             };
 58             """)
 59     void test1() {
 60         Consumer<String> c = MethodReferenceTest::m_s;
 61     }
 62 
 63     @Reflect
 64     @IR("""
 65             func @"test2" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
 66                 %1 : java.type:"java.util.function.BiConsumer<MethodReferenceTest, java.lang.String>" = lambda @lambda.isReflectable=true (%2 : java.type:"MethodReferenceTest", %3 : java.type:"java.lang.String")java.type:"void" -> {
 67                     %4 : Var<java.type:"MethodReferenceTest"> = var %2 @"rec$";
 68                     %5 : Var<java.type:"java.lang.String"> = var %3 @"x$0";
 69                     %6 : java.type:"MethodReferenceTest" = var.load %4;
 70                     %7 : java.type:"java.lang.String" = var.load %5;
 71                     invoke %6 %7 @java.ref:"MethodReferenceTest::m(java.lang.String):void";
 72                     return;
 73                 };
 74                 %8 : Var<java.type:"java.util.function.BiConsumer<MethodReferenceTest, java.lang.String>"> = var %1 @"bc";
 75                 return;
 76             };
 77             """)
 78     void test2() {
 79         BiConsumer<MethodReferenceTest, String> bc = MethodReferenceTest::m;
 80     }
 81 
 82     @Reflect
 83     @IR("""
 84             func @"test3" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
 85                 %1 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda @lambda.isReflectable=true (%2 : java.type:"java.lang.String")java.type:"void" -> {
 86                     %3 : Var<java.type:"java.lang.String"> = var %2 @"x$0";
 87                     %4 : java.type:"java.lang.String" = var.load %3;
 88                     invoke %0 %4 @java.ref:"MethodReferenceTest::m(java.lang.String):void";
 89                     return;
 90                 };
 91                 %5 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %1 @"c";
 92                 return;
 93             };
 94             """)
 95     void test3() {
 96         Consumer<String> c = this::m;
 97     }
 98 
 99     class A<T> {
100         T m(T t) { return t; }
101     }
102 
103     <T> A<T> a(T t) { return null; }
104 
105     @Reflect
106     @IR("""
107             func @"test4" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
108                 %1 : java.type:"java.lang.String" = constant @"s";
109                 %2 : java.type:"MethodReferenceTest::A<java.lang.String>" = invoke %0 %1 @java.ref:"MethodReferenceTest::a(java.lang.Object):MethodReferenceTest::A";
110                 %3 : Var<java.type:"MethodReferenceTest::A<java.lang.String>"> = var %2 @"rec$";
111                 %4 : java.type:"java.util.function.Function<java.lang.String, java.lang.String>" = lambda @lambda.isReflectable=true (%5 : java.type:"java.lang.String")java.type:"java.lang.String" -> {
112                     %6 : Var<java.type:"java.lang.String"> = var %5 @"x$0";
113                     %7 : java.type:"MethodReferenceTest::A<java.lang.String>" = var.load %3;
114                     %8 : java.type:"java.lang.String" = var.load %6;
115                     %9 : java.type:"java.lang.String" = invoke %7 %8 @java.ref:"MethodReferenceTest::A::m(java.lang.Object):java.lang.Object";
116                     %10 : java.type:"java.lang.String" = cast %9 @java.type:"java.lang.String";
117                     return %10;
118                 };
119                 %11 : Var<java.type:"java.util.function.Function<java.lang.String, java.lang.String>"> = var %4 @"f";
120                 return;
121             };
122             """)
123     void test4() {
124         Function<String, String> f = a("s")::m;
125     }
126 
127     @Reflect
128     @IR("""
129             func @"test5" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
130                 %1 : java.type:"java.io.PrintStream" = field.load @java.ref:"java.lang.System::out:java.io.PrintStream";
131                 %2 : Var<java.type:"java.io.PrintStream"> = var %1 @"rec$";
132                 %3 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda @lambda.isReflectable=true (%4 : java.type:"java.lang.String")java.type:"void" -> {
133                     %5 : Var<java.type:"java.lang.String"> = var %4 @"x$0";
134                     %6 : java.type:"java.io.PrintStream" = var.load %2;
135                     %7 : java.type:"java.lang.String" = var.load %5;
136                     invoke %6 %7 @java.ref:"java.io.PrintStream::println(java.lang.String):void";
137                     return;
138                 };
139                 %8 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %3 @"c3";
140                 return;
141             };
142             """)
143     void test5() {
144         Consumer<String> c3 = System.out::println;
145     }
146 
147     static class X {
148         X(int i) {}
149     }
150 
151     @Reflect
152     @IR("""
153             func @"test6" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
154                 %1 : java.type:"java.util.function.Function<java.lang.Integer, MethodReferenceTest$X>" = lambda @lambda.isReflectable=true (%2 : java.type:"java.lang.Integer")java.type:"MethodReferenceTest$X" -> {
155                     %3 : Var<java.type:"java.lang.Integer"> = var %2 @"x$0";
156                     %4 : java.type:"java.lang.Integer" = var.load %3;
157                     %5 : java.type:"int" = invoke %4 @java.ref:"java.lang.Integer::intValue():int";
158                     %6 : java.type:"MethodReferenceTest$X" = new %5 @java.ref:"MethodReferenceTest$X::(int)";
159                     return %6;
160                 };
161                 %7 : Var<java.type:"java.util.function.Function<java.lang.Integer, MethodReferenceTest$X>"> = var %1 @"xNew";
162                 return;
163             };
164             """)
165     void test6() {
166         Function<Integer, X> xNew = X::new;
167     }
168 
169     @Reflect
170     @IR("""
171             func @"test7" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
172                 %1 : java.type:"java.util.function.Supplier<MethodReferenceTest::A<java.lang.String>>" = lambda @lambda.isReflectable=true ()java.type:"MethodReferenceTest::A<java.lang.String>" -> {
173                     %2 : java.type:"MethodReferenceTest::A<java.lang.String>" = new %0 @java.ref:"MethodReferenceTest::A::(MethodReferenceTest)";
174                     return %2;
175                 };
176                 %3 : Var<java.type:"java.util.function.Supplier<MethodReferenceTest::A<java.lang.String>>"> = var %1 @"aNew";
177                 return;
178             };
179             """)
180     void test7() {
181         Supplier<A<String>> aNew = A::new;
182     }
183 
184     @Reflect
185     @IR("""
186             func @"test8" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
187                 %1 : java.type:"java.util.function.IntFunction<MethodReferenceTest::A<java.lang.String>[]>" = lambda @lambda.isReflectable=true (%2 : java.type:"int")java.type:"MethodReferenceTest::A<java.lang.String>[]" -> {
188                     %3 : Var<java.type:"int"> = var %2 @"x$0";
189                     %4 : java.type:"int" = var.load %3;
190                     %5 : java.type:"MethodReferenceTest::A[]" = new %4 @java.ref:"MethodReferenceTest::A[]::(int)";
191                     return %5;
192                 };
193                 %6 : Var<java.type:"java.util.function.IntFunction<MethodReferenceTest::A<java.lang.String>[]>"> = var %1 @"aNewArray";
194                 return;
195             };
196             """)
197     void test8() {
198         IntFunction<A<String>[]> aNewArray = A[]::new;
199     }
200 
201 }