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.3 by dl, Sun Sep 14 20:42:40 2003 UTC vs.
Revision 1.18 by jsr166, Sat Nov 21 10:25:05 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 {
14
17      public static void main(String[] args) {
18 <        junit.textui.TestRunner.run (suite());  
18 >        junit.textui.TestRunner.run (suite());
19      }
18
20      public static Test suite() {
21 <        return new TestSuite(ArrayBlockingQueueTest.class);
21 >        return new TestSuite(ArrayBlockingQueueTest.class);
22      }
23  
24      /**
# Line 27 | 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 <
39 <    public void testConstructor1(){
38 >
39 >    /**
40 >     * A new queue has the indicated capacity
41 >     */
42 >    public void testConstructor1() {
43          assertEquals(SIZE, new ArrayBlockingQueue(SIZE).remainingCapacity());
44      }
45  
46 <    public void testConstructor2(){
46 >    /**
47 >     * Constructor throws IAE if capacity argument nonpositive
48 >     */
49 >    public void testConstructor2() {
50          try {
51              ArrayBlockingQueue q = new ArrayBlockingQueue(0);
52 <            fail("Cannot make zero-sized");
53 <        }
47 <        catch (IllegalArgumentException success) {}
52 >            shouldThrow();
53 >        } catch (IllegalArgumentException success) {}
54      }
55  
56 <    public void testConstructor3(){
57 <
56 >    /**
57 >     * Initializing from null Collection throws NPE
58 >     */
59 >    public void testConstructor3() {
60          try {
61              ArrayBlockingQueue q = new ArrayBlockingQueue(1, true, null);
62 <            fail("Cannot make from null collection");
63 <        }
56 <        catch (NullPointerException success) {}
62 >            shouldThrow();
63 >        } catch (NullPointerException success) {}
64      }
65  
66 <    public void testConstructor4(){
66 >    /**
67 >     * Initializing from Collection of null elements throws NPE
68 >     */
69 >    public void testConstructor4() {
70          try {
71              Integer[] ints = new Integer[SIZE];
72              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, false, Arrays.asList(ints));
73 <            fail("Cannot make with null elements");
74 <        }
65 <        catch (NullPointerException success) {}
73 >            shouldThrow();
74 >        } catch (NullPointerException success) {}
75      }
76  
77 <    public void testConstructor5(){
77 >    /**
78 >     * Initializing from Collection with some null elements throws NPE
79 >     */
80 >    public void testConstructor5() {
81          try {
82              Integer[] ints = new Integer[SIZE];
83              for (int i = 0; i < SIZE-1; ++i)
84                  ints[i] = new Integer(i);
85              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE, false, Arrays.asList(ints));
86 <            fail("Cannot make with null elements");
87 <        }
76 <        catch (NullPointerException success) {}
86 >            shouldThrow();
87 >        } catch (NullPointerException success) {}
88      }
89  
90 <    public void testConstructor6(){
90 >    /**
91 >     * Initializing from too large collection throws IAE
92 >     */
93 >    public void testConstructor6() {
94          try {
95              Integer[] ints = new Integer[SIZE];
96              for (int i = 0; i < SIZE; ++i)
97                  ints[i] = new Integer(i);
98              ArrayBlockingQueue q = new ArrayBlockingQueue(1, false, Arrays.asList(ints));
99 <            fail("Cannot make with insufficient capacity");
100 <        }
87 <        catch (IllegalArgumentException success) {}
99 >            shouldThrow();
100 >        } catch (IllegalArgumentException success) {}
101      }
102  
103 <    public void testConstructor7(){
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)
# Line 99 | Line 115 | public class ArrayBlockingQueueTest exte
115          finally {}
116      }
117  
118 +    /**
119 +     * Queue transitions from empty to full when elements added
120 +     */
121      public void testEmptyFull() {
122          ArrayBlockingQueue q = new ArrayBlockingQueue(2);
123          assertTrue(q.isEmpty());
124 <        assertEquals("should have room for 2", 2, q.remainingCapacity());
124 >        assertEquals(2, q.remainingCapacity());
125          q.add(one);
126          assertFalse(q.isEmpty());
127          q.add(two);
128          assertFalse(q.isEmpty());
129 <        assertEquals("queue should be full", 0, q.remainingCapacity());
130 <        assertFalse("offer should be rejected", q.offer(three));
129 >        assertEquals(0, q.remainingCapacity());
130 >        assertFalse(q.offer(three));
131      }
132  
133 <    public void testRemainingCapacity(){
133 >    /**
134 >     * remainingCapacity decreases on add, increases on remove
135 >     */
136 >    public void testRemainingCapacity() {
137          ArrayBlockingQueue q = populatedQueue(SIZE);
138          for (int i = 0; i < SIZE; ++i) {
139              assertEquals(i, q.remainingCapacity());
# Line 125 | Line 147 | public class ArrayBlockingQueueTest exte
147          }
148      }
149  
150 <    public void testOfferNull(){
151 <        try {
150 >    /**
151 >     *  offer(null) throws NPE
152 >     */
153 >    public void testOfferNull() {
154 >        try {
155              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
156              q.offer(null);
157 <            fail("should throw NPE");
158 <        } catch (NullPointerException success) { }  
157 >            shouldThrow();
158 >        } catch (NullPointerException success) {}
159 >    }
160 >
161 >    /**
162 >     *  add(null) throws NPE
163 >     */
164 >    public void testAddNull() {
165 >        try {
166 >            ArrayBlockingQueue q = new ArrayBlockingQueue(1);
167 >            q.add(null);
168 >            shouldThrow();
169 >        } catch (NullPointerException success) {}
170      }
171  
172 <    public void testOffer(){
172 >    /**
173 >     * Offer succeeds if not full; fails if full
174 >     */
175 >    public void testOffer() {
176          ArrayBlockingQueue q = new ArrayBlockingQueue(1);
177          assertTrue(q.offer(zero));
178          assertFalse(q.offer(one));
179      }
180  
181 <    public void testAdd(){
182 <        try {
181 >    /**
182 >     * add succeeds if not full; throws ISE if full
183 >     */
184 >    public void testAdd() {
185 >        try {
186              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
187              for (int i = 0; i < SIZE; ++i) {
188                  assertTrue(q.add(new Integer(i)));
189              }
190              assertEquals(0, q.remainingCapacity());
191              q.add(new Integer(SIZE));
192 <        } catch (IllegalStateException success){
193 <        }  
192 >            shouldThrow();
193 >        } catch (IllegalStateException success) {}
194      }
195  
196 <    public void testAddAll1(){
196 >    /**
197 >     *  addAll(null) throws NPE
198 >     */
199 >    public void testAddAll1() {
200          try {
201              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
202              q.addAll(null);
203 <            fail("Cannot add null collection");
204 <        }
160 <        catch (NullPointerException success) {}
203 >            shouldThrow();
204 >        } catch (NullPointerException success) {}
205      }
206 <    public void testAddAll2(){
206 >
207 >    /**
208 >     * addAll(this) throws IAE
209 >     */
210 >    public void testAddAllSelf() {
211 >        try {
212 >            ArrayBlockingQueue q = populatedQueue(SIZE);
213 >            q.addAll(q);
214 >            shouldThrow();
215 >        } catch (IllegalArgumentException success) {}
216 >    }
217 >
218 >
219 >    /**
220 >     *  addAll of a collection with null elements throws NPE
221 >     */
222 >    public void testAddAll2() {
223          try {
224              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
225              Integer[] ints = new Integer[SIZE];
226              q.addAll(Arrays.asList(ints));
227 <            fail("Cannot add null elements");
228 <        }
169 <        catch (NullPointerException success) {}
227 >            shouldThrow();
228 >        } catch (NullPointerException success) {}
229      }
230 <    public void testAddAll3(){
230 >    /**
231 >     * addAll of a collection with any null elements throws NPE after
232 >     * possibly adding some elements
233 >     */
234 >    public void testAddAll3() {
235          try {
236              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
237              Integer[] ints = new Integer[SIZE];
238              for (int i = 0; i < SIZE-1; ++i)
239                  ints[i] = new Integer(i);
240              q.addAll(Arrays.asList(ints));
241 <            fail("Cannot add null elements");
242 <        }
180 <        catch (NullPointerException success) {}
241 >            shouldThrow();
242 >        } catch (NullPointerException success) {}
243      }
244 <    public void testAddAll4(){
244 >    /**
245 >     * addAll throws ISE if not enough room
246 >     */
247 >    public void testAddAll4() {
248          try {
249              ArrayBlockingQueue q = new ArrayBlockingQueue(1);
250              Integer[] ints = new Integer[SIZE];
251              for (int i = 0; i < SIZE; ++i)
252                  ints[i] = new Integer(i);
253              q.addAll(Arrays.asList(ints));
254 <            fail("Cannot add with insufficient capacity");
255 <        }
191 <        catch (IllegalStateException success) {}
254 >            shouldThrow();
255 >        } catch (IllegalStateException success) {}
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)
267 <                assertEquals(ints[i], q.poll());
268 <        }
269 <        finally {}
257 >    /**
258 >     * Queue contains all elements, in traversal order, of successful addAll
259 >     */
260 >    public void testAddAll5() {
261 >        Integer[] empty = new Integer[0];
262 >        Integer[] ints = new Integer[SIZE];
263 >        for (int i = 0; i < SIZE; ++i)
264 >            ints[i] = new Integer(i);
265 >        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
266 >        assertFalse(q.addAll(Arrays.asList(empty)));
267 >        assertTrue(q.addAll(Arrays.asList(ints)));
268 >        for (int i = 0; i < SIZE; ++i)
269 >            assertEquals(ints[i], q.poll());
270      }
271  
272 <     public void testPutNull() {
273 <        try {
272 >    /**
273 >     *  put(null) throws NPE
274 >     */
275 >    public void testPutNull() throws InterruptedException {
276 >        try {
277              ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
278              q.put(null);
279 <            fail("put should throw NPE");
280 <        }
214 <        catch (NullPointerException success){
215 <        }  
216 <        catch (InterruptedException ie) {
217 <            fail("Unexpected exception");
218 <        }
279 >            shouldThrow();
280 >        } catch (NullPointerException success) {}
281       }
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);
289 <                 assertTrue(q.contains(I));
290 <             }
291 <             assertEquals(0, q.remainingCapacity());
230 <         }
231 <        catch (InterruptedException ie) {
232 <            fail("Unexpected exception");
283 >    /**
284 >     * all elements successfully put are contained
285 >     */
286 >    public void testPut() throws InterruptedException {
287 >        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
288 >        for (int i = 0; i < SIZE; ++i) {
289 >            Integer I = new Integer(i);
290 >            q.put(I);
291 >            assertTrue(q.contains(I));
292          }
293 +        assertEquals(0, q.remainingCapacity());
294      }
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 <                        threadFail("put should block");
308 <                    } catch (InterruptedException ie){
309 <                        threadAssertEquals(added, SIZE);
310 <                    }  
311 <                }});
312 <        try {
313 <            t.start();
314 <           Thread.sleep(SHORT_DELAY_MS);
315 <           t.interrupt();
316 <           t.join();
317 <        }
318 <        catch (InterruptedException ie) {
259 <            fail("Unexpected exception");
260 <        }
296 >    /**
297 >     * put blocks interruptibly if full
298 >     */
299 >    public void testBlockingPut() throws InterruptedException {
300 >        final ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE);
301 >        Thread t = new Thread(new CheckedRunnable() {
302 >            public void realRun() {
303 >                int added = 0;
304 >                try {
305 >                    for (int i = 0; i < SIZE; ++i) {
306 >                        q.put(new Integer(i));
307 >                        ++added;
308 >                    }
309 >                    q.put(new Integer(SIZE));
310 >                    threadShouldThrow();
311 >                } catch (InterruptedException ie) {
312 >                    threadAssertEquals(added, SIZE);
313 >                }}});
314 >
315 >        t.start();
316 >        Thread.sleep(MEDIUM_DELAY_MS);
317 >        t.interrupt();
318 >        t.join();
319      }
320  
321 <    public void testPutWithTake() {
321 >    /**
322 >     * put blocks waiting for take when full
323 >     */
324 >    public void testPutWithTake() throws InterruptedException {
325          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
326 <        Thread t = new Thread(new Runnable() {
327 <                public void run(){
328 <                    int added = 0;
329 <                    try {
330 <                        q.put(new Object());
331 <                        ++added;
332 <                        q.put(new Object());
333 <                        ++added;
334 <                        q.put(new Object());
335 <                        ++added;
336 <                        q.put(new Object());
337 <                        ++added;
338 <                        threadFail("Should block");
339 <                    } catch (InterruptedException e){
340 <                        threadAssertTrue(added >= 2);
341 <                    }
342 <                }
343 <            });
344 <        try {
345 <            t.start();
346 <            Thread.sleep(SHORT_DELAY_MS);
347 <            q.take();
348 <            t.interrupt();
288 <            t.join();
289 <        } catch (Exception e){
290 <            fail("Unexpected exception");
291 <        }
326 >        Thread t = new Thread(new CheckedRunnable() {
327 >            public void realRun() {
328 >                int added = 0;
329 >                try {
330 >                    q.put(new Object());
331 >                    ++added;
332 >                    q.put(new Object());
333 >                    ++added;
334 >                    q.put(new Object());
335 >                    ++added;
336 >                    q.put(new Object());
337 >                    ++added;
338 >                    threadShouldThrow();
339 >                } catch (InterruptedException e) {
340 >                    threadAssertTrue(added >= 2);
341 >                }
342 >            }});
343 >
344 >        t.start();
345 >        Thread.sleep(SHORT_DELAY_MS);
346 >        q.take();
347 >        t.interrupt();
348 >        t.join();
349      }
350  
351 <    public void testTimedOffer() {
351 >    /**
352 >     * timed offer times out if full and elements not taken
353 >     */
354 >    public void testTimedOffer() throws InterruptedException {
355          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
356 <        Thread t = new Thread(new Runnable() {
357 <                public void run(){
358 <                    try {
359 <                        q.put(new Object());
360 <                        q.put(new Object());
361 <                        threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS/2, TimeUnit.MILLISECONDS));
362 <                        q.offer(new Object(), LONG_DELAY_MS, TimeUnit.MILLISECONDS);
363 <                        threadFail("Should block");
364 <                    } catch (InterruptedException success){}
365 <                }
366 <            });
367 <        
308 <        try {
309 <            t.start();
310 <            Thread.sleep(SHORT_DELAY_MS);
311 <            t.interrupt();
312 <            t.join();
313 <        } catch (Exception e){
314 <            fail("Unexpected exception");
315 <        }
356 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
357 >            public void realRun() throws InterruptedException {
358 >                q.put(new Object());
359 >                q.put(new Object());
360 >                threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS/2, MILLISECONDS));
361 >                q.offer(new Object(), LONG_DELAY_MS, MILLISECONDS);
362 >            }};
363 >
364 >        t.start();
365 >        Thread.sleep(SHORT_DELAY_MS);
366 >        t.interrupt();
367 >        t.join();
368      }
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());
375 <            }
376 <        } catch (InterruptedException e){
377 <            fail("Unexpected exception");
326 <        }  
370 >    /**
371 >     * take retrieves elements in FIFO order
372 >     */
373 >    public void testTake() throws InterruptedException {
374 >        ArrayBlockingQueue q = populatedQueue(SIZE);
375 >        for (int i = 0; i < SIZE; ++i) {
376 >            assertEquals(i, ((Integer)q.take()).intValue());
377 >        }
378      }
379  
380 <    public void testTakeFromEmpty() {
380 >    /**
381 >     * take blocks interruptibly when empty
382 >     */
383 >    public void testTakeFromEmpty() throws InterruptedException {
384          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
385 <        Thread t = new Thread(new Runnable() {
386 <                public void run(){
387 <                    try {
388 <                        q.take();
389 <                        threadFail("Should block");
390 <                    } catch (InterruptedException success){ }                
391 <                }
392 <            });
393 <        try {
340 <            t.start();
341 <            Thread.sleep(SHORT_DELAY_MS);
342 <            t.interrupt();
343 <            t.join();
344 <        } catch (Exception e){
345 <            fail("Unexpected exception");
346 <        }
385 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
386 >            public void realRun() throws InterruptedException {
387 >                q.take();
388 >            }};
389 >
390 >        t.start();
391 >        Thread.sleep(SHORT_DELAY_MS);
392 >        t.interrupt();
393 >        t.join();
394      }
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 <                        threadFail("take should block");
406 <                    } catch (InterruptedException success){
407 <                    }  
408 <                }});
409 <        try {
410 <            t.start();
364 <            Thread.sleep(SHORT_DELAY_MS);
396 >    /**
397 >     * Take removes existing elements until empty, then blocks interruptibly
398 >     */
399 >    public void testBlockingTake() throws InterruptedException {
400 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
401 >            public void realRun() throws InterruptedException {
402 >                ArrayBlockingQueue q = populatedQueue(SIZE);
403 >                for (int i = 0; i < SIZE; ++i) {
404 >                    threadAssertEquals(i, ((Integer)q.take()).intValue());
405 >                }
406 >                q.take();
407 >            }};
408 >
409 >        t.start();
410 >            Thread.sleep(SHORT_DELAY_MS);
411              t.interrupt();
412              t.join();
367        }
368        catch (InterruptedException ie) {
369            fail("Unexpected exception");
370        }
413      }
414  
415  
416 <    public void testPoll(){
416 >    /**
417 >     * poll succeeds unless empty
418 >     */
419 >    public void testPoll() {
420          ArrayBlockingQueue q = populatedQueue(SIZE);
421          for (int i = 0; i < SIZE; ++i) {
422              assertEquals(i, ((Integer)q.poll()).intValue());
423          }
424 <        assertNull(q.poll());
424 >        assertNull(q.poll());
425      }
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 <            }
433 <            assertNull(q.poll(0, TimeUnit.MILLISECONDS));
434 <        } catch (InterruptedException e){
435 <            fail("Unexpected exception");
391 <        }  
427 >    /**
428 >     * timed pool with zero timeout succeeds when non-empty, else times out
429 >     */
430 >    public void testTimedPoll0() throws InterruptedException {
431 >        ArrayBlockingQueue q = populatedQueue(SIZE);
432 >        for (int i = 0; i < SIZE; ++i) {
433 >            assertEquals(i, ((Integer)q.poll(0, MILLISECONDS)).intValue());
434 >        }
435 >        assertNull(q.poll(0, MILLISECONDS));
436      }
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 <            }
444 <            assertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
401 <        } catch (InterruptedException e){
402 <            fail("Unexpected exception");
403 <        }  
404 <    }
405 <
406 <    public void testInterruptedTimedPoll(){
407 <        Thread t = new Thread(new Runnable() {
408 <                public void run() {
409 <                    try {
410 <                        ArrayBlockingQueue q = populatedQueue(SIZE);
411 <                        for (int i = 0; i < SIZE; ++i) {
412 <                            threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
413 <                        }
414 <                        threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
415 <                    } catch (InterruptedException success){
416 <                    }  
417 <                }});
418 <        try {
419 <            t.start();
420 <            Thread.sleep(SHORT_DELAY_MS);
421 <            t.interrupt();
422 <            t.join();
423 <        }
424 <        catch (InterruptedException ie) {
425 <            fail("Unexpected exception");
438 >    /**
439 >     * timed pool with nonzero timeout succeeds when non-empty, else times out
440 >     */
441 >    public void testTimedPoll() throws InterruptedException {
442 >        ArrayBlockingQueue q = populatedQueue(SIZE);
443 >        for (int i = 0; i < SIZE; ++i) {
444 >            assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, MILLISECONDS)).intValue());
445          }
446 +        assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
447 +    }
448 +
449 +    /**
450 +     * Interrupted timed poll throws InterruptedException instead of
451 +     * returning timeout status
452 +     */
453 +    public void testInterruptedTimedPoll() throws InterruptedException {
454 +        Thread t = new ThreadShouldThrow(InterruptedException.class) {
455 +            public void realRun() throws InterruptedException {
456 +                ArrayBlockingQueue q = populatedQueue(SIZE);
457 +                for (int i = 0; i < SIZE; ++i) {
458 +                    threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, MILLISECONDS)).intValue());
459 +                }
460 +                q.poll(MEDIUM_DELAY_MS, MILLISECONDS);
461 +            }};
462 +
463 +        t.start();
464 +        Thread.sleep(SMALL_DELAY_MS);
465 +        t.interrupt();
466 +        t.join();
467      }
468  
469 <    public void testTimedPollWithOffer(){
469 >    /**
470 >     *  timed poll before a delayed offer fails; after offer succeeds;
471 >     *  on interruption throws
472 >     */
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 <                        threadFail("Should block");
482 <                    } catch (InterruptedException success) { }                
483 <                }
484 <            });
485 <        try {
486 <            t.start();
487 <            Thread.sleep(SMALL_DELAY_MS);
444 <            assertTrue(q.offer(zero, SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
445 <            t.interrupt();
446 <            t.join();
447 <        } catch (Exception e){
448 <            fail("Unexpected exception");
449 <        }
450 <    }  
475 >        Thread t = new ThreadShouldThrow(InterruptedException.class) {
476 >            public void realRun() throws InterruptedException {
477 >                threadAssertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
478 >                q.poll(LONG_DELAY_MS, MILLISECONDS);
479 >                q.poll(LONG_DELAY_MS, MILLISECONDS);
480 >            }};
481 >
482 >        t.start();
483 >        Thread.sleep(SMALL_DELAY_MS);
484 >        assertTrue(q.offer(zero, SHORT_DELAY_MS, MILLISECONDS));
485 >        t.interrupt();
486 >        t.join();
487 >    }
488  
489  
490 <    public void testPeek(){
490 >    /**
491 >     * peek returns next element, or null if empty
492 >     */
493 >    public void testPeek() {
494          ArrayBlockingQueue q = populatedQueue(SIZE);
495          for (int i = 0; i < SIZE; ++i) {
496              assertEquals(i, ((Integer)q.peek()).intValue());
# Line 458 | Line 498 | public class ArrayBlockingQueueTest exte
498              assertTrue(q.peek() == null ||
499                         i != ((Integer)q.peek()).intValue());
500          }
501 <        assertNull(q.peek());
501 >        assertNull(q.peek());
502      }
503  
504 <    public void testElement(){
504 >    /**
505 >     * element returns next element, or throws NSEE if empty
506 >     */
507 >    public void testElement() {
508          ArrayBlockingQueue q = populatedQueue(SIZE);
509          for (int i = 0; i < SIZE; ++i) {
510              assertEquals(i, ((Integer)q.element()).intValue());
# Line 469 | Line 512 | public class ArrayBlockingQueueTest exte
512          }
513          try {
514              q.element();
515 <            fail("no such element");
516 <        }
474 <        catch (NoSuchElementException success) {}
515 >            shouldThrow();
516 >        } catch (NoSuchElementException success) {}
517      }
518  
519 <    public void testRemove(){
519 >    /**
520 >     * remove removes next element, or throws NSEE if empty
521 >     */
522 >    public void testRemove() {
523          ArrayBlockingQueue q = populatedQueue(SIZE);
524          for (int i = 0; i < SIZE; ++i) {
525              assertEquals(i, ((Integer)q.remove()).intValue());
526          }
527          try {
528              q.remove();
529 <            fail("remove should throw");
530 <        } catch (NoSuchElementException success){
486 <        }  
529 >            shouldThrow();
530 >        } catch (NoSuchElementException success) {}
531      }
532  
533 <    public void testRemoveElement(){
533 >    /**
534 >     * remove(x) removes x and returns true if present
535 >     */
536 >    public void testRemoveElement() {
537          ArrayBlockingQueue q = populatedQueue(SIZE);
538          for (int i = 1; i < SIZE; i+=2) {
539              assertTrue(q.remove(new Integer(i)));
# Line 497 | Line 544 | public class ArrayBlockingQueueTest exte
544          }
545          assertTrue(q.isEmpty());
546      }
547 <        
548 <    public void testContains(){
547 >
548 >    /**
549 >     * contains(x) reports true when elements added but not yet removed
550 >     */
551 >    public void testContains() {
552          ArrayBlockingQueue q = populatedQueue(SIZE);
553          for (int i = 0; i < SIZE; ++i) {
554              assertTrue(q.contains(new Integer(i)));
# Line 507 | Line 557 | public class ArrayBlockingQueueTest exte
557          }
558      }
559  
560 <    public void testClear(){
560 >    /**
561 >     * clear removes all elements
562 >     */
563 >    public void testClear() {
564          ArrayBlockingQueue q = populatedQueue(SIZE);
565          q.clear();
566          assertTrue(q.isEmpty());
# Line 515 | Line 568 | public class ArrayBlockingQueueTest exte
568          assertEquals(SIZE, q.remainingCapacity());
569          q.add(one);
570          assertFalse(q.isEmpty());
571 +        assertTrue(q.contains(one));
572          q.clear();
573          assertTrue(q.isEmpty());
574      }
575  
576 <    public void testContainsAll(){
576 >    /**
577 >     * containsAll(c) is true when c contains a subset of elements
578 >     */
579 >    public void testContainsAll() {
580          ArrayBlockingQueue q = populatedQueue(SIZE);
581          ArrayBlockingQueue p = new ArrayBlockingQueue(SIZE);
582          for (int i = 0; i < SIZE; ++i) {
# Line 530 | Line 587 | public class ArrayBlockingQueueTest exte
587          assertTrue(p.containsAll(q));
588      }
589  
590 <    public void testRetainAll(){
590 >    /**
591 >     * retainAll(c) retains only those elements of c and reports true if changed
592 >     */
593 >    public void testRetainAll() {
594          ArrayBlockingQueue q = populatedQueue(SIZE);
595          ArrayBlockingQueue p = populatedQueue(SIZE);
596          for (int i = 0; i < SIZE; ++i) {
# Line 546 | Line 606 | public class ArrayBlockingQueueTest exte
606          }
607      }
608  
609 <    public void testRemoveAll(){
609 >    /**
610 >     * removeAll(c) removes only those elements of c and reports true if changed
611 >     */
612 >    public void testRemoveAll() {
613          for (int i = 1; i < SIZE; ++i) {
614              ArrayBlockingQueue q = populatedQueue(SIZE);
615              ArrayBlockingQueue p = populatedQueue(i);
# Line 559 | Line 622 | public class ArrayBlockingQueueTest exte
622          }
623      }
624  
625 +    /**
626 +     *  toArray contains all elements
627 +     */
628 +    public void testToArray() throws InterruptedException {
629 +        ArrayBlockingQueue q = populatedQueue(SIZE);
630 +        Object[] o = q.toArray();
631 +        for (int i = 0; i < o.length; i++)
632 +            assertEquals(o[i], q.take());
633 +    }
634  
635 <    public void testToArray(){
635 >    /**
636 >     * toArray(a) contains all elements
637 >     */
638 >    public void testToArray2() throws InterruptedException {
639          ArrayBlockingQueue q = populatedQueue(SIZE);
640 <        Object[] o = q.toArray();
641 <        try {
642 <        for(int i = 0; i < o.length; i++)
643 <            assertEquals(o[i], q.take());
569 <        } catch (InterruptedException e){
570 <            fail("Unexpected exception");
571 <        }    
572 <    }
573 <
574 <    public void testToArray2(){
575 <        ArrayBlockingQueue q = populatedQueue(SIZE);
576 <        Integer[] ints = new Integer[SIZE];
577 <        ints = (Integer[])q.toArray(ints);
578 <        try {
579 <            for(int i = 0; i < ints.length; i++)
580 <                assertEquals(ints[i], q.take());
581 <        } catch (InterruptedException e){
582 <            fail("Unexpected exception");
583 <        }    
584 <    }
585 <    
586 <    public void testIterator(){
587 <        ArrayBlockingQueue q = populatedQueue(SIZE);
588 <        Iterator it = q.iterator();
589 <        try {
590 <            while(it.hasNext()){
591 <                assertEquals(it.next(), q.take());
592 <            }
593 <        } catch (InterruptedException e){
594 <            fail("Unexpected exception");
595 <        }    
640 >        Integer[] ints = new Integer[SIZE];
641 >        ints = (Integer[])q.toArray(ints);
642 >        for (int i = 0; i < ints.length; i++)
643 >            assertEquals(ints[i], q.take());
644      }
645  
646 <    public void testIteratorOrdering() {
646 >    /**
647 >     * toArray(null) throws NPE
648 >     */
649 >    public void testToArray_BadArg() {
650 >        try {
651 >            ArrayBlockingQueue q = populatedQueue(SIZE);
652 >            Object o[] = q.toArray(null);
653 >            shouldThrow();
654 >        } catch (NullPointerException success) {}
655 >    }
656 >
657 >    /**
658 >     * toArray with incompatible array type throws CCE
659 >     */
660 >    public void testToArray1_BadArg() {
661 >        try {
662 >            ArrayBlockingQueue q = populatedQueue(SIZE);
663 >            Object o[] = q.toArray(new String[10] );
664 >            shouldThrow();
665 >        } catch (ArrayStoreException success) {}
666 >    }
667 >
668 >
669 >    /**
670 >     * iterator iterates through all elements
671 >     */
672 >    public void testIterator() throws InterruptedException {
673 >        ArrayBlockingQueue q = populatedQueue(SIZE);
674 >        Iterator it = q.iterator();
675 >        while (it.hasNext()) {
676 >            assertEquals(it.next(), q.take());
677 >        }
678 >    }
679  
680 +    /**
681 +     * iterator.remove removes current element
682 +     */
683 +    public void testIteratorRemove () {
684          final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
685 +        q.add(two);
686 +        q.add(one);
687 +        q.add(three);
688  
689 +        Iterator it = q.iterator();
690 +        it.next();
691 +        it.remove();
692 +
693 +        it = q.iterator();
694 +        assertEquals(it.next(), one);
695 +        assertEquals(it.next(), three);
696 +        assertFalse(it.hasNext());
697 +    }
698 +
699 +    /**
700 +     * iterator ordering is FIFO
701 +     */
702 +    public void testIteratorOrdering() {
703 +        final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
704          q.add(one);
705          q.add(two);
706          q.add(three);
# Line 608 | Line 710 | public class ArrayBlockingQueueTest exte
710          int k = 0;
711          for (Iterator it = q.iterator(); it.hasNext();) {
712              int i = ((Integer)(it.next())).intValue();
713 <            assertEquals("items should come out in order", ++k, i);
713 >            assertEquals(++k, i);
714          }
715 <
614 <        assertEquals("should go through 3 elements", 3, k);
715 >        assertEquals(3, k);
716      }
717  
718 +    /**
719 +     * Modifications do not cause iterators to fail
720 +     */
721      public void testWeaklyConsistentIteration () {
618
722          final ArrayBlockingQueue q = new ArrayBlockingQueue(3);
620
723          q.add(one);
724          q.add(two);
725          q.add(three);
726 <
727 <        try {
728 <            for (Iterator it = q.iterator(); it.hasNext();) {
627 <                q.remove();
628 <                it.next();
629 <            }
630 <        }
631 <        catch (ConcurrentModificationException e) {
632 <            fail("weakly consistent iterator; should not get CME");
726 >        for (Iterator it = q.iterator(); it.hasNext();) {
727 >            q.remove();
728 >            it.next();
729          }
730 <
635 <        assertEquals("queue should be empty again", 0, q.size());
730 >        assertEquals(0, q.size());
731      }
732  
733  
734 <    public void testToString(){
734 >    /**
735 >     * toString contains toStrings of elements
736 >     */
737 >    public void testToString() {
738          ArrayBlockingQueue q = populatedQueue(SIZE);
739          String s = q.toString();
740          for (int i = 0; i < SIZE; ++i) {
741              assertTrue(s.indexOf(String.valueOf(i)) >= 0);
742          }
743 <    }        
743 >    }
744  
745  
746 +    /**
747 +     * offer transfers elements across Executor tasks
748 +     */
749      public void testOfferInExecutor() {
649
750          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
651
751          q.add(one);
752          q.add(two);
654
753          ExecutorService executor = Executors.newFixedThreadPool(2);
754 <
755 <        executor.execute(new Runnable() {
658 <            public void run() {
754 >        executor.execute(new CheckedRunnable() {
755 >            public void realRun() throws InterruptedException {
756                  threadAssertFalse(q.offer(three));
757 <                try {
758 <                    threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
759 <                    threadAssertEquals(0, q.remainingCapacity());
760 <                }
761 <                catch (InterruptedException e) {
762 <                    threadFail("should not be interrupted");
763 <                }
764 <            }
765 <        });
757 >                threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, MILLISECONDS));
758 >                threadAssertEquals(0, q.remainingCapacity());
759 >            }});
760 >
761 >        executor.execute(new CheckedRunnable() {
762 >            public void realRun() throws InterruptedException {
763 >                Thread.sleep(SMALL_DELAY_MS);
764 >                threadAssertEquals(one, q.take());
765 >            }});
766  
670        executor.execute(new Runnable() {
671            public void run() {
672                try {
673                    Thread.sleep(SMALL_DELAY_MS);
674                    threadAssertEquals(one, q.take());
675                }
676                catch (InterruptedException e) {
677                    threadFail("should not be interrupted");
678                }
679            }
680        });
681        
767          joinPool(executor);
768  
769      }
770  
771 +    /**
772 +     * poll retrieves elements across Executor threads
773 +     */
774      public void testPollInExecutor() {
687
775          final ArrayBlockingQueue q = new ArrayBlockingQueue(2);
689
776          ExecutorService executor = Executors.newFixedThreadPool(2);
777 <
778 <        executor.execute(new Runnable() {
693 <            public void run() {
777 >        executor.execute(new CheckedRunnable() {
778 >            public void realRun() throws InterruptedException {
779                  threadAssertNull(q.poll());
780 <                try {
781 <                    threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
782 <                    threadAssertTrue(q.isEmpty());
783 <                }
784 <                catch (InterruptedException e) {
785 <                    threadFail("should not be interrupted");
786 <                }
787 <            }
788 <        });
780 >                threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, MILLISECONDS));
781 >                threadAssertTrue(q.isEmpty());
782 >            }});
783 >
784 >        executor.execute(new CheckedRunnable() {
785 >            public void realRun() throws InterruptedException {
786 >                Thread.sleep(SMALL_DELAY_MS);
787 >                q.put(one);
788 >            }});
789  
705        executor.execute(new Runnable() {
706            public void run() {
707                try {
708                    Thread.sleep(SMALL_DELAY_MS);
709                    q.put(one);
710                }
711                catch (InterruptedException e) {
712                    threadFail("should not be interrupted");
713                }
714            }
715        });
716        
790          joinPool(executor);
791 +    }
792  
793 +    /**
794 +     * A deserialized serialized queue has same elements in same order
795 +     */
796 +    public void testSerialization() throws Exception {
797 +        ArrayBlockingQueue q = populatedQueue(SIZE);
798 +
799 +        ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
800 +        ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
801 +        out.writeObject(q);
802 +        out.close();
803 +
804 +        ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
805 +        ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
806 +        ArrayBlockingQueue r = (ArrayBlockingQueue)in.readObject();
807 +        assertEquals(q.size(), r.size());
808 +        while (!q.isEmpty())
809 +            assertEquals(q.remove(), r.remove());
810      }
811  
812 <    public void testSerialization() {
812 >    /**
813 >     * drainTo(null) throws NPE
814 >     */
815 >    public void testDrainToNull() {
816          ArrayBlockingQueue q = populatedQueue(SIZE);
817 +        try {
818 +            q.drainTo(null);
819 +            shouldThrow();
820 +        } catch (NullPointerException success) {}
821 +    }
822  
823 +    /**
824 +     * drainTo(this) throws IAE
825 +     */
826 +    public void testDrainToSelf() {
827 +        ArrayBlockingQueue q = populatedQueue(SIZE);
828          try {
829 <            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
830 <            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
831 <            out.writeObject(q);
832 <            out.close();
833 <
834 <            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
835 <            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
836 <            ArrayBlockingQueue r = (ArrayBlockingQueue)in.readObject();
837 <            assertEquals(q.size(), r.size());
838 <            while (!q.isEmpty())
839 <                assertEquals(q.remove(), r.remove());
840 <        } catch(Exception e){
841 <            fail("unexpected exception");
842 <        }
829 >            q.drainTo(q);
830 >            shouldThrow();
831 >        } catch (IllegalArgumentException success) {}
832 >    }
833 >
834 >    /**
835 >     * drainTo(c) empties queue into another collection c
836 >     */
837 >    public void testDrainTo() {
838 >        ArrayBlockingQueue q = populatedQueue(SIZE);
839 >        ArrayList l = new ArrayList();
840 >        q.drainTo(l);
841 >        assertEquals(q.size(), 0);
842 >        assertEquals(l.size(), SIZE);
843 >        for (int i = 0; i < SIZE; ++i)
844 >            assertEquals(l.get(i), new Integer(i));
845 >        q.add(zero);
846 >        q.add(one);
847 >        assertFalse(q.isEmpty());
848 >        assertTrue(q.contains(zero));
849 >        assertTrue(q.contains(one));
850 >        l.clear();
851 >        q.drainTo(l);
852 >        assertEquals(q.size(), 0);
853 >        assertEquals(l.size(), 2);
854 >        for (int i = 0; i < 2; ++i)
855 >            assertEquals(l.get(i), new Integer(i));
856 >    }
857 >
858 >    /**
859 >     * drainTo empties full queue, unblocking a waiting put.
860 >     */
861 >    public void testDrainToWithActivePut() throws InterruptedException {
862 >        final ArrayBlockingQueue q = populatedQueue(SIZE);
863 >        Thread t = new Thread(new CheckedRunnable() {
864 >            public void realRun() throws InterruptedException {
865 >                q.put(new Integer(SIZE+1));
866 >            }});
867 >
868 >        t.start();
869 >        ArrayList l = new ArrayList();
870 >        q.drainTo(l);
871 >        assertTrue(l.size() >= SIZE);
872 >        for (int i = 0; i < SIZE; ++i)
873 >            assertEquals(l.get(i), new Integer(i));
874 >        t.join();
875 >        assertTrue(q.size() + l.size() >= SIZE);
876 >    }
877 >
878 >    /**
879 >     * drainTo(null, n) throws NPE
880 >     */
881 >    public void testDrainToNullN() {
882 >        ArrayBlockingQueue q = populatedQueue(SIZE);
883 >        try {
884 >            q.drainTo(null, 0);
885 >            shouldThrow();
886 >        } catch (NullPointerException success) {}
887      }
888  
889 +    /**
890 +     * drainTo(this, n) throws IAE
891 +     */
892 +    public void testDrainToSelfN() {
893 +        ArrayBlockingQueue q = populatedQueue(SIZE);
894 +        try {
895 +            q.drainTo(q, 0);
896 +            shouldThrow();
897 +        } catch (IllegalArgumentException success) {}
898 +    }
899 +
900 +    /**
901 +     * drainTo(c, n) empties first max {n, size} elements of queue into c
902 +     */
903 +    public void testDrainToN() {
904 +        ArrayBlockingQueue q = new ArrayBlockingQueue(SIZE*2);
905 +        for (int i = 0; i < SIZE + 2; ++i) {
906 +            for (int j = 0; j < SIZE; j++)
907 +                assertTrue(q.offer(new Integer(j)));
908 +            ArrayList l = new ArrayList();
909 +            q.drainTo(l, i);
910 +            int k = (i < SIZE)? i : SIZE;
911 +            assertEquals(l.size(), k);
912 +            assertEquals(q.size(), SIZE-k);
913 +            for (int j = 0; j < k; ++j)
914 +                assertEquals(l.get(j), new Integer(j));
915 +            while (q.poll() != null) ;
916 +        }
917 +    }
918  
919   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines