ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/jsr166/jsr166/src/test/tck/ConcurrentSkipListSetTest.java

Comparing jsr166/src/test/tck/ConcurrentSkipListSetTest.java (file contents):
Revision 1.1 by dl, Tue Dec 28 16:15:59 2004 UTC vs.
Revision 1.46 by jsr166, Sat Mar 11 18:20:46 2017 UTC

#	Line 1 | Line 1
1		/*
2		* Written by Doug Lea with assistance from members of JCP JSR-166
3		* Expert Group and released to the public domain, as explained at
4	<	* http://creativecommons.org/licenses/publicdomain
4	>	* http://creativecommons.org/publicdomain/zero/1.0/
5		*/
6
7	<	import junit.framework.*;
8	<	import java.util.*;
9	<	import java.util.concurrent.*;
10	<	import java.io.*;
7	>	import java.util.Arrays;
8	>	import java.util.BitSet;
9	>	import java.util.Collection;
10	>	import java.util.Comparator;
11	>	import java.util.Iterator;
12	>	import java.util.NavigableSet;
13	>	import java.util.NoSuchElementException;
14	>	import java.util.Random;
15	>	import java.util.Set;
16	>	import java.util.SortedSet;
17	>	import java.util.concurrent.ConcurrentSkipListSet;
18	>
19	>	import junit.framework.Test;
20	>	import junit.framework.TestSuite;
21
22		public class ConcurrentSkipListSetTest extends JSR166TestCase {
23		public static void main(String[] args) {
24	<	junit.textui.TestRunner.run (suite());
24	>	main(suite(), args);
25		}
26		public static Test suite() {
27	<	return new TestSuite(ConcurrentSkipListSetTest.class);
27	>	return new TestSuite(ConcurrentSkipListSetTest.class);
28		}
29
30	<	static class MyReverseComparator implements Comparator {
30	>	static class MyReverseComparator implements Comparator {
31		public int compare(Object x, Object y) {
32	<	int i = ((Integer)x).intValue();
23	<	int j = ((Integer)y).intValue();
24	<	if (i < j) return 1;
25	<	if (i > j) return -1;
26	<	return 0;
32	>	return ((Comparable)y).compareTo(x);
33		}
34		}
35
36		/**
37	<	* Create a set of given size containing consecutive
38	<	* Integers 0 ... n.
37	>	* Returns a new set of given size containing consecutive
38	>	* Integers 0 ... n - 1.
39		*/
40	<	private ConcurrentSkipListSet populatedSet(int n) {
41	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
40	>	private static ConcurrentSkipListSet<Integer> populatedSet(int n) {
41	>	ConcurrentSkipListSet<Integer> q =
42	>	new ConcurrentSkipListSet<Integer>();
43		assertTrue(q.isEmpty());
44	<	for(int i = n-1; i >= 0; i-=2)
45	<	assertTrue(q.add(new Integer(i)));
46	<	for(int i = (n & 1); i < n; i+=2)
47	<	assertTrue(q.add(new Integer(i)));
44	>	for (int i = n - 1; i >= 0; i -= 2)
45	>	assertTrue(q.add(new Integer(i)));
46	>	for (int i = (n & 1); i < n; i += 2)
47	>	assertTrue(q.add(new Integer(i)));
48		assertFalse(q.isEmpty());
49	<	assertEquals(n, q.size());
49	>	assertEquals(n, q.size());
50		return q;
51		}
52
53		/**
54	<	* Create set of first 5 ints
54	>	* Returns a new set of first 5 ints.
55		*/
56	<	private ConcurrentSkipListSet set5() {
56	>	private static ConcurrentSkipListSet set5() {
57		ConcurrentSkipListSet q = new ConcurrentSkipListSet();
58		assertTrue(q.isEmpty());
59		q.add(one);
#	Line 54 | Line 61 | public class ConcurrentSkipListSetTest e
61		q.add(three);
62		q.add(four);
63		q.add(five);
64	<	assertEquals(5, q.size());
64	>	assertEquals(5, q.size());
65		return q;
66		}
67	<
67	>
68		/**
69		* A new set has unbounded capacity
70		*/
#	Line 70 | Line 77 | public class ConcurrentSkipListSetTest e
77		*/
78		public void testConstructor3() {
79		try {
80	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet((Collection)null);
80	>	new ConcurrentSkipListSet((Collection)null);
81		shouldThrow();
82	<	}
76	<	catch (NullPointerException success) {}
82	>	} catch (NullPointerException success) {}
83		}
84
85		/**
#	Line 81 | Line 87 | public class ConcurrentSkipListSetTest e
87		*/
88		public void testConstructor4() {
89		try {
90	<	Integer[] ints = new Integer[SIZE];
85	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet(Arrays.asList(ints));
90	>	new ConcurrentSkipListSet(Arrays.asList(new Integer[SIZE]));
91		shouldThrow();
92	<	}
88	<	catch (NullPointerException success) {}
92	>	} catch (NullPointerException success) {}
93		}
94
95		/**
96		* Initializing from Collection with some null elements throws NPE
97		*/
98		public void testConstructor5() {
99	+	Integer[] ints = new Integer[SIZE];
100	+	for (int i = 0; i < SIZE - 1; ++i)
101	+	ints[i] = new Integer(i);
102		try {
103	<	Integer[] ints = new Integer[SIZE];
97	<	for (int i = 0; i < SIZE-1; ++i)
98	<	ints[i] = new Integer(i);
99	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet(Arrays.asList(ints));
103	>	new ConcurrentSkipListSet(Arrays.asList(ints));
104		shouldThrow();
105	<	}
102	<	catch (NullPointerException success) {}
105	>	} catch (NullPointerException success) {}
106		}
107
108		/**
109		* Set contains all elements of collection used to initialize
110		*/
111		public void testConstructor6() {
112	<	try {
113	<	Integer[] ints = new Integer[SIZE];
114	<	for (int i = 0; i < SIZE; ++i)
115	<	ints[i] = new Integer(i);
116	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet(Arrays.asList(ints));
117	<	for (int i = 0; i < SIZE; ++i)
115	<	assertEquals(ints[i], q.pollFirst());
116	<	}
117	<	finally {}
112	>	Integer[] ints = new Integer[SIZE];
113	>	for (int i = 0; i < SIZE; ++i)
114	>	ints[i] = new Integer(i);
115	>	ConcurrentSkipListSet q = new ConcurrentSkipListSet(Arrays.asList(ints));
116	>	for (int i = 0; i < SIZE; ++i)
117	>	assertEquals(ints[i], q.pollFirst());
118		}
119
120		/**
121		* The comparator used in constructor is used
122		*/
123		public void testConstructor7() {
124	<	try {
125	<	MyReverseComparator cmp = new MyReverseComparator();
126	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet(cmp);
127	<	assertEquals(cmp, q.comparator());
128	<	Integer[] ints = new Integer[SIZE];
129	<	for (int i = 0; i < SIZE; ++i)
130	<	ints[i] = new Integer(i);
131	<	q.addAll(Arrays.asList(ints));
132	<	for (int i = SIZE-1; i >= 0; --i)
133	<	assertEquals(ints[i], q.pollFirst());
134	<	}
135	<	finally {}
124	>	MyReverseComparator cmp = new MyReverseComparator();
125	>	ConcurrentSkipListSet q = new ConcurrentSkipListSet(cmp);
126	>	assertEquals(cmp, q.comparator());
127	>	Integer[] ints = new Integer[SIZE];
128	>	for (int i = 0; i < SIZE; ++i)
129	>	ints[i] = new Integer(i);
130	>	q.addAll(Arrays.asList(ints));
131	>	for (int i = SIZE - 1; i >= 0; --i)
132	>	assertEquals(ints[i], q.pollFirst());
133		}
134
135		/**
#	Line 155 | Line 152 | public class ConcurrentSkipListSetTest e
152		public void testSize() {
153		ConcurrentSkipListSet q = populatedSet(SIZE);
154		for (int i = 0; i < SIZE; ++i) {
155	<	assertEquals(SIZE-i, q.size());
155	>	assertEquals(SIZE - i, q.size());
156		q.pollFirst();
157		}
158		for (int i = 0; i < SIZE; ++i) {
#	Line 168 | Line 165 | public class ConcurrentSkipListSetTest e
165		* add(null) throws NPE
166		*/
167		public void testAddNull() {
168	<	try {
169	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
168	>	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
169	>	try {
170		q.add(null);
171		shouldThrow();
172	<	} catch (NullPointerException success) { }
172	>	} catch (NullPointerException success) {}
173		}
174
175		/**
#	Line 197 | Line 194 | public class ConcurrentSkipListSetTest e
194		* Add of non-Comparable throws CCE
195		*/
196		public void testAddNonComparable() {
197	+	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
198		try {
201	–	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
202	–	q.add(new Object());
199		q.add(new Object());
200		q.add(new Object());
201		shouldThrow();
202	+	} catch (ClassCastException success) {
203	+	assertTrue(q.size() < 2);
204	+	for (int i = 0, size = q.size(); i < size; i++)
205	+	assertTrue(q.pollFirst().getClass() == Object.class);
206	+	assertNull(q.pollFirst());
207	+	assertTrue(q.isEmpty());
208	+	assertEquals(0, q.size());
209		}
207	–	catch(ClassCastException success) {}
210		}
211
212		/**
213		* addAll(null) throws NPE
214		*/
215		public void testAddAll1() {
216	+	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
217		try {
215	–	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
218		q.addAll(null);
219		shouldThrow();
220	<	}
219	<	catch (NullPointerException success) {}
220	>	} catch (NullPointerException success) {}
221		}
222	+
223		/**
224		* addAll of a collection with null elements throws NPE
225		*/
226		public void testAddAll2() {
227	+	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
228	+	Integer[] ints = new Integer[SIZE];
229		try {
226	–	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
227	–	Integer[] ints = new Integer[SIZE];
230		q.addAll(Arrays.asList(ints));
231		shouldThrow();
232	<	}
231	<	catch (NullPointerException success) {}
232	>	} catch (NullPointerException success) {}
233		}
234	+
235		/**
236		* addAll of a collection with any null elements throws NPE after
237		* possibly adding some elements
238		*/
239		public void testAddAll3() {
240	+	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
241	+	Integer[] ints = new Integer[SIZE];
242	+	for (int i = 0; i < SIZE - 1; ++i)
243	+	ints[i] = new Integer(i);
244		try {
239	–	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
240	–	Integer[] ints = new Integer[SIZE];
241	–	for (int i = 0; i < SIZE-1; ++i)
242	–	ints[i] = new Integer(i);
245		q.addAll(Arrays.asList(ints));
246		shouldThrow();
247	<	}
246	<	catch (NullPointerException success) {}
247	>	} catch (NullPointerException success) {}
248		}
249
250		/**
251		* Set contains all elements of successful addAll
252		*/
253		public void testAddAll5() {
254	<	try {
255	<	Integer[] empty = new Integer[0];
256	<	Integer[] ints = new Integer[SIZE];
257	<	for (int i = 0; i < SIZE; ++i)
258	<	ints[i] = new Integer(SIZE-1-i);
259	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
260	<	assertFalse(q.addAll(Arrays.asList(empty)));
261	<	assertTrue(q.addAll(Arrays.asList(ints)));
262	<	for (int i = 0; i < SIZE; ++i)
262	<	assertEquals(new Integer(i), q.pollFirst());
263	<	}
264	<	finally {}
254	>	Integer[] empty = new Integer[0];
255	>	Integer[] ints = new Integer[SIZE];
256	>	for (int i = 0; i < SIZE; ++i)
257	>	ints[i] = new Integer(SIZE - 1 - i);
258	>	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
259	>	assertFalse(q.addAll(Arrays.asList(empty)));
260	>	assertTrue(q.addAll(Arrays.asList(ints)));
261	>	for (int i = 0; i < SIZE; ++i)
262	>	assertEquals(i, q.pollFirst());
263		}
264
265		/**
#	Line 270 | Line 268 | public class ConcurrentSkipListSetTest e
268		public void testPollFirst() {
269		ConcurrentSkipListSet q = populatedSet(SIZE);
270		for (int i = 0; i < SIZE; ++i) {
271	<	assertEquals(i, ((Integer)q.pollFirst()).intValue());
271	>	assertEquals(i, q.pollFirst());
272		}
273	<	assertNull(q.pollFirst());
273	>	assertNull(q.pollFirst());
274		}
275
276		/**
#	Line 280 | Line 278 | public class ConcurrentSkipListSetTest e
278		*/
279		public void testPollLast() {
280		ConcurrentSkipListSet q = populatedSet(SIZE);
281	<	for (int i = SIZE-1; i >= 0; --i) {
282	<	assertEquals(i, ((Integer)q.pollLast()).intValue());
281	>	for (int i = SIZE - 1; i >= 0; --i) {
282	>	assertEquals(i, q.pollLast());
283		}
284	<	assertNull(q.pollFirst());
284	>	assertNull(q.pollFirst());
285		}
286
289	–
287		/**
288		* remove(x) removes x and returns true if present
289		*/
290		public void testRemoveElement() {
291		ConcurrentSkipListSet q = populatedSet(SIZE);
292	<	for (int i = 1; i < SIZE; i+=2) {
293	<	assertTrue(q.remove(new Integer(i)));
294	<	}
295	<	for (int i = 0; i < SIZE; i+=2) {
296	<	assertTrue(q.remove(new Integer(i)));
297	<	assertFalse(q.remove(new Integer(i+1)));
292	>	for (int i = 1; i < SIZE; i += 2) {
293	>	assertTrue(q.contains(i));
294	>	assertTrue(q.remove(i));
295	>	assertFalse(q.contains(i));
296	>	assertTrue(q.contains(i - 1));
297	>	}
298	>	for (int i = 0; i < SIZE; i += 2) {
299	>	assertTrue(q.contains(i));
300	>	assertTrue(q.remove(i));
301	>	assertFalse(q.contains(i));
302	>	assertFalse(q.remove(i + 1));
303	>	assertFalse(q.contains(i + 1));
304		}
305		assertTrue(q.isEmpty());
306		}
307	<
307	>
308		/**
309		* contains(x) reports true when elements added but not yet removed
310		*/
#	Line 356 | Line 359 | public class ConcurrentSkipListSetTest e
359		assertTrue(changed);
360
361		assertTrue(q.containsAll(p));
362	<	assertEquals(SIZE-i, q.size());
362	>	assertEquals(SIZE - i, q.size());
363		p.pollFirst();
364		}
365		}
#	Line 369 | Line 372 | public class ConcurrentSkipListSetTest e
372		ConcurrentSkipListSet q = populatedSet(SIZE);
373		ConcurrentSkipListSet p = populatedSet(i);
374		assertTrue(q.removeAll(p));
375	<	assertEquals(SIZE-i, q.size());
375	>	assertEquals(SIZE - i, q.size());
376		for (int j = 0; j < i; ++j) {
377	<	Integer I = (Integer)(p.pollFirst());
378	<	assertFalse(q.contains(I));
377	>	Integer x = (Integer)(p.pollFirst());
378	>	assertFalse(q.contains(x));
379		}
380		}
381		}
382
380	–
381	–
383		/**
384		* lower returns preceding element
385		*/
#	Line 395 | Line 396 | public class ConcurrentSkipListSetTest e
396
397		Object e4 = q.lower(zero);
398		assertNull(e4);
398	–
399		}
400
401		/**
#	Line 414 | Line 414 | public class ConcurrentSkipListSetTest e
414
415		Object e4 = q.higher(six);
416		assertNull(e4);
417	–
417		}
418
419		/**
#	Line 433 | Line 432 | public class ConcurrentSkipListSetTest e
432
433		Object e4 = q.floor(zero);
434		assertNull(e4);
436	–
435		}
436
437		/**
#	Line 452 | Line 450 | public class ConcurrentSkipListSetTest e
450
451		Object e4 = q.ceiling(six);
452		assertNull(e4);
455	–
453		}
454
455		/**
456	<	* toArray contains all elements
456	>	* toArray contains all elements in sorted order
457		*/
458		public void testToArray() {
459		ConcurrentSkipListSet q = populatedSet(SIZE);
460	<	Object[] o = q.toArray();
461	<	Arrays.sort(o);
462	<	for(int i = 0; i < o.length; i++)
466	<	assertEquals(o[i], q.pollFirst());
460	>	Object[] o = q.toArray();
461	>	for (int i = 0; i < o.length; i++)
462	>	assertSame(o[i], q.pollFirst());
463		}
464
465		/**
466	<	* toArray(a) contains all elements
466	>	* toArray(a) contains all elements in sorted order
467		*/
468		public void testToArray2() {
469	<	ConcurrentSkipListSet q = populatedSet(SIZE);
470	<	Integer[] ints = new Integer[SIZE];
471	<	ints = (Integer[])q.toArray(ints);
472	<	Arrays.sort(ints);
473	<	for(int i = 0; i < ints.length; i++)
478	<	assertEquals(ints[i], q.pollFirst());
469	>	ConcurrentSkipListSet<Integer> q = populatedSet(SIZE);
470	>	Integer[] ints = new Integer[SIZE];
471	>	assertSame(ints, q.toArray(ints));
472	>	for (int i = 0; i < ints.length; i++)
473	>	assertSame(ints[i], q.pollFirst());
474		}
475	<
475	>
476		/**
477		* iterator iterates through all elements
478		*/
479		public void testIterator() {
480		ConcurrentSkipListSet q = populatedSet(SIZE);
481	<	int i = 0;
482	<	Iterator it = q.iterator();
483	<	while(it.hasNext()) {
481	>	Iterator it = q.iterator();
482	>	int i;
483	>	for (i = 0; it.hasNext(); i++)
484		assertTrue(q.contains(it.next()));
490	–	++i;
491	–	}
485		assertEquals(i, SIZE);
486	+	assertIteratorExhausted(it);
487		}
488
489		/**
490		* iterator of empty set has no elements
491		*/
492		public void testEmptyIterator() {
493	<	ConcurrentSkipListSet q = new ConcurrentSkipListSet();
494	<	int i = 0;
495	<	Iterator it = q.iterator();
502	<	while(it.hasNext()) {
503	<	assertTrue(q.contains(it.next()));
504	<	++i;
505	<	}
506	<	assertEquals(i, 0);
493	>	NavigableSet s = new ConcurrentSkipListSet();
494	>	assertIteratorExhausted(s.iterator());
495	>	assertIteratorExhausted(s.descendingSet().iterator());
496		}
497
498		/**
499		* iterator.remove removes current element
500		*/
501	<	public void testIteratorRemove () {
501	>	public void testIteratorRemove() {
502		final ConcurrentSkipListSet q = new ConcurrentSkipListSet();
503		q.add(new Integer(2));
504		q.add(new Integer(1));
#	Line 525 | Line 514 | public class ConcurrentSkipListSetTest e
514		assertFalse(it.hasNext());
515		}
516
528	–
517		/**
518		* toString contains toStrings of elements
519		*/
#	Line 533 | Line 521 | public class ConcurrentSkipListSetTest e
521		ConcurrentSkipListSet q = populatedSet(SIZE);
522		String s = q.toString();
523		for (int i = 0; i < SIZE; ++i) {
524	<	assertTrue(s.indexOf(String.valueOf(i)) >= 0);
524	>	assertTrue(s.contains(String.valueOf(i)));
525		}
526	<	}
526	>	}
527
528		/**
529	<	* A deserialized serialized set has same elements
529	>	* A deserialized serialized set has same elements
530		*/
531	<	public void testSerialization() {
532	<	ConcurrentSkipListSet q = populatedSet(SIZE);
533	<	try {
534	<	ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
535	<	ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
536	<	out.writeObject(q);
537	<	out.close();
538	<
539	<	ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
540	<	ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
541	<	ConcurrentSkipListSet r = (ConcurrentSkipListSet)in.readObject();
554	<	assertEquals(q.size(), r.size());
555	<	while (!q.isEmpty())
556	<	assertEquals(q.pollFirst(), r.pollFirst());
557	<	} catch(Exception e){
558	<	e.printStackTrace();
559	<	unexpectedException();
531	>	public void testSerialization() throws Exception {
532	>	NavigableSet x = populatedSet(SIZE);
533	>	NavigableSet y = serialClone(x);
534	>
535	>	assertNotSame(x, y);
536	>	assertEquals(x.size(), y.size());
537	>	assertEquals(x, y);
538	>	assertEquals(y, x);
539	>	while (!x.isEmpty()) {
540	>	assertFalse(y.isEmpty());
541	>	assertEquals(x.pollFirst(), y.pollFirst());
542		}
543	+	assertTrue(y.isEmpty());
544		}
545
546		/**
#	Line 679 | Line 662 | public class ConcurrentSkipListSetTest e
662		assertEquals(4, set.size());
663		}
664
665	+	Random rnd = new Random(666);
666	+
667	+	/**
668	+	* Subsets of subsets subdivide correctly
669	+	*/
670	+	public void testRecursiveSubSets() throws Exception {
671	+	int setSize = expensiveTests ? 1000 : 100;
672	+	Class cl = ConcurrentSkipListSet.class;
673	+
674	+	NavigableSet<Integer> set = newSet(cl);
675	+	BitSet bs = new BitSet(setSize);
676	+
677	+	populate(set, setSize, bs);
678	+	check(set,                 0, setSize - 1, true, bs);
679	+	check(set.descendingSet(), 0, setSize - 1, false, bs);
680	+
681	+	mutateSet(set, 0, setSize - 1, bs);
682	+	check(set,                 0, setSize - 1, true, bs);
683	+	check(set.descendingSet(), 0, setSize - 1, false, bs);
684	+
685	+	bashSubSet(set.subSet(0, true, setSize, false),
686	+	0, setSize - 1, true, bs);
687	+	}
688	+
689	+	/**
690	+	* addAll is idempotent
691	+	*/
692	+	public void testAddAll_idempotent() throws Exception {
693	+	Set x = populatedSet(SIZE);
694	+	Set y = new ConcurrentSkipListSet(x);
695	+	y.addAll(x);
696	+	assertEquals(x, y);
697	+	assertEquals(y, x);
698	+	}
699	+
700	+	static NavigableSet<Integer> newSet(Class cl) throws Exception {
701	+	NavigableSet<Integer> result =
702	+	(NavigableSet<Integer>) cl.getConstructor().newInstance();
703	+	assertEquals(0, result.size());
704	+	assertFalse(result.iterator().hasNext());
705	+	return result;
706	+	}
707	+
708	+	void populate(NavigableSet<Integer> set, int limit, BitSet bs) {
709	+	for (int i = 0, n = 2 * limit / 3; i < n; i++) {
710	+	int element = rnd.nextInt(limit);
711	+	put(set, element, bs);
712	+	}
713	+	}
714	+
715	+	void mutateSet(NavigableSet<Integer> set, int min, int max, BitSet bs) {
716	+	int size = set.size();
717	+	int rangeSize = max - min + 1;
718	+
719	+	// Remove a bunch of entries directly
720	+	for (int i = 0, n = rangeSize / 2; i < n; i++) {
721	+	remove(set, min - 5 + rnd.nextInt(rangeSize + 10), bs);
722	+	}
723	+
724	+	// Remove a bunch of entries with iterator
725	+	for (Iterator<Integer> it = set.iterator(); it.hasNext(); ) {
726	+	if (rnd.nextBoolean()) {
727	+	bs.clear(it.next());
728	+	it.remove();
729	+	}
730	+	}
731	+
732	+	// Add entries till we're back to original size
733	+	while (set.size() < size) {
734	+	int element = min + rnd.nextInt(rangeSize);
735	+	assertTrue(element >= min && element <= max);
736	+	put(set, element, bs);
737	+	}
738	+	}
739	+
740	+	void mutateSubSet(NavigableSet<Integer> set, int min, int max,
741	+	BitSet bs) {
742	+	int size = set.size();
743	+	int rangeSize = max - min + 1;
744	+
745	+	// Remove a bunch of entries directly
746	+	for (int i = 0, n = rangeSize / 2; i < n; i++) {
747	+	remove(set, min - 5 + rnd.nextInt(rangeSize + 10), bs);
748	+	}
749	+
750	+	// Remove a bunch of entries with iterator
751	+	for (Iterator<Integer> it = set.iterator(); it.hasNext(); ) {
752	+	if (rnd.nextBoolean()) {
753	+	bs.clear(it.next());
754	+	it.remove();
755	+	}
756	+	}
757	+
758	+	// Add entries till we're back to original size
759	+	while (set.size() < size) {
760	+	int element = min - 5 + rnd.nextInt(rangeSize + 10);
761	+	if (element >= min && element <= max) {
762	+	put(set, element, bs);
763	+	} else {
764	+	try {
765	+	set.add(element);
766	+	shouldThrow();
767	+	} catch (IllegalArgumentException success) {}
768	+	}
769	+	}
770	+	}
771	+
772	+	void put(NavigableSet<Integer> set, int element, BitSet bs) {
773	+	if (set.add(element))
774	+	bs.set(element);
775	+	}
776	+
777	+	void remove(NavigableSet<Integer> set, int element, BitSet bs) {
778	+	if (set.remove(element))
779	+	bs.clear(element);
780	+	}
781	+
782	+	void bashSubSet(NavigableSet<Integer> set,
783	+	int min, int max, boolean ascending,
784	+	BitSet bs) {
785	+	check(set, min, max, ascending, bs);
786	+	check(set.descendingSet(), min, max, !ascending, bs);
787	+
788	+	mutateSubSet(set, min, max, bs);
789	+	check(set, min, max, ascending, bs);
790	+	check(set.descendingSet(), min, max, !ascending, bs);
791	+
792	+	// Recurse
793	+	if (max - min < 2)
794	+	return;
795	+	int midPoint = (min + max) / 2;
796	+
797	+	// headSet - pick direction and endpoint inclusion randomly
798	+	boolean incl = rnd.nextBoolean();
799	+	NavigableSet<Integer> hm = set.headSet(midPoint, incl);
800	+	if (ascending) {
801	+	if (rnd.nextBoolean())
802	+	bashSubSet(hm, min, midPoint - (incl ? 0 : 1), true, bs);
803	+	else
804	+	bashSubSet(hm.descendingSet(), min, midPoint - (incl ? 0 : 1),
805	+	false, bs);
806	+	} else {
807	+	if (rnd.nextBoolean())
808	+	bashSubSet(hm, midPoint + (incl ? 0 : 1), max, false, bs);
809	+	else
810	+	bashSubSet(hm.descendingSet(), midPoint + (incl ? 0 : 1), max,
811	+	true, bs);
812	+	}
813	+
814	+	// tailSet - pick direction and endpoint inclusion randomly
815	+	incl = rnd.nextBoolean();
816	+	NavigableSet<Integer> tm = set.tailSet(midPoint,incl);
817	+	if (ascending) {
818	+	if (rnd.nextBoolean())
819	+	bashSubSet(tm, midPoint + (incl ? 0 : 1), max, true, bs);
820	+	else
821	+	bashSubSet(tm.descendingSet(), midPoint + (incl ? 0 : 1), max,
822	+	false, bs);
823	+	} else {
824	+	if (rnd.nextBoolean()) {
825	+	bashSubSet(tm, min, midPoint - (incl ? 0 : 1), false, bs);
826	+	} else {
827	+	bashSubSet(tm.descendingSet(), min, midPoint - (incl ? 0 : 1),
828	+	true, bs);
829	+	}
830	+	}
831	+
832	+	// subSet - pick direction and endpoint inclusion randomly
833	+	int rangeSize = max - min + 1;
834	+	int[] endpoints = new int[2];
835	+	endpoints[0] = min + rnd.nextInt(rangeSize);
836	+	endpoints[1] = min + rnd.nextInt(rangeSize);
837	+	Arrays.sort(endpoints);
838	+	boolean lowIncl = rnd.nextBoolean();
839	+	boolean highIncl = rnd.nextBoolean();
840	+	if (ascending) {
841	+	NavigableSet<Integer> sm = set.subSet(
842	+	endpoints[0], lowIncl, endpoints[1], highIncl);
843	+	if (rnd.nextBoolean())
844	+	bashSubSet(sm, endpoints[0] + (lowIncl ? 0 : 1),
845	+	endpoints[1] - (highIncl ? 0 : 1), true, bs);
846	+	else
847	+	bashSubSet(sm.descendingSet(), endpoints[0] + (lowIncl ? 0 : 1),
848	+	endpoints[1] - (highIncl ? 0 : 1), false, bs);
849	+	} else {
850	+	NavigableSet<Integer> sm = set.subSet(
851	+	endpoints[1], highIncl, endpoints[0], lowIncl);
852	+	if (rnd.nextBoolean())
853	+	bashSubSet(sm, endpoints[0] + (lowIncl ? 0 : 1),
854	+	endpoints[1] - (highIncl ? 0 : 1), false, bs);
855	+	else
856	+	bashSubSet(sm.descendingSet(), endpoints[0] + (lowIncl ? 0 : 1),
857	+	endpoints[1] - (highIncl ? 0 : 1), true, bs);
858	+	}
859	+	}
860	+
861	+	/**
862	+	* min and max are both inclusive.  If max < min, interval is empty.
863	+	*/
864	+	void check(NavigableSet<Integer> set,
865	+	final int min, final int max, final boolean ascending,
866	+	final BitSet bs) {
867	+	class ReferenceSet {
868	+	int lower(int element) {
869	+	return ascending ?
870	+	lowerAscending(element) : higherAscending(element);
871	+	}
872	+	int floor(int element) {
873	+	return ascending ?
874	+	floorAscending(element) : ceilingAscending(element);
875	+	}
876	+	int ceiling(int element) {
877	+	return ascending ?
878	+	ceilingAscending(element) : floorAscending(element);
879	+	}
880	+	int higher(int element) {
881	+	return ascending ?
882	+	higherAscending(element) : lowerAscending(element);
883	+	}
884	+	int first() {
885	+	return ascending ? firstAscending() : lastAscending();
886	+	}
887	+	int last() {
888	+	return ascending ? lastAscending() : firstAscending();
889	+	}
890	+	int lowerAscending(int element) {
891	+	return floorAscending(element - 1);
892	+	}
893	+	int floorAscending(int element) {
894	+	if (element < min)
895	+	return -1;
896	+	else if (element > max)
897	+	element = max;
898	+
899	+	// BitSet should support this! Test would run much faster
900	+	while (element >= min) {
901	+	if (bs.get(element))
902	+	return element;
903	+	element--;
904	+	}
905	+	return -1;
906	+	}
907	+	int ceilingAscending(int element) {
908	+	if (element < min)
909	+	element = min;
910	+	else if (element > max)
911	+	return -1;
912	+	int result = bs.nextSetBit(element);
913	+	return result > max ? -1 : result;
914	+	}
915	+	int higherAscending(int element) {
916	+	return ceilingAscending(element + 1);
917	+	}
918	+	private int firstAscending() {
919	+	int result = ceilingAscending(min);
920	+	return result > max ? -1 : result;
921	+	}
922	+	private int lastAscending() {
923	+	int result = floorAscending(max);
924	+	return result < min ? -1 : result;
925	+	}
926	+	}
927	+	ReferenceSet rs = new ReferenceSet();
928	+
929	+	// Test contents using containsElement
930	+	int size = 0;
931	+	for (int i = min; i <= max; i++) {
932	+	boolean bsContainsI = bs.get(i);
933	+	assertEquals(bsContainsI, set.contains(i));
934	+	if (bsContainsI)
935	+	size++;
936	+	}
937	+	assertEquals(size, set.size());
938	+
939	+	// Test contents using contains elementSet iterator
940	+	int size2 = 0;
941	+	int previousElement = -1;
942	+	for (int element : set) {
943	+	assertTrue(bs.get(element));
944	+	size2++;
945	+	assertTrue(previousElement < 0 || (ascending ?
946	+	element - previousElement > 0 : element - previousElement < 0));
947	+	previousElement = element;
948	+	}
949	+	assertEquals(size2, size);
950	+
951	+	// Test navigation ops
952	+	for (int element = min - 1; element <= max + 1; element++) {
953	+	assertEq(set.lower(element), rs.lower(element));
954	+	assertEq(set.floor(element), rs.floor(element));
955	+	assertEq(set.higher(element), rs.higher(element));
956	+	assertEq(set.ceiling(element), rs.ceiling(element));
957	+	}
958	+
959	+	// Test extrema
960	+	if (set.size() != 0) {
961	+	assertEq(set.first(), rs.first());
962	+	assertEq(set.last(), rs.last());
963	+	} else {
964	+	assertEq(rs.first(), -1);
965	+	assertEq(rs.last(),  -1);
966	+	try {
967	+	set.first();
968	+	shouldThrow();
969	+	} catch (NoSuchElementException success) {}
970	+	try {
971	+	set.last();
972	+	shouldThrow();
973	+	} catch (NoSuchElementException success) {}
974	+	}
975	+	}
976	+
977	+	static void assertEq(Integer i, int j) {
978	+	if (i == null)
979	+	assertEquals(j, -1);
980	+	else
981	+	assertEquals((int) i, j);
982	+	}
983	+
984	+	static boolean eq(Integer i, int j) {
985	+	return (i == null) ? j == -1 : i == j;
986	+	}
987	+
988		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines