blob: 982bfe90054f375053810e469c51ac36eebcef1c [file] [log] [blame]
jrose900bafd2010-10-30 21:08:23 -07001/*
alanb0d058232012-11-02 15:50:11 +00002 * Copyright (c) 2009, 2012, Oracle and/or its affiliates. All rights reserved.
jrose900bafd2010-10-30 21:08:23 -07003 * 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. Oracle designates this
8 * particular file as subject to the "Classpath" exception as provided
9 * by Oracle in the LICENSE file that accompanied this code.
10 *
11 * This code is distributed in the hope that it will be useful, but WITHOUT
12 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
13 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
14 * version 2 for more details (a copy is included in the LICENSE file that
15 * accompanied this code).
16 *
17 * You should have received a copy of the GNU General Public License version
18 * 2 along with this work; if not, write to the Free Software Foundation,
19 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
20 *
21 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
22 * or visit www.oracle.com if you need additional information or have any
23 * questions.
24 */
25
26/* @test
jroseada69fa2011-03-23 23:02:31 -070027 * @summary example code used in javadoc for java.lang.invoke API
jrose9633f5f2011-04-07 22:07:06 -070028 * @compile JavaDocExamplesTest.java
henryjenf6613f42013-08-12 12:11:04 -070029 * @run testng/othervm test.java.lang.invoke.JavaDocExamplesTest
jrose900bafd2010-10-30 21:08:23 -070030 */
31
jroseada69fa2011-03-23 23:02:31 -070032package test.java.lang.invoke;
jrose900bafd2010-10-30 21:08:23 -070033
jroseada69fa2011-03-23 23:02:31 -070034import java.lang.invoke.*;
35import static java.lang.invoke.MethodHandles.*;
36import static java.lang.invoke.MethodType.*;
jrose900bafd2010-10-30 21:08:23 -070037
jrose900bafd2010-10-30 21:08:23 -070038import java.util.*;
39
henryjenf6613f42013-08-12 12:11:04 -070040import org.testng.*;
41import static org.testng.AssertJUnit.*;
42import org.testng.annotations.*;
jrose900bafd2010-10-30 21:08:23 -070043
44/**
45 * @author jrose
46 */
47public class JavaDocExamplesTest {
henryjenf6613f42013-08-12 12:11:04 -070048 /** Wrapper for running the TestNG tests in this module.
49 * Put TestNG on the classpath!
jrose900bafd2010-10-30 21:08:23 -070050 */
jrose9b82ad62011-05-26 17:37:36 -070051 public static void main(String... ignore) throws Throwable {
jrose9b82ad62011-05-26 17:37:36 -070052 new JavaDocExamplesTest().run();
53 }
henryjenf6613f42013-08-12 12:11:04 -070054
jrose9b82ad62011-05-26 17:37:36 -070055 public void run() throws Throwable {
jrose942f3d32013-10-05 05:30:39 -070056 testMisc();
57 testFindStatic();
58 testFindConstructor();
jrose9b82ad62011-05-26 17:37:36 -070059 testFindVirtual();
jrose942f3d32013-10-05 05:30:39 -070060 testFindSpecial();
jrose9b82ad62011-05-26 17:37:36 -070061 testPermuteArguments();
62 testDropArguments();
63 testFilterArguments();
64 testFoldArguments();
jroseb4be0262011-07-16 15:44:33 -070065 testFoldArguments2();
jrose9b82ad62011-05-26 17:37:36 -070066 testMethodHandlesSummary();
67 testAsSpreader();
68 testAsCollector();
69 testAsVarargsCollector();
70 testAsFixedArity();
71 testAsTypeCornerCases();
72 testMutableCallSite();
jrose900bafd2010-10-30 21:08:23 -070073 }
74 // How much output?
jrose9b82ad62011-05-26 17:37:36 -070075 static final Class<?> THIS_CLASS = JavaDocExamplesTest.class;
76 static int verbosity = Integer.getInteger(THIS_CLASS.getSimpleName()+".verbosity", 0);
77
jrose900bafd2010-10-30 21:08:23 -070078
79{}
80static final private Lookup LOOKUP = lookup();
81// static final private MethodHandle CONCAT_1 = LOOKUP.findVirtual(String.class,
82// "concat", methodType(String.class, String.class));
83// static final private MethodHandle HASHCODE_1 = LOOKUP.findVirtual(Object.class,
84// "hashCode", methodType(int.class));
85
jrosef15905c2011-02-11 01:26:32 -080086// form required if ReflectiveOperationException is intercepted:
jrose9b82ad62011-05-26 17:37:36 -070087 static final private MethodHandle CONCAT_2, HASHCODE_2, ADD_2, SUB_2;
jrose900bafd2010-10-30 21:08:23 -070088static {
89 try {
jrose9b82ad62011-05-26 17:37:36 -070090 Class<?> THIS_CLASS = LOOKUP.lookupClass();
jrose900bafd2010-10-30 21:08:23 -070091 CONCAT_2 = LOOKUP.findVirtual(String.class,
92 "concat", methodType(String.class, String.class));
93 HASHCODE_2 = LOOKUP.findVirtual(Object.class,
94 "hashCode", methodType(int.class));
jrose9b82ad62011-05-26 17:37:36 -070095 ADD_2 = LOOKUP.findStatic(THIS_CLASS, "add", methodType(int.class, int.class, int.class));
96 SUB_2 = LOOKUP.findStatic(THIS_CLASS, "sub", methodType(int.class, int.class, int.class));
jrosef15905c2011-02-11 01:26:32 -080097 } catch (ReflectiveOperationException ex) {
jrose900bafd2010-10-30 21:08:23 -070098 throw new RuntimeException(ex);
99 }
100}
jrose9b82ad62011-05-26 17:37:36 -0700101 static int add(int x, int y) { return x + y; }
102 static int sub(int x, int y) { return x - y; }
103
jrose900bafd2010-10-30 21:08:23 -0700104{}
105
jrose942f3d32013-10-05 05:30:39 -0700106 @Test public void testMisc() throws Throwable {
107// Extra tests, not from javadoc:
jrose900bafd2010-10-30 21:08:23 -0700108{}
109MethodHandle CONCAT_3 = LOOKUP.findVirtual(String.class,
110 "concat", methodType(String.class, String.class));
111MethodHandle HASHCODE_3 = LOOKUP.findVirtual(Object.class,
112 "hashCode", methodType(int.class));
113//assertEquals("xy", (String) CONCAT_1.invokeExact("x", "y"));
jrose1c1bfac2010-11-22 22:41:31 -0800114assertEquals("xy", (String) CONCAT_2.invokeExact("x", "y"));
115assertEquals("xy", (String) CONCAT_3.invokeExact("x", "y"));
116//assertEquals("xy".hashCode(), (int) HASHCODE_1.invokeExact((Object)"xy"));
117assertEquals("xy".hashCode(), (int) HASHCODE_2.invokeExact((Object)"xy"));
118assertEquals("xy".hashCode(), (int) HASHCODE_3.invokeExact((Object)"xy"));
jrose900bafd2010-10-30 21:08:23 -0700119{}
120 }
jrose9b82ad62011-05-26 17:37:36 -0700121
jrose942f3d32013-10-05 05:30:39 -0700122 @Test public void testFindStatic() throws Throwable {
123{}
124MethodHandle MH_asList = publicLookup().findStatic(Arrays.class,
125 "asList", methodType(List.class, Object[].class));
126assertEquals("[x, y]", MH_asList.invoke("x", "y").toString());
127{}
128 }
129
130 @Test public void testFindVirtual() throws Throwable {
131{}
132MethodHandle MH_concat = publicLookup().findVirtual(String.class,
133 "concat", methodType(String.class, String.class));
134MethodHandle MH_hashCode = publicLookup().findVirtual(Object.class,
135 "hashCode", methodType(int.class));
136MethodHandle MH_hashCode_String = publicLookup().findVirtual(String.class,
137 "hashCode", methodType(int.class));
138assertEquals("xy", (String) MH_concat.invokeExact("x", "y"));
139assertEquals("xy".hashCode(), (int) MH_hashCode.invokeExact((Object)"xy"));
140assertEquals("xy".hashCode(), (int) MH_hashCode_String.invokeExact("xy"));
141// interface method:
142MethodHandle MH_subSequence = publicLookup().findVirtual(CharSequence.class,
143 "subSequence", methodType(CharSequence.class, int.class, int.class));
144assertEquals("def", MH_subSequence.invoke("abcdefghi", 3, 6).toString());
145// constructor "internal method" must be accessed differently:
146MethodType MT_newString = methodType(void.class); //()V for new String()
147try { assertEquals("impossible", lookup()
148 .findVirtual(String.class, "<init>", MT_newString));
149 } catch (NoSuchMethodException ex) { } // OK
150MethodHandle MH_newString = publicLookup()
151 .findConstructor(String.class, MT_newString);
152assertEquals("", (String) MH_newString.invokeExact());
153{}
154 }
155
156 @Test public void testFindConstructor() throws Throwable {
157{}
158MethodHandle MH_newArrayList = publicLookup().findConstructor(
159 ArrayList.class, methodType(void.class, Collection.class));
160Collection orig = Arrays.asList("x", "y");
161Collection copy = (ArrayList) MH_newArrayList.invokeExact(orig);
162assert(orig != copy);
163assertEquals(orig, copy);
164// a variable-arity constructor:
165MethodHandle MH_newProcessBuilder = publicLookup().findConstructor(
166 ProcessBuilder.class, methodType(void.class, String[].class));
167ProcessBuilder pb = (ProcessBuilder)
168 MH_newProcessBuilder.invoke("x", "y", "z");
169assertEquals("[x, y, z]", pb.command().toString());
170{}
171 }
172
173// for testFindSpecial
174{}
175static class Listie extends ArrayList {
176 public String toString() { return "[wee Listie]"; }
177 static Lookup lookup() { return MethodHandles.lookup(); }
178}
179{}
180
181 @Test public void testFindSpecial() throws Throwable {
182{}
183// no access to constructor via invokeSpecial:
184MethodHandle MH_newListie = Listie.lookup()
185 .findConstructor(Listie.class, methodType(void.class));
186Listie l = (Listie) MH_newListie.invokeExact();
187try { assertEquals("impossible", Listie.lookup().findSpecial(
188 Listie.class, "<init>", methodType(void.class), Listie.class));
189 } catch (NoSuchMethodException ex) { } // OK
190// access to super and self methods via invokeSpecial:
191MethodHandle MH_super = Listie.lookup().findSpecial(
192 ArrayList.class, "toString" , methodType(String.class), Listie.class);
193MethodHandle MH_this = Listie.lookup().findSpecial(
194 Listie.class, "toString" , methodType(String.class), Listie.class);
195MethodHandle MH_duper = Listie.lookup().findSpecial(
196 Object.class, "toString" , methodType(String.class), Listie.class);
197assertEquals("[]", (String) MH_super.invokeExact(l));
198assertEquals(""+l, (String) MH_this.invokeExact(l));
199assertEquals("[]", (String) MH_duper.invokeExact(l)); // ArrayList method
200try { assertEquals("inaccessible", Listie.lookup().findSpecial(
201 String.class, "toString", methodType(String.class), Listie.class));
202 } catch (IllegalAccessException ex) { } // OK
203Listie subl = new Listie() { public String toString() { return "[subclass]"; } };
204assertEquals(""+l, (String) MH_this.invokeExact(subl)); // Listie method
205{}
206 }
207
jrose9b82ad62011-05-26 17:37:36 -0700208 @Test public void testPermuteArguments() throws Throwable {
209 {{
210{} /// JAVADOC
211MethodType intfn1 = methodType(int.class, int.class);
212MethodType intfn2 = methodType(int.class, int.class, int.class);
213MethodHandle sub = SUB_2;// ... {int x, int y => x-y} ...;
214assert(sub.type().equals(intfn2));
215MethodHandle sub1 = permuteArguments(sub, intfn2, 0, 1);
216MethodHandle rsub = permuteArguments(sub, intfn2, 1, 0);
217assert((int)rsub.invokeExact(1, 100) == 99);
218MethodHandle add = ADD_2;// ... {int x, int y => x+y} ...;
219assert(add.type().equals(intfn2));
220MethodHandle twice = permuteArguments(add, intfn1, 0, 0);
221assert(twice.type().equals(intfn1));
222assert((int)twice.invokeExact(21) == 42);
223 }}
224 {{
225{} /// JAVADOC
226MethodHandle cat = lookup().findVirtual(String.class,
227 "concat", methodType(String.class, String.class));
228assertEquals("xy", (String) cat.invokeExact("x", "y"));
229MethodHandle d0 = dropArguments(cat, 0, String.class);
230assertEquals("yz", (String) d0.invokeExact("x", "y", "z"));
231MethodHandle d1 = dropArguments(cat, 1, String.class);
232assertEquals("xz", (String) d1.invokeExact("x", "y", "z"));
233MethodHandle d2 = dropArguments(cat, 2, String.class);
234assertEquals("xy", (String) d2.invokeExact("x", "y", "z"));
235MethodHandle d12 = dropArguments(cat, 1, int.class, boolean.class);
236assertEquals("xz", (String) d12.invokeExact("x", 12, true, "z"));
237 }}
238 }
239
jrose900bafd2010-10-30 21:08:23 -0700240 @Test public void testDropArguments() throws Throwable {
241 {{
242{} /// JAVADOC
243MethodHandle cat = lookup().findVirtual(String.class,
244 "concat", methodType(String.class, String.class));
jroseb90d2e72010-12-16 15:59:27 -0800245assertEquals("xy", (String) cat.invokeExact("x", "y"));
jroseada69fa2011-03-23 23:02:31 -0700246MethodType bigType = cat.type().insertParameterTypes(0, int.class, String.class);
247MethodHandle d0 = dropArguments(cat, 0, bigType.parameterList().subList(0,2));
248assertEquals(bigType, d0.type());
249assertEquals("yz", (String) d0.invokeExact(123, "x", "y", "z"));
250 }}
251 {{
252{} /// JAVADOC
253MethodHandle cat = lookup().findVirtual(String.class,
254 "concat", methodType(String.class, String.class));
255assertEquals("xy", (String) cat.invokeExact("x", "y"));
jrose900bafd2010-10-30 21:08:23 -0700256MethodHandle d0 = dropArguments(cat, 0, String.class);
jroseb90d2e72010-12-16 15:59:27 -0800257assertEquals("yz", (String) d0.invokeExact("x", "y", "z"));
jrose900bafd2010-10-30 21:08:23 -0700258MethodHandle d1 = dropArguments(cat, 1, String.class);
jroseb90d2e72010-12-16 15:59:27 -0800259assertEquals("xz", (String) d1.invokeExact("x", "y", "z"));
jrose900bafd2010-10-30 21:08:23 -0700260MethodHandle d2 = dropArguments(cat, 2, String.class);
jroseb90d2e72010-12-16 15:59:27 -0800261assertEquals("xy", (String) d2.invokeExact("x", "y", "z"));
jrose900bafd2010-10-30 21:08:23 -0700262MethodHandle d12 = dropArguments(cat, 1, int.class, boolean.class);
jroseb90d2e72010-12-16 15:59:27 -0800263assertEquals("xz", (String) d12.invokeExact("x", 12, true, "z"));
jrose900bafd2010-10-30 21:08:23 -0700264 }}
265 }
266
267 @Test public void testFilterArguments() throws Throwable {
268 {{
269{} /// JAVADOC
270MethodHandle cat = lookup().findVirtual(String.class,
271 "concat", methodType(String.class, String.class));
jrose900bafd2010-10-30 21:08:23 -0700272MethodHandle upcase = lookup().findVirtual(String.class,
273 "toUpperCase", methodType(String.class));
jroseb90d2e72010-12-16 15:59:27 -0800274assertEquals("xy", (String) cat.invokeExact("x", "y"));
jrose900bafd2010-10-30 21:08:23 -0700275MethodHandle f0 = filterArguments(cat, 0, upcase);
jroseb90d2e72010-12-16 15:59:27 -0800276assertEquals("Xy", (String) f0.invokeExact("x", "y")); // Xy
jrose900bafd2010-10-30 21:08:23 -0700277MethodHandle f1 = filterArguments(cat, 1, upcase);
jroseb90d2e72010-12-16 15:59:27 -0800278assertEquals("xY", (String) f1.invokeExact("x", "y")); // xY
jrose900bafd2010-10-30 21:08:23 -0700279MethodHandle f2 = filterArguments(cat, 0, upcase, upcase);
jroseb90d2e72010-12-16 15:59:27 -0800280assertEquals("XY", (String) f2.invokeExact("x", "y")); // XY
jrose900bafd2010-10-30 21:08:23 -0700281 }}
282 }
283
jrose9b82ad62011-05-26 17:37:36 -0700284 @Test public void testFoldArguments() throws Throwable {
285 {{
286{} /// JAVADOC
287MethodHandle trace = publicLookup().findVirtual(java.io.PrintStream.class,
288 "println", methodType(void.class, String.class))
289 .bindTo(System.out);
290MethodHandle cat = lookup().findVirtual(String.class,
291 "concat", methodType(String.class, String.class));
292assertEquals("boojum", (String) cat.invokeExact("boo", "jum"));
293MethodHandle catTrace = foldArguments(cat, trace);
294// also prints "boo":
295assertEquals("boojum", (String) catTrace.invokeExact("boo", "jum"));
296 }}
297 }
298
jrose900bafd2010-10-30 21:08:23 -0700299 static void assertEquals(Object exp, Object act) {
300 if (verbosity > 0)
301 System.out.println("result: "+act);
302 Assert.assertEquals(exp, act);
303 }
304
jrosef108fc02011-02-11 01:26:24 -0800305 @Test public void testMethodHandlesSummary() throws Throwable {
jrose900bafd2010-10-30 21:08:23 -0700306 {{
307{} /// JAVADOC
jrosef108fc02011-02-11 01:26:24 -0800308Object x, y; String s; int i;
309MethodType mt; MethodHandle mh;
310MethodHandles.Lookup lookup = MethodHandles.lookup();
311// mt is (char,char)String
312mt = MethodType.methodType(String.class, char.class, char.class);
313mh = lookup.findVirtual(String.class, "replace", mt);
jrosef108fc02011-02-11 01:26:24 -0800314s = (String) mh.invokeExact("daddy",'d','n');
jroseadc650a2011-02-11 01:26:28 -0800315// invokeExact(Ljava/lang/String;CC)Ljava/lang/String;
jrose9b82ad62011-05-26 17:37:36 -0700316assertEquals(s, "nanny");
jrosef108fc02011-02-11 01:26:24 -0800317// weakly typed invocation (using MHs.invoke)
318s = (String) mh.invokeWithArguments("sappy", 'p', 'v');
jrose9b82ad62011-05-26 17:37:36 -0700319assertEquals(s, "savvy");
jrosef108fc02011-02-11 01:26:24 -0800320// mt is (Object[])List
321mt = MethodType.methodType(java.util.List.class, Object[].class);
322mh = lookup.findStatic(java.util.Arrays.class, "asList", mt);
323assert(mh.isVarargsCollector());
jrose79e0a6c2011-05-12 19:27:33 -0700324x = mh.invoke("one", "two");
325// invoke(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/Object;
jrose9b82ad62011-05-26 17:37:36 -0700326assertEquals(x, java.util.Arrays.asList("one","two"));
jrosef108fc02011-02-11 01:26:24 -0800327// mt is (Object,Object,Object)Object
328mt = MethodType.genericMethodType(3);
jroseadc650a2011-02-11 01:26:28 -0800329mh = mh.asType(mt);
jrosef108fc02011-02-11 01:26:24 -0800330x = mh.invokeExact((Object)1, (Object)2, (Object)3);
jroseadc650a2011-02-11 01:26:28 -0800331// invokeExact(Ljava/lang/Object;Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object;
jrose9b82ad62011-05-26 17:37:36 -0700332assertEquals(x, java.util.Arrays.asList(1,2,3));
333// mt is ()int
jrosef108fc02011-02-11 01:26:24 -0800334mt = MethodType.methodType(int.class);
335mh = lookup.findVirtual(java.util.List.class, "size", mt);
jrosef108fc02011-02-11 01:26:24 -0800336i = (int) mh.invokeExact(java.util.Arrays.asList(1,2,3));
jroseadc650a2011-02-11 01:26:28 -0800337// invokeExact(Ljava/util/List;)I
jrosef108fc02011-02-11 01:26:24 -0800338assert(i == 3);
339mt = MethodType.methodType(void.class, String.class);
340mh = lookup.findVirtual(java.io.PrintStream.class, "println", mt);
341mh.invokeExact(System.out, "Hello, world.");
jroseadc650a2011-02-11 01:26:28 -0800342// invokeExact(Ljava/io/PrintStream;Ljava/lang/String;)V
jrosef108fc02011-02-11 01:26:24 -0800343{}
jrose900bafd2010-10-30 21:08:23 -0700344 }}
345 }
346
jrose9b82ad62011-05-26 17:37:36 -0700347 @Test public void testAsSpreader() throws Throwable {
348 {{
349{} /// JAVADOC
350MethodHandle equals = publicLookup()
351 .findVirtual(String.class, "equals", methodType(boolean.class, Object.class));
352assert( (boolean) equals.invokeExact("me", (Object)"me"));
353assert(!(boolean) equals.invokeExact("me", (Object)"thee"));
354// spread both arguments from a 2-array:
355MethodHandle eq2 = equals.asSpreader(Object[].class, 2);
356assert( (boolean) eq2.invokeExact(new Object[]{ "me", "me" }));
357assert(!(boolean) eq2.invokeExact(new Object[]{ "me", "thee" }));
jrose70feaa92013-10-05 05:30:39 -0700358// try to spread from anything but a 2-array:
359for (int n = 0; n <= 10; n++) {
360 Object[] badArityArgs = (n == 2 ? null : new Object[n]);
361 try { assert((boolean) eq2.invokeExact(badArityArgs) && false); }
362 catch (IllegalArgumentException ex) { } // OK
363}
jrose9b82ad62011-05-26 17:37:36 -0700364// spread both arguments from a String array:
365MethodHandle eq2s = equals.asSpreader(String[].class, 2);
366assert( (boolean) eq2s.invokeExact(new String[]{ "me", "me" }));
367assert(!(boolean) eq2s.invokeExact(new String[]{ "me", "thee" }));
368// spread second arguments from a 1-array:
369MethodHandle eq1 = equals.asSpreader(Object[].class, 1);
370assert( (boolean) eq1.invokeExact("me", new Object[]{ "me" }));
371assert(!(boolean) eq1.invokeExact("me", new Object[]{ "thee" }));
372// spread no arguments from a 0-array or null:
373MethodHandle eq0 = equals.asSpreader(Object[].class, 0);
374assert( (boolean) eq0.invokeExact("me", (Object)"me", new Object[0]));
375assert(!(boolean) eq0.invokeExact("me", (Object)"thee", (Object[])null));
376// asSpreader and asCollector are approximate inverses:
377for (int n = 0; n <= 2; n++) {
378 for (Class<?> a : new Class<?>[]{Object[].class, String[].class, CharSequence[].class}) {
379 MethodHandle equals2 = equals.asSpreader(a, n).asCollector(a, n);
380 assert( (boolean) equals2.invokeWithArguments("me", "me"));
381 assert(!(boolean) equals2.invokeWithArguments("me", "thee"));
382 }
383}
384MethodHandle caToString = publicLookup()
385 .findStatic(Arrays.class, "toString", methodType(String.class, char[].class));
386assertEquals("[A, B, C]", (String) caToString.invokeExact("ABC".toCharArray()));
387MethodHandle caString3 = caToString.asCollector(char[].class, 3);
388assertEquals("[A, B, C]", (String) caString3.invokeExact('A', 'B', 'C'));
389MethodHandle caToString2 = caString3.asSpreader(char[].class, 2);
390assertEquals("[A, B, C]", (String) caToString2.invokeExact('A', "BC".toCharArray()));
391 }}
392 }
393
394 @Test public void testAsCollector() throws Throwable {
395 {{
396{} /// JAVADOC
397MethodHandle deepToString = publicLookup()
398 .findStatic(Arrays.class, "deepToString", methodType(String.class, Object[].class));
399assertEquals("[won]", (String) deepToString.invokeExact(new Object[]{"won"}));
400MethodHandle ts1 = deepToString.asCollector(Object[].class, 1);
401assertEquals(methodType(String.class, Object.class), ts1.type());
402//assertEquals("[won]", (String) ts1.invokeExact( new Object[]{"won"})); //FAIL
403assertEquals("[[won]]", (String) ts1.invokeExact((Object) new Object[]{"won"}));
404// arrayType can be a subtype of Object[]
405MethodHandle ts2 = deepToString.asCollector(String[].class, 2);
406assertEquals(methodType(String.class, String.class, String.class), ts2.type());
407assertEquals("[two, too]", (String) ts2.invokeExact("two", "too"));
408MethodHandle ts0 = deepToString.asCollector(Object[].class, 0);
409assertEquals("[]", (String) ts0.invokeExact());
410// collectors can be nested, Lisp-style
411MethodHandle ts22 = deepToString.asCollector(Object[].class, 3).asCollector(String[].class, 2);
412assertEquals("[A, B, [C, D]]", ((String) ts22.invokeExact((Object)'A', (Object)"B", "C", "D")));
413// arrayType can be any primitive array type
414MethodHandle bytesToString = publicLookup()
415 .findStatic(Arrays.class, "toString", methodType(String.class, byte[].class))
416 .asCollector(byte[].class, 3);
417assertEquals("[1, 2, 3]", (String) bytesToString.invokeExact((byte)1, (byte)2, (byte)3));
418MethodHandle longsToString = publicLookup()
419 .findStatic(Arrays.class, "toString", methodType(String.class, long[].class))
420 .asCollector(long[].class, 1);
421assertEquals("[123]", (String) longsToString.invokeExact((long)123));
422 }}
423 }
424
jrose49494522012-01-18 17:34:29 -0800425 @SuppressWarnings("rawtypes")
jrosef108fc02011-02-11 01:26:24 -0800426 @Test public void testAsVarargsCollector() throws Throwable {
427 {{
428{} /// JAVADOC
jrose9b82ad62011-05-26 17:37:36 -0700429MethodHandle deepToString = publicLookup()
430 .findStatic(Arrays.class, "deepToString", methodType(String.class, Object[].class));
431MethodHandle ts1 = deepToString.asVarargsCollector(Object[].class);
432assertEquals("[won]", (String) ts1.invokeExact( new Object[]{"won"}));
433assertEquals("[won]", (String) ts1.invoke( new Object[]{"won"}));
434assertEquals("[won]", (String) ts1.invoke( "won" ));
435assertEquals("[[won]]", (String) ts1.invoke((Object) new Object[]{"won"}));
436// findStatic of Arrays.asList(...) produces a variable arity method handle:
jrosef108fc02011-02-11 01:26:24 -0800437MethodHandle asList = publicLookup()
jrose9b82ad62011-05-26 17:37:36 -0700438 .findStatic(Arrays.class, "asList", methodType(List.class, Object[].class));
439assertEquals(methodType(List.class, Object[].class), asList.type());
440assert(asList.isVarargsCollector());
jrose79e0a6c2011-05-12 19:27:33 -0700441assertEquals("[]", asList.invoke().toString());
442assertEquals("[1]", asList.invoke(1).toString());
443assertEquals("[two, too]", asList.invoke("two", "too").toString());
jrose9b82ad62011-05-26 17:37:36 -0700444String[] argv = { "three", "thee", "tee" };
jrose79e0a6c2011-05-12 19:27:33 -0700445assertEquals("[three, thee, tee]", asList.invoke(argv).toString());
jrose9b82ad62011-05-26 17:37:36 -0700446assertEquals("[three, thee, tee]", asList.invoke((Object[])argv).toString());
jrose79e0a6c2011-05-12 19:27:33 -0700447List ls = (List) asList.invoke((Object)argv);
jrosef108fc02011-02-11 01:26:24 -0800448assertEquals(1, ls.size());
449assertEquals("[three, thee, tee]", Arrays.toString((Object[])ls.get(0)));
450 }}
451 }
jrose900bafd2010-10-30 21:08:23 -0700452
jrose9b82ad62011-05-26 17:37:36 -0700453 @Test public void testAsFixedArity() throws Throwable {
jrosef108fc02011-02-11 01:26:24 -0800454 {{
455{} /// JAVADOC
jrose9b82ad62011-05-26 17:37:36 -0700456MethodHandle asListVar = publicLookup()
jrosef108fc02011-02-11 01:26:24 -0800457 .findStatic(Arrays.class, "asList", methodType(List.class, Object[].class))
458 .asVarargsCollector(Object[].class);
jrose9b82ad62011-05-26 17:37:36 -0700459MethodHandle asListFix = asListVar.asFixedArity();
460assertEquals("[1]", asListVar.invoke(1).toString());
461Exception caught = null;
462try { asListFix.invoke((Object)1); }
463catch (Exception ex) { caught = ex; }
464assert(caught instanceof ClassCastException);
465assertEquals("[two, too]", asListVar.invoke("two", "too").toString());
466try { asListFix.invoke("two", "too"); }
467catch (Exception ex) { caught = ex; }
468assert(caught instanceof WrongMethodTypeException);
469Object[] argv = { "three", "thee", "tee" };
470assertEquals("[three, thee, tee]", asListVar.invoke(argv).toString());
471assertEquals("[three, thee, tee]", asListFix.invoke(argv).toString());
472assertEquals(1, ((List) asListVar.invoke((Object)argv)).size());
473assertEquals("[three, thee, tee]", asListFix.invoke((Object)argv).toString());
474 }}
475 }
476
477 @Test public void testAsTypeCornerCases() throws Throwable {
478 {{
479{} /// JAVADOC
480MethodHandle i2s = publicLookup()
481 .findVirtual(Integer.class, "toString", methodType(String.class));
482i2s = i2s.asType(i2s.type().unwrap());
483MethodHandle l2s = publicLookup()
484 .findVirtual(Long.class, "toString", methodType(String.class));
485l2s = l2s.asType(l2s.type().unwrap());
486
487Exception caught = null;
488try { i2s.asType(methodType(String.class, String.class)); }
489catch (Exception ex) { caught = ex; }
490assert(caught instanceof WrongMethodTypeException);
491
492i2s.asType(methodType(String.class, byte.class));
493i2s.asType(methodType(String.class, Byte.class));
494i2s.asType(methodType(String.class, Character.class));
495i2s.asType(methodType(String.class, Integer.class));
496l2s.asType(methodType(String.class, byte.class));
497l2s.asType(methodType(String.class, Byte.class));
498l2s.asType(methodType(String.class, Character.class));
499l2s.asType(methodType(String.class, Integer.class));
500l2s.asType(methodType(String.class, Long.class));
501
502caught = null;
503try { i2s.asType(methodType(String.class, Long.class)); }
504catch (Exception ex) { caught = ex; }
505assert(caught instanceof WrongMethodTypeException);
506
507MethodHandle i2sGen = i2s.asType(methodType(String.class, Object.class));
508MethodHandle l2sGen = l2s.asType(methodType(String.class, Object.class));
509
510i2sGen.invoke(42); // int -> Integer -> Object -> Integer -> int
511i2sGen.invoke((byte)4); // byte -> Byte -> Object -> Byte -> byte -> int
512l2sGen.invoke(42); // int -> Integer -> Object -> Integer -> int
513l2sGen.invoke((byte)4); // byte -> Byte -> Object -> Byte -> byte -> int
514l2sGen.invoke(0x420000000L);
515
516caught = null;
517try { i2sGen.invoke(0x420000000L); } // long -> Long -> Object -> Integer CCE
518catch (Exception ex) { caught = ex; }
519assert(caught instanceof ClassCastException);
520
521caught = null;
522try { i2sGen.invoke("asdf"); } // String -> Object -> Integer CCE
523catch (Exception ex) { caught = ex; }
524assert(caught instanceof ClassCastException);
jrosef108fc02011-02-11 01:26:24 -0800525{}
526 }}
527 }
jrose9b82ad62011-05-26 17:37:36 -0700528
529 @Test public void testMutableCallSite() throws Throwable {
530 {{
531{} /// JAVADOC
532MutableCallSite name = new MutableCallSite(MethodType.methodType(String.class));
533MethodHandle MH_name = name.dynamicInvoker();
534MethodType MT_str1 = MethodType.methodType(String.class);
535MethodHandle MH_upcase = MethodHandles.lookup()
536 .findVirtual(String.class, "toUpperCase", MT_str1);
537MethodHandle worker1 = MethodHandles.filterReturnValue(MH_name, MH_upcase);
538name.setTarget(MethodHandles.constant(String.class, "Rocky"));
539assertEquals("ROCKY", (String) worker1.invokeExact());
540name.setTarget(MethodHandles.constant(String.class, "Fred"));
541assertEquals("FRED", (String) worker1.invokeExact());
542// (mutation can be continued indefinitely)
543/*
544 * </pre></blockquote>
545 * <p>
546 * The same call site may be used in several places at once.
547 * <blockquote><pre>
548 */
549MethodType MT_str2 = MethodType.methodType(String.class, String.class);
550MethodHandle MH_cat = lookup().findVirtual(String.class,
551 "concat", methodType(String.class, String.class));
552MethodHandle MH_dear = MethodHandles.insertArguments(MH_cat, 1, ", dear?");
553MethodHandle worker2 = MethodHandles.filterReturnValue(MH_name, MH_dear);
554assertEquals("Fred, dear?", (String) worker2.invokeExact());
555name.setTarget(MethodHandles.constant(String.class, "Wilma"));
556assertEquals("WILMA", (String) worker1.invokeExact());
557assertEquals("Wilma, dear?", (String) worker2.invokeExact());
558{}
559 }}
560 }
561
562 @Test public void testSwitchPoint() throws Throwable {
563 {{
564{} /// JAVADOC
565MethodHandle MH_strcat = MethodHandles.lookup()
566 .findVirtual(String.class, "concat", MethodType.methodType(String.class, String.class));
567SwitchPoint spt = new SwitchPoint();
jrosecd1d97b2011-06-03 11:20:20 -0700568assert(!spt.hasBeenInvalidated());
jrose9b82ad62011-05-26 17:37:36 -0700569// the following steps may be repeated to re-use the same switch point:
570MethodHandle worker1 = MH_strcat;
571MethodHandle worker2 = MethodHandles.permuteArguments(MH_strcat, MH_strcat.type(), 1, 0);
572MethodHandle worker = spt.guardWithTest(worker1, worker2);
573assertEquals("method", (String) worker.invokeExact("met", "hod"));
574SwitchPoint.invalidateAll(new SwitchPoint[]{ spt });
jrosecd1d97b2011-06-03 11:20:20 -0700575assert(spt.hasBeenInvalidated());
jrose9b82ad62011-05-26 17:37:36 -0700576assertEquals("hodmet", (String) worker.invokeExact("met", "hod"));
577{}
578 }}
579 }
580
jroseb4be0262011-07-16 15:44:33 -0700581 @Test public void testFoldArguments2() throws Throwable {
582 {{
583{} /// JAVADOC
584// argument-based dispatch for methods of the form boolean x.___(y: String)
585Lookup lookup = publicLookup();
586// first, a tracing hack:
587MethodHandle println = lookup.findVirtual(java.io.PrintStream.class, "println", methodType(void.class, String.class));
588MethodHandle arrayToString = lookup.findStatic(Arrays.class, "toString", methodType(String.class, Object[].class));
589MethodHandle concat = lookup.findVirtual(String.class, "concat", methodType(String.class, String.class));
590MethodHandle arrayToString_DIS = filterReturnValue(arrayToString, concat.bindTo("DIS:"));
591MethodHandle arrayToString_INV = filterReturnValue(arrayToString, concat.bindTo("INV:"));
592MethodHandle printArgs_DIS = filterReturnValue(arrayToString_DIS, println.bindTo(System.out)).asVarargsCollector(Object[].class);
593MethodHandle printArgs_INV = filterReturnValue(arrayToString_INV, println.bindTo(System.out)).asVarargsCollector(Object[].class);
594// metaobject protocol:
595MethodType mtype = methodType(boolean.class, String.class);
596MethodHandle findVirtual = lookup.findVirtual(Lookup.class,
597 "findVirtual", methodType(MethodHandle.class, Class.class, String.class, MethodType.class));
598MethodHandle getClass = lookup.findVirtual(Object.class,
599 "getClass", methodType(Class.class));
600MethodHandle dispatch = findVirtual;
601dispatch = filterArguments(dispatch, 1, getClass);
602dispatch = insertArguments(dispatch, 3, mtype);
603dispatch = dispatch.bindTo(lookup);
604assertEquals(methodType(MethodHandle.class, Object.class, String.class), dispatch.type());
605MethodHandle invoker = invoker(mtype.insertParameterTypes(0, Object.class));
606// wrap tracing around the dispatch and invoke steps:
607dispatch = foldArguments(dispatch, printArgs_DIS.asType(dispatch.type().changeReturnType(void.class)));
608invoker = foldArguments(invoker, printArgs_INV.asType(invoker.type().changeReturnType(void.class)));
609invoker = dropArguments(invoker, 2, String.class); // ignore selector
610// compose the dispatcher and the invoker:
611MethodHandle invokeDispatched = foldArguments(invoker, dispatch);
612Object x = "football", y = new java.util.Scanner("bar");
613assert( (boolean) invokeDispatched.invokeExact(x, "startsWith", "foo"));
614assert(!(boolean) invokeDispatched.invokeExact(x, "startsWith", "#"));
615assert( (boolean) invokeDispatched.invokeExact(x, "endsWith", "all"));
616assert(!(boolean) invokeDispatched.invokeExact(x, "endsWith", "foo"));
617assert( (boolean) invokeDispatched.invokeExact(y, "hasNext", "[abc]+[rst]"));
618assert(!(boolean) invokeDispatched.invokeExact(y, "hasNext", "[123]+[789]"));
619 }}
620 }
621
jrose9b82ad62011-05-26 17:37:36 -0700622 /* ---- TEMPLATE ----
623 @Test public void testFoo() throws Throwable {
624 {{
625{} /// JAVADOC
626{}
627 }}
628 }
629 */
jrose900bafd2010-10-30 21:08:23 -0700630}