Path: blob/aarch64-shenandoah-jdk8u272-b10/jdk/test/java/util/Spliterator/SpliteratorTraversingAndSplittingTest.java
38811 views
/*1* Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.2* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.3*4* This code is free software; you can redistribute it and/or modify it5* under the terms of the GNU General Public License version 2 only, as6* published by the Free Software Foundation.7*8* This code is distributed in the hope that it will be useful, but WITHOUT9* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or10* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License11* version 2 for more details (a copy is included in the LICENSE file that12* accompanied this code).13*14* You should have received a copy of the GNU General Public License version15* 2 along with this work; if not, write to the Free Software Foundation,16* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.17*18* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA19* or visit www.oracle.com if you need additional information or have any20* questions.21*/2223/**24* @test25* @summary Spliterator traversing and splitting tests26* @run testng SpliteratorTraversingAndSplittingTest27* @bug 802001628*/2930import org.testng.annotations.DataProvider;31import org.testng.annotations.Test;3233import java.util.AbstractCollection;34import java.util.AbstractList;35import java.util.AbstractSet;36import java.util.ArrayDeque;37import java.util.ArrayList;38import java.util.Arrays;39import java.util.Collection;40import java.util.Collections;41import java.util.Comparator;42import java.util.Deque;43import java.util.HashMap;44import java.util.HashSet;45import java.util.IdentityHashMap;46import java.util.Iterator;47import java.util.LinkedHashMap;48import java.util.LinkedHashSet;49import java.util.LinkedList;50import java.util.List;51import java.util.Map;52import java.util.PriorityQueue;53import java.util.Set;54import java.util.SortedSet;55import java.util.Spliterator;56import java.util.Spliterators;57import java.util.Stack;58import java.util.TreeMap;59import java.util.TreeSet;60import java.util.Vector;61import java.util.WeakHashMap;62import java.util.concurrent.ArrayBlockingQueue;63import java.util.concurrent.ConcurrentHashMap;64import java.util.concurrent.ConcurrentLinkedQueue;65import java.util.concurrent.ConcurrentSkipListMap;66import java.util.concurrent.ConcurrentSkipListSet;67import java.util.concurrent.CopyOnWriteArrayList;68import java.util.concurrent.CopyOnWriteArraySet;69import java.util.concurrent.LinkedBlockingDeque;70import java.util.concurrent.LinkedBlockingQueue;71import java.util.concurrent.LinkedTransferQueue;72import java.util.concurrent.PriorityBlockingQueue;73import java.util.function.Consumer;74import java.util.function.DoubleConsumer;75import java.util.function.Function;76import java.util.function.IntConsumer;77import java.util.function.LongConsumer;78import java.util.function.Supplier;79import java.util.function.UnaryOperator;8081import static org.testng.Assert.*;82import static org.testng.Assert.assertEquals;8384@Test85public class SpliteratorTraversingAndSplittingTest {8687private static List<Integer> SIZES = Arrays.asList(0, 1, 10, 100, 1000);8889private static class SpliteratorDataBuilder<T> {90List<Object[]> data;9192List<T> exp;9394Map<T, T> mExp;9596SpliteratorDataBuilder(List<Object[]> data, List<T> exp) {97this.data = data;98this.exp = exp;99this.mExp = createMap(exp);100}101102Map<T, T> createMap(List<T> l) {103Map<T, T> m = new LinkedHashMap<>();104for (T t : l) {105m.put(t, t);106}107return m;108}109110void add(String description, Collection<?> expected, Supplier<Spliterator<?>> s) {111description = joiner(description).toString();112data.add(new Object[]{description, expected, s});113}114115void add(String description, Supplier<Spliterator<?>> s) {116add(description, exp, s);117}118119void addCollection(Function<Collection<T>, ? extends Collection<T>> c) {120add("new " + c.apply(Collections.<T>emptyList()).getClass().getName() + ".spliterator()",121() -> c.apply(exp).spliterator());122}123124void addList(Function<Collection<T>, ? extends List<T>> l) {125// @@@ If collection is instance of List then add sub-list tests126addCollection(l);127}128129void addMap(Function<Map<T, T>, ? extends Map<T, T>> m) {130String description = "new " + m.apply(Collections.<T, T>emptyMap()).getClass().getName();131addMap(m, description);132}133134void addMap(Function<Map<T, T>, ? extends Map<T, T>> m, String description) {135add(description + ".keySet().spliterator()", () -> m.apply(mExp).keySet().spliterator());136add(description + ".values().spliterator()", () -> m.apply(mExp).values().spliterator());137add(description + ".entrySet().spliterator()", mExp.entrySet(), () -> m.apply(mExp).entrySet().spliterator());138}139140StringBuilder joiner(String description) {141return new StringBuilder(description).142append(" {").143append("size=").append(exp.size()).144append("}");145}146}147148static Object[][] spliteratorDataProvider;149150@DataProvider(name = "Spliterator<Integer>")151public static Object[][] spliteratorDataProvider() {152if (spliteratorDataProvider != null) {153return spliteratorDataProvider;154}155156List<Object[]> data = new ArrayList<>();157for (int size : SIZES) {158List<Integer> exp = listIntRange(size);159SpliteratorDataBuilder<Integer> db = new SpliteratorDataBuilder<>(data, exp);160161// Direct spliterator methods162163db.add("Spliterators.spliterator(Collection, ...)",164() -> Spliterators.spliterator(exp, 0));165166db.add("Spliterators.spliterator(Iterator, ...)",167() -> Spliterators.spliterator(exp.iterator(), exp.size(), 0));168169db.add("Spliterators.spliteratorUnknownSize(Iterator, ...)",170() -> Spliterators.spliteratorUnknownSize(exp.iterator(), 0));171172db.add("Spliterators.spliterator(Spliterators.iteratorFromSpliterator(Spliterator ), ...)",173() -> Spliterators.spliterator(Spliterators.iterator(exp.spliterator()), exp.size(), 0));174175db.add("Spliterators.spliterator(T[], ...)",176() -> Spliterators.spliterator(exp.toArray(new Integer[0]), 0));177178db.add("Arrays.spliterator(T[], ...)",179() -> Arrays.spliterator(exp.toArray(new Integer[0])));180181class SpliteratorFromIterator extends Spliterators.AbstractSpliterator<Integer> {182Iterator<Integer> it;183184SpliteratorFromIterator(Iterator<Integer> it, long est) {185super(est, Spliterator.SIZED);186this.it = it;187}188189@Override190public boolean tryAdvance(Consumer<? super Integer> action) {191if (action == null)192throw new NullPointerException();193if (it.hasNext()) {194action.accept(it.next());195return true;196}197else {198return false;199}200}201}202db.add("new Spliterators.AbstractSpliterator()",203() -> new SpliteratorFromIterator(exp.iterator(), exp.size()));204205// Collections206207// default method implementations208209class AbstractCollectionImpl extends AbstractCollection<Integer> {210Collection<Integer> c;211212AbstractCollectionImpl(Collection<Integer> c) {213this.c = c;214}215216@Override217public Iterator<Integer> iterator() {218return c.iterator();219}220221@Override222public int size() {223return c.size();224}225}226db.addCollection(227c -> new AbstractCollectionImpl(c));228229class AbstractListImpl extends AbstractList<Integer> {230List<Integer> l;231232AbstractListImpl(Collection<Integer> c) {233this.l = new ArrayList<>(c);234}235236@Override237public Integer get(int index) {238return l.get(index);239}240241@Override242public int size() {243return l.size();244}245}246db.addCollection(247c -> new AbstractListImpl(c));248249class AbstractSetImpl extends AbstractSet<Integer> {250Set<Integer> s;251252AbstractSetImpl(Collection<Integer> c) {253this.s = new HashSet<>(c);254}255256@Override257public Iterator<Integer> iterator() {258return s.iterator();259}260261@Override262public int size() {263return s.size();264}265}266db.addCollection(267c -> new AbstractSetImpl(c));268269class AbstractSortedSetImpl extends AbstractSet<Integer> implements SortedSet<Integer> {270SortedSet<Integer> s;271272AbstractSortedSetImpl(Collection<Integer> c) {273this.s = new TreeSet<>(c);274}275276@Override277public Iterator<Integer> iterator() {278return s.iterator();279}280281@Override282public int size() {283return s.size();284}285286@Override287public Comparator<? super Integer> comparator() {288return s.comparator();289}290291@Override292public SortedSet<Integer> subSet(Integer fromElement, Integer toElement) {293return s.subSet(fromElement, toElement);294}295296@Override297public SortedSet<Integer> headSet(Integer toElement) {298return s.headSet(toElement);299}300301@Override302public SortedSet<Integer> tailSet(Integer fromElement) {303return s.tailSet(fromElement);304}305306@Override307public Integer first() {308return s.first();309}310311@Override312public Integer last() {313return s.last();314}315316@Override317public Spliterator<Integer> spliterator() {318return SortedSet.super.spliterator();319}320}321db.addCollection(322c -> new AbstractSortedSetImpl(c));323324class IterableWrapper implements Iterable<Integer> {325final Iterable<Integer> it;326327IterableWrapper(Iterable<Integer> it) {328this.it = it;329}330331@Override332public Iterator<Integer> iterator() {333return it.iterator();334}335}336db.add("new Iterable.spliterator()",337() -> new IterableWrapper(exp).spliterator());338339//340341db.add("Arrays.asList().spliterator()",342() -> Spliterators.spliterator(Arrays.asList(exp.toArray(new Integer[0])), 0));343344db.addList(ArrayList::new);345346db.addList(LinkedList::new);347348db.addList(Vector::new);349350351db.addCollection(HashSet::new);352353db.addCollection(LinkedHashSet::new);354355db.addCollection(TreeSet::new);356357358db.addCollection(c -> { Stack<Integer> s = new Stack<>(); s.addAll(c); return s;});359360db.addCollection(PriorityQueue::new);361362db.addCollection(ArrayDeque::new);363364365db.addCollection(ConcurrentSkipListSet::new);366367if (size > 0) {368db.addCollection(c -> {369ArrayBlockingQueue<Integer> abq = new ArrayBlockingQueue<>(size);370abq.addAll(c);371return abq;372});373}374375db.addCollection(PriorityBlockingQueue::new);376377db.addCollection(LinkedBlockingQueue::new);378379db.addCollection(LinkedTransferQueue::new);380381db.addCollection(ConcurrentLinkedQueue::new);382383db.addCollection(LinkedBlockingDeque::new);384385db.addCollection(CopyOnWriteArrayList::new);386387db.addCollection(CopyOnWriteArraySet::new);388389if (size == 0) {390db.addCollection(c -> Collections.<Integer>emptySet());391db.addList(c -> Collections.<Integer>emptyList());392}393else if (size == 1) {394db.addCollection(c -> Collections.singleton(exp.get(0)));395db.addCollection(c -> Collections.singletonList(exp.get(0)));396}397398{399Integer[] ai = new Integer[size];400Arrays.fill(ai, 1);401db.add(String.format("Collections.nCopies(%d, 1)", exp.size()),402Arrays.asList(ai),403() -> Collections.nCopies(exp.size(), 1).spliterator());404}405406// Collections.synchronized/unmodifiable/checked wrappers407db.addCollection(Collections::unmodifiableCollection);408db.addCollection(c -> Collections.unmodifiableSet(new HashSet<>(c)));409db.addCollection(c -> Collections.unmodifiableSortedSet(new TreeSet<>(c)));410db.addList(c -> Collections.unmodifiableList(new ArrayList<>(c)));411db.addMap(Collections::unmodifiableMap);412db.addMap(m -> Collections.unmodifiableSortedMap(new TreeMap<>(m)));413414db.addCollection(Collections::synchronizedCollection);415db.addCollection(c -> Collections.synchronizedSet(new HashSet<>(c)));416db.addCollection(c -> Collections.synchronizedSortedSet(new TreeSet<>(c)));417db.addList(c -> Collections.synchronizedList(new ArrayList<>(c)));418db.addMap(Collections::synchronizedMap);419db.addMap(m -> Collections.synchronizedSortedMap(new TreeMap<>(m)));420421db.addCollection(c -> Collections.checkedCollection(c, Integer.class));422db.addCollection(c -> Collections.checkedQueue(new ArrayDeque<>(c), Integer.class));423db.addCollection(c -> Collections.checkedSet(new HashSet<>(c), Integer.class));424db.addCollection(c -> Collections.checkedSortedSet(new TreeSet<>(c), Integer.class));425db.addList(c -> Collections.checkedList(new ArrayList<>(c), Integer.class));426db.addMap(c -> Collections.checkedMap(c, Integer.class, Integer.class));427db.addMap(m -> Collections.checkedSortedMap(new TreeMap<>(m), Integer.class, Integer.class));428429// Maps430431db.addMap(HashMap::new);432433db.addMap(m -> {434// Create a Map ensuring that for large sizes435// buckets will contain 2 or more entries436HashMap<Integer, Integer> cm = new HashMap<>(1, m.size() + 1);437// Don't use putAll which inflates the table by438// m.size() * loadFactor, thus creating a very sparse439// map for 1000 entries defeating the purpose of this test,440// in addition it will cause the split until null test to fail441// because the number of valid splits is larger than the442// threshold443for (Map.Entry<Integer, Integer> e : m.entrySet())444cm.put(e.getKey(), e.getValue());445return cm;446}, "new java.util.HashMap(1, size + 1)");447448db.addMap(LinkedHashMap::new);449450db.addMap(IdentityHashMap::new);451452db.addMap(WeakHashMap::new);453454db.addMap(m -> {455// Create a Map ensuring that for large sizes456// buckets will be consist of 2 or more entries457WeakHashMap<Integer, Integer> cm = new WeakHashMap<>(1, m.size() + 1);458for (Map.Entry<Integer, Integer> e : m.entrySet())459cm.put(e.getKey(), e.getValue());460return cm;461}, "new java.util.WeakHashMap(1, size + 1)");462463// @@@ Descending maps etc464db.addMap(TreeMap::new);465466db.addMap(ConcurrentHashMap::new);467468db.addMap(ConcurrentSkipListMap::new);469470if (size == 0) {471db.addMap(m -> Collections.<Integer, Integer>emptyMap());472}473else if (size == 1) {474db.addMap(m -> Collections.singletonMap(exp.get(0), exp.get(0)));475}476}477478return spliteratorDataProvider = data.toArray(new Object[0][]);479}480481private static List<Integer> listIntRange(int upTo) {482List<Integer> exp = new ArrayList<>();483for (int i = 0; i < upTo; i++)484exp.add(i);485return Collections.unmodifiableList(exp);486}487488@Test(dataProvider = "Spliterator<Integer>")489@SuppressWarnings({"unchecked", "rawtypes"})490public void testNullPointerException(String description, Collection exp, Supplier<Spliterator> s) {491executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining(null));492executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance(null));493}494495@Test(dataProvider = "Spliterator<Integer>")496@SuppressWarnings({"unchecked", "rawtypes"})497public void testForEach(String description, Collection exp, Supplier<Spliterator> s) {498testForEach(exp, s, (Consumer<Object> b) -> b);499}500501@Test(dataProvider = "Spliterator<Integer>")502@SuppressWarnings({"unchecked", "rawtypes"})503public void testTryAdvance(String description, Collection exp, Supplier<Spliterator> s) {504testTryAdvance(exp, s, (Consumer<Object> b) -> b);505}506507@Test(dataProvider = "Spliterator<Integer>")508@SuppressWarnings({"unchecked", "rawtypes"})509public void testMixedTryAdvanceForEach(String description, Collection exp, Supplier<Spliterator> s) {510testMixedTryAdvanceForEach(exp, s, (Consumer<Object> b) -> b);511}512513@Test(dataProvider = "Spliterator<Integer>")514@SuppressWarnings({"unchecked", "rawtypes"})515public void testSplitAfterFullTraversal(String description, Collection exp, Supplier<Spliterator> s) {516testSplitAfterFullTraversal(s, (Consumer<Object> b) -> b);517}518519@Test(dataProvider = "Spliterator<Integer>")520@SuppressWarnings({"unchecked", "rawtypes"})521public void testSplitOnce(String description, Collection exp, Supplier<Spliterator> s) {522testSplitOnce(exp, s, (Consumer<Object> b) -> b);523}524525@Test(dataProvider = "Spliterator<Integer>")526@SuppressWarnings({"unchecked", "rawtypes"})527public void testSplitSixDeep(String description, Collection exp, Supplier<Spliterator> s) {528testSplitSixDeep(exp, s, (Consumer<Object> b) -> b);529}530531@Test(dataProvider = "Spliterator<Integer>")532@SuppressWarnings({"unchecked", "rawtypes"})533public void testSplitUntilNull(String description, Collection exp, Supplier<Spliterator> s) {534testSplitUntilNull(exp, s, (Consumer<Object> b) -> b);535}536537//538539private static class SpliteratorOfIntDataBuilder {540List<Object[]> data;541542List<Integer> exp;543544SpliteratorOfIntDataBuilder(List<Object[]> data, List<Integer> exp) {545this.data = data;546this.exp = exp;547}548549void add(String description, List<Integer> expected, Supplier<Spliterator.OfInt> s) {550description = joiner(description).toString();551data.add(new Object[]{description, expected, s});552}553554void add(String description, Supplier<Spliterator.OfInt> s) {555add(description, exp, s);556}557558StringBuilder joiner(String description) {559return new StringBuilder(description).560append(" {").561append("size=").append(exp.size()).562append("}");563}564}565566static Object[][] spliteratorOfIntDataProvider;567568@DataProvider(name = "Spliterator.OfInt")569public static Object[][] spliteratorOfIntDataProvider() {570if (spliteratorOfIntDataProvider != null) {571return spliteratorOfIntDataProvider;572}573574List<Object[]> data = new ArrayList<>();575for (int size : SIZES) {576int exp[] = arrayIntRange(size);577SpliteratorOfIntDataBuilder db = new SpliteratorOfIntDataBuilder(data, listIntRange(size));578579db.add("Spliterators.spliterator(int[], ...)",580() -> Spliterators.spliterator(exp, 0));581582db.add("Arrays.spliterator(int[], ...)",583() -> Arrays.spliterator(exp));584585db.add("Spliterators.spliterator(PrimitiveIterator.OfInt, ...)",586() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));587588db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfInt, ...)",589() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));590591class IntSpliteratorFromArray extends Spliterators.AbstractIntSpliterator {592int[] a;593int index = 0;594595IntSpliteratorFromArray(int[] a) {596super(a.length, Spliterator.SIZED);597this.a = a;598}599600@Override601public boolean tryAdvance(IntConsumer action) {602if (action == null)603throw new NullPointerException();604if (index < a.length) {605action.accept(a[index++]);606return true;607}608else {609return false;610}611}612}613db.add("new Spliterators.AbstractIntAdvancingSpliterator()",614() -> new IntSpliteratorFromArray(exp));615}616617return spliteratorOfIntDataProvider = data.toArray(new Object[0][]);618}619620private static int[] arrayIntRange(int upTo) {621int[] exp = new int[upTo];622for (int i = 0; i < upTo; i++)623exp[i] = i;624return exp;625}626627private static UnaryOperator<Consumer<Integer>> intBoxingConsumer() {628class BoxingAdapter implements Consumer<Integer>, IntConsumer {629private final Consumer<Integer> b;630631BoxingAdapter(Consumer<Integer> b) {632this.b = b;633}634635@Override636public void accept(Integer value) {637throw new IllegalStateException();638}639640@Override641public void accept(int value) {642b.accept(value);643}644}645646return b -> new BoxingAdapter(b);647}648649@Test(dataProvider = "Spliterator.OfInt")650public void testIntNullPointerException(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {651executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((IntConsumer) null));652executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((IntConsumer) null));653}654655@Test(dataProvider = "Spliterator.OfInt")656public void testIntForEach(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {657testForEach(exp, s, intBoxingConsumer());658}659660@Test(dataProvider = "Spliterator.OfInt")661public void testIntTryAdvance(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {662testTryAdvance(exp, s, intBoxingConsumer());663}664665@Test(dataProvider = "Spliterator.OfInt")666public void testIntMixedTryAdvanceForEach(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {667testMixedTryAdvanceForEach(exp, s, intBoxingConsumer());668}669670@Test(dataProvider = "Spliterator.OfInt")671public void testIntSplitAfterFullTraversal(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {672testSplitAfterFullTraversal(s, intBoxingConsumer());673}674675@Test(dataProvider = "Spliterator.OfInt")676public void testIntSplitOnce(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {677testSplitOnce(exp, s, intBoxingConsumer());678}679680@Test(dataProvider = "Spliterator.OfInt")681public void testIntSplitSixDeep(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {682testSplitSixDeep(exp, s, intBoxingConsumer());683}684685@Test(dataProvider = "Spliterator.OfInt")686public void testIntSplitUntilNull(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {687testSplitUntilNull(exp, s, intBoxingConsumer());688}689690//691692private static class SpliteratorOfLongDataBuilder {693List<Object[]> data;694695List<Long> exp;696697SpliteratorOfLongDataBuilder(List<Object[]> data, List<Long> exp) {698this.data = data;699this.exp = exp;700}701702void add(String description, List<Long> expected, Supplier<Spliterator.OfLong> s) {703description = joiner(description).toString();704data.add(new Object[]{description, expected, s});705}706707void add(String description, Supplier<Spliterator.OfLong> s) {708add(description, exp, s);709}710711StringBuilder joiner(String description) {712return new StringBuilder(description).713append(" {").714append("size=").append(exp.size()).715append("}");716}717}718719static Object[][] spliteratorOfLongDataProvider;720721@DataProvider(name = "Spliterator.OfLong")722public static Object[][] spliteratorOfLongDataProvider() {723if (spliteratorOfLongDataProvider != null) {724return spliteratorOfLongDataProvider;725}726727List<Object[]> data = new ArrayList<>();728for (int size : SIZES) {729long exp[] = arrayLongRange(size);730SpliteratorOfLongDataBuilder db = new SpliteratorOfLongDataBuilder(data, listLongRange(size));731732db.add("Spliterators.spliterator(long[], ...)",733() -> Spliterators.spliterator(exp, 0));734735db.add("Arrays.spliterator(long[], ...)",736() -> Arrays.spliterator(exp));737738db.add("Spliterators.spliterator(PrimitiveIterator.OfLong, ...)",739() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));740741db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfLong, ...)",742() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));743744class LongSpliteratorFromArray extends Spliterators.AbstractLongSpliterator {745long[] a;746int index = 0;747748LongSpliteratorFromArray(long[] a) {749super(a.length, Spliterator.SIZED);750this.a = a;751}752753@Override754public boolean tryAdvance(LongConsumer action) {755if (action == null)756throw new NullPointerException();757if (index < a.length) {758action.accept(a[index++]);759return true;760}761else {762return false;763}764}765}766db.add("new Spliterators.AbstractLongAdvancingSpliterator()",767() -> new LongSpliteratorFromArray(exp));768}769770return spliteratorOfLongDataProvider = data.toArray(new Object[0][]);771}772773private static List<Long> listLongRange(int upTo) {774List<Long> exp = new ArrayList<>();775for (long i = 0; i < upTo; i++)776exp.add(i);777return Collections.unmodifiableList(exp);778}779780private static long[] arrayLongRange(int upTo) {781long[] exp = new long[upTo];782for (int i = 0; i < upTo; i++)783exp[i] = i;784return exp;785}786787private static UnaryOperator<Consumer<Long>> longBoxingConsumer() {788class BoxingAdapter implements Consumer<Long>, LongConsumer {789private final Consumer<Long> b;790791BoxingAdapter(Consumer<Long> b) {792this.b = b;793}794795@Override796public void accept(Long value) {797throw new IllegalStateException();798}799800@Override801public void accept(long value) {802b.accept(value);803}804}805806return b -> new BoxingAdapter(b);807}808809@Test(dataProvider = "Spliterator.OfLong")810public void testLongNullPointerException(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {811executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((LongConsumer) null));812executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((LongConsumer) null));813}814815@Test(dataProvider = "Spliterator.OfLong")816public void testLongForEach(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {817testForEach(exp, s, longBoxingConsumer());818}819820@Test(dataProvider = "Spliterator.OfLong")821public void testLongTryAdvance(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {822testTryAdvance(exp, s, longBoxingConsumer());823}824825@Test(dataProvider = "Spliterator.OfLong")826public void testLongMixedTryAdvanceForEach(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {827testMixedTryAdvanceForEach(exp, s, longBoxingConsumer());828}829830@Test(dataProvider = "Spliterator.OfLong")831public void testLongSplitAfterFullTraversal(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {832testSplitAfterFullTraversal(s, longBoxingConsumer());833}834835@Test(dataProvider = "Spliterator.OfLong")836public void testLongSplitOnce(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {837testSplitOnce(exp, s, longBoxingConsumer());838}839840@Test(dataProvider = "Spliterator.OfLong")841public void testLongSplitSixDeep(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {842testSplitSixDeep(exp, s, longBoxingConsumer());843}844845@Test(dataProvider = "Spliterator.OfLong")846public void testLongSplitUntilNull(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {847testSplitUntilNull(exp, s, longBoxingConsumer());848}849850//851852private static class SpliteratorOfDoubleDataBuilder {853List<Object[]> data;854855List<Double> exp;856857SpliteratorOfDoubleDataBuilder(List<Object[]> data, List<Double> exp) {858this.data = data;859this.exp = exp;860}861862void add(String description, List<Double> expected, Supplier<Spliterator.OfDouble> s) {863description = joiner(description).toString();864data.add(new Object[]{description, expected, s});865}866867void add(String description, Supplier<Spliterator.OfDouble> s) {868add(description, exp, s);869}870871StringBuilder joiner(String description) {872return new StringBuilder(description).873append(" {").874append("size=").append(exp.size()).875append("}");876}877}878879static Object[][] spliteratorOfDoubleDataProvider;880881@DataProvider(name = "Spliterator.OfDouble")882public static Object[][] spliteratorOfDoubleDataProvider() {883if (spliteratorOfDoubleDataProvider != null) {884return spliteratorOfDoubleDataProvider;885}886887List<Object[]> data = new ArrayList<>();888for (int size : SIZES) {889double exp[] = arrayDoubleRange(size);890SpliteratorOfDoubleDataBuilder db = new SpliteratorOfDoubleDataBuilder(data, listDoubleRange(size));891892db.add("Spliterators.spliterator(double[], ...)",893() -> Spliterators.spliterator(exp, 0));894895db.add("Arrays.spliterator(double[], ...)",896() -> Arrays.spliterator(exp));897898db.add("Spliterators.spliterator(PrimitiveIterator.OfDouble, ...)",899() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));900901db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfDouble, ...)",902() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));903904class DoubleSpliteratorFromArray extends Spliterators.AbstractDoubleSpliterator {905double[] a;906int index = 0;907908DoubleSpliteratorFromArray(double[] a) {909super(a.length, Spliterator.SIZED);910this.a = a;911}912913@Override914public boolean tryAdvance(DoubleConsumer action) {915if (action == null)916throw new NullPointerException();917if (index < a.length) {918action.accept(a[index++]);919return true;920}921else {922return false;923}924}925}926db.add("new Spliterators.AbstractDoubleAdvancingSpliterator()",927() -> new DoubleSpliteratorFromArray(exp));928}929930return spliteratorOfDoubleDataProvider = data.toArray(new Object[0][]);931}932933private static List<Double> listDoubleRange(int upTo) {934List<Double> exp = new ArrayList<>();935for (double i = 0; i < upTo; i++)936exp.add(i);937return Collections.unmodifiableList(exp);938}939940private static double[] arrayDoubleRange(int upTo) {941double[] exp = new double[upTo];942for (int i = 0; i < upTo; i++)943exp[i] = i;944return exp;945}946947private static UnaryOperator<Consumer<Double>> doubleBoxingConsumer() {948class BoxingAdapter implements Consumer<Double>, DoubleConsumer {949private final Consumer<Double> b;950951BoxingAdapter(Consumer<Double> b) {952this.b = b;953}954955@Override956public void accept(Double value) {957throw new IllegalStateException();958}959960@Override961public void accept(double value) {962b.accept(value);963}964}965966return b -> new BoxingAdapter(b);967}968969@Test(dataProvider = "Spliterator.OfDouble")970public void testDoubleNullPointerException(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {971executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((DoubleConsumer) null));972executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((DoubleConsumer) null));973}974975@Test(dataProvider = "Spliterator.OfDouble")976public void testDoubleForEach(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {977testForEach(exp, s, doubleBoxingConsumer());978}979980@Test(dataProvider = "Spliterator.OfDouble")981public void testDoubleTryAdvance(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {982testTryAdvance(exp, s, doubleBoxingConsumer());983}984985@Test(dataProvider = "Spliterator.OfDouble")986public void testDoubleMixedTryAdvanceForEach(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {987testMixedTryAdvanceForEach(exp, s, doubleBoxingConsumer());988}989990@Test(dataProvider = "Spliterator.OfDouble")991public void testDoubleSplitAfterFullTraversal(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {992testSplitAfterFullTraversal(s, doubleBoxingConsumer());993}994995@Test(dataProvider = "Spliterator.OfDouble")996public void testDoubleSplitOnce(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {997testSplitOnce(exp, s, doubleBoxingConsumer());998}9991000@Test(dataProvider = "Spliterator.OfDouble")1001public void testDoubleSplitSixDeep(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {1002testSplitSixDeep(exp, s, doubleBoxingConsumer());1003}10041005@Test(dataProvider = "Spliterator.OfDouble")1006public void testDoubleSplitUntilNull(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {1007testSplitUntilNull(exp, s, doubleBoxingConsumer());1008}10091010//10111012private static <T, S extends Spliterator<T>> void testForEach(1013Collection<T> exp,1014Supplier<S> supplier,1015UnaryOperator<Consumer<T>> boxingAdapter) {1016S spliterator = supplier.get();1017long sizeIfKnown = spliterator.getExactSizeIfKnown();1018boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);10191020ArrayList<T> fromForEach = new ArrayList<>();1021spliterator = supplier.get();1022Consumer<T> addToFromForEach = boxingAdapter.apply(fromForEach::add);1023spliterator.forEachRemaining(addToFromForEach);10241025// Assert that forEach now produces no elements1026spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));1027// Assert that tryAdvance now produce no elements1028spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));10291030// assert that size, tryAdvance, and forEach are consistent1031if (sizeIfKnown >= 0) {1032assertEquals(sizeIfKnown, exp.size());1033}1034assertEquals(fromForEach.size(), exp.size());10351036assertContents(fromForEach, exp, isOrdered);1037}10381039private static <T, S extends Spliterator<T>> void testTryAdvance(1040Collection<T> exp,1041Supplier<S> supplier,1042UnaryOperator<Consumer<T>> boxingAdapter) {1043S spliterator = supplier.get();1044long sizeIfKnown = spliterator.getExactSizeIfKnown();1045boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);10461047spliterator = supplier.get();1048ArrayList<T> fromTryAdvance = new ArrayList<>();1049Consumer<T> addToFromTryAdvance = boxingAdapter.apply(fromTryAdvance::add);1050while (spliterator.tryAdvance(addToFromTryAdvance)) { }10511052// Assert that forEach now produces no elements1053spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));1054// Assert that tryAdvance now produce no elements1055spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));10561057// assert that size, tryAdvance, and forEach are consistent1058if (sizeIfKnown >= 0) {1059assertEquals(sizeIfKnown, exp.size());1060}1061assertEquals(fromTryAdvance.size(), exp.size());10621063assertContents(fromTryAdvance, exp, isOrdered);1064}10651066private static <T, S extends Spliterator<T>> void testMixedTryAdvanceForEach(1067Collection<T> exp,1068Supplier<S> supplier,1069UnaryOperator<Consumer<T>> boxingAdapter) {1070S spliterator = supplier.get();1071long sizeIfKnown = spliterator.getExactSizeIfKnown();1072boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);10731074// tryAdvance first few elements, then forEach rest1075ArrayList<T> dest = new ArrayList<>();1076spliterator = supplier.get();1077Consumer<T> addToDest = boxingAdapter.apply(dest::add);1078for (int i = 0; i < 10 && spliterator.tryAdvance(addToDest); i++) { }1079spliterator.forEachRemaining(addToDest);10801081// Assert that forEach now produces no elements1082spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));1083// Assert that tryAdvance now produce no elements1084spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));10851086if (sizeIfKnown >= 0) {1087assertEquals(sizeIfKnown, dest.size());1088}1089assertEquals(dest.size(), exp.size());10901091if (isOrdered) {1092assertEquals(dest, exp);1093}1094else {1095assertContentsUnordered(dest, exp);1096}1097}10981099private static <T, S extends Spliterator<T>> void testSplitAfterFullTraversal(1100Supplier<S> supplier,1101UnaryOperator<Consumer<T>> boxingAdapter) {1102// Full traversal using tryAdvance1103Spliterator<T> spliterator = supplier.get();1104while (spliterator.tryAdvance(boxingAdapter.apply(e -> { }))) { }1105Spliterator<T> split = spliterator.trySplit();1106assertNull(split);11071108// Full traversal using forEach1109spliterator = supplier.get();1110spliterator.forEachRemaining(boxingAdapter.apply(e -> {1111}));1112split = spliterator.trySplit();1113assertNull(split);11141115// Full traversal using tryAdvance then forEach1116spliterator = supplier.get();1117spliterator.tryAdvance(boxingAdapter.apply(e -> { }));1118spliterator.forEachRemaining(boxingAdapter.apply(e -> {1119}));1120split = spliterator.trySplit();1121assertNull(split);1122}11231124private static <T, S extends Spliterator<T>> void testSplitOnce(1125Collection<T> exp,1126Supplier<S> supplier,1127UnaryOperator<Consumer<T>> boxingAdapter) {1128S spliterator = supplier.get();1129long sizeIfKnown = spliterator.getExactSizeIfKnown();1130boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);11311132ArrayList<T> fromSplit = new ArrayList<>();1133Spliterator<T> s1 = supplier.get();1134Spliterator<T> s2 = s1.trySplit();1135long s1Size = s1.getExactSizeIfKnown();1136long s2Size = (s2 != null) ? s2.getExactSizeIfKnown() : 0;1137Consumer<T> addToFromSplit = boxingAdapter.apply(fromSplit::add);1138if (s2 != null)1139s2.forEachRemaining(addToFromSplit);1140s1.forEachRemaining(addToFromSplit);11411142if (sizeIfKnown >= 0) {1143assertEquals(sizeIfKnown, fromSplit.size());1144if (s1Size >= 0 && s2Size >= 0)1145assertEquals(sizeIfKnown, s1Size + s2Size);1146}1147assertContents(fromSplit, exp, isOrdered);1148}11491150private static <T, S extends Spliterator<T>> void testSplitSixDeep(1151Collection<T> exp,1152Supplier<S> supplier,1153UnaryOperator<Consumer<T>> boxingAdapter) {1154S spliterator = supplier.get();1155boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);11561157for (int depth=0; depth < 6; depth++) {1158List<T> dest = new ArrayList<>();1159spliterator = supplier.get();11601161assertRootSpliterator(spliterator);11621163// verify splitting with forEach1164visit(depth, 0, dest, spliterator, boxingAdapter, spliterator.characteristics(), false);1165assertContents(dest, exp, isOrdered);11661167// verify splitting with tryAdvance1168dest.clear();1169spliterator = supplier.get();1170visit(depth, 0, dest, spliterator, boxingAdapter, spliterator.characteristics(), true);1171assertContents(dest, exp, isOrdered);1172}1173}11741175private static <T, S extends Spliterator<T>> void visit(int depth, int curLevel,1176List<T> dest, S spliterator, UnaryOperator<Consumer<T>> boxingAdapter,1177int rootCharacteristics, boolean useTryAdvance) {1178if (curLevel < depth) {1179long beforeSize = spliterator.getExactSizeIfKnown();1180Spliterator<T> split = spliterator.trySplit();1181if (split != null) {1182assertSpliterator(split, rootCharacteristics);1183assertSpliterator(spliterator, rootCharacteristics);11841185if ((rootCharacteristics & Spliterator.SUBSIZED) != 0 &&1186(rootCharacteristics & Spliterator.SIZED) != 0) {1187assertEquals(beforeSize, split.estimateSize() + spliterator.estimateSize());1188}1189visit(depth, curLevel + 1, dest, split, boxingAdapter, rootCharacteristics, useTryAdvance);1190}1191visit(depth, curLevel + 1, dest, spliterator, boxingAdapter, rootCharacteristics, useTryAdvance);1192}1193else {1194long sizeIfKnown = spliterator.getExactSizeIfKnown();1195if (useTryAdvance) {1196Consumer<T> addToDest = boxingAdapter.apply(dest::add);1197int count = 0;1198while (spliterator.tryAdvance(addToDest)) {1199++count;1200}12011202if (sizeIfKnown >= 0)1203assertEquals(sizeIfKnown, count);12041205// Assert that forEach now produces no elements1206spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));12071208Spliterator<T> split = spliterator.trySplit();1209assertNull(split);1210}1211else {1212List<T> leafDest = new ArrayList<>();1213Consumer<T> addToLeafDest = boxingAdapter.apply(leafDest::add);1214spliterator.forEachRemaining(addToLeafDest);12151216if (sizeIfKnown >= 0)1217assertEquals(sizeIfKnown, leafDest.size());12181219// Assert that forEach now produces no elements1220spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));12211222Spliterator<T> split = spliterator.trySplit();1223assertNull(split);12241225dest.addAll(leafDest);1226}1227}1228}12291230private static <T, S extends Spliterator<T>> void testSplitUntilNull(1231Collection<T> exp,1232Supplier<S> supplier,1233UnaryOperator<Consumer<T>> boxingAdapter) {1234Spliterator<T> s = supplier.get();1235boolean isOrdered = s.hasCharacteristics(Spliterator.ORDERED);1236assertRootSpliterator(s);12371238List<T> splits = new ArrayList<>();1239Consumer<T> c = boxingAdapter.apply(splits::add);12401241testSplitUntilNull(new SplitNode<T>(c, s));1242assertContents(splits, exp, isOrdered);1243}12441245private static class SplitNode<T> {1246// Constant for every node1247final Consumer<T> c;1248final int rootCharacteristics;12491250final Spliterator<T> s;12511252SplitNode(Consumer<T> c, Spliterator<T> s) {1253this(c, s.characteristics(), s);1254}12551256private SplitNode(Consumer<T> c, int rootCharacteristics, Spliterator<T> s) {1257this.c = c;1258this.rootCharacteristics = rootCharacteristics;1259this.s = s;1260}12611262SplitNode<T> fromSplit(Spliterator<T> split) {1263return new SplitNode<>(c, rootCharacteristics, split);1264}1265}12661267/**1268* Set the maximum stack capacity to 0.25MB. This should be more than enough to detect a bad spliterator1269* while not unduly disrupting test infrastructure given the test data sizes that are used are small.1270* Note that j.u.c.ForkJoinPool sets the max queue size to 64M (1 << 26).1271*/1272private static final int MAXIMUM_STACK_CAPACITY = 1 << 18; // 0.25MB12731274private static <T> void testSplitUntilNull(SplitNode<T> e) {1275// Use an explicit stack to avoid a StackOverflowException when testing a Spliterator1276// that when repeatedly split produces a right-balanced (and maybe degenerate) tree, or1277// for a spliterator that is badly behaved.1278Deque<SplitNode<T>> stack = new ArrayDeque<>();1279stack.push(e);12801281int iteration = 0;1282while (!stack.isEmpty()) {1283assertTrue(iteration++ < MAXIMUM_STACK_CAPACITY, "Exceeded maximum stack modification count of 1 << 18");12841285e = stack.pop();1286Spliterator<T> parentAndRightSplit = e.s;12871288long parentEstimateSize = parentAndRightSplit.estimateSize();1289assertTrue(parentEstimateSize >= 0,1290String.format("Split size estimate %d < 0", parentEstimateSize));12911292long parentSize = parentAndRightSplit.getExactSizeIfKnown();1293Spliterator<T> leftSplit = parentAndRightSplit.trySplit();1294if (leftSplit == null) {1295parentAndRightSplit.forEachRemaining(e.c);1296continue;1297}12981299assertSpliterator(leftSplit, e.rootCharacteristics);1300assertSpliterator(parentAndRightSplit, e.rootCharacteristics);13011302if (parentEstimateSize != Long.MAX_VALUE && leftSplit.estimateSize() > 0 && parentAndRightSplit.estimateSize() > 0) {1303assertTrue(leftSplit.estimateSize() < parentEstimateSize,1304String.format("Left split size estimate %d >= parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));1305assertTrue(parentAndRightSplit.estimateSize() < parentEstimateSize,1306String.format("Right split size estimate %d >= parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));1307}1308else {1309assertTrue(leftSplit.estimateSize() <= parentEstimateSize,1310String.format("Left split size estimate %d > parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));1311assertTrue(parentAndRightSplit.estimateSize() <= parentEstimateSize,1312String.format("Right split size estimate %d > parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));1313}13141315long leftSize = leftSplit.getExactSizeIfKnown();1316long rightSize = parentAndRightSplit.getExactSizeIfKnown();1317if (parentSize >= 0 && leftSize >= 0 && rightSize >= 0)1318assertEquals(parentSize, leftSize + rightSize,1319String.format("exact left split size %d + exact right split size %d != parent exact split size %d",1320leftSize, rightSize, parentSize));13211322// Add right side to stack first so left side is popped off first1323stack.push(e.fromSplit(parentAndRightSplit));1324stack.push(e.fromSplit(leftSplit));1325}1326}13271328private static void assertRootSpliterator(Spliterator<?> s) {1329assertFalse(s.hasCharacteristics(Spliterator.SIZED | Spliterator.CONCURRENT),1330"Root spliterator should not be SIZED and CONCURRENT");13311332assertSpliterator(s);1333}13341335private static void assertSpliterator(Spliterator<?> s, int rootCharacteristics) {1336if ((rootCharacteristics & Spliterator.SUBSIZED) != 0) {1337assertTrue(s.hasCharacteristics(Spliterator.SUBSIZED),1338"Child split is not SUBSIZED when root split is SUBSIZED");1339}1340assertSpliterator(s);1341}13421343private static void assertSpliterator(Spliterator<?> s) {1344if (s.hasCharacteristics(Spliterator.SUBSIZED)) {1345assertTrue(s.hasCharacteristics(Spliterator.SIZED));1346}1347if (s.hasCharacteristics(Spliterator.SIZED)) {1348assertTrue(s.estimateSize() != Long.MAX_VALUE);1349assertTrue(s.getExactSizeIfKnown() >= 0);1350}1351try {1352s.getComparator();1353assertTrue(s.hasCharacteristics(Spliterator.SORTED));1354} catch (IllegalStateException e) {1355assertFalse(s.hasCharacteristics(Spliterator.SORTED));1356}1357}13581359private static<T> void assertContents(Collection<T> actual, Collection<T> expected, boolean isOrdered) {1360if (isOrdered) {1361assertEquals(actual, expected);1362}1363else {1364assertContentsUnordered(actual, expected);1365}1366}13671368private static<T> void assertContentsUnordered(Iterable<T> actual, Iterable<T> expected) {1369assertEquals(toBoxedMultiset(actual), toBoxedMultiset(expected));1370}13711372private static <T> Map<T, Integer> toBoxedMultiset(Iterable<T> c) {1373Map<T, Integer> result = new HashMap<>();1374c.forEach(e -> {1375if (result.containsKey(e)) result.put(e, result.get(e) + 1);1376else result.put(e, 1);1377});1378return result;1379}13801381private void executeAndCatch(Class<? extends Exception> expected, Runnable r) {1382Exception caught = null;1383try {1384r.run();1385}1386catch (Exception e) {1387caught = e;1388}13891390assertNotNull(caught,1391String.format("No Exception was thrown, expected an Exception of %s to be thrown",1392expected.getName()));1393assertTrue(expected.isInstance(caught),1394String.format("Exception thrown %s not an instance of %s",1395caught.getClass().getName(), expected.getName()));1396}13971398}139914001401