Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
PojavLauncherTeam
GitHub Repository: PojavLauncherTeam/openjdk-multiarch-jdk8u
Path: blob/aarch64-shenandoah-jdk8u272-b10/jdk/test/java/util/Spliterator/SpliteratorTraversingAndSplittingTest.java
38811 views
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
24
/**
25
* @test
26
* @summary Spliterator traversing and splitting tests
27
* @run testng SpliteratorTraversingAndSplittingTest
28
* @bug 8020016
29
*/
30
31
import org.testng.annotations.DataProvider;
32
import org.testng.annotations.Test;
33
34
import java.util.AbstractCollection;
35
import java.util.AbstractList;
36
import java.util.AbstractSet;
37
import java.util.ArrayDeque;
38
import java.util.ArrayList;
39
import java.util.Arrays;
40
import java.util.Collection;
41
import java.util.Collections;
42
import java.util.Comparator;
43
import java.util.Deque;
44
import java.util.HashMap;
45
import java.util.HashSet;
46
import java.util.IdentityHashMap;
47
import java.util.Iterator;
48
import java.util.LinkedHashMap;
49
import java.util.LinkedHashSet;
50
import java.util.LinkedList;
51
import java.util.List;
52
import java.util.Map;
53
import java.util.PriorityQueue;
54
import java.util.Set;
55
import java.util.SortedSet;
56
import java.util.Spliterator;
57
import java.util.Spliterators;
58
import java.util.Stack;
59
import java.util.TreeMap;
60
import java.util.TreeSet;
61
import java.util.Vector;
62
import java.util.WeakHashMap;
63
import java.util.concurrent.ArrayBlockingQueue;
64
import java.util.concurrent.ConcurrentHashMap;
65
import java.util.concurrent.ConcurrentLinkedQueue;
66
import java.util.concurrent.ConcurrentSkipListMap;
67
import java.util.concurrent.ConcurrentSkipListSet;
68
import java.util.concurrent.CopyOnWriteArrayList;
69
import java.util.concurrent.CopyOnWriteArraySet;
70
import java.util.concurrent.LinkedBlockingDeque;
71
import java.util.concurrent.LinkedBlockingQueue;
72
import java.util.concurrent.LinkedTransferQueue;
73
import java.util.concurrent.PriorityBlockingQueue;
74
import java.util.function.Consumer;
75
import java.util.function.DoubleConsumer;
76
import java.util.function.Function;
77
import java.util.function.IntConsumer;
78
import java.util.function.LongConsumer;
79
import java.util.function.Supplier;
80
import java.util.function.UnaryOperator;
81
82
import static org.testng.Assert.*;
83
import static org.testng.Assert.assertEquals;
84
85
@Test
86
public class SpliteratorTraversingAndSplittingTest {
87
88
private static List<Integer> SIZES = Arrays.asList(0, 1, 10, 100, 1000);
89
90
private static class SpliteratorDataBuilder<T> {
91
List<Object[]> data;
92
93
List<T> exp;
94
95
Map<T, T> mExp;
96
97
SpliteratorDataBuilder(List<Object[]> data, List<T> exp) {
98
this.data = data;
99
this.exp = exp;
100
this.mExp = createMap(exp);
101
}
102
103
Map<T, T> createMap(List<T> l) {
104
Map<T, T> m = new LinkedHashMap<>();
105
for (T t : l) {
106
m.put(t, t);
107
}
108
return m;
109
}
110
111
void add(String description, Collection<?> expected, Supplier<Spliterator<?>> s) {
112
description = joiner(description).toString();
113
data.add(new Object[]{description, expected, s});
114
}
115
116
void add(String description, Supplier<Spliterator<?>> s) {
117
add(description, exp, s);
118
}
119
120
void addCollection(Function<Collection<T>, ? extends Collection<T>> c) {
121
add("new " + c.apply(Collections.<T>emptyList()).getClass().getName() + ".spliterator()",
122
() -> c.apply(exp).spliterator());
123
}
124
125
void addList(Function<Collection<T>, ? extends List<T>> l) {
126
// @@@ If collection is instance of List then add sub-list tests
127
addCollection(l);
128
}
129
130
void addMap(Function<Map<T, T>, ? extends Map<T, T>> m) {
131
String description = "new " + m.apply(Collections.<T, T>emptyMap()).getClass().getName();
132
addMap(m, description);
133
}
134
135
void addMap(Function<Map<T, T>, ? extends Map<T, T>> m, String description) {
136
add(description + ".keySet().spliterator()", () -> m.apply(mExp).keySet().spliterator());
137
add(description + ".values().spliterator()", () -> m.apply(mExp).values().spliterator());
138
add(description + ".entrySet().spliterator()", mExp.entrySet(), () -> m.apply(mExp).entrySet().spliterator());
139
}
140
141
StringBuilder joiner(String description) {
142
return new StringBuilder(description).
143
append(" {").
144
append("size=").append(exp.size()).
145
append("}");
146
}
147
}
148
149
static Object[][] spliteratorDataProvider;
150
151
@DataProvider(name = "Spliterator<Integer>")
152
public static Object[][] spliteratorDataProvider() {
153
if (spliteratorDataProvider != null) {
154
return spliteratorDataProvider;
155
}
156
157
List<Object[]> data = new ArrayList<>();
158
for (int size : SIZES) {
159
List<Integer> exp = listIntRange(size);
160
SpliteratorDataBuilder<Integer> db = new SpliteratorDataBuilder<>(data, exp);
161
162
// Direct spliterator methods
163
164
db.add("Spliterators.spliterator(Collection, ...)",
165
() -> Spliterators.spliterator(exp, 0));
166
167
db.add("Spliterators.spliterator(Iterator, ...)",
168
() -> Spliterators.spliterator(exp.iterator(), exp.size(), 0));
169
170
db.add("Spliterators.spliteratorUnknownSize(Iterator, ...)",
171
() -> Spliterators.spliteratorUnknownSize(exp.iterator(), 0));
172
173
db.add("Spliterators.spliterator(Spliterators.iteratorFromSpliterator(Spliterator ), ...)",
174
() -> Spliterators.spliterator(Spliterators.iterator(exp.spliterator()), exp.size(), 0));
175
176
db.add("Spliterators.spliterator(T[], ...)",
177
() -> Spliterators.spliterator(exp.toArray(new Integer[0]), 0));
178
179
db.add("Arrays.spliterator(T[], ...)",
180
() -> Arrays.spliterator(exp.toArray(new Integer[0])));
181
182
class SpliteratorFromIterator extends Spliterators.AbstractSpliterator<Integer> {
183
Iterator<Integer> it;
184
185
SpliteratorFromIterator(Iterator<Integer> it, long est) {
186
super(est, Spliterator.SIZED);
187
this.it = it;
188
}
189
190
@Override
191
public boolean tryAdvance(Consumer<? super Integer> action) {
192
if (action == null)
193
throw new NullPointerException();
194
if (it.hasNext()) {
195
action.accept(it.next());
196
return true;
197
}
198
else {
199
return false;
200
}
201
}
202
}
203
db.add("new Spliterators.AbstractSpliterator()",
204
() -> new SpliteratorFromIterator(exp.iterator(), exp.size()));
205
206
// Collections
207
208
// default method implementations
209
210
class AbstractCollectionImpl extends AbstractCollection<Integer> {
211
Collection<Integer> c;
212
213
AbstractCollectionImpl(Collection<Integer> c) {
214
this.c = c;
215
}
216
217
@Override
218
public Iterator<Integer> iterator() {
219
return c.iterator();
220
}
221
222
@Override
223
public int size() {
224
return c.size();
225
}
226
}
227
db.addCollection(
228
c -> new AbstractCollectionImpl(c));
229
230
class AbstractListImpl extends AbstractList<Integer> {
231
List<Integer> l;
232
233
AbstractListImpl(Collection<Integer> c) {
234
this.l = new ArrayList<>(c);
235
}
236
237
@Override
238
public Integer get(int index) {
239
return l.get(index);
240
}
241
242
@Override
243
public int size() {
244
return l.size();
245
}
246
}
247
db.addCollection(
248
c -> new AbstractListImpl(c));
249
250
class AbstractSetImpl extends AbstractSet<Integer> {
251
Set<Integer> s;
252
253
AbstractSetImpl(Collection<Integer> c) {
254
this.s = new HashSet<>(c);
255
}
256
257
@Override
258
public Iterator<Integer> iterator() {
259
return s.iterator();
260
}
261
262
@Override
263
public int size() {
264
return s.size();
265
}
266
}
267
db.addCollection(
268
c -> new AbstractSetImpl(c));
269
270
class AbstractSortedSetImpl extends AbstractSet<Integer> implements SortedSet<Integer> {
271
SortedSet<Integer> s;
272
273
AbstractSortedSetImpl(Collection<Integer> c) {
274
this.s = new TreeSet<>(c);
275
}
276
277
@Override
278
public Iterator<Integer> iterator() {
279
return s.iterator();
280
}
281
282
@Override
283
public int size() {
284
return s.size();
285
}
286
287
@Override
288
public Comparator<? super Integer> comparator() {
289
return s.comparator();
290
}
291
292
@Override
293
public SortedSet<Integer> subSet(Integer fromElement, Integer toElement) {
294
return s.subSet(fromElement, toElement);
295
}
296
297
@Override
298
public SortedSet<Integer> headSet(Integer toElement) {
299
return s.headSet(toElement);
300
}
301
302
@Override
303
public SortedSet<Integer> tailSet(Integer fromElement) {
304
return s.tailSet(fromElement);
305
}
306
307
@Override
308
public Integer first() {
309
return s.first();
310
}
311
312
@Override
313
public Integer last() {
314
return s.last();
315
}
316
317
@Override
318
public Spliterator<Integer> spliterator() {
319
return SortedSet.super.spliterator();
320
}
321
}
322
db.addCollection(
323
c -> new AbstractSortedSetImpl(c));
324
325
class IterableWrapper implements Iterable<Integer> {
326
final Iterable<Integer> it;
327
328
IterableWrapper(Iterable<Integer> it) {
329
this.it = it;
330
}
331
332
@Override
333
public Iterator<Integer> iterator() {
334
return it.iterator();
335
}
336
}
337
db.add("new Iterable.spliterator()",
338
() -> new IterableWrapper(exp).spliterator());
339
340
//
341
342
db.add("Arrays.asList().spliterator()",
343
() -> Spliterators.spliterator(Arrays.asList(exp.toArray(new Integer[0])), 0));
344
345
db.addList(ArrayList::new);
346
347
db.addList(LinkedList::new);
348
349
db.addList(Vector::new);
350
351
352
db.addCollection(HashSet::new);
353
354
db.addCollection(LinkedHashSet::new);
355
356
db.addCollection(TreeSet::new);
357
358
359
db.addCollection(c -> { Stack<Integer> s = new Stack<>(); s.addAll(c); return s;});
360
361
db.addCollection(PriorityQueue::new);
362
363
db.addCollection(ArrayDeque::new);
364
365
366
db.addCollection(ConcurrentSkipListSet::new);
367
368
if (size > 0) {
369
db.addCollection(c -> {
370
ArrayBlockingQueue<Integer> abq = new ArrayBlockingQueue<>(size);
371
abq.addAll(c);
372
return abq;
373
});
374
}
375
376
db.addCollection(PriorityBlockingQueue::new);
377
378
db.addCollection(LinkedBlockingQueue::new);
379
380
db.addCollection(LinkedTransferQueue::new);
381
382
db.addCollection(ConcurrentLinkedQueue::new);
383
384
db.addCollection(LinkedBlockingDeque::new);
385
386
db.addCollection(CopyOnWriteArrayList::new);
387
388
db.addCollection(CopyOnWriteArraySet::new);
389
390
if (size == 0) {
391
db.addCollection(c -> Collections.<Integer>emptySet());
392
db.addList(c -> Collections.<Integer>emptyList());
393
}
394
else if (size == 1) {
395
db.addCollection(c -> Collections.singleton(exp.get(0)));
396
db.addCollection(c -> Collections.singletonList(exp.get(0)));
397
}
398
399
{
400
Integer[] ai = new Integer[size];
401
Arrays.fill(ai, 1);
402
db.add(String.format("Collections.nCopies(%d, 1)", exp.size()),
403
Arrays.asList(ai),
404
() -> Collections.nCopies(exp.size(), 1).spliterator());
405
}
406
407
// Collections.synchronized/unmodifiable/checked wrappers
408
db.addCollection(Collections::unmodifiableCollection);
409
db.addCollection(c -> Collections.unmodifiableSet(new HashSet<>(c)));
410
db.addCollection(c -> Collections.unmodifiableSortedSet(new TreeSet<>(c)));
411
db.addList(c -> Collections.unmodifiableList(new ArrayList<>(c)));
412
db.addMap(Collections::unmodifiableMap);
413
db.addMap(m -> Collections.unmodifiableSortedMap(new TreeMap<>(m)));
414
415
db.addCollection(Collections::synchronizedCollection);
416
db.addCollection(c -> Collections.synchronizedSet(new HashSet<>(c)));
417
db.addCollection(c -> Collections.synchronizedSortedSet(new TreeSet<>(c)));
418
db.addList(c -> Collections.synchronizedList(new ArrayList<>(c)));
419
db.addMap(Collections::synchronizedMap);
420
db.addMap(m -> Collections.synchronizedSortedMap(new TreeMap<>(m)));
421
422
db.addCollection(c -> Collections.checkedCollection(c, Integer.class));
423
db.addCollection(c -> Collections.checkedQueue(new ArrayDeque<>(c), Integer.class));
424
db.addCollection(c -> Collections.checkedSet(new HashSet<>(c), Integer.class));
425
db.addCollection(c -> Collections.checkedSortedSet(new TreeSet<>(c), Integer.class));
426
db.addList(c -> Collections.checkedList(new ArrayList<>(c), Integer.class));
427
db.addMap(c -> Collections.checkedMap(c, Integer.class, Integer.class));
428
db.addMap(m -> Collections.checkedSortedMap(new TreeMap<>(m), Integer.class, Integer.class));
429
430
// Maps
431
432
db.addMap(HashMap::new);
433
434
db.addMap(m -> {
435
// Create a Map ensuring that for large sizes
436
// buckets will contain 2 or more entries
437
HashMap<Integer, Integer> cm = new HashMap<>(1, m.size() + 1);
438
// Don't use putAll which inflates the table by
439
// m.size() * loadFactor, thus creating a very sparse
440
// map for 1000 entries defeating the purpose of this test,
441
// in addition it will cause the split until null test to fail
442
// because the number of valid splits is larger than the
443
// threshold
444
for (Map.Entry<Integer, Integer> e : m.entrySet())
445
cm.put(e.getKey(), e.getValue());
446
return cm;
447
}, "new java.util.HashMap(1, size + 1)");
448
449
db.addMap(LinkedHashMap::new);
450
451
db.addMap(IdentityHashMap::new);
452
453
db.addMap(WeakHashMap::new);
454
455
db.addMap(m -> {
456
// Create a Map ensuring that for large sizes
457
// buckets will be consist of 2 or more entries
458
WeakHashMap<Integer, Integer> cm = new WeakHashMap<>(1, m.size() + 1);
459
for (Map.Entry<Integer, Integer> e : m.entrySet())
460
cm.put(e.getKey(), e.getValue());
461
return cm;
462
}, "new java.util.WeakHashMap(1, size + 1)");
463
464
// @@@ Descending maps etc
465
db.addMap(TreeMap::new);
466
467
db.addMap(ConcurrentHashMap::new);
468
469
db.addMap(ConcurrentSkipListMap::new);
470
471
if (size == 0) {
472
db.addMap(m -> Collections.<Integer, Integer>emptyMap());
473
}
474
else if (size == 1) {
475
db.addMap(m -> Collections.singletonMap(exp.get(0), exp.get(0)));
476
}
477
}
478
479
return spliteratorDataProvider = data.toArray(new Object[0][]);
480
}
481
482
private static List<Integer> listIntRange(int upTo) {
483
List<Integer> exp = new ArrayList<>();
484
for (int i = 0; i < upTo; i++)
485
exp.add(i);
486
return Collections.unmodifiableList(exp);
487
}
488
489
@Test(dataProvider = "Spliterator<Integer>")
490
@SuppressWarnings({"unchecked", "rawtypes"})
491
public void testNullPointerException(String description, Collection exp, Supplier<Spliterator> s) {
492
executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining(null));
493
executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance(null));
494
}
495
496
@Test(dataProvider = "Spliterator<Integer>")
497
@SuppressWarnings({"unchecked", "rawtypes"})
498
public void testForEach(String description, Collection exp, Supplier<Spliterator> s) {
499
testForEach(exp, s, (Consumer<Object> b) -> b);
500
}
501
502
@Test(dataProvider = "Spliterator<Integer>")
503
@SuppressWarnings({"unchecked", "rawtypes"})
504
public void testTryAdvance(String description, Collection exp, Supplier<Spliterator> s) {
505
testTryAdvance(exp, s, (Consumer<Object> b) -> b);
506
}
507
508
@Test(dataProvider = "Spliterator<Integer>")
509
@SuppressWarnings({"unchecked", "rawtypes"})
510
public void testMixedTryAdvanceForEach(String description, Collection exp, Supplier<Spliterator> s) {
511
testMixedTryAdvanceForEach(exp, s, (Consumer<Object> b) -> b);
512
}
513
514
@Test(dataProvider = "Spliterator<Integer>")
515
@SuppressWarnings({"unchecked", "rawtypes"})
516
public void testSplitAfterFullTraversal(String description, Collection exp, Supplier<Spliterator> s) {
517
testSplitAfterFullTraversal(s, (Consumer<Object> b) -> b);
518
}
519
520
@Test(dataProvider = "Spliterator<Integer>")
521
@SuppressWarnings({"unchecked", "rawtypes"})
522
public void testSplitOnce(String description, Collection exp, Supplier<Spliterator> s) {
523
testSplitOnce(exp, s, (Consumer<Object> b) -> b);
524
}
525
526
@Test(dataProvider = "Spliterator<Integer>")
527
@SuppressWarnings({"unchecked", "rawtypes"})
528
public void testSplitSixDeep(String description, Collection exp, Supplier<Spliterator> s) {
529
testSplitSixDeep(exp, s, (Consumer<Object> b) -> b);
530
}
531
532
@Test(dataProvider = "Spliterator<Integer>")
533
@SuppressWarnings({"unchecked", "rawtypes"})
534
public void testSplitUntilNull(String description, Collection exp, Supplier<Spliterator> s) {
535
testSplitUntilNull(exp, s, (Consumer<Object> b) -> b);
536
}
537
538
//
539
540
private static class SpliteratorOfIntDataBuilder {
541
List<Object[]> data;
542
543
List<Integer> exp;
544
545
SpliteratorOfIntDataBuilder(List<Object[]> data, List<Integer> exp) {
546
this.data = data;
547
this.exp = exp;
548
}
549
550
void add(String description, List<Integer> expected, Supplier<Spliterator.OfInt> s) {
551
description = joiner(description).toString();
552
data.add(new Object[]{description, expected, s});
553
}
554
555
void add(String description, Supplier<Spliterator.OfInt> s) {
556
add(description, exp, s);
557
}
558
559
StringBuilder joiner(String description) {
560
return new StringBuilder(description).
561
append(" {").
562
append("size=").append(exp.size()).
563
append("}");
564
}
565
}
566
567
static Object[][] spliteratorOfIntDataProvider;
568
569
@DataProvider(name = "Spliterator.OfInt")
570
public static Object[][] spliteratorOfIntDataProvider() {
571
if (spliteratorOfIntDataProvider != null) {
572
return spliteratorOfIntDataProvider;
573
}
574
575
List<Object[]> data = new ArrayList<>();
576
for (int size : SIZES) {
577
int exp[] = arrayIntRange(size);
578
SpliteratorOfIntDataBuilder db = new SpliteratorOfIntDataBuilder(data, listIntRange(size));
579
580
db.add("Spliterators.spliterator(int[], ...)",
581
() -> Spliterators.spliterator(exp, 0));
582
583
db.add("Arrays.spliterator(int[], ...)",
584
() -> Arrays.spliterator(exp));
585
586
db.add("Spliterators.spliterator(PrimitiveIterator.OfInt, ...)",
587
() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));
588
589
db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfInt, ...)",
590
() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));
591
592
class IntSpliteratorFromArray extends Spliterators.AbstractIntSpliterator {
593
int[] a;
594
int index = 0;
595
596
IntSpliteratorFromArray(int[] a) {
597
super(a.length, Spliterator.SIZED);
598
this.a = a;
599
}
600
601
@Override
602
public boolean tryAdvance(IntConsumer action) {
603
if (action == null)
604
throw new NullPointerException();
605
if (index < a.length) {
606
action.accept(a[index++]);
607
return true;
608
}
609
else {
610
return false;
611
}
612
}
613
}
614
db.add("new Spliterators.AbstractIntAdvancingSpliterator()",
615
() -> new IntSpliteratorFromArray(exp));
616
}
617
618
return spliteratorOfIntDataProvider = data.toArray(new Object[0][]);
619
}
620
621
private static int[] arrayIntRange(int upTo) {
622
int[] exp = new int[upTo];
623
for (int i = 0; i < upTo; i++)
624
exp[i] = i;
625
return exp;
626
}
627
628
private static UnaryOperator<Consumer<Integer>> intBoxingConsumer() {
629
class BoxingAdapter implements Consumer<Integer>, IntConsumer {
630
private final Consumer<Integer> b;
631
632
BoxingAdapter(Consumer<Integer> b) {
633
this.b = b;
634
}
635
636
@Override
637
public void accept(Integer value) {
638
throw new IllegalStateException();
639
}
640
641
@Override
642
public void accept(int value) {
643
b.accept(value);
644
}
645
}
646
647
return b -> new BoxingAdapter(b);
648
}
649
650
@Test(dataProvider = "Spliterator.OfInt")
651
public void testIntNullPointerException(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
652
executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((IntConsumer) null));
653
executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((IntConsumer) null));
654
}
655
656
@Test(dataProvider = "Spliterator.OfInt")
657
public void testIntForEach(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
658
testForEach(exp, s, intBoxingConsumer());
659
}
660
661
@Test(dataProvider = "Spliterator.OfInt")
662
public void testIntTryAdvance(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
663
testTryAdvance(exp, s, intBoxingConsumer());
664
}
665
666
@Test(dataProvider = "Spliterator.OfInt")
667
public void testIntMixedTryAdvanceForEach(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
668
testMixedTryAdvanceForEach(exp, s, intBoxingConsumer());
669
}
670
671
@Test(dataProvider = "Spliterator.OfInt")
672
public void testIntSplitAfterFullTraversal(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
673
testSplitAfterFullTraversal(s, intBoxingConsumer());
674
}
675
676
@Test(dataProvider = "Spliterator.OfInt")
677
public void testIntSplitOnce(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
678
testSplitOnce(exp, s, intBoxingConsumer());
679
}
680
681
@Test(dataProvider = "Spliterator.OfInt")
682
public void testIntSplitSixDeep(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
683
testSplitSixDeep(exp, s, intBoxingConsumer());
684
}
685
686
@Test(dataProvider = "Spliterator.OfInt")
687
public void testIntSplitUntilNull(String description, Collection<Integer> exp, Supplier<Spliterator.OfInt> s) {
688
testSplitUntilNull(exp, s, intBoxingConsumer());
689
}
690
691
//
692
693
private static class SpliteratorOfLongDataBuilder {
694
List<Object[]> data;
695
696
List<Long> exp;
697
698
SpliteratorOfLongDataBuilder(List<Object[]> data, List<Long> exp) {
699
this.data = data;
700
this.exp = exp;
701
}
702
703
void add(String description, List<Long> expected, Supplier<Spliterator.OfLong> s) {
704
description = joiner(description).toString();
705
data.add(new Object[]{description, expected, s});
706
}
707
708
void add(String description, Supplier<Spliterator.OfLong> s) {
709
add(description, exp, s);
710
}
711
712
StringBuilder joiner(String description) {
713
return new StringBuilder(description).
714
append(" {").
715
append("size=").append(exp.size()).
716
append("}");
717
}
718
}
719
720
static Object[][] spliteratorOfLongDataProvider;
721
722
@DataProvider(name = "Spliterator.OfLong")
723
public static Object[][] spliteratorOfLongDataProvider() {
724
if (spliteratorOfLongDataProvider != null) {
725
return spliteratorOfLongDataProvider;
726
}
727
728
List<Object[]> data = new ArrayList<>();
729
for (int size : SIZES) {
730
long exp[] = arrayLongRange(size);
731
SpliteratorOfLongDataBuilder db = new SpliteratorOfLongDataBuilder(data, listLongRange(size));
732
733
db.add("Spliterators.spliterator(long[], ...)",
734
() -> Spliterators.spliterator(exp, 0));
735
736
db.add("Arrays.spliterator(long[], ...)",
737
() -> Arrays.spliterator(exp));
738
739
db.add("Spliterators.spliterator(PrimitiveIterator.OfLong, ...)",
740
() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));
741
742
db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfLong, ...)",
743
() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));
744
745
class LongSpliteratorFromArray extends Spliterators.AbstractLongSpliterator {
746
long[] a;
747
int index = 0;
748
749
LongSpliteratorFromArray(long[] a) {
750
super(a.length, Spliterator.SIZED);
751
this.a = a;
752
}
753
754
@Override
755
public boolean tryAdvance(LongConsumer action) {
756
if (action == null)
757
throw new NullPointerException();
758
if (index < a.length) {
759
action.accept(a[index++]);
760
return true;
761
}
762
else {
763
return false;
764
}
765
}
766
}
767
db.add("new Spliterators.AbstractLongAdvancingSpliterator()",
768
() -> new LongSpliteratorFromArray(exp));
769
}
770
771
return spliteratorOfLongDataProvider = data.toArray(new Object[0][]);
772
}
773
774
private static List<Long> listLongRange(int upTo) {
775
List<Long> exp = new ArrayList<>();
776
for (long i = 0; i < upTo; i++)
777
exp.add(i);
778
return Collections.unmodifiableList(exp);
779
}
780
781
private static long[] arrayLongRange(int upTo) {
782
long[] exp = new long[upTo];
783
for (int i = 0; i < upTo; i++)
784
exp[i] = i;
785
return exp;
786
}
787
788
private static UnaryOperator<Consumer<Long>> longBoxingConsumer() {
789
class BoxingAdapter implements Consumer<Long>, LongConsumer {
790
private final Consumer<Long> b;
791
792
BoxingAdapter(Consumer<Long> b) {
793
this.b = b;
794
}
795
796
@Override
797
public void accept(Long value) {
798
throw new IllegalStateException();
799
}
800
801
@Override
802
public void accept(long value) {
803
b.accept(value);
804
}
805
}
806
807
return b -> new BoxingAdapter(b);
808
}
809
810
@Test(dataProvider = "Spliterator.OfLong")
811
public void testLongNullPointerException(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
812
executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((LongConsumer) null));
813
executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((LongConsumer) null));
814
}
815
816
@Test(dataProvider = "Spliterator.OfLong")
817
public void testLongForEach(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
818
testForEach(exp, s, longBoxingConsumer());
819
}
820
821
@Test(dataProvider = "Spliterator.OfLong")
822
public void testLongTryAdvance(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
823
testTryAdvance(exp, s, longBoxingConsumer());
824
}
825
826
@Test(dataProvider = "Spliterator.OfLong")
827
public void testLongMixedTryAdvanceForEach(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
828
testMixedTryAdvanceForEach(exp, s, longBoxingConsumer());
829
}
830
831
@Test(dataProvider = "Spliterator.OfLong")
832
public void testLongSplitAfterFullTraversal(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
833
testSplitAfterFullTraversal(s, longBoxingConsumer());
834
}
835
836
@Test(dataProvider = "Spliterator.OfLong")
837
public void testLongSplitOnce(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
838
testSplitOnce(exp, s, longBoxingConsumer());
839
}
840
841
@Test(dataProvider = "Spliterator.OfLong")
842
public void testLongSplitSixDeep(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
843
testSplitSixDeep(exp, s, longBoxingConsumer());
844
}
845
846
@Test(dataProvider = "Spliterator.OfLong")
847
public void testLongSplitUntilNull(String description, Collection<Long> exp, Supplier<Spliterator.OfLong> s) {
848
testSplitUntilNull(exp, s, longBoxingConsumer());
849
}
850
851
//
852
853
private static class SpliteratorOfDoubleDataBuilder {
854
List<Object[]> data;
855
856
List<Double> exp;
857
858
SpliteratorOfDoubleDataBuilder(List<Object[]> data, List<Double> exp) {
859
this.data = data;
860
this.exp = exp;
861
}
862
863
void add(String description, List<Double> expected, Supplier<Spliterator.OfDouble> s) {
864
description = joiner(description).toString();
865
data.add(new Object[]{description, expected, s});
866
}
867
868
void add(String description, Supplier<Spliterator.OfDouble> s) {
869
add(description, exp, s);
870
}
871
872
StringBuilder joiner(String description) {
873
return new StringBuilder(description).
874
append(" {").
875
append("size=").append(exp.size()).
876
append("}");
877
}
878
}
879
880
static Object[][] spliteratorOfDoubleDataProvider;
881
882
@DataProvider(name = "Spliterator.OfDouble")
883
public static Object[][] spliteratorOfDoubleDataProvider() {
884
if (spliteratorOfDoubleDataProvider != null) {
885
return spliteratorOfDoubleDataProvider;
886
}
887
888
List<Object[]> data = new ArrayList<>();
889
for (int size : SIZES) {
890
double exp[] = arrayDoubleRange(size);
891
SpliteratorOfDoubleDataBuilder db = new SpliteratorOfDoubleDataBuilder(data, listDoubleRange(size));
892
893
db.add("Spliterators.spliterator(double[], ...)",
894
() -> Spliterators.spliterator(exp, 0));
895
896
db.add("Arrays.spliterator(double[], ...)",
897
() -> Arrays.spliterator(exp));
898
899
db.add("Spliterators.spliterator(PrimitiveIterator.OfDouble, ...)",
900
() -> Spliterators.spliterator(Spliterators.iterator(Arrays.spliterator(exp)), exp.length, 0));
901
902
db.add("Spliterators.spliteratorUnknownSize(PrimitiveIterator.OfDouble, ...)",
903
() -> Spliterators.spliteratorUnknownSize(Spliterators.iterator(Arrays.spliterator(exp)), 0));
904
905
class DoubleSpliteratorFromArray extends Spliterators.AbstractDoubleSpliterator {
906
double[] a;
907
int index = 0;
908
909
DoubleSpliteratorFromArray(double[] a) {
910
super(a.length, Spliterator.SIZED);
911
this.a = a;
912
}
913
914
@Override
915
public boolean tryAdvance(DoubleConsumer action) {
916
if (action == null)
917
throw new NullPointerException();
918
if (index < a.length) {
919
action.accept(a[index++]);
920
return true;
921
}
922
else {
923
return false;
924
}
925
}
926
}
927
db.add("new Spliterators.AbstractDoubleAdvancingSpliterator()",
928
() -> new DoubleSpliteratorFromArray(exp));
929
}
930
931
return spliteratorOfDoubleDataProvider = data.toArray(new Object[0][]);
932
}
933
934
private static List<Double> listDoubleRange(int upTo) {
935
List<Double> exp = new ArrayList<>();
936
for (double i = 0; i < upTo; i++)
937
exp.add(i);
938
return Collections.unmodifiableList(exp);
939
}
940
941
private static double[] arrayDoubleRange(int upTo) {
942
double[] exp = new double[upTo];
943
for (int i = 0; i < upTo; i++)
944
exp[i] = i;
945
return exp;
946
}
947
948
private static UnaryOperator<Consumer<Double>> doubleBoxingConsumer() {
949
class BoxingAdapter implements Consumer<Double>, DoubleConsumer {
950
private final Consumer<Double> b;
951
952
BoxingAdapter(Consumer<Double> b) {
953
this.b = b;
954
}
955
956
@Override
957
public void accept(Double value) {
958
throw new IllegalStateException();
959
}
960
961
@Override
962
public void accept(double value) {
963
b.accept(value);
964
}
965
}
966
967
return b -> new BoxingAdapter(b);
968
}
969
970
@Test(dataProvider = "Spliterator.OfDouble")
971
public void testDoubleNullPointerException(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
972
executeAndCatch(NullPointerException.class, () -> s.get().forEachRemaining((DoubleConsumer) null));
973
executeAndCatch(NullPointerException.class, () -> s.get().tryAdvance((DoubleConsumer) null));
974
}
975
976
@Test(dataProvider = "Spliterator.OfDouble")
977
public void testDoubleForEach(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
978
testForEach(exp, s, doubleBoxingConsumer());
979
}
980
981
@Test(dataProvider = "Spliterator.OfDouble")
982
public void testDoubleTryAdvance(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
983
testTryAdvance(exp, s, doubleBoxingConsumer());
984
}
985
986
@Test(dataProvider = "Spliterator.OfDouble")
987
public void testDoubleMixedTryAdvanceForEach(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
988
testMixedTryAdvanceForEach(exp, s, doubleBoxingConsumer());
989
}
990
991
@Test(dataProvider = "Spliterator.OfDouble")
992
public void testDoubleSplitAfterFullTraversal(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
993
testSplitAfterFullTraversal(s, doubleBoxingConsumer());
994
}
995
996
@Test(dataProvider = "Spliterator.OfDouble")
997
public void testDoubleSplitOnce(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
998
testSplitOnce(exp, s, doubleBoxingConsumer());
999
}
1000
1001
@Test(dataProvider = "Spliterator.OfDouble")
1002
public void testDoubleSplitSixDeep(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
1003
testSplitSixDeep(exp, s, doubleBoxingConsumer());
1004
}
1005
1006
@Test(dataProvider = "Spliterator.OfDouble")
1007
public void testDoubleSplitUntilNull(String description, Collection<Double> exp, Supplier<Spliterator.OfDouble> s) {
1008
testSplitUntilNull(exp, s, doubleBoxingConsumer());
1009
}
1010
1011
//
1012
1013
private static <T, S extends Spliterator<T>> void testForEach(
1014
Collection<T> exp,
1015
Supplier<S> supplier,
1016
UnaryOperator<Consumer<T>> boxingAdapter) {
1017
S spliterator = supplier.get();
1018
long sizeIfKnown = spliterator.getExactSizeIfKnown();
1019
boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);
1020
1021
ArrayList<T> fromForEach = new ArrayList<>();
1022
spliterator = supplier.get();
1023
Consumer<T> addToFromForEach = boxingAdapter.apply(fromForEach::add);
1024
spliterator.forEachRemaining(addToFromForEach);
1025
1026
// Assert that forEach now produces no elements
1027
spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));
1028
// Assert that tryAdvance now produce no elements
1029
spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));
1030
1031
// assert that size, tryAdvance, and forEach are consistent
1032
if (sizeIfKnown >= 0) {
1033
assertEquals(sizeIfKnown, exp.size());
1034
}
1035
assertEquals(fromForEach.size(), exp.size());
1036
1037
assertContents(fromForEach, exp, isOrdered);
1038
}
1039
1040
private static <T, S extends Spliterator<T>> void testTryAdvance(
1041
Collection<T> exp,
1042
Supplier<S> supplier,
1043
UnaryOperator<Consumer<T>> boxingAdapter) {
1044
S spliterator = supplier.get();
1045
long sizeIfKnown = spliterator.getExactSizeIfKnown();
1046
boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);
1047
1048
spliterator = supplier.get();
1049
ArrayList<T> fromTryAdvance = new ArrayList<>();
1050
Consumer<T> addToFromTryAdvance = boxingAdapter.apply(fromTryAdvance::add);
1051
while (spliterator.tryAdvance(addToFromTryAdvance)) { }
1052
1053
// Assert that forEach now produces no elements
1054
spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));
1055
// Assert that tryAdvance now produce no elements
1056
spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));
1057
1058
// assert that size, tryAdvance, and forEach are consistent
1059
if (sizeIfKnown >= 0) {
1060
assertEquals(sizeIfKnown, exp.size());
1061
}
1062
assertEquals(fromTryAdvance.size(), exp.size());
1063
1064
assertContents(fromTryAdvance, exp, isOrdered);
1065
}
1066
1067
private static <T, S extends Spliterator<T>> void testMixedTryAdvanceForEach(
1068
Collection<T> exp,
1069
Supplier<S> supplier,
1070
UnaryOperator<Consumer<T>> boxingAdapter) {
1071
S spliterator = supplier.get();
1072
long sizeIfKnown = spliterator.getExactSizeIfKnown();
1073
boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);
1074
1075
// tryAdvance first few elements, then forEach rest
1076
ArrayList<T> dest = new ArrayList<>();
1077
spliterator = supplier.get();
1078
Consumer<T> addToDest = boxingAdapter.apply(dest::add);
1079
for (int i = 0; i < 10 && spliterator.tryAdvance(addToDest); i++) { }
1080
spliterator.forEachRemaining(addToDest);
1081
1082
// Assert that forEach now produces no elements
1083
spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));
1084
// Assert that tryAdvance now produce no elements
1085
spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));
1086
1087
if (sizeIfKnown >= 0) {
1088
assertEquals(sizeIfKnown, dest.size());
1089
}
1090
assertEquals(dest.size(), exp.size());
1091
1092
if (isOrdered) {
1093
assertEquals(dest, exp);
1094
}
1095
else {
1096
assertContentsUnordered(dest, exp);
1097
}
1098
}
1099
1100
private static <T, S extends Spliterator<T>> void testSplitAfterFullTraversal(
1101
Supplier<S> supplier,
1102
UnaryOperator<Consumer<T>> boxingAdapter) {
1103
// Full traversal using tryAdvance
1104
Spliterator<T> spliterator = supplier.get();
1105
while (spliterator.tryAdvance(boxingAdapter.apply(e -> { }))) { }
1106
Spliterator<T> split = spliterator.trySplit();
1107
assertNull(split);
1108
1109
// Full traversal using forEach
1110
spliterator = supplier.get();
1111
spliterator.forEachRemaining(boxingAdapter.apply(e -> {
1112
}));
1113
split = spliterator.trySplit();
1114
assertNull(split);
1115
1116
// Full traversal using tryAdvance then forEach
1117
spliterator = supplier.get();
1118
spliterator.tryAdvance(boxingAdapter.apply(e -> { }));
1119
spliterator.forEachRemaining(boxingAdapter.apply(e -> {
1120
}));
1121
split = spliterator.trySplit();
1122
assertNull(split);
1123
}
1124
1125
private static <T, S extends Spliterator<T>> void testSplitOnce(
1126
Collection<T> exp,
1127
Supplier<S> supplier,
1128
UnaryOperator<Consumer<T>> boxingAdapter) {
1129
S spliterator = supplier.get();
1130
long sizeIfKnown = spliterator.getExactSizeIfKnown();
1131
boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);
1132
1133
ArrayList<T> fromSplit = new ArrayList<>();
1134
Spliterator<T> s1 = supplier.get();
1135
Spliterator<T> s2 = s1.trySplit();
1136
long s1Size = s1.getExactSizeIfKnown();
1137
long s2Size = (s2 != null) ? s2.getExactSizeIfKnown() : 0;
1138
Consumer<T> addToFromSplit = boxingAdapter.apply(fromSplit::add);
1139
if (s2 != null)
1140
s2.forEachRemaining(addToFromSplit);
1141
s1.forEachRemaining(addToFromSplit);
1142
1143
if (sizeIfKnown >= 0) {
1144
assertEquals(sizeIfKnown, fromSplit.size());
1145
if (s1Size >= 0 && s2Size >= 0)
1146
assertEquals(sizeIfKnown, s1Size + s2Size);
1147
}
1148
assertContents(fromSplit, exp, isOrdered);
1149
}
1150
1151
private static <T, S extends Spliterator<T>> void testSplitSixDeep(
1152
Collection<T> exp,
1153
Supplier<S> supplier,
1154
UnaryOperator<Consumer<T>> boxingAdapter) {
1155
S spliterator = supplier.get();
1156
boolean isOrdered = spliterator.hasCharacteristics(Spliterator.ORDERED);
1157
1158
for (int depth=0; depth < 6; depth++) {
1159
List<T> dest = new ArrayList<>();
1160
spliterator = supplier.get();
1161
1162
assertRootSpliterator(spliterator);
1163
1164
// verify splitting with forEach
1165
visit(depth, 0, dest, spliterator, boxingAdapter, spliterator.characteristics(), false);
1166
assertContents(dest, exp, isOrdered);
1167
1168
// verify splitting with tryAdvance
1169
dest.clear();
1170
spliterator = supplier.get();
1171
visit(depth, 0, dest, spliterator, boxingAdapter, spliterator.characteristics(), true);
1172
assertContents(dest, exp, isOrdered);
1173
}
1174
}
1175
1176
private static <T, S extends Spliterator<T>> void visit(int depth, int curLevel,
1177
List<T> dest, S spliterator, UnaryOperator<Consumer<T>> boxingAdapter,
1178
int rootCharacteristics, boolean useTryAdvance) {
1179
if (curLevel < depth) {
1180
long beforeSize = spliterator.getExactSizeIfKnown();
1181
Spliterator<T> split = spliterator.trySplit();
1182
if (split != null) {
1183
assertSpliterator(split, rootCharacteristics);
1184
assertSpliterator(spliterator, rootCharacteristics);
1185
1186
if ((rootCharacteristics & Spliterator.SUBSIZED) != 0 &&
1187
(rootCharacteristics & Spliterator.SIZED) != 0) {
1188
assertEquals(beforeSize, split.estimateSize() + spliterator.estimateSize());
1189
}
1190
visit(depth, curLevel + 1, dest, split, boxingAdapter, rootCharacteristics, useTryAdvance);
1191
}
1192
visit(depth, curLevel + 1, dest, spliterator, boxingAdapter, rootCharacteristics, useTryAdvance);
1193
}
1194
else {
1195
long sizeIfKnown = spliterator.getExactSizeIfKnown();
1196
if (useTryAdvance) {
1197
Consumer<T> addToDest = boxingAdapter.apply(dest::add);
1198
int count = 0;
1199
while (spliterator.tryAdvance(addToDest)) {
1200
++count;
1201
}
1202
1203
if (sizeIfKnown >= 0)
1204
assertEquals(sizeIfKnown, count);
1205
1206
// Assert that forEach now produces no elements
1207
spliterator.forEachRemaining(boxingAdapter.apply(e -> fail("Spliterator.forEach produced an element after spliterator exhausted: " + e)));
1208
1209
Spliterator<T> split = spliterator.trySplit();
1210
assertNull(split);
1211
}
1212
else {
1213
List<T> leafDest = new ArrayList<>();
1214
Consumer<T> addToLeafDest = boxingAdapter.apply(leafDest::add);
1215
spliterator.forEachRemaining(addToLeafDest);
1216
1217
if (sizeIfKnown >= 0)
1218
assertEquals(sizeIfKnown, leafDest.size());
1219
1220
// Assert that forEach now produces no elements
1221
spliterator.tryAdvance(boxingAdapter.apply(e -> fail("Spliterator.tryAdvance produced an element after spliterator exhausted: " + e)));
1222
1223
Spliterator<T> split = spliterator.trySplit();
1224
assertNull(split);
1225
1226
dest.addAll(leafDest);
1227
}
1228
}
1229
}
1230
1231
private static <T, S extends Spliterator<T>> void testSplitUntilNull(
1232
Collection<T> exp,
1233
Supplier<S> supplier,
1234
UnaryOperator<Consumer<T>> boxingAdapter) {
1235
Spliterator<T> s = supplier.get();
1236
boolean isOrdered = s.hasCharacteristics(Spliterator.ORDERED);
1237
assertRootSpliterator(s);
1238
1239
List<T> splits = new ArrayList<>();
1240
Consumer<T> c = boxingAdapter.apply(splits::add);
1241
1242
testSplitUntilNull(new SplitNode<T>(c, s));
1243
assertContents(splits, exp, isOrdered);
1244
}
1245
1246
private static class SplitNode<T> {
1247
// Constant for every node
1248
final Consumer<T> c;
1249
final int rootCharacteristics;
1250
1251
final Spliterator<T> s;
1252
1253
SplitNode(Consumer<T> c, Spliterator<T> s) {
1254
this(c, s.characteristics(), s);
1255
}
1256
1257
private SplitNode(Consumer<T> c, int rootCharacteristics, Spliterator<T> s) {
1258
this.c = c;
1259
this.rootCharacteristics = rootCharacteristics;
1260
this.s = s;
1261
}
1262
1263
SplitNode<T> fromSplit(Spliterator<T> split) {
1264
return new SplitNode<>(c, rootCharacteristics, split);
1265
}
1266
}
1267
1268
/**
1269
* Set the maximum stack capacity to 0.25MB. This should be more than enough to detect a bad spliterator
1270
* while not unduly disrupting test infrastructure given the test data sizes that are used are small.
1271
* Note that j.u.c.ForkJoinPool sets the max queue size to 64M (1 << 26).
1272
*/
1273
private static final int MAXIMUM_STACK_CAPACITY = 1 << 18; // 0.25MB
1274
1275
private static <T> void testSplitUntilNull(SplitNode<T> e) {
1276
// Use an explicit stack to avoid a StackOverflowException when testing a Spliterator
1277
// that when repeatedly split produces a right-balanced (and maybe degenerate) tree, or
1278
// for a spliterator that is badly behaved.
1279
Deque<SplitNode<T>> stack = new ArrayDeque<>();
1280
stack.push(e);
1281
1282
int iteration = 0;
1283
while (!stack.isEmpty()) {
1284
assertTrue(iteration++ < MAXIMUM_STACK_CAPACITY, "Exceeded maximum stack modification count of 1 << 18");
1285
1286
e = stack.pop();
1287
Spliterator<T> parentAndRightSplit = e.s;
1288
1289
long parentEstimateSize = parentAndRightSplit.estimateSize();
1290
assertTrue(parentEstimateSize >= 0,
1291
String.format("Split size estimate %d < 0", parentEstimateSize));
1292
1293
long parentSize = parentAndRightSplit.getExactSizeIfKnown();
1294
Spliterator<T> leftSplit = parentAndRightSplit.trySplit();
1295
if (leftSplit == null) {
1296
parentAndRightSplit.forEachRemaining(e.c);
1297
continue;
1298
}
1299
1300
assertSpliterator(leftSplit, e.rootCharacteristics);
1301
assertSpliterator(parentAndRightSplit, e.rootCharacteristics);
1302
1303
if (parentEstimateSize != Long.MAX_VALUE && leftSplit.estimateSize() > 0 && parentAndRightSplit.estimateSize() > 0) {
1304
assertTrue(leftSplit.estimateSize() < parentEstimateSize,
1305
String.format("Left split size estimate %d >= parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));
1306
assertTrue(parentAndRightSplit.estimateSize() < parentEstimateSize,
1307
String.format("Right split size estimate %d >= parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));
1308
}
1309
else {
1310
assertTrue(leftSplit.estimateSize() <= parentEstimateSize,
1311
String.format("Left split size estimate %d > parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));
1312
assertTrue(parentAndRightSplit.estimateSize() <= parentEstimateSize,
1313
String.format("Right split size estimate %d > parent split size estimate %d", leftSplit.estimateSize(), parentEstimateSize));
1314
}
1315
1316
long leftSize = leftSplit.getExactSizeIfKnown();
1317
long rightSize = parentAndRightSplit.getExactSizeIfKnown();
1318
if (parentSize >= 0 && leftSize >= 0 && rightSize >= 0)
1319
assertEquals(parentSize, leftSize + rightSize,
1320
String.format("exact left split size %d + exact right split size %d != parent exact split size %d",
1321
leftSize, rightSize, parentSize));
1322
1323
// Add right side to stack first so left side is popped off first
1324
stack.push(e.fromSplit(parentAndRightSplit));
1325
stack.push(e.fromSplit(leftSplit));
1326
}
1327
}
1328
1329
private static void assertRootSpliterator(Spliterator<?> s) {
1330
assertFalse(s.hasCharacteristics(Spliterator.SIZED | Spliterator.CONCURRENT),
1331
"Root spliterator should not be SIZED and CONCURRENT");
1332
1333
assertSpliterator(s);
1334
}
1335
1336
private static void assertSpliterator(Spliterator<?> s, int rootCharacteristics) {
1337
if ((rootCharacteristics & Spliterator.SUBSIZED) != 0) {
1338
assertTrue(s.hasCharacteristics(Spliterator.SUBSIZED),
1339
"Child split is not SUBSIZED when root split is SUBSIZED");
1340
}
1341
assertSpliterator(s);
1342
}
1343
1344
private static void assertSpliterator(Spliterator<?> s) {
1345
if (s.hasCharacteristics(Spliterator.SUBSIZED)) {
1346
assertTrue(s.hasCharacteristics(Spliterator.SIZED));
1347
}
1348
if (s.hasCharacteristics(Spliterator.SIZED)) {
1349
assertTrue(s.estimateSize() != Long.MAX_VALUE);
1350
assertTrue(s.getExactSizeIfKnown() >= 0);
1351
}
1352
try {
1353
s.getComparator();
1354
assertTrue(s.hasCharacteristics(Spliterator.SORTED));
1355
} catch (IllegalStateException e) {
1356
assertFalse(s.hasCharacteristics(Spliterator.SORTED));
1357
}
1358
}
1359
1360
private static<T> void assertContents(Collection<T> actual, Collection<T> expected, boolean isOrdered) {
1361
if (isOrdered) {
1362
assertEquals(actual, expected);
1363
}
1364
else {
1365
assertContentsUnordered(actual, expected);
1366
}
1367
}
1368
1369
private static<T> void assertContentsUnordered(Iterable<T> actual, Iterable<T> expected) {
1370
assertEquals(toBoxedMultiset(actual), toBoxedMultiset(expected));
1371
}
1372
1373
private static <T> Map<T, Integer> toBoxedMultiset(Iterable<T> c) {
1374
Map<T, Integer> result = new HashMap<>();
1375
c.forEach(e -> {
1376
if (result.containsKey(e)) result.put(e, result.get(e) + 1);
1377
else result.put(e, 1);
1378
});
1379
return result;
1380
}
1381
1382
private void executeAndCatch(Class<? extends Exception> expected, Runnable r) {
1383
Exception caught = null;
1384
try {
1385
r.run();
1386
}
1387
catch (Exception e) {
1388
caught = e;
1389
}
1390
1391
assertNotNull(caught,
1392
String.format("No Exception was thrown, expected an Exception of %s to be thrown",
1393
expected.getName()));
1394
assertTrue(expected.isInstance(caught),
1395
String.format("Exception thrown %s not an instance of %s",
1396
caught.getClass().getName(), expected.getName()));
1397
}
1398
1399
}
1400
1401