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