14c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin/* 24c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Copyright (c) 2012, 2013, Oracle and/or its affiliates. All rights reserved. 34c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 44c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 54c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * This code is free software; you can redistribute it and/or modify it 64c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * under the terms of the GNU General Public License version 2 only, as 74c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * published by the Free Software Foundation. 84c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 94c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * This code is distributed in the hope that it will be useful, but WITHOUT 104c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 114c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 124c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * version 2 for more details (a copy is included in the LICENSE file that 134c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * accompanied this code). 144c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 154c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * You should have received a copy of the GNU General Public License version 164c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 2 along with this work; if not, write to the Free Software Foundation, 174c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 184c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * 194c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 204c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * or visit www.oracle.com if you need additional information or have any 214c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin * questions. 224c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin */ 234c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinpackage java.util.stream; 244c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 254c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.ArrayList; 264c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.Arrays; 274c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.Collections; 284c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.Iterator; 294c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.List; 304c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport java.util.function.Function; 314c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 324c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport org.testng.annotations.DataProvider; 334c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinimport org.testng.annotations.Test; 344c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 354c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin@Test 364c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkinpublic class NodeTest extends OpTestCase { 374c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 384c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @DataProvider(name = "nodes") 394c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public Object[][] createSizes() { 404c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Object[]> params = new ArrayList<>(); 414c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 424c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int size : Arrays.asList(0, 1, 4, 15, 16, 17, 127, 128, 129, 1000)) { 434c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Integer[] array = new Integer[size]; 444c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i = 0; i < array.length; i++) { 454c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin array[i] = i; 464c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 474c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 484c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Node<Integer>> nodes = new ArrayList<>(); 494c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(Nodes.node(array)); 504c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(Nodes.node(Arrays.asList(array))); 514c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(degenerateTree(Arrays.asList(array).iterator())); 524c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(tree(Arrays.asList(array), l -> Nodes.node(l.toArray(new Integer[l.size()])))); 534c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(tree(Arrays.asList(array), l -> Nodes.node(l))); 544c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(fill(array, Nodes.builder(array.length, LambdaTestHelpers.integerArrayGenerator))); 554c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nodes.add(fill(array, Nodes.builder())); 564c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 574c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int i = 0; i < nodes.size(); i++) { 584c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin params.add(new Object[]{array, nodes.get(i)}); 594c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 604c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 614c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 624c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 634c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return params.toArray(new Object[0][]); 644c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 654c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 664c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Node<Integer> fill(Integer[] array, Node.Builder<Integer> nb) { 674c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.begin(array.length); 684c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (Integer i : array) { 694c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.accept(i); 704c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 714c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin nb.end(); 724c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return nb.build(); 734c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 744c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 754c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Node<Integer> degenerateTree(Iterator<Integer> it) { 764c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (!it.hasNext()) { 774c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return Nodes.node(Collections.emptyList()); 784c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 794c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 804c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Integer i = it.next(); 814c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (it.hasNext()) { 824c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return new Nodes.ConcNode<Integer>(Nodes.node(new Integer[] {i}), degenerateTree(it)); 834c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 844c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin else { 854c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return Nodes.node(new Integer[]{i}); 864c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 874c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 884c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 894c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Node<Integer> tree(List<Integer> l, Function<List<Integer>, Node<Integer>> m) { 904c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (l.size() < 3) { 914c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return m.apply(l); 924c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 934c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin else { 944c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin return new Nodes.ConcNode<>( 954c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin tree(l.subList(0, l.size() / 2), m), 964c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin tree(l.subList(l.size() / 2, l.size()), m )); 974c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 984c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 994c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1004c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1014c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testAsArray(Integer[] array, Node<Integer> n) { 1024c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(n.asArray(LambdaTestHelpers.integerArrayGenerator), array); 1034c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1044c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1054c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1064c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testFlattenAsArray(Integer[] array, Node<Integer> n) { 1074c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(Nodes.flatten(n, LambdaTestHelpers.integerArrayGenerator) 1084c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin .asArray(LambdaTestHelpers.integerArrayGenerator), array); 1094c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1104c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1114c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1124c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testCopyTo(Integer[] array, Node<Integer> n) { 1134c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Integer[] copy = new Integer[(int) n.count()]; 1144c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin n.copyInto(copy, 0); 1154c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1164c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(copy, array); 1174c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1184c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1194c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes", groups = { "serialization-hostile" }) 1204c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testForEach(Integer[] array, Node<Integer> n) { 1214c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin List<Integer> l = new ArrayList<>((int) n.count()); 1224c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin n.forEach(e -> l.add(e)); 1234c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1244c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(l.toArray(), array); 1254c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1264c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1274c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1284c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testStreams(Integer[] array, Node<Integer> n) { 1294c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin TestData<Integer, Stream<Integer>> data = TestData.Factory.ofRefNode("Node", n); 1304c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1314c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin exerciseOps(data, s -> s); 1324c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1334c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin exerciseTerminalOps(data, s -> s.toArray()); 1344c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1354c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1364c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1374c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testSpliterator(Integer[] array, Node<Integer> n) { 1384c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin SpliteratorTestHelper.testSpliterator(n::spliterator); 1394c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1404c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin 1414c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin @Test(dataProvider = "nodes") 1424c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin public void testTruncate(Integer[] array, Node<Integer> n) { 1434c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin int[] nums = new int[] { 0, 1, array.length / 2, array.length - 1, array.length }; 1444c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int start : nums) 1454c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int end : nums) { 1464c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin if (start < 0 || end < 0 || end < start || end > array.length) 1474c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin continue; 1484c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Node<Integer> slice = n.truncate(start, end, Integer[]::new); 1494c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin Integer[] asArray = slice.asArray(Integer[]::new); 1504c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin for (int k = start; k < end; k++) 1514c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin assertEquals(array[k], asArray[k - start]); 1524c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1534c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin } 1544c65c01a4fcfe8c68cd73914370bd269712b91d9Igor Murashkin} 155