ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/ArrayBlockingQueueTest.java
(Generate patch)

Comparing jsr166/src/test/tck/ArrayBlockingQueueTest.java (file contents):
Revision 1.5 by dl, Thu Sep 25 11:02:41 2003 UTC vs.
Revision 1.23 by jsr166, Sat Nov 21 21:00:34 2009 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/licenses/publicdomain
5 > * Other contributors include Andrew Wright, Jeffrey Hayes,
6 > * Pat Fisher, Mike Judd.
7   */
8  
9  
10   import junit.framework.*;
11   import java.util.*;
12   import java.util.concurrent.*;
13 + import static java.util.concurrent.TimeUnit.MILLISECONDS;
14   import java.io.*;
15  
16   public class ArrayBlockingQueueTest extends JSR166TestCase {
17      public static void main(String[] args) {
18 <        junit.textui.TestRunner.run (suite());  
18 >        junit.textui.TestRunner.run (suite());
19      }
20      public static Test suite() {
21 <        return new TestSuite(ArrayBlockingQueueTest.class);
21 >        return new TestSuite(ArrayBlockingQueueTest.class);
22      }
23  
24      /**
# Line 26 | Line 28 | public class ArrayBlockingQueueTest exte
28      private ArrayBlockingQueue populatedQueue(int n) {
29          ArrayBlockingQueue q = new ArrayBlockingQueue(n);
30          assertTrue(q.isEmpty());
31 <        for(int i = 0; i < n; i++)
32 <            assertTrue(q.offer(new Integer(i)));
31 >        for (int i = 0; i < n; i++)
32 >            assertTrue(q.offer(new Integer(i)));
33          assertFalse(q.isEmpty());
34          assertEquals(0, q.remainingCapacity());
35 <        assertEquals(n, q.size());
35 >        assertEquals(n, q.size());
36          return q;
37      }
38 <
38 >
39      /**
40       * A new queue has the indicated capacity
41       */
# Line 42 | Line 44 | public class ArrayBlockingQueueTest exte
44      }
45  
46      /**
47 <     * Constructor throws IAE if  capacity argument nonpositive
47 >     * Constructor throws IAE if capacity argument nonpositive
48       */
49      public void testConstructor2() {
50          try {
51              ArrayBlockingQueue q = new ArrayBlockingQueue(0);
52              shouldThrow();
53 <        }
52 <        catch (IllegalArgumentException success) {}
53 >        } catch (IllegalArgumentException success) {}
54      }
55  
56      /**
# Line 59 | Line 60 | public class ArrayBlockingQueueTest exte
60          try {
61              ArrayBlockingQueue q = new ArrayBlockingQueue(1, true, null);
62              shouldThrow();
63 <        }
63 <        catch (NullPointerException success) {}
63 >        } catch (NullPointerException success) {}
64      }
65  
66      /**
# Line 71 | Line 71 | public class ArrayBlockingQueueTest exte
71              Integer[] ints = new Integer[SIZE];
72              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, false, Arrays.asList(ints));
73              shouldThrow();
74 <        }
75 <        catch (NullPointerException success) {}
74 >        } catch (NullPointerException success) {}
75      }
76  
77      /**
# Line 85 | Line 84 | public class ArrayBlockingQueueTest exte
84                  ints[i] = new Integer(i);
85              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, false, Arrays.asList(ints));
86              shouldThrow();
87 <        }
89 <        catch (NullPointerException success) {}
87 >        } catch (NullPointerException success) {}
88      }
89  
90      /**
# Line 99 | Line 97 | public class ArrayBlockingQueueTest exte
97                  ints[i] = new Integer(i);
98              ArrayBlockingQueue q = new ArrayBlockingQueue(1, false, Arrays.asList(ints));
99              shouldThrow();
100 <        }
103 <        catch (IllegalArgumentException success) {}
100 >        } catch (IllegalArgumentException success) {}
101      }
102  
103      /**
104       * Queue contains all elements of collection used to initialize
105       */
106      public void testConstructor7() {
107 <        try {
108 <            Integer[] ints = new Integer[SIZE];
109 <            for (int i = 0; i < SIZE; ++i)
110 <                ints[i] = new Integer(i);
111 <            ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, true, Arrays.asList(ints));
112 <            for (int i = 0; i < SIZE; ++i)
116 <                assertEquals(ints[i], q.poll());
117 <        }
118 <        finally {}
107 >        Integer[] ints = new Integer[SIZE];
108 >        for (int i = 0; i < SIZE; ++i)
109 >            ints[i] = new Integer(i);
110 >        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, true, Arrays.asList(ints));
111 >        for (int i = 0; i < SIZE; ++i)
112 >            assertEquals(ints[i], q.poll());
113      }
114  
115      /**
# Line 154 | Line 148 | public class ArrayBlockingQueueTest exte
148       *  offer(null) throws NPE
149       */
150      public void testOfferNull() {
151 <        try {
151 >        try {
152              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
153              q.offer(null);
154              shouldThrow();
155 <        } catch (NullPointerException success) { }  
155 >        } catch (NullPointerException success) {}
156 >    }
157 >
158 >    /**
159 >     *  add(null) throws NPE
160 >     */
161 >    public void testAddNull() {
162 >        try {
163 >            ArrayBlockingQueue q = new ArrayBlockingQueue(1);
164 >            q.add(null);
165 >            shouldThrow();
166 >        } catch (NullPointerException success) {}
167      }
168  
169      /**
# Line 174 | Line 179 | public class ArrayBlockingQueueTest exte
179       * add succeeds if not full; throws ISE if full
180       */
181      public void testAdd() {
182 <        try {
182 >        try {
183              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
184              for (int i = 0; i < SIZE; ++i) {
185                  assertTrue(q.add(new Integer(i)));
186              }
187              assertEquals(0, q.remainingCapacity());
188              q.add(new Integer(SIZE));
189 <        } catch (IllegalStateException success){
190 <        }  
189 >            shouldThrow();
190 >        } catch (IllegalStateException success) {}
191      }
192  
193      /**
# Line 193 | Line 198 | public class ArrayBlockingQueueTest exte
198              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
199              q.addAll(null);
200              shouldThrow();
201 <        }
197 <        catch (NullPointerException success) {}
201 >        } catch (NullPointerException success) {}
202      }
203 +
204 +    /**
205 +     * addAll(this) throws IAE
206 +     */
207 +    public void testAddAllSelf() {
208 +        try {
209 +            ArrayBlockingQueue q = populatedQueue(SIZE);
210 +            q.addAll(q);
211 +            shouldThrow();
212 +        } catch (IllegalArgumentException success) {}
213 +    }
214 +
215 +
216      /**
217       *  addAll of a collection with null elements throws NPE
218       */
# Line 205 | Line 222 | public class ArrayBlockingQueueTest exte
222              Integer[] ints = new Integer[SIZE];
223              q.addAll(Arrays.asList(ints));
224              shouldThrow();
225 <        }
209 <        catch (NullPointerException success) {}
225 >        } catch (NullPointerException success) {}
226      }
227      /**
228       * addAll of a collection with any null elements throws NPE after
# Line 220 | Line 236 | public class ArrayBlockingQueueTest exte
236                  ints[i] = new Integer(i);
237              q.addAll(Arrays.asList(ints));
238              shouldThrow();
239 <        }
224 <        catch (NullPointerException success) {}
239 >        } catch (NullPointerException success) {}
240      }
241      /**
242       * addAll throws ISE if not enough room
# Line 234 | Line 249 | public class ArrayBlockingQueueTest exte
249                  ints[i] = new Integer(i);
250              q.addAll(Arrays.asList(ints));
251              shouldThrow();
252 <        }
238 <        catch (IllegalStateException success) {}
252 >        } catch (IllegalStateException success) {}
253      }
254      /**
255       * Queue contains all elements, in traversal order, of successful addAll
256       */
257      public void testAddAll5() {
258 <        try {
259 <            Integer[] empty = new Integer[0];
260 <            Integer[] ints = new Integer[SIZE];
261 <            for (int i = 0; i < SIZE; ++i)
262 <                ints[i] = new Integer(i);
263 <            ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
264 <            assertFalse(q.addAll(Arrays.asList(empty)));
265 <            assertTrue(q.addAll(Arrays.asList(ints)));
266 <            for (int i = 0; i < SIZE; ++i)
253 <                assertEquals(ints[i], q.poll());
254 <        }
255 <        finally {}
258 >        Integer[] empty = new Integer[0];
259 >        Integer[] ints = new Integer[SIZE];
260 >        for (int i = 0; i < SIZE; ++i)
261 >            ints[i] = new Integer(i);
262 >        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
263 >        assertFalse(q.addAll(Arrays.asList(empty)));
264 >        assertTrue(q.addAll(Arrays.asList(ints)));
265 >        for (int i = 0; i < SIZE; ++i)
266 >            assertEquals(ints[i], q.poll());
267      }
268  
269      /**
270       *  put(null) throws NPE
271       */
272 <     public void testPutNull() {
273 <        try {
272 >    public void testPutNull() throws InterruptedException {
273 >        try {
274              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
275              q.put(null);
276              shouldThrow();
277 <        }
267 <        catch (NullPointerException success){
268 <        }  
269 <        catch (InterruptedException ie) {
270 <            unexpectedException();
271 <        }
277 >        } catch (NullPointerException success) {}
278       }
279  
280      /**
281       * all elements successfully put are contained
282       */
283 <     public void testPut() {
284 <         try {
285 <             ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
286 <             for (int i = 0; i < SIZE; ++i) {
287 <                 Integer I = new Integer(i);
288 <                 q.put(I);
283 <                 assertTrue(q.contains(I));
284 <             }
285 <             assertEquals(0, q.remainingCapacity());
286 <         }
287 <        catch (InterruptedException ie) {
288 <            unexpectedException();
283 >    public void testPut() throws InterruptedException {
284 >        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
285 >        for (int i = 0; i < SIZE; ++i) {
286 >            Integer I = new Integer(i);
287 >            q.put(I);
288 >            assertTrue(q.contains(I));
289          }
290 +        assertEquals(0, q.remainingCapacity());
291      }
292  
293      /**
294       * put blocks interruptibly if full
295       */
296 <    public void testBlockingPut() {
297 <        Thread t = new Thread(new Runnable() {
298 <                public void run() {
299 <                    int added = 0;
300 <                    try {
301 <                        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
302 <                        for (int i = 0; i < SIZE; ++i) {
303 <                            q.put(new Integer(i));
304 <                            ++added;
305 <                        }
306 <                        q.put(new Integer(SIZE));
307 <                        threadShouldThrow();
308 <                    } catch (InterruptedException ie){
309 <                        threadAssertEquals(added, SIZE);
310 <                    }  
311 <                }});
312 <        try {
313 <            t.start();
314 <           Thread.sleep(SHORT_DELAY_MS);
315 <           t.interrupt();
315 <           t.join();
316 <        }
317 <        catch (InterruptedException ie) {
318 <            unexpectedException();
319 <        }
296 >    public void testBlockingPut() throws InterruptedException {
297 >        final ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
298 >        Thread t = new Thread(new CheckedRunnable() {
299 >            public void realRun() {
300 >                int added = 0;
301 >                try {
302 >                    for (int i = 0; i < SIZE; ++i) {
303 >                        q.put(new Integer(i));
304 >                        ++added;
305 >                    }
306 >                    q.put(new Integer(SIZE));
307 >                    threadShouldThrow();
308 >                } catch (InterruptedException success) {
309 >                    threadAssertEquals(added, SIZE);
310 >                }}});
311 >
312 >        t.start();
313 >        Thread.sleep(MEDIUM_DELAY_MS);
314 >        t.interrupt();
315 >        t.join();
316      }
317  
318      /**
319       * put blocks waiting for take when full
320       */
321 <    public void testPutWithTake() {
321 >    public void testPutWithTake() throws InterruptedException {
322          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
323 <        Thread t = new Thread(new Runnable() {
324 <                public void run() {
325 <                    int added = 0;
326 <                    try {
327 <                        q.put(new Object());
328 <                        ++added;
329 <                        q.put(new Object());
330 <                        ++added;
331 <                        q.put(new Object());
332 <                        ++added;
333 <                        q.put(new Object());
334 <                        ++added;
335 <                        threadShouldThrow();
336 <                    } catch (InterruptedException e){
337 <                        threadAssertTrue(added >= 2);
338 <                    }
339 <                }
340 <            });
341 <        try {
342 <            t.start();
343 <            Thread.sleep(SHORT_DELAY_MS);
344 <            q.take();
345 <            t.interrupt();
350 <            t.join();
351 <        } catch (Exception e){
352 <            unexpectedException();
353 <        }
323 >        Thread t = new Thread(new CheckedRunnable() {
324 >            public void realRun() {
325 >                int added = 0;
326 >                try {
327 >                    q.put(new Object());
328 >                    ++added;
329 >                    q.put(new Object());
330 >                    ++added;
331 >                    q.put(new Object());
332 >                    ++added;
333 >                    q.put(new Object());
334 >                    ++added;
335 >                    threadShouldThrow();
336 >                } catch (InterruptedException success) {
337 >                    threadAssertTrue(added >= 2);
338 >                }
339 >            }});
340 >
341 >        t.start();
342 >        Thread.sleep(SHORT_DELAY_MS);
343 >        q.take();
344 >        t.interrupt();
345 >        t.join();
346      }
347  
348      /**
349       * timed offer times out if full and elements not taken
350       */
351 <    public void testTimedOffer() {
351 >    public void testTimedOffer() throws InterruptedException {
352          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
353 <        Thread t = new Thread(new Runnable() {
354 <                public void run() {
355 <                    try {
356 <                        q.put(new Object());
357 <                        q.put(new Object());
358 <                        threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS/2, TimeUnit.MILLISECONDS));
359 <                        q.offer(new Object(), LONG_DELAY_MS, TimeUnit.MILLISECONDS);
360 <                        threadShouldThrow();
361 <                    } catch (InterruptedException success){}
362 <                }
363 <            });
364 <        
373 <        try {
374 <            t.start();
375 <            Thread.sleep(SHORT_DELAY_MS);
376 <            t.interrupt();
377 <            t.join();
378 <        } catch (Exception e){
379 <            unexpectedException();
380 <        }
353 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
354 >            public void realRun() throws InterruptedException {
355 >                q.put(new Object());
356 >                q.put(new Object());
357 >                threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS/2, MILLISECONDS));
358 >                q.offer(new Object(), LONG_DELAY_MS, MILLISECONDS);
359 >            }};
360 >
361 >        t.start();
362 >        Thread.sleep(SHORT_DELAY_MS);
363 >        t.interrupt();
364 >        t.join();
365      }
366  
367      /**
368       * take retrieves elements in FIFO order
369       */
370 <    public void testTake() {
371 <        try {
372 <            ArrayBlockingQueue q = populatedQueue(SIZE);
373 <            for (int i = 0; i < SIZE; ++i) {
374 <                assertEquals(i, ((Integer)q.take()).intValue());
391 <            }
392 <        } catch (InterruptedException e){
393 <            unexpectedException();
394 <        }  
370 >    public void testTake() throws InterruptedException {
371 >        ArrayBlockingQueue q = populatedQueue(SIZE);
372 >        for (int i = 0; i < SIZE; ++i) {
373 >            assertEquals(i, ((Integer)q.take()).intValue());
374 >        }
375      }
376  
377      /**
378       * take blocks interruptibly when empty
379       */
380 <    public void testTakeFromEmpty() {
380 >    public void testTakeFromEmpty() throws InterruptedException {
381          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
382 <        Thread t = new Thread(new Runnable() {
383 <                public void run() {
384 <                    try {
385 <                        q.take();
386 <                        threadShouldThrow();
387 <                    } catch (InterruptedException success){ }                
388 <                }
389 <            });
390 <        try {
411 <            t.start();
412 <            Thread.sleep(SHORT_DELAY_MS);
413 <            t.interrupt();
414 <            t.join();
415 <        } catch (Exception e){
416 <            unexpectedException();
417 <        }
382 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
383 >            public void realRun() throws InterruptedException {
384 >                q.take();
385 >            }};
386 >
387 >        t.start();
388 >        Thread.sleep(SHORT_DELAY_MS);
389 >        t.interrupt();
390 >        t.join();
391      }
392  
393      /**
394       * Take removes existing elements until empty, then blocks interruptibly
395       */
396 <    public void testBlockingTake() {
397 <        Thread t = new Thread(new Runnable() {
398 <                public void run() {
399 <                    try {
400 <                        ArrayBlockingQueue q = populatedQueue(SIZE);
401 <                        for (int i = 0; i < SIZE; ++i) {
402 <                            threadAssertEquals(i, ((Integer)q.take()).intValue());
403 <                        }
404 <                        q.take();
405 <                        threadShouldThrow();
406 <                    } catch (InterruptedException success){
407 <                    }  
435 <                }});
436 <        try {
437 <            t.start();
438 <            Thread.sleep(SHORT_DELAY_MS);
396 >    public void testBlockingTake() throws InterruptedException {
397 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
398 >            public void realRun() throws InterruptedException {
399 >                ArrayBlockingQueue q = populatedQueue(SIZE);
400 >                for (int i = 0; i < SIZE; ++i) {
401 >                    threadAssertEquals(i, ((Integer)q.take()).intValue());
402 >                }
403 >                q.take();
404 >            }};
405 >
406 >        t.start();
407 >            Thread.sleep(SHORT_DELAY_MS);
408              t.interrupt();
409              t.join();
441        }
442        catch (InterruptedException ie) {
443            unexpectedException();
444        }
410      }
411  
412  
# Line 453 | Line 418 | public class ArrayBlockingQueueTest exte
418          for (int i = 0; i < SIZE; ++i) {
419              assertEquals(i, ((Integer)q.poll()).intValue());
420          }
421 <        assertNull(q.poll());
421 >        assertNull(q.poll());
422      }
423  
424      /**
425       * timed pool with zero timeout succeeds when non-empty, else times out
426       */
427 <    public void testTimedPoll0() {
428 <        try {
429 <            ArrayBlockingQueue q = populatedQueue(SIZE);
430 <            for (int i = 0; i < SIZE; ++i) {
431 <                assertEquals(i, ((Integer)q.poll(0, TimeUnit.MILLISECONDS)).intValue());
432 <            }
468 <            assertNull(q.poll(0, TimeUnit.MILLISECONDS));
469 <        } catch (InterruptedException e){
470 <            unexpectedException();
471 <        }  
427 >    public void testTimedPoll0() throws InterruptedException {
428 >        ArrayBlockingQueue q = populatedQueue(SIZE);
429 >        for (int i = 0; i < SIZE; ++i) {
430 >            assertEquals(i, ((Integer)q.poll(0, MILLISECONDS)).intValue());
431 >        }
432 >        assertNull(q.poll(0, MILLISECONDS));
433      }
434  
435      /**
436       * timed pool with nonzero timeout succeeds when non-empty, else times out
437       */
438 <    public void testTimedPoll() {
439 <        try {
440 <            ArrayBlockingQueue q = populatedQueue(SIZE);
441 <            for (int i = 0; i < SIZE; ++i) {
442 <                assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
443 <            }
483 <            assertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
484 <        } catch (InterruptedException e){
485 <            unexpectedException();
486 <        }  
438 >    public void testTimedPoll() throws InterruptedException {
439 >        ArrayBlockingQueue q = populatedQueue(SIZE);
440 >        for (int i = 0; i < SIZE; ++i) {
441 >            assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, MILLISECONDS)).intValue());
442 >        }
443 >        assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
444      }
445  
446      /**
447       * Interrupted timed poll throws InterruptedException instead of
448       * returning timeout status
449       */
450 <    public void testInterruptedTimedPoll() {
451 <        Thread t = new Thread(new Runnable() {
452 <                public void run() {
453 <                    try {
454 <                        ArrayBlockingQueue q = populatedQueue(SIZE);
455 <                        for (int i = 0; i < SIZE; ++i) {
456 <                            threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
457 <                        }
458 <                        threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
459 <                    } catch (InterruptedException success){
460 <                    }  
461 <                }});
462 <        try {
463 <            t.start();
464 <            Thread.sleep(SHORT_DELAY_MS);
465 <            t.interrupt();
466 <            t.join();
510 <        }
511 <        catch (InterruptedException ie) {
512 <            unexpectedException();
513 <        }
450 >    public void testInterruptedTimedPoll() throws InterruptedException {
451 >        Thread t = new Thread(new CheckedRunnable() {
452 >            public void realRun() throws InterruptedException {
453 >                ArrayBlockingQueue q = populatedQueue(SIZE);
454 >                for (int i = 0; i < SIZE; ++i) {
455 >                    assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, MILLISECONDS)).intValue());
456 >                }
457 >                try {
458 >                    q.poll(SMALL_DELAY_MS, MILLISECONDS);
459 >                    shouldThrow();
460 >                } catch (InterruptedException success) {}
461 >            }});
462 >
463 >        t.start();
464 >        Thread.sleep(SHORT_DELAY_MS);
465 >        t.interrupt();
466 >        t.join();
467      }
468  
469      /**
470       *  timed poll before a delayed offer fails; after offer succeeds;
471       *  on interruption throws
472       */
473 <    public void testTimedPollWithOffer() {
473 >    public void testTimedPollWithOffer() throws InterruptedException {
474          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
475 <        Thread t = new Thread(new Runnable() {
476 <                public void run() {
477 <                    try {
478 <                        threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
479 <                        q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
480 <                        q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
481 <                        threadShouldThrow();
482 <                    } catch (InterruptedException success) { }                
483 <                }
484 <            });
485 <        try {
486 <            t.start();
487 <            Thread.sleep(SMALL_DELAY_MS);
488 <            assertTrue(q.offer(zero, SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
489 <            t.interrupt();
490 <            t.join();
538 <        } catch (Exception e){
539 <            unexpectedException();
540 <        }
541 <    }  
475 >        Thread t = new Thread(new CheckedRunnable() {
476 >            public void realRun() throws InterruptedException {
477 >                assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
478 >                assertSame(zero, q.poll(LONG_DELAY_MS, MILLISECONDS));
479 >                try {
480 >                    q.poll(LONG_DELAY_MS, MILLISECONDS);
481 >                    shouldThrow();
482 >                } catch (InterruptedException success) {}
483 >            }});
484 >
485 >        t.start();
486 >        Thread.sleep(SMALL_DELAY_MS);
487 >        assertTrue(q.offer(zero, SHORT_DELAY_MS, MILLISECONDS));
488 >        t.interrupt();
489 >        t.join();
490 >    }
491  
492  
493      /**
# Line 552 | Line 501 | public class ArrayBlockingQueueTest exte
501              assertTrue(q.peek() == null ||
502                         i != ((Integer)q.peek()).intValue());
503          }
504 <        assertNull(q.peek());
504 >        assertNull(q.peek());
505      }
506  
507      /**
# Line 567 | Line 516 | public class ArrayBlockingQueueTest exte
516          try {
517              q.element();
518              shouldThrow();
519 <        }
571 <        catch (NoSuchElementException success) {}
519 >        } catch (NoSuchElementException success) {}
520      }
521  
522      /**
# Line 582 | Line 530 | public class ArrayBlockingQueueTest exte
530          try {
531              q.remove();
532              shouldThrow();
533 <        } catch (NoSuchElementException success){
586 <        }  
533 >        } catch (NoSuchElementException success) {}
534      }
535  
536      /**
# Line 600 | Line 547 | public class ArrayBlockingQueueTest exte
547          }
548          assertTrue(q.isEmpty());
549      }
550 <        
550 >
551      /**
552       * contains(x) reports true when elements added but not yet removed
553       */
# Line 624 | Line 571 | public class ArrayBlockingQueueTest exte
571          assertEquals(SIZE, q.remainingCapacity());
572          q.add(one);
573          assertFalse(q.isEmpty());
574 +        assertTrue(q.contains(one));
575          q.clear();
576          assertTrue(q.isEmpty());
577      }
# Line 680 | Line 628 | public class ArrayBlockingQueueTest exte
628      /**
629       *  toArray contains all elements
630       */
631 <    public void testToArray() {
631 >    public void testToArray() throws InterruptedException {
632          ArrayBlockingQueue q = populatedQueue(SIZE);
633 <        Object[] o = q.toArray();
634 <        try {
635 <        for(int i = 0; i < o.length; i++)
688 <            assertEquals(o[i], q.take());
689 <        } catch (InterruptedException e){
690 <            unexpectedException();
691 <        }    
633 >        Object[] o = q.toArray();
634 >        for (int i = 0; i < o.length; i++)
635 >            assertEquals(o[i], q.take());
636      }
637  
638      /**
639       * toArray(a) contains all elements
640       */
641 <    public void testToArray2() {
641 >    public void testToArray2() throws InterruptedException {
642          ArrayBlockingQueue q = populatedQueue(SIZE);
643 <        Integer[] ints = new Integer[SIZE];
644 <        ints = (Integer[])q.toArray(ints);
645 <        try {
646 <            for(int i = 0; i < ints.length; i++)
647 <                assertEquals(ints[i], q.take());
648 <        } catch (InterruptedException e){
649 <            unexpectedException();
650 <        }    
643 >        Integer[] ints = new Integer[SIZE];
644 >        ints = (Integer[])q.toArray(ints);
645 >        for (int i = 0; i < ints.length; i++)
646 >            assertEquals(ints[i], q.take());
647 >    }
648 >
649 >    /**
650 >     * toArray(null) throws NPE
651 >     */
652 >    public void testToArray_BadArg() {
653 >        try {
654 >            ArrayBlockingQueue q = populatedQueue(SIZE);
655 >            Object o[] = q.toArray(null);
656 >            shouldThrow();
657 >        } catch (NullPointerException success) {}
658 >    }
659 >
660 >    /**
661 >     * toArray with incompatible array type throws CCE
662 >     */
663 >    public void testToArray1_BadArg() {
664 >        try {
665 >            ArrayBlockingQueue q = populatedQueue(SIZE);
666 >            Object o[] = q.toArray(new String[10] );
667 >            shouldThrow();
668 >        } catch (ArrayStoreException success) {}
669      }
670 <    
670 >
671 >
672      /**
673       * iterator iterates through all elements
674       */
675 <    public void testIterator() {
675 >    public void testIterator() throws InterruptedException {
676          ArrayBlockingQueue q = populatedQueue(SIZE);
677 <        Iterator it = q.iterator();
678 <        try {
679 <            while(it.hasNext()){
680 <                assertEquals(it.next(), q.take());
718 <            }
719 <        } catch (InterruptedException e){
720 <            unexpectedException();
721 <        }    
677 >        Iterator it = q.iterator();
678 >        while (it.hasNext()) {
679 >            assertEquals(it.next(), q.take());
680 >        }
681      }
682  
683      /**
# Line 733 | Line 692 | public class ArrayBlockingQueueTest exte
692          Iterator it = q.iterator();
693          it.next();
694          it.remove();
695 <        
695 >
696          it = q.iterator();
697          assertEquals(it.next(), one);
698          assertEquals(it.next(), three);
# Line 767 | Line 726 | public class ArrayBlockingQueueTest exte
726          q.add(one);
727          q.add(two);
728          q.add(three);
729 <        try {
730 <            for (Iterator it = q.iterator(); it.hasNext();) {
731 <                q.remove();
773 <                it.next();
774 <            }
775 <        }
776 <        catch (ConcurrentModificationException e) {
777 <            unexpectedException();
729 >        for (Iterator it = q.iterator(); it.hasNext();) {
730 >            q.remove();
731 >            it.next();
732          }
733          assertEquals(0, q.size());
734      }
# Line 789 | Line 743 | public class ArrayBlockingQueueTest exte
743          for (int i = 0; i < SIZE; ++i) {
744              assertTrue(s.indexOf(String.valueOf(i)) >= 0);
745          }
746 <    }        
746 >    }
747  
748  
749      /**
# Line 800 | Line 754 | public class ArrayBlockingQueueTest exte
754          q.add(one);
755          q.add(two);
756          ExecutorService executor = Executors.newFixedThreadPool(2);
757 <        executor.execute(new Runnable() {
758 <            public void run() {
757 >        executor.execute(new CheckedRunnable() {
758 >            public void realRun() throws InterruptedException {
759                  threadAssertFalse(q.offer(three));
760 <                try {
761 <                    threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
762 <                    threadAssertEquals(0, q.remainingCapacity());
763 <                }
764 <                catch (InterruptedException e) {
765 <                    threadUnexpectedException();
766 <                }
767 <            }
768 <        });
760 >                threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, MILLISECONDS));
761 >                threadAssertEquals(0, q.remainingCapacity());
762 >            }});
763 >
764 >        executor.execute(new CheckedRunnable() {
765 >            public void realRun() throws InterruptedException {
766 >                Thread.sleep(SMALL_DELAY_MS);
767 >                threadAssertEquals(one, q.take());
768 >            }});
769  
816        executor.execute(new Runnable() {
817            public void run() {
818                try {
819                    Thread.sleep(SMALL_DELAY_MS);
820                    threadAssertEquals(one, q.take());
821                }
822                catch (InterruptedException e) {
823                    threadUnexpectedException();
824                }
825            }
826        });
827        
770          joinPool(executor);
771  
772      }
# Line 835 | Line 777 | public class ArrayBlockingQueueTest exte
777      public void testPollInExecutor() {
778          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
779          ExecutorService executor = Executors.newFixedThreadPool(2);
780 <        executor.execute(new Runnable() {
781 <            public void run() {
780 >        executor.execute(new CheckedRunnable() {
781 >            public void realRun() throws InterruptedException {
782                  threadAssertNull(q.poll());
783 <                try {
784 <                    threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
785 <                    threadAssertTrue(q.isEmpty());
786 <                }
787 <                catch (InterruptedException e) {
788 <                    threadUnexpectedException();
789 <                }
790 <            }
791 <        });
783 >                threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, MILLISECONDS));
784 >                threadAssertTrue(q.isEmpty());
785 >            }});
786 >
787 >        executor.execute(new CheckedRunnable() {
788 >            public void realRun() throws InterruptedException {
789 >                Thread.sleep(SMALL_DELAY_MS);
790 >                q.put(one);
791 >            }});
792  
851        executor.execute(new Runnable() {
852            public void run() {
853                try {
854                    Thread.sleep(SMALL_DELAY_MS);
855                    q.put(one);
856                }
857                catch (InterruptedException e) {
858                    threadUnexpectedException();
859                }
860            }
861        });
862        
793          joinPool(executor);
794      }
795  
796      /**
797       * A deserialized serialized queue has same elements in same order
798       */
799 <    public void testSerialization() {
799 >    public void testSerialization() throws Exception {
800          ArrayBlockingQueue q = populatedQueue(SIZE);
801  
802 +        ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
803 +        ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
804 +        out.writeObject(q);
805 +        out.close();
806 +
807 +        ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
808 +        ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
809 +        ArrayBlockingQueue r = (ArrayBlockingQueue)in.readObject();
810 +        assertEquals(q.size(), r.size());
811 +        while (!q.isEmpty())
812 +            assertEquals(q.remove(), r.remove());
813 +    }
814 +
815 +    /**
816 +     * drainTo(null) throws NPE
817 +     */
818 +    public void testDrainToNull() {
819 +        ArrayBlockingQueue q = populatedQueue(SIZE);
820          try {
821 <            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
822 <            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
823 <            out.writeObject(q);
824 <            out.close();
825 <
826 <            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
827 <            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
828 <            ArrayBlockingQueue r = (ArrayBlockingQueue)in.readObject();
829 <            assertEquals(q.size(), r.size());
830 <            while (!q.isEmpty())
831 <                assertEquals(q.remove(), r.remove());
832 <        } catch(Exception e){
833 <            unexpectedException();
834 <        }
821 >            q.drainTo(null);
822 >            shouldThrow();
823 >        } catch (NullPointerException success) {}
824 >    }
825 >
826 >    /**
827 >     * drainTo(this) throws IAE
828 >     */
829 >    public void testDrainToSelf() {
830 >        ArrayBlockingQueue q = populatedQueue(SIZE);
831 >        try {
832 >            q.drainTo(q);
833 >            shouldThrow();
834 >        } catch (IllegalArgumentException success) {}
835 >    }
836 >
837 >    /**
838 >     * drainTo(c) empties queue into another collection c
839 >     */
840 >    public void testDrainTo() {
841 >        ArrayBlockingQueue q = populatedQueue(SIZE);
842 >        ArrayList l = new ArrayList();
843 >        q.drainTo(l);
844 >        assertEquals(q.size(), 0);
845 >        assertEquals(l.size(), SIZE);
846 >        for (int i = 0; i < SIZE; ++i)
847 >            assertEquals(l.get(i), new Integer(i));
848 >        q.add(zero);
849 >        q.add(one);
850 >        assertFalse(q.isEmpty());
851 >        assertTrue(q.contains(zero));
852 >        assertTrue(q.contains(one));
853 >        l.clear();
854 >        q.drainTo(l);
855 >        assertEquals(q.size(), 0);
856 >        assertEquals(l.size(), 2);
857 >        for (int i = 0; i < 2; ++i)
858 >            assertEquals(l.get(i), new Integer(i));
859 >    }
860 >
861 >    /**
862 >     * drainTo empties full queue, unblocking a waiting put.
863 >     */
864 >    public void testDrainToWithActivePut() throws InterruptedException {
865 >        final ArrayBlockingQueue q = populatedQueue(SIZE);
866 >        Thread t = new Thread(new CheckedRunnable() {
867 >            public void realRun() throws InterruptedException {
868 >                q.put(new Integer(SIZE+1));
869 >            }});
870 >
871 >        t.start();
872 >        ArrayList l = new ArrayList();
873 >        q.drainTo(l);
874 >        assertTrue(l.size() >= SIZE);
875 >        for (int i = 0; i < SIZE; ++i)
876 >            assertEquals(l.get(i), new Integer(i));
877 >        t.join();
878 >        assertTrue(q.size() + l.size() >= SIZE);
879      }
880  
881 +    /**
882 +     * drainTo(null, n) throws NPE
883 +     */
884 +    public void testDrainToNullN() {
885 +        ArrayBlockingQueue q = populatedQueue(SIZE);
886 +        try {
887 +            q.drainTo(null, 0);
888 +            shouldThrow();
889 +        } catch (NullPointerException success) {}
890 +    }
891 +
892 +    /**
893 +     * drainTo(this, n) throws IAE
894 +     */
895 +    public void testDrainToSelfN() {
896 +        ArrayBlockingQueue q = populatedQueue(SIZE);
897 +        try {
898 +            q.drainTo(q, 0);
899 +            shouldThrow();
900 +        } catch (IllegalArgumentException success) {}
901 +    }
902 +
903 +    /**
904 +     * drainTo(c, n) empties first max {n, size} elements of queue into c
905 +     */
906 +    public void testDrainToN() {
907 +        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE*2);
908 +        for (int i = 0; i < SIZE + 2; ++i) {
909 +            for (int j = 0; j < SIZE; j++)
910 +                assertTrue(q.offer(new Integer(j)));
911 +            ArrayList l = new ArrayList();
912 +            q.drainTo(l, i);
913 +            int k = (i < SIZE)? i : SIZE;
914 +            assertEquals(l.size(), k);
915 +            assertEquals(q.size(), SIZE-k);
916 +            for (int j = 0; j < k; ++j)
917 +                assertEquals(l.get(j), new Integer(j));
918 +            while (q.poll() != null) ;
919 +        }
920 +    }
921  
922   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines