1 /*
2  * Copyright (c) 2015, 2016, 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.  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
27  * @bug 8139885
28  * @bug 8143798
29  * @run testng/othervm -ea -esa test.java.lang.invoke.SpreadCollectTest
30  */
31 
32 package test.java.lang.invoke;
33 
34 import java.io.StringWriter;
35 import java.lang.invoke.MethodHandle;
36 import java.lang.invoke.MethodHandles;
37 import java.lang.invoke.MethodHandles.Lookup;
38 import java.lang.invoke.MethodType;
39 import java.lang.invoke.WrongMethodTypeException;
40 import java.util.*;
41 
42 import static java.lang.invoke.MethodType.methodType;
43 
44 import static org.testng.AssertJUnit.*;
45 
46 import org.testng.annotations.*;
47 
48 /**
49  * Tests for the new asSpreader/asCollector API added in JEP 274.
50  */
51 public class SpreadCollectTest {
52 
53     static final Lookup LOOKUP = MethodHandles.lookup();
54 
55     @Test
testAsSpreader()56     public static void testAsSpreader() throws Throwable {
57         MethodHandle spreader = SpreadCollect.MH_forSpreading.asSpreader(1, int[].class, 3);
58         assertEquals(SpreadCollect.MT_spreader, spreader.type());
59         assertEquals("A456B", (String) spreader.invoke("A", new int[]{4, 5, 6}, "B"));
60     }
61 
62     @Test
testAsSpreaderExample()63     public static void testAsSpreaderExample() throws Throwable {
64         // test the JavaDoc asSpreader-with-pos example
65         MethodHandle compare = LOOKUP.findStatic(Objects.class, "compare", methodType(int.class, Object.class, Object.class, Comparator.class));
66         MethodHandle compare2FromArray = compare.asSpreader(0, Object[].class, 2);
67         Object[] ints = new Object[]{3, 9, 7, 7};
68         Comparator<Integer> cmp = (a, b) -> a - b;
69         assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 0, 2), cmp) < 0);
70         assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 1, 3), cmp) > 0);
71         assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 2, 4), cmp) == 0);
72     }
73 
74     @DataProvider
asSpreaderIllegalPositions()75     static Object[][] asSpreaderIllegalPositions() {
76         return new Object[][]{{-7}, {3}, {19}};
77     }
78 
79     @Test(dataProvider = "asSpreaderIllegalPositions")
testAsSpreaderIllegalPos(int p)80     public static void testAsSpreaderIllegalPos(int p) throws Throwable {
81         boolean caught = false;
82         try {
83             SpreadCollect.MH_forSpreading.asSpreader(p, Object[].class, 3);
84         } catch (IllegalArgumentException iae) {
85             assertEquals("bad spread position", iae.getMessage());
86             caught = true;
87         }
88         assertTrue(caught);
89     }
90 
91     @Test(expectedExceptions = {WrongMethodTypeException.class})
testAsSpreaderIllegalMethodType()92     public static void testAsSpreaderIllegalMethodType() {
93         MethodHandle h = MethodHandles.dropArguments(MethodHandles.constant(String.class, ""), 0, int.class, int.class);
94         MethodHandle s = h.asSpreader(String[].class, 1);
95     }
96 
97     @Test(expectedExceptions = {NullPointerException.class})
testAsSpreaderNullArrayType()98     public static void testAsSpreaderNullArrayType() {
99         SpreadCollect.MH_forSpreading.asSpreader(null, 0);
100     }
101 
102     @Test(expectedExceptions = {NullPointerException.class})
testAsSpreaderNullArrayNonZeroLength()103     public static void testAsSpreaderNullArrayNonZeroLength() {
104         SpreadCollect.MH_forSpreading.asSpreader(null, 1);
105     }
106 
107     @Test(expectedExceptions = {IllegalArgumentException.class})
testAsSpreaderTooManyParams()108     public static void testAsSpreaderTooManyParams() throws Throwable {
109         SpreadCollect.MH_forSpreading.asSpreader(1, int[].class, 6);
110     }
111 
112     @Test
testAsCollector()113     public static void testAsCollector() throws Throwable {
114         MethodHandle collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 1);
115         assertEquals(SpreadCollect.MT_collector1, collector.type());
116         assertEquals("A4B", (String) collector.invoke("A", 4, "B"));
117         collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 2);
118         assertEquals(SpreadCollect.MT_collector2, collector.type());
119         assertEquals("A45B", (String) collector.invoke("A", 4, 5, "B"));
120         collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 3);
121         assertEquals(SpreadCollect.MT_collector3, collector.type());
122         assertEquals("A456B", (String) collector.invoke("A", 4, 5, 6, "B"));
123     }
124 
125     @Test
testAsCollectorInvokeWithArguments()126     public static void testAsCollectorInvokeWithArguments() throws Throwable {
127         MethodHandle collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 1);
128         assertEquals(SpreadCollect.MT_collector1, collector.type());
129         assertEquals("A4B", (String) collector.invokeWithArguments("A", 4, "B"));
130         collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 2);
131         assertEquals(SpreadCollect.MT_collector2, collector.type());
132         assertEquals("A45B", (String) collector.invokeWithArguments("A", 4, 5, "B"));
133         collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 3);
134         assertEquals(SpreadCollect.MT_collector3, collector.type());
135         assertEquals("A456B", (String) collector.invokeWithArguments("A", 4, 5, 6, "B"));
136     }
137 
138     @Test
testAsCollectorLeading()139     public static void testAsCollectorLeading() throws Throwable {
140         MethodHandle collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 1);
141         assertEquals(SpreadCollect.MT_collectorLeading1, collector.type());
142         assertEquals("7Q", (String) collector.invoke(7, "Q"));
143         collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 2);
144         assertEquals(SpreadCollect.MT_collectorLeading2, collector.type());
145         assertEquals("78Q", (String) collector.invoke(7, 8, "Q"));
146         collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 3);
147         assertEquals(SpreadCollect.MT_collectorLeading3, collector.type());
148         assertEquals("789Q", (String) collector.invoke(7, 8, 9, "Q"));
149     }
150 
151     @Test
testAsCollectorLeadingInvokeWithArguments()152     public static void testAsCollectorLeadingInvokeWithArguments() throws Throwable {
153         MethodHandle collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 1);
154         assertEquals(SpreadCollect.MT_collectorLeading1, collector.type());
155         assertEquals("7Q", (String) collector.invokeWithArguments(7, "Q"));
156         collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 2);
157         assertEquals(SpreadCollect.MT_collectorLeading2, collector.type());
158         assertEquals("78Q", (String) collector.invokeWithArguments(7, 8, "Q"));
159         collector = SpreadCollect.MH_forCollectingLeading.asCollector(0, int[].class, 3);
160         assertEquals(SpreadCollect.MT_collectorLeading3, collector.type());
161         assertEquals("789Q", (String) collector.invokeWithArguments(7, 8, 9, "Q"));
162     }
163 
164     @Test
testAsCollectorNone()165     public static void testAsCollectorNone() throws Throwable {
166         MethodHandle collector = SpreadCollect.MH_forCollecting.asCollector(1, int[].class, 0);
167         assertEquals(SpreadCollect.MT_collector0, collector.type());
168         assertEquals("AB", (String) collector.invoke("A", "B"));
169     }
170 
171     @DataProvider
asCollectorIllegalPositions()172     static Object[][] asCollectorIllegalPositions() {
173         return new Object[][]{{-1}, {17}};
174     }
175 
176     @Test(dataProvider = "asCollectorIllegalPositions")
testAsCollectorIllegalPos(int p)177     public static void testAsCollectorIllegalPos(int p) {
178         boolean caught = false;
179         try {
180             SpreadCollect.MH_forCollecting.asCollector(p, int[].class, 0);
181         } catch (IllegalArgumentException iae) {
182             assertEquals("bad collect position", iae.getMessage());
183             caught = true;
184         }
185         assertTrue(caught);
186     }
187 
188     @Test
testAsCollectorExample()189     public static void testAsCollectorExample() throws Throwable {
190         // test the JavaDoc asCollector-with-pos example
191         StringWriter swr = new StringWriter();
192         MethodHandle swWrite = LOOKUP.
193                 findVirtual(StringWriter.class, "write", methodType(void.class, char[].class, int.class, int.class)).
194                 bindTo(swr);
195         MethodHandle swWrite4 = swWrite.asCollector(0, char[].class, 4);
196         swWrite4.invoke('A', 'B', 'C', 'D', 1, 2);
197         assertEquals("BC", swr.toString());
198         swWrite4.invoke('P', 'Q', 'R', 'S', 0, 4);
199         assertEquals("BCPQRS", swr.toString());
200         swWrite4.invoke('W', 'X', 'Y', 'Z', 3, 1);
201         assertEquals("BCPQRSZ", swr.toString());
202     }
203 
204     static class SpreadCollect {
205 
forSpreading(String s1, int i1, int i2, int i3, String s2)206         static String forSpreading(String s1, int i1, int i2, int i3, String s2) {
207             return s1 + i1 + i2 + i3 + s2;
208         }
209 
forCollecting(String s1, int[] is, String s2)210         static String forCollecting(String s1, int[] is, String s2) {
211             StringBuilder sb = new StringBuilder(s1);
212             for (int i : is) {
213                 sb.append(i);
214             }
215             return sb.append(s2).toString();
216         }
217 
forCollectingLeading(int[] is, String s)218         static String forCollectingLeading(int[] is, String s) {
219             return forCollecting("", is, s);
220         }
221 
222         static final Class<SpreadCollect> SPREAD_COLLECT = SpreadCollect.class;
223 
224         static final MethodType MT_forSpreading = methodType(String.class, String.class, int.class, int.class, int.class, String.class);
225         static final MethodType MT_forCollecting = methodType(String.class, String.class, int[].class, String.class);
226         static final MethodType MT_forCollectingLeading = methodType(String.class, int[].class, String.class);
227 
228         static final MethodHandle MH_forSpreading;
229         static final MethodHandle MH_forCollecting;
230         static final MethodHandle MH_forCollectingLeading;
231 
232         static final MethodType MT_spreader = methodType(String.class, String.class, int[].class, String.class);
233         static final MethodType MT_collector0 = methodType(String.class, String.class, String.class);
234         static final MethodType MT_collector1 = methodType(String.class, String.class, int.class, String.class);
235         static final MethodType MT_collector2 = methodType(String.class, String.class, int.class, int.class, String.class);
236         static final MethodType MT_collector3 = methodType(String.class, String.class, int.class, int.class, int.class, String.class);
237         static final MethodType MT_collectorLeading1 = methodType(String.class, int.class, String.class);
238         static final MethodType MT_collectorLeading2 = methodType(String.class, int.class, int.class, String.class);
239         static final MethodType MT_collectorLeading3 = methodType(String.class, int.class, int.class, int.class, String.class);
240 
241         static final String NONE_ERROR = "zero array length in MethodHandle.asCollector";
242 
243         static {
244             try {
245                 MH_forSpreading = LOOKUP.findStatic(SPREAD_COLLECT, "forSpreading", MT_forSpreading);
246                 MH_forCollecting = LOOKUP.findStatic(SPREAD_COLLECT, "forCollecting", MT_forCollecting);
247                 MH_forCollectingLeading = LOOKUP.findStatic(SPREAD_COLLECT, "forCollectingLeading", MT_forCollectingLeading);
248             } catch (Exception e) {
249                 throw new ExceptionInInitializerError(e);
250             }
251         }
252 
253     }
254 
255 }
256