ViewVC Help

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

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

Comparing jsr166/src/test/tck/PriorityBlockingQueueTest.java (file contents):
Revision 1.3 by dl, Sun Sep 14 20:42:40 2003 UTC vs.
Revision 1.67 by jsr166, Wed Jun 1 16:08:04 2016 UTC

#	Line 1 | Line 1
1		/*
2	<	* Written by members of JCP JSR-166 Expert Group and released to the
3	<	* public domain. Use, modify, and redistribute this code in any way
4	<	* without acknowledgement. Other contributors include Andrew Wright,
5	<	* Jeffrey Hayes, Pat Fischer, Mike Judd.
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/publicdomain/zero/1.0/
5	>	* Other contributors include Andrew Wright, Jeffrey Hayes,
6	>	* Pat Fisher, Mike Judd.
7		*/
8
9	<	import junit.framework.*;
10	<	import java.util.*;
11	<	import java.util.concurrent.*;
12	<	import java.io.*;
9	>	import static java.util.concurrent.TimeUnit.MILLISECONDS;
10	>
11	>	import java.util.ArrayList;
12	>	import java.util.Arrays;
13	>	import java.util.Collection;
14	>	import java.util.Comparator;
15	>	import java.util.Iterator;
16	>	import java.util.NoSuchElementException;
17	>	import java.util.Queue;
18	>	import java.util.concurrent.BlockingQueue;
19	>	import java.util.concurrent.CountDownLatch;
20	>	import java.util.concurrent.Executors;
21	>	import java.util.concurrent.ExecutorService;
22	>	import java.util.concurrent.PriorityBlockingQueue;
23	>
24	>	import junit.framework.Test;
25
26		public class PriorityBlockingQueueTest extends JSR166TestCase {
27	+
28	+	public static class Generic extends BlockingQueueTest {
29	+	protected BlockingQueue emptyCollection() {
30	+	return new PriorityBlockingQueue();
31	+	}
32	+	}
33	+
34	+	public static class InitialCapacity extends BlockingQueueTest {
35	+	protected BlockingQueue emptyCollection() {
36	+	return new PriorityBlockingQueue(SIZE);
37	+	}
38	+	}
39	+
40		public static void main(String[] args) {
41	<	junit.textui.TestRunner.run (suite());
41	>	main(suite(), args);
42		}
43
44		public static Test suite() {
45	<	return new TestSuite(PriorityBlockingQueueTest.class);
45	>	return newTestSuite(PriorityBlockingQueueTest.class,
46	>	new Generic().testSuite(),
47	>	new InitialCapacity().testSuite());
48		}
49
22	–	private static final int NOCAP = Integer.MAX_VALUE;
23	–
50		/** Sample Comparator */
51	<	static class MyReverseComparator implements Comparator {
51	>	static class MyReverseComparator implements Comparator {
52		public int compare(Object x, Object y) {
53	<	int i = ((Integer)x).intValue();
28	<	int j = ((Integer)y).intValue();
29	<	if (i < j) return 1;
30	<	if (i > j) return -1;
31	<	return 0;
53	>	return ((Comparable)y).compareTo(x);
54		}
55		}
56
35	–
57		/**
58	<	* Create a queue of given size containing consecutive
58	>	* Returns a new queue of given size containing consecutive
59		* Integers 0 ... n.
60		*/
61	<	private PriorityBlockingQueue populatedQueue(int n) {
62	<	PriorityBlockingQueue q = new PriorityBlockingQueue(n);
61	>	private PriorityBlockingQueue<Integer> populatedQueue(int n) {
62	>	PriorityBlockingQueue<Integer> q =
63	>	new PriorityBlockingQueue<Integer>(n);
64		assertTrue(q.isEmpty());
65	<	for(int i = n-1; i >= 0; i-=2)
66	<	assertTrue(q.offer(new Integer(i)));
67	<	for(int i = (n & 1); i < n; i+=2)
68	<	assertTrue(q.offer(new Integer(i)));
65	>	for (int i = n - 1; i >= 0; i -= 2)
66	>	assertTrue(q.offer(new Integer(i)));
67	>	for (int i = (n & 1); i < n; i += 2)
68	>	assertTrue(q.offer(new Integer(i)));
69		assertFalse(q.isEmpty());
70	<	assertEquals(NOCAP, q.remainingCapacity());
71	<	assertEquals(n, q.size());
70	>	assertEquals(Integer.MAX_VALUE, q.remainingCapacity());
71	>	assertEquals(n, q.size());
72		return q;
73		}
74	<
75	<	public void testConstructor1(){
76	<	assertEquals(NOCAP, new PriorityBlockingQueue(SIZE).remainingCapacity());
74	>
75	>	/**
76	>	* A new queue has unbounded capacity
77	>	*/
78	>	public void testConstructor1() {
79	>	assertEquals(Integer.MAX_VALUE,
80	>	new PriorityBlockingQueue(SIZE).remainingCapacity());
81		}
82
83	<	public void testConstructor2(){
83	>	/**
84	>	* Constructor throws IAE if capacity argument nonpositive
85	>	*/
86	>	public void testConstructor2() {
87		try {
88	<	PriorityBlockingQueue q = new PriorityBlockingQueue(0);
89	<	fail("Cannot make zero-sized");
90	<	}
62	<	catch (IllegalArgumentException success) {}
88	>	new PriorityBlockingQueue(0);
89	>	shouldThrow();
90	>	} catch (IllegalArgumentException success) {}
91		}
92
93	<	public void testConstructor3(){
94	<
93	>	/**
94	>	* Initializing from null Collection throws NPE
95	>	*/
96	>	public void testConstructor3() {
97		try {
98	<	PriorityBlockingQueue q = new PriorityBlockingQueue(null);
99	<	fail("Cannot make from null collection");
100	<	}
71	<	catch (NullPointerException success) {}
98	>	new PriorityBlockingQueue(null);
99	>	shouldThrow();
100	>	} catch (NullPointerException success) {}
101		}
102
103	<	public void testConstructor4(){
103	>	/**
104	>	* Initializing from Collection of null elements throws NPE
105	>	*/
106	>	public void testConstructor4() {
107	>	Collection<Integer> elements = Arrays.asList(new Integer[SIZE]);
108		try {
109	<	Integer[] ints = new Integer[SIZE];
110	<	PriorityBlockingQueue q = new PriorityBlockingQueue(Arrays.asList(ints));
111	<	fail("Cannot make with null elements");
79	<	}
80	<	catch (NullPointerException success) {}
109	>	new PriorityBlockingQueue(elements);
110	>	shouldThrow();
111	>	} catch (NullPointerException success) {}
112		}
113
114	<	public void testConstructor5(){
115	<	try {
116	<	Integer[] ints = new Integer[SIZE];
117	<	for (int i = 0; i < SIZE-1; ++i)
118	<	ints[i] = new Integer(i);
119	<	PriorityBlockingQueue q = new PriorityBlockingQueue(Arrays.asList(ints));
120	<	fail("Cannot make with null elements");
121	<	}
122	<	catch (NullPointerException success) {}
114	>	/**
115	>	* Initializing from Collection with some null elements throws NPE
116	>	*/
117	>	public void testConstructor5() {
118	>	Integer[] ints = new Integer[SIZE];
119	>	for (int i = 0; i < SIZE - 1; ++i)
120	>	ints[i] = i;
121	>	Collection<Integer> elements = Arrays.asList(ints);
122	>	try {
123	>	new PriorityBlockingQueue(elements);
124	>	shouldThrow();
125	>	} catch (NullPointerException success) {}
126		}
127
128	<	public void testConstructor6(){
129	<	try {
130	<	Integer[] ints = new Integer[SIZE];
131	<	for (int i = 0; i < SIZE; ++i)
132	<	ints[i] = new Integer(i);
133	<	PriorityBlockingQueue q = new PriorityBlockingQueue(Arrays.asList(ints));
134	<	for (int i = 0; i < SIZE; ++i)
135	<	assertEquals(ints[i], q.poll());
136	<	}
137	<	finally {}
128	>	/**
129	>	* Queue contains all elements of collection used to initialize
130	>	*/
131	>	public void testConstructor6() {
132	>	Integer[] ints = new Integer[SIZE];
133	>	for (int i = 0; i < SIZE; ++i)
134	>	ints[i] = i;
135	>	PriorityBlockingQueue q = new PriorityBlockingQueue(Arrays.asList(ints));
136	>	for (int i = 0; i < SIZE; ++i)
137	>	assertEquals(ints[i], q.poll());
138		}
139
140	<	public void testConstructor7(){
141	<	try {
142	<	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE, new MyReverseComparator());
143	<	Integer[] ints = new Integer[SIZE];
144	<	for (int i = 0; i < SIZE; ++i)
145	<	ints[i] = new Integer(i);
146	<	q.addAll(Arrays.asList(ints));
147	<	for (int i = SIZE-1; i >= 0; --i)
148	<	assertEquals(ints[i], q.poll());
149	<	}
150	<	finally {}
140	>	/**
141	>	* The comparator used in constructor is used
142	>	*/
143	>	public void testConstructor7() {
144	>	MyReverseComparator cmp = new MyReverseComparator();
145	>	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE, cmp);
146	>	assertEquals(cmp, q.comparator());
147	>	Integer[] ints = new Integer[SIZE];
148	>	for (int i = 0; i < SIZE; ++i)
149	>	ints[i] = new Integer(i);
150	>	q.addAll(Arrays.asList(ints));
151	>	for (int i = SIZE - 1; i >= 0; --i)
152	>	assertEquals(ints[i], q.poll());
153		}
154
155	+	/**
156	+	* isEmpty is true before add, false after
157	+	*/
158		public void testEmpty() {
159		PriorityBlockingQueue q = new PriorityBlockingQueue(2);
160		assertTrue(q.isEmpty());
161	<	assertEquals(NOCAP, q.remainingCapacity());
162	<	q.add(new Integer(1));
161	>	assertEquals(Integer.MAX_VALUE, q.remainingCapacity());
162	>	q.add(one);
163		assertFalse(q.isEmpty());
164	<	q.add(new Integer(2));
164	>	q.add(two);
165		q.remove();
166		q.remove();
167		assertTrue(q.isEmpty());
168		}
169
170	<	public void testRemainingCapacity(){
171	<	PriorityBlockingQueue q = populatedQueue(SIZE);
170	>	/**
171	>	* remainingCapacity() always returns Integer.MAX_VALUE
172	>	*/
173	>	public void testRemainingCapacity() {
174	>	BlockingQueue q = populatedQueue(SIZE);
175		for (int i = 0; i < SIZE; ++i) {
176	<	assertEquals(NOCAP, q.remainingCapacity());
177	<	assertEquals(SIZE-i, q.size());
178	<	q.remove();
176	>	assertEquals(Integer.MAX_VALUE, q.remainingCapacity());
177	>	assertEquals(SIZE - i, q.size());
178	>	assertEquals(i, q.remove());
179		}
180		for (int i = 0; i < SIZE; ++i) {
181	<	assertEquals(NOCAP, q.remainingCapacity());
181	>	assertEquals(Integer.MAX_VALUE, q.remainingCapacity());
182		assertEquals(i, q.size());
183	<	q.add(new Integer(i));
183	>	assertTrue(q.add(i));
184		}
185		}
186
187	<	public void testOfferNull(){
188	<	try {
189	<	PriorityBlockingQueue q = new PriorityBlockingQueue(1);
148	<	q.offer(null);
149	<	fail("should throw NPE");
150	<	} catch (NullPointerException success) { }
151	<	}
152	<
187	>	/**
188	>	* Offer of comparable element succeeds
189	>	*/
190		public void testOffer() {
191		PriorityBlockingQueue q = new PriorityBlockingQueue(1);
192	<	assertTrue(q.offer(new Integer(0)));
193	<	assertTrue(q.offer(new Integer(1)));
192	>	assertTrue(q.offer(zero));
193	>	assertTrue(q.offer(one));
194		}
195
196	+	/**
197	+	* Offer of non-Comparable throws CCE
198	+	*/
199		public void testOfferNonComparable() {
200	+	PriorityBlockingQueue q = new PriorityBlockingQueue(1);
201		try {
161	–	PriorityBlockingQueue q = new PriorityBlockingQueue(1);
162	–	q.offer(new Object());
202		q.offer(new Object());
203	<	q.offer(new Object());
204	<	fail("should throw CCE");
203	>	shouldThrow();
204	>	} catch (ClassCastException success) {
205	>	assertTrue(q.isEmpty());
206	>	assertEquals(0, q.size());
207	>	assertNull(q.poll());
208		}
167	–	catch(ClassCastException success) {}
209		}
210
211	<	public void testAdd(){
211	>	/**
212	>	* add of comparable succeeds
213	>	*/
214	>	public void testAdd() {
215		PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
216		for (int i = 0; i < SIZE; ++i) {
217		assertEquals(i, q.size());
#	Line 175 | Line 219 | public class PriorityBlockingQueueTest e
219		}
220		}
221
222	<	public void testAddAll1(){
222	>	/**
223	>	* addAll(this) throws IAE
224	>	*/
225	>	public void testAddAllSelf() {
226	>	PriorityBlockingQueue q = populatedQueue(SIZE);
227		try {
228	<	PriorityBlockingQueue q = new PriorityBlockingQueue(1);
229	<	q.addAll(null);
230	<	fail("Cannot add null collection");
183	<	}
184	<	catch (NullPointerException success) {}
228	>	q.addAll(q);
229	>	shouldThrow();
230	>	} catch (IllegalArgumentException success) {}
231		}
232	<	public void testAddAll2(){
232	>
233	>	/**
234	>	* addAll of a collection with any null elements throws NPE after
235	>	* possibly adding some elements
236	>	*/
237	>	public void testAddAll3() {
238	>	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
239	>	Integer[] ints = new Integer[SIZE];
240	>	for (int i = 0; i < SIZE - 1; ++i)
241	>	ints[i] = new Integer(i);
242		try {
188	–	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
189	–	Integer[] ints = new Integer[SIZE];
243		q.addAll(Arrays.asList(ints));
244	<	fail("Cannot add null elements");
245	<	}
193	<	catch (NullPointerException success) {}
244	>	shouldThrow();
245	>	} catch (NullPointerException success) {}
246		}
247	<	public void testAddAll3(){
248	<	try {
249	<	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
250	<	Integer[] ints = new Integer[SIZE];
251	<	for (int i = 0; i < SIZE-1; ++i)
252	<	ints[i] = new Integer(i);
253	<	q.addAll(Arrays.asList(ints));
254	<	fail("Cannot add null elements");
255	<	}
256	<	catch (NullPointerException success) {}
247	>
248	>	/**
249	>	* Queue contains all elements of successful addAll
250	>	*/
251	>	public void testAddAll5() {
252	>	Integer[] empty = new Integer[0];
253	>	Integer[] ints = new Integer[SIZE];
254	>	for (int i = SIZE - 1; i >= 0; --i)
255	>	ints[i] = new Integer(i);
256	>	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
257	>	assertFalse(q.addAll(Arrays.asList(empty)));
258	>	assertTrue(q.addAll(Arrays.asList(ints)));
259	>	for (int i = 0; i < SIZE; ++i)
260	>	assertEquals(ints[i], q.poll());
261		}
262
263	<	public void testAddAll5(){
264	<	try {
265	<	Integer[] empty = new Integer[0];
266	<	Integer[] ints = new Integer[SIZE];
267	<	for (int i = SIZE-1; i >= 0; --i)
268	<	ints[i] = new Integer(i);
269	<	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
270	<	assertFalse(q.addAll(Arrays.asList(empty)));
271	<	assertTrue(q.addAll(Arrays.asList(ints)));
216	<	for (int i = 0; i < SIZE; ++i)
217	<	assertEquals(ints[i], q.poll());
218	<	}
219	<	finally {}
220	<	}
221	<
222	<	public void testPutNull() {
223	<	try {
224	<	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
225	<	q.put(null);
226	<	fail("put should throw NPE");
227	<	}
228	<	catch (NullPointerException success){
229	<	}
230	<	}
231	<
232	<	public void testPut() {
233	<	try {
234	<	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
235	<	for (int i = 0; i < SIZE; ++i) {
236	<	Integer I = new Integer(i);
237	<	q.put(I);
238	<	assertTrue(q.contains(I));
239	<	}
240	<	assertEquals(SIZE, q.size());
241	<	}
242	<	finally {
263	>	/**
264	>	* all elements successfully put are contained
265	>	*/
266	>	public void testPut() {
267	>	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE);
268	>	for (int i = 0; i < SIZE; ++i) {
269	>	Integer x = new Integer(i);
270	>	q.put(x);
271	>	assertTrue(q.contains(x));
272		}
273	+	assertEquals(SIZE, q.size());
274		}
275
276	<	public void testPutWithTake() {
276	>	/**
277	>	* put doesn't block waiting for take
278	>	*/
279	>	public void testPutWithTake() throws InterruptedException {
280		final PriorityBlockingQueue q = new PriorityBlockingQueue(2);
281	<	Thread t = new Thread(new Runnable() {
282	<	public void run(){
283	<	int added = 0;
284	<	try {
285	<	q.put(new Integer(0));
286	<	++added;
287	<	q.put(new Integer(0));
288	<	++added;
289	<	q.put(new Integer(0));
290	<	++added;
258	<	q.put(new Integer(0));
259	<	++added;
260	<	threadAssertTrue(added == 4);
261	<	} finally {
262	<	}
263	<	}
264	<	});
265	<	try {
266	<	t.start();
267	<	Thread.sleep(SHORT_DELAY_MS);
268	<	q.take();
269	<	t.interrupt();
270	<	t.join();
271	<	} catch (Exception e){
272	<	fail("Unexpected exception");
273	<	}
281	>	final int size = 4;
282	>	Thread t = newStartedThread(new CheckedRunnable() {
283	>	public void realRun() {
284	>	for (int i = 0; i < size; i++)
285	>	q.put(new Integer(0));
286	>	}});
287	>
288	>	awaitTermination(t);
289	>	assertEquals(size, q.size());
290	>	q.take();
291		}
292
293	<	public void testTimedOffer() {
293	>	/**
294	>	* timed offer does not time out
295	>	*/
296	>	public void testTimedOffer() throws InterruptedException {
297		final PriorityBlockingQueue q = new PriorityBlockingQueue(2);
298	<	Thread t = new Thread(new Runnable() {
299	<	public void run(){
300	<	try {
301	<	q.put(new Integer(0));
302	<	q.put(new Integer(0));
303	<	threadAssertTrue(q.offer(new Integer(0), SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
304	<	threadAssertTrue(q.offer(new Integer(0), LONG_DELAY_MS, TimeUnit.MILLISECONDS));
285	<	} finally { }
286	<	}
287	<	});
288	<
289	<	try {
290	<	t.start();
291	<	Thread.sleep(SMALL_DELAY_MS);
292	<	t.interrupt();
293	<	t.join();
294	<	} catch (Exception e){
295	<	fail("Unexpected exception");
296	<	}
297	<	}
298	>	Thread t = newStartedThread(new CheckedRunnable() {
299	>	public void realRun() {
300	>	q.put(new Integer(0));
301	>	q.put(new Integer(0));
302	>	assertTrue(q.offer(new Integer(0), SHORT_DELAY_MS, MILLISECONDS));
303	>	assertTrue(q.offer(new Integer(0), LONG_DELAY_MS, MILLISECONDS));
304	>	}});
305
306	<	public void testTake(){
300	<	try {
301	<	PriorityBlockingQueue q = populatedQueue(SIZE);
302	<	for (int i = 0; i < SIZE; ++i) {
303	<	assertEquals(i, ((Integer)q.take()).intValue());
304	<	}
305	<	} catch (InterruptedException e){
306	<	fail("Unexpected exception");
307	<	}
306	>	awaitTermination(t);
307		}
308
309	<	public void testTakeFromEmpty() {
310	<	final PriorityBlockingQueue q = new PriorityBlockingQueue(2);
311	<	Thread t = new Thread(new Runnable() {
312	<	public void run(){
313	<	try {
314	<	q.take();
315	<	threadFail("Should block");
317	<	} catch (InterruptedException success){ }
318	<	}
319	<	});
320	<	try {
321	<	t.start();
322	<	Thread.sleep(SHORT_DELAY_MS);
323	<	t.interrupt();
324	<	t.join();
325	<	} catch (Exception e){
326	<	fail("Unexpected exception");
309	>	/**
310	>	* take retrieves elements in priority order
311	>	*/
312	>	public void testTake() throws InterruptedException {
313	>	PriorityBlockingQueue q = populatedQueue(SIZE);
314	>	for (int i = 0; i < SIZE; ++i) {
315	>	assertEquals(i, q.take());
316		}
317		}
318
319	<	public void testBlockingTake(){
320	<	Thread t = new Thread(new Runnable() {
321	<	public void run() {
322	<	try {
323	<	PriorityBlockingQueue q = populatedQueue(SIZE);
324	<	for (int i = 0; i < SIZE; ++i) {
325	<	threadAssertEquals(i, ((Integer)q.take()).intValue());
326	<	}
327	<	q.take();
328	<	threadFail("take should block");
329	<	} catch (InterruptedException success){
341	<	}
342	<	}});
343	<	t.start();
344	<	try {
345	<	Thread.sleep(SHORT_DELAY_MS);
346	<	t.interrupt();
347	<	t.join();
348	<	}
349	<	catch (InterruptedException ie) {
350	<	fail("Unexpected exception");
351	<	}
352	<	}
319	>	/**
320	>	* Take removes existing elements until empty, then blocks interruptibly
321	>	*/
322	>	public void testBlockingTake() throws InterruptedException {
323	>	final PriorityBlockingQueue q = populatedQueue(SIZE);
324	>	final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
325	>	Thread t = newStartedThread(new CheckedRunnable() {
326	>	public void realRun() throws InterruptedException {
327	>	for (int i = 0; i < SIZE; ++i) {
328	>	assertEquals(i, q.take());
329	>	}
330
331	+	Thread.currentThread().interrupt();
332	+	try {
333	+	q.take();
334	+	shouldThrow();
335	+	} catch (InterruptedException success) {}
336	+	assertFalse(Thread.interrupted());
337	+
338	+	pleaseInterrupt.countDown();
339	+	try {
340	+	q.take();
341	+	shouldThrow();
342	+	} catch (InterruptedException success) {}
343	+	assertFalse(Thread.interrupted());
344	+	}});
345	+
346	+	await(pleaseInterrupt);
347	+	assertThreadStaysAlive(t);
348	+	t.interrupt();
349	+	awaitTermination(t);
350	+	}
351
352	<	public void testPoll(){
352	>	/**
353	>	* poll succeeds unless empty
354	>	*/
355	>	public void testPoll() {
356		PriorityBlockingQueue q = populatedQueue(SIZE);
357		for (int i = 0; i < SIZE; ++i) {
358	<	assertEquals(i, ((Integer)q.poll()).intValue());
358	>	assertEquals(i, q.poll());
359		}
360	<	assertNull(q.poll());
360	>	assertNull(q.poll());
361		}
362
363	<	public void testTimedPoll0() {
364	<	try {
365	<	PriorityBlockingQueue q = populatedQueue(SIZE);
366	<	for (int i = 0; i < SIZE; ++i) {
367	<	assertEquals(i, ((Integer)q.poll(0, TimeUnit.MILLISECONDS)).intValue());
368	<	}
369	<	assertNull(q.poll(0, TimeUnit.MILLISECONDS));
370	<	} catch (InterruptedException e){
371	<	fail("Unexpected exception");
372	<	}
363	>	/**
364	>	* timed poll with zero timeout succeeds when non-empty, else times out
365	>	*/
366	>	public void testTimedPoll0() throws InterruptedException {
367	>	PriorityBlockingQueue q = populatedQueue(SIZE);
368	>	for (int i = 0; i < SIZE; ++i) {
369	>	assertEquals(i, q.poll(0, MILLISECONDS));
370	>	}
371	>	assertNull(q.poll(0, MILLISECONDS));
372		}
373
374	<	public void testTimedPoll() {
375	<	try {
376	<	PriorityBlockingQueue q = populatedQueue(SIZE);
377	<	for (int i = 0; i < SIZE; ++i) {
378	<	assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
379	<	}
380	<	assertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
381	<	} catch (InterruptedException e){
382	<	fail("Unexpected exception");
383	<	}
384	<	}
385	<
386	<	public void testInterruptedTimedPoll(){
387	<	Thread t = new Thread(new Runnable() {
389	<	public void run() {
390	<	try {
391	<	PriorityBlockingQueue q = populatedQueue(SIZE);
392	<	for (int i = 0; i < SIZE; ++i) {
393	<	threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
394	<	}
395	<	threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
396	<	} catch (InterruptedException success){
397	<	}
398	<	}});
399	<	t.start();
400	<	try {
401	<	Thread.sleep(SHORT_DELAY_MS);
402	<	t.interrupt();
403	<	t.join();
404	<	}
405	<	catch (InterruptedException ie) {
406	<	fail("Unexpected exception");
407	<	}
374	>	/**
375	>	* timed poll with nonzero timeout succeeds when non-empty, else times out
376	>	*/
377	>	public void testTimedPoll() throws InterruptedException {
378	>	PriorityBlockingQueue<Integer> q = populatedQueue(SIZE);
379	>	for (int i = 0; i < SIZE; ++i) {
380	>	long startTime = System.nanoTime();
381	>	assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
382	>	assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
383	>	}
384	>	long startTime = System.nanoTime();
385	>	assertNull(q.poll(timeoutMillis(), MILLISECONDS));
386	>	assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
387	>	checkEmpty(q);
388		}
389
390	<	public void testTimedPollWithOffer(){
391	<	final PriorityBlockingQueue q = new PriorityBlockingQueue(2);
392	<	Thread t = new Thread(new Runnable() {
393	<	public void run(){
394	<	try {
395	<	threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
396	<	q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
397	<	q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
398	<	threadFail("Should block");
399	<	} catch (InterruptedException success) { }
390	>	/**
391	>	* Interrupted timed poll throws InterruptedException instead of
392	>	* returning timeout status
393	>	*/
394	>	public void testInterruptedTimedPoll() throws InterruptedException {
395	>	final BlockingQueue<Integer> q = populatedQueue(SIZE);
396	>	final CountDownLatch aboutToWait = new CountDownLatch(1);
397	>	Thread t = newStartedThread(new CheckedRunnable() {
398	>	public void realRun() throws InterruptedException {
399	>	long startTime = System.nanoTime();
400	>	for (int i = 0; i < SIZE; ++i) {
401	>	assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
402		}
403	<	});
404	<	try {
405	<	t.start();
406	<	Thread.sleep(SMALL_DELAY_MS);
407	<	assertTrue(q.offer(new Integer(0), SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
408	<	t.interrupt();
409	<	t.join();
410	<	} catch (Exception e){
429	<	fail("Unexpected exception");
430	<	}
431	<	}
403	>	aboutToWait.countDown();
404	>	try {
405	>	q.poll(LONG_DELAY_MS, MILLISECONDS);
406	>	shouldThrow();
407	>	} catch (InterruptedException success) {
408	>	assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
409	>	}
410	>	}});
411
412	+	aboutToWait.await();
413	+	waitForThreadToEnterWaitState(t, LONG_DELAY_MS);
414	+	t.interrupt();
415	+	awaitTermination(t);
416	+	}
417
418	<	public void testPeek(){
418	>	/**
419	>	* peek returns next element, or null if empty
420	>	*/
421	>	public void testPeek() {
422		PriorityBlockingQueue q = populatedQueue(SIZE);
423		for (int i = 0; i < SIZE; ++i) {
424	<	assertEquals(i, ((Integer)q.peek()).intValue());
425	<	q.poll();
424	>	assertEquals(i, q.peek());
425	>	assertEquals(i, q.poll());
426		assertTrue(q.peek() == null ||
427	<	i != ((Integer)q.peek()).intValue());
427	>	!q.peek().equals(i));
428		}
429	<	assertNull(q.peek());
429	>	assertNull(q.peek());
430		}
431
432	<	public void testElement(){
432	>	/**
433	>	* element returns next element, or throws NSEE if empty
434	>	*/
435	>	public void testElement() {
436		PriorityBlockingQueue q = populatedQueue(SIZE);
437		for (int i = 0; i < SIZE; ++i) {
438	<	assertEquals(i, ((Integer)q.element()).intValue());
439	<	q.poll();
438	>	assertEquals(i, q.element());
439	>	assertEquals(i, q.poll());
440		}
441		try {
442		q.element();
443	<	fail("no such element");
444	<	}
455	<	catch (NoSuchElementException success) {}
443	>	shouldThrow();
444	>	} catch (NoSuchElementException success) {}
445		}
446
447	<	public void testRemove(){
447	>	/**
448	>	* remove removes next element, or throws NSEE if empty
449	>	*/
450	>	public void testRemove() {
451		PriorityBlockingQueue q = populatedQueue(SIZE);
452		for (int i = 0; i < SIZE; ++i) {
453	<	assertEquals(i, ((Integer)q.remove()).intValue());
453	>	assertEquals(i, q.remove());
454		}
455		try {
456		q.remove();
457	<	fail("remove should throw");
458	<	} catch (NoSuchElementException success){
467	<	}
457	>	shouldThrow();
458	>	} catch (NoSuchElementException success) {}
459		}
460
461	<	public void testRemoveElement(){
462	<	PriorityBlockingQueue q = populatedQueue(SIZE);
463	<	for (int i = 1; i < SIZE; i+=2) {
464	<	assertTrue(q.remove(new Integer(i)));
474	<	}
475	<	for (int i = 0; i < SIZE; i+=2) {
476	<	assertTrue(q.remove(new Integer(i)));
477	<	assertFalse(q.remove(new Integer(i+1)));
478	<	}
479	<	assertTrue(q.isEmpty());
480	<	}
481	<
482	<	public void testContains(){
461	>	/**
462	>	* contains(x) reports true when elements added but not yet removed
463	>	*/
464	>	public void testContains() {
465		PriorityBlockingQueue q = populatedQueue(SIZE);
466		for (int i = 0; i < SIZE; ++i) {
467		assertTrue(q.contains(new Integer(i)));
#	Line 488 | Line 470 | public class PriorityBlockingQueueTest e
470		}
471		}
472
473	<	public void testClear(){
473	>	/**
474	>	* clear removes all elements
475	>	*/
476	>	public void testClear() {
477		PriorityBlockingQueue q = populatedQueue(SIZE);
478		q.clear();
479		assertTrue(q.isEmpty());
480		assertEquals(0, q.size());
481	<	assertEquals(NOCAP, q.remainingCapacity());
497	<	q.add(new Integer(1));
481	>	q.add(one);
482		assertFalse(q.isEmpty());
483	+	assertTrue(q.contains(one));
484		q.clear();
485		assertTrue(q.isEmpty());
486		}
487
488	<	public void testContainsAll(){
488	>	/**
489	>	* containsAll(c) is true when c contains a subset of elements
490	>	*/
491	>	public void testContainsAll() {
492		PriorityBlockingQueue q = populatedQueue(SIZE);
493		PriorityBlockingQueue p = new PriorityBlockingQueue(SIZE);
494		for (int i = 0; i < SIZE; ++i) {
#	Line 511 | Line 499 | public class PriorityBlockingQueueTest e
499		assertTrue(p.containsAll(q));
500		}
501
502	<	public void testRetainAll(){
502	>	/**
503	>	* retainAll(c) retains only those elements of c and reports true if changed
504	>	*/
505	>	public void testRetainAll() {
506		PriorityBlockingQueue q = populatedQueue(SIZE);
507		PriorityBlockingQueue p = populatedQueue(SIZE);
508		for (int i = 0; i < SIZE; ++i) {
#	Line 522 | Line 513 | public class PriorityBlockingQueueTest e
513		assertTrue(changed);
514
515		assertTrue(q.containsAll(p));
516	<	assertEquals(SIZE-i, q.size());
516	>	assertEquals(SIZE - i, q.size());
517		p.remove();
518		}
519		}
520
521	<	public void testRemoveAll(){
521	>	/**
522	>	* removeAll(c) removes only those elements of c and reports true if changed
523	>	*/
524	>	public void testRemoveAll() {
525		for (int i = 1; i < SIZE; ++i) {
526		PriorityBlockingQueue q = populatedQueue(SIZE);
527		PriorityBlockingQueue p = populatedQueue(i);
528		assertTrue(q.removeAll(p));
529	<	assertEquals(SIZE-i, q.size());
529	>	assertEquals(SIZE - i, q.size());
530		for (int j = 0; j < i; ++j) {
531	<	Integer I = (Integer)(p.remove());
532	<	assertFalse(q.contains(I));
531	>	Integer x = (Integer)(p.remove());
532	>	assertFalse(q.contains(x));
533		}
534		}
535		}
536
537	<	public void testToArray(){
537	>	/**
538	>	* toArray contains all elements
539	>	*/
540	>	public void testToArray() throws InterruptedException {
541		PriorityBlockingQueue q = populatedQueue(SIZE);
542	<	Object[] o = q.toArray();
542	>	Object[] o = q.toArray();
543		Arrays.sort(o);
544	<	try {
545	<	for(int i = 0; i < o.length; i++)
549	<	assertEquals(o[i], q.take());
550	<	} catch (InterruptedException e){
551	<	fail("Unexpected exception");
552	<	}
544	>	for (int i = 0; i < o.length; i++)
545	>	assertSame(o[i], q.take());
546		}
547
548	<	public void testToArray2(){
549	<	PriorityBlockingQueue q = populatedQueue(SIZE);
550	<	Integer[] ints = new Integer[SIZE];
551	<	ints = (Integer[])q.toArray(ints);
548	>	/**
549	>	* toArray(a) contains all elements
550	>	*/
551	>	public void testToArray2() throws InterruptedException {
552	>	PriorityBlockingQueue<Integer> q = populatedQueue(SIZE);
553	>	Integer[] ints = new Integer[SIZE];
554	>	Integer[] array = q.toArray(ints);
555	>	assertSame(ints, array);
556		Arrays.sort(ints);
557	<	try {
558	<	for(int i = 0; i < ints.length; i++)
559	<	assertEquals(ints[i], q.take());
560	<	} catch (InterruptedException e){
561	<	fail("Unexpected exception");
562	<	}
563	<	}
564	<
565	<	public void testIterator(){
566	<	PriorityBlockingQueue q = populatedQueue(SIZE);
567	<	int i = 0;
568	<	Iterator it = q.iterator();
569	<	while(it.hasNext()) {
557	>	for (int i = 0; i < ints.length; i++)
558	>	assertSame(ints[i], q.take());
559	>	}
560	>
561	>	/**
562	>	* toArray(incompatible array type) throws ArrayStoreException
563	>	*/
564	>	public void testToArray1_BadArg() {
565	>	PriorityBlockingQueue q = populatedQueue(SIZE);
566	>	try {
567	>	q.toArray(new String[10]);
568	>	shouldThrow();
569	>	} catch (ArrayStoreException success) {}
570	>	}
571	>
572	>	/**
573	>	* iterator iterates through all elements
574	>	*/
575	>	public void testIterator() {
576	>	PriorityBlockingQueue q = populatedQueue(SIZE);
577	>	Iterator it = q.iterator();
578	>	int i;
579	>	for (i = 0; it.hasNext(); i++)
580		assertTrue(q.contains(it.next()));
574	–	++i;
575	–	}
581		assertEquals(i, SIZE);
582	+	assertIteratorExhausted(it);
583		}
584
585	<	public void testIteratorRemove () {
585	>	/**
586	>	* iterator of empty collection has no elements
587	>	*/
588	>	public void testEmptyIterator() {
589	>	assertIteratorExhausted(new PriorityBlockingQueue().iterator());
590	>	}
591
592	+	/**
593	+	* iterator.remove removes current element
594	+	*/
595	+	public void testIteratorRemove() {
596		final PriorityBlockingQueue q = new PriorityBlockingQueue(3);
582	–
597		q.add(new Integer(2));
598		q.add(new Integer(1));
599		q.add(new Integer(3));
#	Line 594 | Line 608 | public class PriorityBlockingQueueTest e
608		assertFalse(it.hasNext());
609		}
610
611	<
612	<	public void testToString(){
611	>	/**
612	>	* toString contains toStrings of elements
613	>	*/
614	>	public void testToString() {
615		PriorityBlockingQueue q = populatedQueue(SIZE);
616		String s = q.toString();
617		for (int i = 0; i < SIZE; ++i) {
618	<	assertTrue(s.indexOf(String.valueOf(i)) >= 0);
618	>	assertTrue(s.contains(String.valueOf(i)));
619		}
620	<	}
620	>	}
621
622	+	/**
623	+	* timed poll transfers elements across Executor tasks
624	+	*/
625		public void testPollInExecutor() {
607	–
626		final PriorityBlockingQueue q = new PriorityBlockingQueue(2);
627	+	final CheckedBarrier threadsStarted = new CheckedBarrier(2);
628	+	final ExecutorService executor = Executors.newFixedThreadPool(2);
629	+	try (PoolCleaner cleaner = cleaner(executor)) {
630	+	executor.execute(new CheckedRunnable() {
631	+	public void realRun() throws InterruptedException {
632	+	assertNull(q.poll());
633	+	threadsStarted.await();
634	+	assertSame(one, q.poll(LONG_DELAY_MS, MILLISECONDS));
635	+	checkEmpty(q);
636	+	}});
637
638	<	ExecutorService executor = Executors.newFixedThreadPool(2);
638	>	executor.execute(new CheckedRunnable() {
639	>	public void realRun() throws InterruptedException {
640	>	threadsStarted.await();
641	>	q.put(one);
642	>	}});
643	>	}
644	>	}
645
646	<	executor.execute(new Runnable() {
647	<	public void run() {
648	<	threadAssertNull(q.poll());
649	<	try {
650	<	threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
651	<	threadAssertTrue(q.isEmpty());
652	<	}
653	<	catch (InterruptedException e) {
654	<	threadFail("should not be interrupted");
655	<	}
656	<	}
657	<	});
646	>	/**
647	>	* A deserialized serialized queue has same elements
648	>	*/
649	>	public void testSerialization() throws Exception {
650	>	Queue x = populatedQueue(SIZE);
651	>	Queue y = serialClone(x);
652	>
653	>	assertNotSame(x, y);
654	>	assertEquals(x.size(), y.size());
655	>	while (!x.isEmpty()) {
656	>	assertFalse(y.isEmpty());
657	>	assertEquals(x.remove(), y.remove());
658	>	}
659	>	assertTrue(y.isEmpty());
660	>	}
661
662	<	executor.execute(new Runnable() {
663	<	public void run() {
664	<	try {
665	<	Thread.sleep(SMALL_DELAY_MS);
666	<	q.put(new Integer(1));
667	<	}
668	<	catch (InterruptedException e) {
669	<	threadFail("should not be interrupted");
670	<	}
671	<	}
672	<	});
673	<
674	<	joinPool(executor);
662	>	/**
663	>	* drainTo(c) empties queue into another collection c
664	>	*/
665	>	public void testDrainTo() {
666	>	PriorityBlockingQueue q = populatedQueue(SIZE);
667	>	ArrayList l = new ArrayList();
668	>	q.drainTo(l);
669	>	assertEquals(0, q.size());
670	>	assertEquals(SIZE, l.size());
671	>	for (int i = 0; i < SIZE; ++i)
672	>	assertEquals(l.get(i), new Integer(i));
673	>	q.add(zero);
674	>	q.add(one);
675	>	assertFalse(q.isEmpty());
676	>	assertTrue(q.contains(zero));
677	>	assertTrue(q.contains(one));
678	>	l.clear();
679	>	q.drainTo(l);
680	>	assertEquals(0, q.size());
681	>	assertEquals(2, l.size());
682	>	for (int i = 0; i < 2; ++i)
683	>	assertEquals(l.get(i), new Integer(i));
684	>	}
685	>
686	>	/**
687	>	* drainTo empties queue
688	>	*/
689	>	public void testDrainToWithActivePut() throws InterruptedException {
690	>	final PriorityBlockingQueue q = populatedQueue(SIZE);
691	>	Thread t = new Thread(new CheckedRunnable() {
692	>	public void realRun() {
693	>	q.put(new Integer(SIZE + 1));
694	>	}});
695
696	+	t.start();
697	+	ArrayList l = new ArrayList();
698	+	q.drainTo(l);
699	+	assertTrue(l.size() >= SIZE);
700	+	for (int i = 0; i < SIZE; ++i)
701	+	assertEquals(l.get(i), new Integer(i));
702	+	t.join();
703	+	assertTrue(q.size() + l.size() >= SIZE);
704		}
705
706	<	public void testSerialization() {
707	<	PriorityBlockingQueue q = populatedQueue(SIZE);
708	<	try {
709	<	ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
710	<	ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
711	<	out.writeObject(q);
712	<	out.close();
713	<
714	<	ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
715	<	ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
716	<	PriorityBlockingQueue r = (PriorityBlockingQueue)in.readObject();
717	<	assertEquals(q.size(), r.size());
718	<	while (!q.isEmpty())
719	<	assertEquals(q.remove(), r.remove());
720	<	} catch(Exception e){
721	<	fail("unexpected exception");
706	>	/**
707	>	* drainTo(c, n) empties first min(n, size) elements of queue into c
708	>	*/
709	>	public void testDrainToN() {
710	>	PriorityBlockingQueue q = new PriorityBlockingQueue(SIZE * 2);
711	>	for (int i = 0; i < SIZE + 2; ++i) {
712	>	for (int j = 0; j < SIZE; j++)
713	>	assertTrue(q.offer(new Integer(j)));
714	>	ArrayList l = new ArrayList();
715	>	q.drainTo(l, i);
716	>	int k = (i < SIZE) ? i : SIZE;
717	>	assertEquals(k, l.size());
718	>	assertEquals(SIZE - k, q.size());
719	>	for (int j = 0; j < k; ++j)
720	>	assertEquals(l.get(j), new Integer(j));
721	>	do {} while (q.poll() != null);
722	>	}
723	>	}
724	>
725	>	/**
726	>	* remove(null), contains(null) always return false
727	>	*/
728	>	public void testNeverContainsNull() {
729	>	Collection<?>[] qs = {
730	>	new PriorityBlockingQueue<Object>(),
731	>	populatedQueue(2),
732	>	};
733	>
734	>	for (Collection<?> q : qs) {
735	>	assertFalse(q.contains(null));
736	>	assertFalse(q.remove(null));
737		}
738		}
739

Diff Legend

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