1 /* 2 * Copyright (c) 2013, 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 package java.util.stream; 24 25 import org.testng.annotations.DataProvider; 26 27 import java.util.*; 28 import java.util.Spliterators; 29 import java.util.function.Supplier; 30 31 /** TestNG DataProvider for long-valued streams */ 32 public class LongStreamTestDataProvider { 33 private static final long[] to0 = new long[0]; 34 private static final long[] to1 = new long[1]; 35 private static final long[] to10 = new long[10]; 36 private static final long[] to100 = new long[100]; 37 private static final long[] to1000 = new long[1000]; 38 private static final long[] reversed = new long[100]; 39 private static final long[] ones = new long[100]; 40 private static final long[] twice = new long[200]; 41 private static final long[] pseudoRandom; 42 43 private static final Object[][] testData; 44 private static final Object[][] spliteratorTestData; 45 46 static { 47 long[][] arrays = {to0, to1, to10, to100, to1000}; 48 for (long[] arr : arrays) { 49 for (int i = 0; i < arr.length; i++) { 50 arr[i] = i; 51 } 52 } 53 for (int i = 0; i < reversed.length; i++) { 54 reversed[i] = reversed.length - i; 55 } 56 for (int i = 0; i < ones.length; i++) { 57 ones[i] = 1; 58 } 59 System.arraycopy(to100, 0, twice, 0, to100.length); 60 System.arraycopy(to100, 0, twice, to100.length, to100.length); 61 pseudoRandom = new long[LambdaTestHelpers.LONG_STRING.length()]; 62 for (int i = 0; i < LambdaTestHelpers.LONG_STRING.length(); i++) { 63 pseudoRandom[i] = (long) LambdaTestHelpers.LONG_STRING.charAt(i); 64 } 65 } 66 67 static final Object[][] arrays = { 68 {"empty", to0}, 69 {"0..1", to1}, 70 {"0..10", to10}, 71 {"0..100", to100}, 72 {"0..1000", to1000}, 73 {"100x[1]", ones}, 74 {"2x[0..100]", twice}, 75 {"reverse 0..100", reversed}, 76 {"pseudorandom", pseudoRandom} 77 }; 78 79 static { 80 { 81 List<Object[]> list = new ArrayList<>(); 82 for (Object[] data : arrays) { 83 final Object name = data[0]; 84 final long[] longs = (long[]) data[1]; 85 86 list.add(new Object[]{"array:" + name, 87 TestData.Factory.ofArray("array:" + name, longs)}); 88 89 SpinedBuffer.OfLong isl = new SpinedBuffer.OfLong(); 90 for (long i : longs) { 91 isl.accept(i); 92 } 93 list.add(new Object[]{"SpinedList:" + name, 94 TestData.Factory.ofSpinedBuffer("SpinedList:" + name, isl)}); 95 96 list.add(streamDataDescr("LongStream.longRange(0,l): " + longs.length, 97 () -> LongStream.range(0, longs.length))); 98 list.add(streamDataDescr("LongStream.longRangeClosed(0,l): " + longs.length, 99 () -> LongStream.rangeClosed(0, longs.length))); 100 } 101 testData = list.toArray(new Object[0][]); 102 } 103 104 { 105 List<Object[]> spliterators = new ArrayList<>(); 106 for (Object[] data : arrays) { 107 final Object name = data[0]; 108 final long[] longs = (long[]) data[1]; 109 110 SpinedBuffer.OfLong isl = new SpinedBuffer.OfLong(); 111 for (long i : longs) { 112 isl.accept(i); 113 } 114 115 spliterators.add(splitDescr("Arrays.s(array):" + name, 116 () -> Arrays.spliterator(longs))); 117 spliterators.add(splitDescr("Arrays.s(array,o,l):" + name, 118 () -> Arrays.spliterator(longs, 0, longs.length / 2))); 119 120 spliterators.add(splitDescr("SpinedBuffer.s():" + name, 121 () -> isl.spliterator())); 122 123 spliterators.add(splitDescr("Primitives.s(SpinedBuffer.iterator(), size):" + name, 124 () -> Spliterators.spliterator(isl.iterator(), longs.length, 0))); 125 spliterators.add(splitDescr("Primitives.s(SpinedBuffer.iterator()):" + name, 126 () -> Spliterators.spliteratorUnknownSize(isl.iterator(), 0))); 127 128 spliterators.add(splitDescr("LongStream.longRange(0,l):" + name, 129 () -> LongStream.range(0, longs.length).spliterator())); 130 spliterators.add(splitDescr("LongStream.longRangeClosed(0,l):" + name, 131 () -> LongStream.rangeClosed(0, longs.length).spliterator())); 132 // Need more! 133 } 134 spliteratorTestData = spliterators.toArray(new Object[0][]); 135 } 136 137 } 138 139 static <T> Object[] streamDataDescr(String description, Supplier<LongStream> s) { 140 return new Object[] { description, TestData.Factory.ofLongSupplier(description, s) }; 141 } 142 143 static <T> Object[] splitDescr(String description, Supplier<Spliterator.OfLong> s) { 144 return new Object[] { description, s }; 145 } 146 147 // Return an array of ( String name, LongStreamTestData ) 148 @DataProvider(name = "LongStreamTestData") 149 public static Object[][] makeLongStreamTestData() { 150 return testData; 151 } 152 153 // returns an array of (String name, Supplier<PrimitiveSpliterator<Long>>) 154 @DataProvider(name = "LongSpliterator") 155 public static Object[][] spliteratorProvider() { 156 return spliteratorTestData; 157 } 158 } 159
