1 /*
2 * Copyright (c) 2013, 2024, 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 java.util.ArrayList;
25 import java.util.Arrays;
26 import java.util.ConcurrentModificationException;
27 import java.util.HashMap;
28 import java.util.HashSet;
29 import java.util.LinkedHashMap;
30 import java.util.LinkedHashSet;
31 import java.util.LinkedList;
32 import java.util.List;
33 import java.util.PriorityQueue;
34 import java.util.Spliterator;
35 import java.util.Stack;
36 import java.util.TreeMap;
37 import java.util.TreeSet;
38 import java.util.Vector;
39 import java.util.WeakHashMap;
40 import java.util.function.Supplier;
41
42 import org.testng.Assert.ThrowingRunnable;
43 import org.testng.annotations.DataProvider;
44 import org.testng.annotations.Test;
45
46 import static org.testng.Assert.assertThrows;
47
48 /**
49 * @test
50 * @bug 8148748 8336672
51 * @summary Spliterator fail-fast tests
52 * @run testng SpliteratorFailFastTest
53 */
54
55 @Test
56 public class SpliteratorFailFastTest extends SpliteratorLateBindingFailFastHelper {
57
58 static Object[][] spliteratorDataProvider;
59
60 @DataProvider(name = "Source")
61 public static Object[][] spliteratorDataProvider() {
62 if (spliteratorDataProvider != null) {
63 return spliteratorDataProvider;
64 }
65
66 List<Object[]> data = new ArrayList<>();
67 SpliteratorDataBuilder<String> db =
68 new SpliteratorDataBuilder<>(data, "Z", Arrays.asList("A", "B", "C", "D"));
69
70 // Collections
71
72 db.addList(ArrayList::new);
73
74 db.addList(LinkedList::new);
75
76 db.addList(Vector::new);
77
78 db.addList(AbstractRandomAccessListImpl::new);
79
80 db.addCollection(HashSet::new);
81
82 db.addCollection(LinkedHashSet::new);
83
84 db.addCollection(TreeSet::new);
85
86 db.addCollection(c -> {
87 Stack<String> s = new Stack<>();
88 s.addAll(c);
89 return s;
90 });
91
92 db.addCollection(PriorityQueue::new);
93
94 // ArrayDeque fails some tests since its fail-fast support is weaker
95 // than other collections and limited to detecting most, but not all,
96 // removals. It probably requires its own test since it is difficult
97 // to abstract out the conditions under which it fails-fast.
98 // db.addCollection(ArrayDeque::new);
99
100 // Maps
101
102 db.addMap(HashMap::new);
103
104 db.addMap(LinkedHashMap::new);
105
106 // This fails when run through jtreg but passes when run through
107 // ant
108 // db.addMap(IdentityHashMap::new);
109 // BUG: Assumes identity
110 db.addMap(WeakHashMap::new);
111
112 // @@@ Descending maps etc
113 db.addMap(TreeMap::new);
114
115 return spliteratorDataProvider = data.toArray(new Object[0][]);
116 }
117
118 @Test(dataProvider = "Source")
119 public <T> void testTryAdvance(String description, Supplier<Source<T>> ss) {
120 {
121 Source<T> source = ss.get();
122 Spliterator<T> s = source.spliterator();
123
124 s.tryAdvance(e -> {
125 });
126 source.update();
127
128 assertThrowsCME(() -> s.tryAdvance(e -> {
129 }));
130 }
131
132 {
133 Source<T> source = ss.get();
134 Spliterator<T> s = source.spliterator();
135
136 s.tryAdvance(e -> {
137 });
138 source.update();
139
140 assertThrowsCME(() -> s.forEachRemaining(e -> {
141 }));
142 }
143 }
144
145 @Test(dataProvider = "Source")
146 public <T> void testForEach(String description, Supplier<Source<T>> ss) {
147 Source<T> source = ss.get();
148 Spliterator<T> s = source.spliterator();
149
150 assertThrowsCME(() -> s.forEachRemaining(e -> {
151 source.update();
152 }));
153 }
154
155 @Test(dataProvider = "Source")
156 public <T> void testEstimateSize(String description, Supplier<Source<T>> ss) {
157 {
158 Source<T> source = ss.get();
159 Spliterator<T> s = source.spliterator();
160
161 s.estimateSize();
162 source.update();
163
164 assertThrowsCME(() -> s.tryAdvance(e -> {
165 }));
166 }
167
168 {
169 Source<T> source = ss.get();
170 Spliterator<T> s = source.spliterator();
171
172 s.estimateSize();
173 source.update();
174
175 assertThrowsCME(() -> s.forEachRemaining(e -> {
176 }));
177 }
178 }
179
180 private void assertThrowsCME(ThrowingRunnable r) {
181 assertThrows(ConcurrentModificationException.class, r);
182 }
183
184 }