util/Spliterator/SpliteratorLateBindingTest.java

/*
 * Copyright (c) 2013, 2016, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import java.nio.CharBuffer;
import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.BitSet;
import java.util.HashMap;
import java.util.HashSet;
import java.util.IdentityHashMap;
import java.util.LinkedHashMap;
import java.util.LinkedHashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.PriorityQueue;
import java.util.Set;
import java.util.Spliterator;
import java.util.Stack;
import java.util.TreeMap;
import java.util.TreeSet;
import java.util.Vector;
import java.util.WeakHashMap;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Stream;

import static org.testng.Assert.assertEquals;

/**
 * @test
 * @bug 8148748 8170155
 * @summary Spliterator last-binding tests
 * @run testng SpliteratorLateBindingTest
 */

@Test
public class SpliteratorLateBindingTest extends SpliteratorLateBindingFailFastHelper {

    static Object[][] spliteratorDataProvider;

    @DataProvider(name = "Source")
    public static Object[][] sourceDataProvider() {
        if (spliteratorDataProvider != null) {
            return spliteratorDataProvider;
        }

        List<Object[]> data = new ArrayList<>();
        SpliteratorDataBuilder<Integer> db =
                new SpliteratorDataBuilder<>(data, 5, Arrays.asList(1, 2, 3, 4));

        // Collections

        db.addList(ArrayList::new);

        db.addList(LinkedList::new);

        db.addList(Vector::new);

        db.addList(AbstractRandomAccessListImpl::new);

        db.addCollection(HashSet::new);

        db.addCollection(LinkedHashSet::new);

        db.addCollection(TreeSet::new);

        db.addCollection(c -> {
            Stack<Integer> s = new Stack<>();
            s.addAll(c);
            return s;
        });

        db.addCollection(PriorityQueue::new);

        db.addCollection(ArrayDeque::new);

        // Maps

        db.addMap(HashMap::new);

        db.addMap(LinkedHashMap::new);

        db.addMap(IdentityHashMap::new);

        db.addMap(WeakHashMap::new);

        // @@@  Descending maps etc
        db.addMap(TreeMap::new);

        // BitSet

        List<Integer> bits = List.of(0, 1, 2);
        Function<BitSet, Spliterator.OfInt> bitsSource = bs -> bs.stream().spliterator();
        db.add("new BitSet.stream().spliterator() ADD",
               () -> new IntSource<>(toBitSet(bits), bitsSource, bs -> bs.set(3)));
        db.add("new BitSet.stream().spliterator() REMOVE",
               () -> new IntSource<>(toBitSet(bits), bitsSource, bs -> bs.clear(2)));

        // CharSequence

        Function<CharSequence, Spliterator.OfInt> charsSource = sb -> sb.chars().spliterator();
        Function<CharSequence, Spliterator.OfInt> pointsSource = sb -> sb.codePoints().spliterator();

        db.add("new StringBuilder.chars().spliterator() ADD",
               () -> new IntSource<>(new StringBuilder("ABC"), charsSource, bs -> bs.append("D"), true));
        db.add("new StringBuilder.chars().spliterator() REMOVE",
               () -> new IntSource<>(new StringBuilder("ABC"), charsSource, bs -> bs.deleteCharAt(2), true));
        db.add("new StringBuilder.codePoints().spliterator() ADD",
               () -> new IntSource<>(new StringBuilder("ABC"), pointsSource, bs -> bs.append("D"), true));
        db.add("new StringBuilder.codePoints().spliterator() REMOVE",
               () -> new IntSource<>(new StringBuilder("ABC"), pointsSource, bs -> bs.deleteCharAt(2), true));

        db.add("new StringBuffer.chars().spliterator() ADD",
               () -> new IntSource<>(new StringBuffer("ABC"), charsSource, bs -> bs.append("D"), true));
        db.add("new StringBuffer.chars().spliterator() REMOVE",
               () -> new IntSource<>(new StringBuffer("ABC"), charsSource, bs -> bs.deleteCharAt(2), true));
        db.add("new StringBuffer.codePoints().spliterator() ADD",
               () -> new IntSource<>(new StringBuffer("ABC"), pointsSource, bs -> bs.append("D"), true));
        db.add("new StringBuffer.codePoints().spliterator() REMOVE",
               () -> new IntSource<>(new StringBuffer("ABC"), pointsSource, bs -> bs.deleteCharAt(2), true));

        db.add("CharBuffer.wrap().chars().spliterator() ADD",
               () -> new IntSource<>(CharBuffer.wrap("ABCD").limit(3), charsSource, bs -> bs.limit(4), true));
        db.add("CharBuffer.wrap().chars().spliterator() REMOVE",
               () -> new IntSource<>(CharBuffer.wrap("ABCD"), charsSource, bs -> bs.limit(3), true));
        db.add("CharBuffer.wrap().codePoints().spliterator() ADD",
               () -> new IntSource<>(CharBuffer.wrap("ABCD").limit(3), pointsSource, bs -> bs.limit(4), true));
        db.add("CharBuffer.wrap().codePoints().spliterator() REMOVE",
               () -> new IntSource<>(CharBuffer.wrap("ABCD"), pointsSource, bs -> bs.limit(3), true));

        return spliteratorDataProvider = data.toArray(new Object[0][]);
    }


    @DataProvider(name = "Source.Non.Binding.Characteristics")
    public static Object[][] sourceCharacteristicsDataProvider() {
        return Stream.of(sourceDataProvider()).filter(tc -> {
            @SuppressWarnings("unchecked")
            Supplier<Source<?>> s = (Supplier<Source<?>>) tc[1];
            return !s.get().bindOnCharacteristics();
        }).toArray(Object[][]::new);
    }

    static BitSet toBitSet(List<Integer> bits) {
        BitSet bs = new BitSet();
        bits.forEach(bs::set);
        return bs;
    }


    @Test(dataProvider = "Source")
    public <T> void testForEach(String description, Supplier<Source<T>> ss) {
        Source<T> source = ss.get();
        Spliterator<T> s = source.spliterator();

        source.update();

        Set<T> a = new HashSet<>();
        s.forEachRemaining(a::add);

        Set<T> e = new HashSet<>();
        source.spliterator().forEachRemaining(e::add);
        assertEquals(a, e);
    }

    @Test(dataProvider = "Source")
    public <T> void testTryAdvance(String description, Supplier<Source<T>> ss) {
        Source<T> source = ss.get();
        Spliterator<T> s = source.spliterator();

        source.update();

        Set<T> a = new HashSet<>();
        while (s.tryAdvance(a::add)) {
        }

        Set<T> e = new HashSet<>();
        source.spliterator().forEachRemaining(e::add);
        assertEquals(a, e);
    }

    @Test(dataProvider = "Source.Non.Binding.Characteristics")
    public <T> void testCharacteristics(String description, Supplier<Source<T>> ss) {
        Source<T> source = ss.get();
        Spliterator<T> s = source.spliterator();

        s.characteristics();
        source.update();

        Set<T> a = new HashSet<>();
        s.forEachRemaining(a::add);

        Set<T> e = new HashSet<>();
        source.spliterator().forEachRemaining(e::add);
        assertEquals(a, e);
    }
}
Revision:	1.1
Committed:	Fri Dec 9 04:46:51 2016 UTC (7 years, 5 months ago) by jsr166
Branch:	MAIN
CVS Tags:	HEAD
Log Message:	sync Spliterator tests from upstream
#	User	Rev	Content
1	jsr166	1.1	/*
2			* Copyright (c) 2013, 2016, Oracle and/or its affiliates. All rights reserved.
3			* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4			*
5			* This code is free software; you can redistribute it and/or modify it
6			* under the terms of the GNU General Public License version 2 only, as
7			* published by the Free Software Foundation.
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			import org.testng.annotations.DataProvider;
25			import org.testng.annotations.Test;
26
27			import java.nio.CharBuffer;
28			import java.util.ArrayDeque;
29			import java.util.ArrayList;
30			import java.util.Arrays;
31			import java.util.BitSet;
32			import java.util.HashMap;
33			import java.util.HashSet;
34			import java.util.IdentityHashMap;
35			import java.util.LinkedHashMap;
36			import java.util.LinkedHashSet;
37			import java.util.LinkedList;
38			import java.util.List;
39			import java.util.PriorityQueue;
40			import java.util.Set;
41			import java.util.Spliterator;
42			import java.util.Stack;
43			import java.util.TreeMap;
44			import java.util.TreeSet;
45			import java.util.Vector;
46			import java.util.WeakHashMap;
47			import java.util.function.Function;
48			import java.util.function.Supplier;
49			import java.util.stream.Stream;
50
51			import static org.testng.Assert.assertEquals;
52
53			/**
54			* @test
55			* @bug 8148748 8170155
56			* @summary Spliterator last-binding tests
57			* @run testng SpliteratorLateBindingTest
58			*/
59
60			@Test
61			public class SpliteratorLateBindingTest extends SpliteratorLateBindingFailFastHelper {
62
63			static Object[][] spliteratorDataProvider;
64
65			@DataProvider(name = "Source")
66			public static Object[][] sourceDataProvider() {
67			if (spliteratorDataProvider != null) {
68			return spliteratorDataProvider;
69			}
70
71			List<Object[]> data = new ArrayList<>();
72			SpliteratorDataBuilder<Integer> db =
73			new SpliteratorDataBuilder<>(data, 5, Arrays.asList(1, 2, 3, 4));
74
75			// Collections
76
77			db.addList(ArrayList::new);
78
79			db.addList(LinkedList::new);
80
81			db.addList(Vector::new);
82
83			db.addList(AbstractRandomAccessListImpl::new);
84
85			db.addCollection(HashSet::new);
86
87			db.addCollection(LinkedHashSet::new);
88
89			db.addCollection(TreeSet::new);
90
91			db.addCollection(c -> {
92			Stack<Integer> s = new Stack<>();
93			s.addAll(c);
94			return s;
95			});
96
97			db.addCollection(PriorityQueue::new);
98
99			db.addCollection(ArrayDeque::new);
100
101			// Maps
102
103			db.addMap(HashMap::new);
104
105			db.addMap(LinkedHashMap::new);
106
107			db.addMap(IdentityHashMap::new);
108
109			db.addMap(WeakHashMap::new);
110
111			// @@@ Descending maps etc
112			db.addMap(TreeMap::new);
113
114			// BitSet
115
116			List<Integer> bits = List.of(0, 1, 2);
117			Function<BitSet, Spliterator.OfInt> bitsSource = bs -> bs.stream().spliterator();
118			db.add("new BitSet.stream().spliterator() ADD",
119			() -> new IntSource<>(toBitSet(bits), bitsSource, bs -> bs.set(3)));
120			db.add("new BitSet.stream().spliterator() REMOVE",
121			() -> new IntSource<>(toBitSet(bits), bitsSource, bs -> bs.clear(2)));
122
123			// CharSequence
124
125			Function<CharSequence, Spliterator.OfInt> charsSource = sb -> sb.chars().spliterator();
126			Function<CharSequence, Spliterator.OfInt> pointsSource = sb -> sb.codePoints().spliterator();
127
128			db.add("new StringBuilder.chars().spliterator() ADD",
129			() -> new IntSource<>(new StringBuilder("ABC"), charsSource, bs -> bs.append("D"), true));
130			db.add("new StringBuilder.chars().spliterator() REMOVE",
131			() -> new IntSource<>(new StringBuilder("ABC"), charsSource, bs -> bs.deleteCharAt(2), true));
132			db.add("new StringBuilder.codePoints().spliterator() ADD",
133			() -> new IntSource<>(new StringBuilder("ABC"), pointsSource, bs -> bs.append("D"), true));
134			db.add("new StringBuilder.codePoints().spliterator() REMOVE",
135			() -> new IntSource<>(new StringBuilder("ABC"), pointsSource, bs -> bs.deleteCharAt(2), true));
136
137			db.add("new StringBuffer.chars().spliterator() ADD",
138			() -> new IntSource<>(new StringBuffer("ABC"), charsSource, bs -> bs.append("D"), true));
139			db.add("new StringBuffer.chars().spliterator() REMOVE",
140			() -> new IntSource<>(new StringBuffer("ABC"), charsSource, bs -> bs.deleteCharAt(2), true));
141			db.add("new StringBuffer.codePoints().spliterator() ADD",
142			() -> new IntSource<>(new StringBuffer("ABC"), pointsSource, bs -> bs.append("D"), true));
143			db.add("new StringBuffer.codePoints().spliterator() REMOVE",
144			() -> new IntSource<>(new StringBuffer("ABC"), pointsSource, bs -> bs.deleteCharAt(2), true));
145
146			db.add("CharBuffer.wrap().chars().spliterator() ADD",
147			() -> new IntSource<>(CharBuffer.wrap("ABCD").limit(3), charsSource, bs -> bs.limit(4), true));
148			db.add("CharBuffer.wrap().chars().spliterator() REMOVE",
149			() -> new IntSource<>(CharBuffer.wrap("ABCD"), charsSource, bs -> bs.limit(3), true));
150			db.add("CharBuffer.wrap().codePoints().spliterator() ADD",
151			() -> new IntSource<>(CharBuffer.wrap("ABCD").limit(3), pointsSource, bs -> bs.limit(4), true));
152			db.add("CharBuffer.wrap().codePoints().spliterator() REMOVE",
153			() -> new IntSource<>(CharBuffer.wrap("ABCD"), pointsSource, bs -> bs.limit(3), true));
154
155			return spliteratorDataProvider = data.toArray(new Object[0][]);
156			}
157
158
159			@DataProvider(name = "Source.Non.Binding.Characteristics")
160			public static Object[][] sourceCharacteristicsDataProvider() {
161			return Stream.of(sourceDataProvider()).filter(tc -> {
162			@SuppressWarnings("unchecked")
163			Supplier<Source<?>> s = (Supplier<Source<?>>) tc[1];
164			return !s.get().bindOnCharacteristics();
165			}).toArray(Object[][]::new);
166			}
167
168			static BitSet toBitSet(List<Integer> bits) {
169			BitSet bs = new BitSet();
170			bits.forEach(bs::set);
171			return bs;
172			}
173
174
175			@Test(dataProvider = "Source")
176			public <T> void testForEach(String description, Supplier<Source<T>> ss) {
177			Source<T> source = ss.get();
178			Spliterator<T> s = source.spliterator();
179
180			source.update();
181
182			Set<T> a = new HashSet<>();
183			s.forEachRemaining(a::add);
184
185			Set<T> e = new HashSet<>();
186			source.spliterator().forEachRemaining(e::add);
187			assertEquals(a, e);
188			}
189
190			@Test(dataProvider = "Source")
191			public <T> void testTryAdvance(String description, Supplier<Source<T>> ss) {
192			Source<T> source = ss.get();
193			Spliterator<T> s = source.spliterator();
194
195			source.update();
196
197			Set<T> a = new HashSet<>();
198			while (s.tryAdvance(a::add)) {
199			}
200
201			Set<T> e = new HashSet<>();
202			source.spliterator().forEachRemaining(e::add);
203			assertEquals(a, e);
204			}
205
206			@Test(dataProvider = "Source.Non.Binding.Characteristics")
207			public <T> void testCharacteristics(String description, Supplier<Source<T>> ss) {
208			Source<T> source = ss.get();
209			Spliterator<T> s = source.spliterator();
210
211			s.characteristics();
212			source.update();
213
214			Set<T> a = new HashSet<>();
215			s.forEachRemaining(a::add);
216
217			Set<T> e = new HashSet<>();
218			source.spliterator().forEachRemaining(e::add);
219			assertEquals(a, e);
220			}
221			}