ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/PriorityBlockingQueueTest.java
Revision: 1.40
Committed: Sun Nov 28 08:43:53 2010 UTC (13 years, 5 months ago) by jsr166
Branch: MAIN
Changes since 1.39: +15 -8 lines
Log Message:
update testInterruptedTimedPoll

File Contents

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