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

Comparing jsr166/src/test/tck/LinkedBlockingQueueTest.java (file contents):
Revision 1.5 by dl, Thu Sep 25 11:02:41 2003 UTC vs.
Revision 1.45 by jsr166, Mon May 30 22:43:20 2011 UTC

# Line 1 | Line 1
1   /*
2 < * Written by members of JCP JSR-166 Expert Group and released to the
3 < * public domain. Use, modify, and redistribute this code in any way
4 < * without acknowledgement. Other contributors include Andrew Wright,
5 < * Jeffrey Hayes, Pat Fischer, Mike Judd.
2 > * Written by Doug Lea with assistance from members of JCP JSR-166
3 > * Expert Group and released to the public domain, as explained at
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5 > * Other contributors include Andrew Wright, Jeffrey Hayes,
6 > * Pat Fisher, Mike Judd.
7   */
8  
9   import junit.framework.*;
10 < import java.util.*;
11 < import java.util.concurrent.*;
10 > import java.util.Arrays;
11 > import java.util.ArrayList;
12 > import java.util.Collection;
13 > import java.util.Iterator;
14 > import java.util.NoSuchElementException;
15 > import java.util.concurrent.BlockingQueue;
16 > import java.util.concurrent.CountDownLatch;
17 > import java.util.concurrent.LinkedBlockingQueue;
18 > import java.util.concurrent.Executors;
19 > import java.util.concurrent.ExecutorService;
20 > import static java.util.concurrent.TimeUnit.MILLISECONDS;
21   import java.io.*;
22  
23   public class LinkedBlockingQueueTest extends JSR166TestCase {
24  
25 +    public static class Unbounded extends BlockingQueueTest {
26 +        protected BlockingQueue emptyCollection() {
27 +            return new LinkedBlockingQueue();
28 +        }
29 +    }
30 +
31 +    public static class Bounded extends BlockingQueueTest {
32 +        protected BlockingQueue emptyCollection() {
33 +            return new LinkedBlockingQueue(20);
34 +        }
35 +    }
36 +
37      public static void main(String[] args) {
38 <        junit.textui.TestRunner.run (suite());  
38 >        junit.textui.TestRunner.run(suite());
39      }
40  
41      public static Test suite() {
42 <        return new TestSuite(LinkedBlockingQueueTest.class);
42 >        return newTestSuite(LinkedBlockingQueueTest.class,
43 >                            new Unbounded().testSuite(),
44 >                            new Bounded().testSuite());
45      }
46  
23
47      /**
48       * Create a queue of given size containing consecutive
49       * Integers 0 ... n.
50       */
51 <    private LinkedBlockingQueue populatedQueue(int n) {
52 <        LinkedBlockingQueue q = new LinkedBlockingQueue(n);
51 >    private LinkedBlockingQueue<Integer> populatedQueue(int n) {
52 >        LinkedBlockingQueue<Integer> q =
53 >            new LinkedBlockingQueue<Integer>(n);
54          assertTrue(q.isEmpty());
55 <        for(int i = 0; i < n; i++)
56 <            assertTrue(q.offer(new Integer(i)));
55 >        for (int i = 0; i < n; i++)
56 >            assertTrue(q.offer(new Integer(i)));
57          assertFalse(q.isEmpty());
58          assertEquals(0, q.remainingCapacity());
59 <        assertEquals(n, q.size());
59 >        assertEquals(n, q.size());
60          return q;
61      }
62 <
62 >
63      /**
64       * A new queue has the indicated capacity, or Integer.MAX_VALUE if
65       * none given
# Line 46 | Line 70 | public class LinkedBlockingQueueTest ext
70      }
71  
72      /**
73 <     * Constructor throws IAE if  capacity argument nonpositive
73 >     * Constructor throws IllegalArgumentException if capacity argument nonpositive
74       */
75      public void testConstructor2() {
76          try {
77 <            LinkedBlockingQueue q = new LinkedBlockingQueue(0);
77 >            new LinkedBlockingQueue(0);
78              shouldThrow();
79 <        }
56 <        catch (IllegalArgumentException success) {}
79 >        } catch (IllegalArgumentException success) {}
80      }
81  
82      /**
83 <     * Initializing from null Collection throws NPE
83 >     * Initializing from null Collection throws NullPointerException
84       */
85      public void testConstructor3() {
86          try {
87 <            LinkedBlockingQueue q = new LinkedBlockingQueue(null);
87 >            new LinkedBlockingQueue(null);
88              shouldThrow();
89 <        }
67 <        catch (NullPointerException success) {}
89 >        } catch (NullPointerException success) {}
90      }
91  
92      /**
93 <     * Initializing from Collection of null elements throws NPE
93 >     * Initializing from Collection of null elements throws NullPointerException
94       */
95      public void testConstructor4() {
96 +        Collection<Integer> elements = Arrays.asList(new Integer[SIZE]);
97          try {
98 <            Integer[] ints = new Integer[SIZE];
76 <            LinkedBlockingQueue q = new LinkedBlockingQueue(Arrays.asList(ints));
98 >            new LinkedBlockingQueue(elements);
99              shouldThrow();
100 <        }
79 <        catch (NullPointerException success) {}
100 >        } catch (NullPointerException success) {}
101      }
102  
103      /**
104 <     * Initializing from Collection with some null elements throws NPE
104 >     * Initializing from Collection with some null elements throws
105 >     * NullPointerException
106       */
107      public void testConstructor5() {
108 +        Integer[] ints = new Integer[SIZE];
109 +        for (int i = 0; i < SIZE-1; ++i)
110 +            ints[i] = new Integer(i);
111 +        Collection<Integer> elements = Arrays.asList(ints);
112          try {
113 <            Integer[] ints = new Integer[SIZE];
88 <            for (int i = 0; i < SIZE-1; ++i)
89 <                ints[i] = new Integer(i);
90 <            LinkedBlockingQueue q = new LinkedBlockingQueue(Arrays.asList(ints));
113 >            new LinkedBlockingQueue(elements);
114              shouldThrow();
115 <        }
93 <        catch (NullPointerException success) {}
115 >        } catch (NullPointerException success) {}
116      }
117  
118      /**
119       * Queue contains all elements of collection used to initialize
120       */
121      public void testConstructor6() {
122 <        try {
123 <            Integer[] ints = new Integer[SIZE];
124 <            for (int i = 0; i < SIZE; ++i)
125 <                ints[i] = new Integer(i);
126 <            LinkedBlockingQueue q = new LinkedBlockingQueue(Arrays.asList(ints));
127 <            for (int i = 0; i < SIZE; ++i)
106 <                assertEquals(ints[i], q.poll());
107 <        }
108 <        finally {}
122 >        Integer[] ints = new Integer[SIZE];
123 >        for (int i = 0; i < SIZE; ++i)
124 >            ints[i] = new Integer(i);
125 >        LinkedBlockingQueue q = new LinkedBlockingQueue(Arrays.asList(ints));
126 >        for (int i = 0; i < SIZE; ++i)
127 >            assertEquals(ints[i], q.poll());
128      }
129  
130      /**
# Line 141 | Line 160 | public class LinkedBlockingQueueTest ext
160      }
161  
162      /**
144     * offer(null) throws NPE
145     */
146    public void testOfferNull() {
147        try {
148            LinkedBlockingQueue q = new LinkedBlockingQueue(1);
149            q.offer(null);
150            shouldThrow();
151        } catch (NullPointerException success) { }  
152    }
153
154    /**
163       * Offer succeeds if not full; fails if full
164       */
165      public void testOffer() {
# Line 161 | Line 169 | public class LinkedBlockingQueueTest ext
169      }
170  
171      /**
172 <     * add succeeds if not full; throws ISE if full
172 >     * add succeeds if not full; throws IllegalStateException if full
173       */
174      public void testAdd() {
175 <        try {
176 <            LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
177 <            for (int i = 0; i < SIZE; ++i) {
178 <                assertTrue(q.add(new Integer(i)));
171 <            }
172 <            assertEquals(0, q.remainingCapacity());
173 <            q.add(new Integer(SIZE));
174 <        } catch (IllegalStateException success){
175 <        }  
176 <    }
177 <
178 <    /**
179 <     * addAll(null) throws NPE
180 <     */
181 <    public void testAddAll1() {
175 >        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
176 >        for (int i = 0; i < SIZE; ++i)
177 >            assertTrue(q.add(new Integer(i)));
178 >        assertEquals(0, q.remainingCapacity());
179          try {
180 <            LinkedBlockingQueue q = new LinkedBlockingQueue(1);
184 <            q.addAll(null);
180 >            q.add(new Integer(SIZE));
181              shouldThrow();
182 <        }
187 <        catch (NullPointerException success) {}
182 >        } catch (IllegalStateException success) {}
183      }
184 +
185      /**
186 <     * addAll of a collection with null elements throws NPE
186 >     * addAll(this) throws IllegalArgumentException
187       */
188 <    public void testAddAll2() {
188 >    public void testAddAllSelf() {
189 >        LinkedBlockingQueue q = populatedQueue(SIZE);
190          try {
191 <            LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
195 <            Integer[] ints = new Integer[SIZE];
196 <            q.addAll(Arrays.asList(ints));
191 >            q.addAll(q);
192              shouldThrow();
193 <        }
199 <        catch (NullPointerException success) {}
193 >        } catch (IllegalArgumentException success) {}
194      }
195 +
196      /**
197       * addAll of a collection with any null elements throws NPE after
198       * possibly adding some elements
199       */
200      public void testAddAll3() {
201 +        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
202 +        Integer[] ints = new Integer[SIZE];
203 +        for (int i = 0; i < SIZE-1; ++i)
204 +            ints[i] = new Integer(i);
205 +        Collection<Integer> elements = Arrays.asList(ints);
206          try {
207 <            LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
208 <            Integer[] ints = new Integer[SIZE];
209 <            for (int i = 0; i < SIZE-1; ++i)
210 <                ints[i] = new Integer(i);
211 <            q.addAll(Arrays.asList(ints));
207 >            q.addAll(elements);
208              shouldThrow();
209 <        }
214 <        catch (NullPointerException success) {}
209 >        } catch (NullPointerException success) {}
210      }
211 +
212      /**
213 <     * addAll throws ISE if not enough room
213 >     * addAll throws IllegalStateException if not enough room
214       */
215      public void testAddAll4() {
216 +        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE - 1);
217 +        Integer[] ints = new Integer[SIZE];
218 +        for (int i = 0; i < SIZE; ++i)
219 +            ints[i] = new Integer(i);
220 +        Collection<Integer> elements = Arrays.asList(ints);
221          try {
222 <            LinkedBlockingQueue q = new LinkedBlockingQueue(1);
222 <            Integer[] ints = new Integer[SIZE];
223 <            for (int i = 0; i < SIZE; ++i)
224 <                ints[i] = new Integer(i);
225 <            q.addAll(Arrays.asList(ints));
222 >            q.addAll(elements);
223              shouldThrow();
224 <        }
228 <        catch (IllegalStateException success) {}
224 >        } catch (IllegalStateException success) {}
225      }
226 +
227      /**
228       * Queue contains all elements, in traversal order, of successful addAll
229       */
230      public void testAddAll5() {
231 <        try {
232 <            Integer[] empty = new Integer[0];
233 <            Integer[] ints = new Integer[SIZE];
234 <            for (int i = 0; i < SIZE; ++i)
235 <                ints[i] = new Integer(i);
236 <            LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
237 <            assertFalse(q.addAll(Arrays.asList(empty)));
238 <            assertTrue(q.addAll(Arrays.asList(ints)));
239 <            for (int i = 0; i < SIZE; ++i)
243 <                assertEquals(ints[i], q.poll());
244 <        }
245 <        finally {}
231 >        Integer[] empty = new Integer[0];
232 >        Integer[] ints = new Integer[SIZE];
233 >        for (int i = 0; i < SIZE; ++i)
234 >            ints[i] = new Integer(i);
235 >        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
236 >        assertFalse(q.addAll(Arrays.asList(empty)));
237 >        assertTrue(q.addAll(Arrays.asList(ints)));
238 >        for (int i = 0; i < SIZE; ++i)
239 >            assertEquals(ints[i], q.poll());
240      }
241  
242      /**
249     * put(null) throws NPE
250     */
251     public void testPutNull() {
252        try {
253            LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
254            q.put(null);
255            shouldThrow();
256        }
257        catch (NullPointerException success){
258        }  
259        catch (InterruptedException ie) {
260            unexpectedException();
261        }
262     }
263
264    /**
243       * all elements successfully put are contained
244       */
245 <     public void testPut() {
246 <         try {
247 <             LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
248 <             for (int i = 0; i < SIZE; ++i) {
249 <                 Integer I = new Integer(i);
250 <                 q.put(I);
273 <                 assertTrue(q.contains(I));
274 <             }
275 <             assertEquals(0, q.remainingCapacity());
276 <         }
277 <        catch (InterruptedException ie) {
278 <            unexpectedException();
245 >    public void testPut() throws InterruptedException {
246 >        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
247 >        for (int i = 0; i < SIZE; ++i) {
248 >            Integer I = new Integer(i);
249 >            q.put(I);
250 >            assertTrue(q.contains(I));
251          }
252 +        assertEquals(0, q.remainingCapacity());
253      }
254  
255      /**
256       * put blocks interruptibly if full
257       */
258 <    public void testBlockingPut() {
259 <        Thread t = new Thread(new Runnable() {
260 <                public void run() {
261 <                    int added = 0;
262 <                    try {
263 <                        LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
264 <                        for (int i = 0; i < SIZE; ++i) {
265 <                            q.put(new Integer(i));
266 <                            ++added;
267 <                        }
268 <                        q.put(new Integer(SIZE));
269 <                        threadShouldThrow();
270 <                    } catch (InterruptedException ie){
271 <                        threadAssertEquals(added, SIZE);
272 <                    }  
273 <                }});
274 <        t.start();
275 <        try {
276 <           Thread.sleep(SHORT_DELAY_MS);
277 <           t.interrupt();
278 <           t.join();
279 <        }
280 <        catch (InterruptedException ie) {
281 <            unexpectedException();
282 <        }
258 >    public void testBlockingPut() throws InterruptedException {
259 >        final LinkedBlockingQueue q = new LinkedBlockingQueue(SIZE);
260 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
261 >        Thread t = newStartedThread(new CheckedRunnable() {
262 >            public void realRun() throws InterruptedException {
263 >                for (int i = 0; i < SIZE; ++i)
264 >                    q.put(i);
265 >                assertEquals(SIZE, q.size());
266 >                assertEquals(0, q.remainingCapacity());
267 >
268 >                Thread.currentThread().interrupt();
269 >                try {
270 >                    q.put(99);
271 >                    shouldThrow();
272 >                } catch (InterruptedException success) {}
273 >                assertFalse(Thread.interrupted());
274 >
275 >                pleaseInterrupt.countDown();
276 >                try {
277 >                    q.put(99);
278 >                    shouldThrow();
279 >                } catch (InterruptedException success) {}
280 >                assertFalse(Thread.interrupted());
281 >            }});
282 >
283 >        await(pleaseInterrupt);
284 >        assertThreadStaysAlive(t);
285 >        t.interrupt();
286 >        awaitTermination(t);
287 >        assertEquals(SIZE, q.size());
288 >        assertEquals(0, q.remainingCapacity());
289      }
290  
291      /**
292 <     * put blocks waiting for take when full
292 >     * put blocks interruptibly waiting for take when full
293       */
294 <    public void testPutWithTake() {
294 >    public void testPutWithTake() throws InterruptedException {
295 >        final int capacity = 2;
296          final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
297 <        Thread t = new Thread(new Runnable() {
298 <                public void run() {
299 <                    int added = 0;
300 <                    try {
301 <                        q.put(new Object());
302 <                        ++added;
303 <                        q.put(new Object());
304 <                        ++added;
305 <                        q.put(new Object());
306 <                        ++added;
307 <                        q.put(new Object());
308 <                        ++added;
309 <                        threadShouldThrow();
310 <                    } catch (InterruptedException e){
311 <                        threadAssertTrue(added >= 2);
312 <                    }
313 <                }
314 <            });
315 <        try {
316 <            t.start();
317 <            Thread.sleep(SHORT_DELAY_MS);
318 <            q.take();
319 <            t.interrupt();
320 <            t.join();
321 <        } catch (Exception e){
322 <            unexpectedException();
343 <        }
297 >        final CountDownLatch pleaseTake = new CountDownLatch(1);
298 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
299 >        Thread t = newStartedThread(new CheckedRunnable() {
300 >            public void realRun() throws InterruptedException {
301 >                for (int i = 0; i < capacity; i++)
302 >                    q.put(i);
303 >                pleaseTake.countDown();
304 >                q.put(86);
305 >
306 >                pleaseInterrupt.countDown();
307 >                try {
308 >                    q.put(99);
309 >                    shouldThrow();
310 >                } catch (InterruptedException success) {}
311 >                assertFalse(Thread.interrupted());
312 >            }});
313 >
314 >        await(pleaseTake);
315 >        assertEquals(q.remainingCapacity(), 0);
316 >        assertEquals(0, q.take());
317 >
318 >        await(pleaseInterrupt);
319 >        assertThreadStaysAlive(t);
320 >        t.interrupt();
321 >        awaitTermination(t);
322 >        assertEquals(q.remainingCapacity(), 0);
323      }
324  
325      /**
# Line 348 | Line 327 | public class LinkedBlockingQueueTest ext
327       */
328      public void testTimedOffer() {
329          final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
330 <        Thread t = new Thread(new Runnable() {
331 <                public void run() {
332 <                    try {
333 <                        q.put(new Object());
334 <                        q.put(new Object());
335 <                        threadAssertFalse(q.offer(new Object(), SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
336 <                        q.offer(new Object(), LONG_DELAY_MS, TimeUnit.MILLISECONDS);
337 <                        threadShouldThrow();
338 <                    } catch (InterruptedException success){}
339 <                }
340 <            });
341 <        
342 <        try {
343 <            t.start();
344 <            Thread.sleep(SMALL_DELAY_MS);
345 <            t.interrupt();
346 <            t.join();
347 <        } catch (Exception e){
348 <            unexpectedException();
370 <        }
330 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
331 >        Thread t = newStartedThread(new CheckedRunnable() {
332 >            public void realRun() throws InterruptedException {
333 >                q.put(new Object());
334 >                q.put(new Object());
335 >                long startTime = System.nanoTime();
336 >                assertFalse(q.offer(new Object(), timeoutMillis(), MILLISECONDS));
337 >                assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
338 >                pleaseInterrupt.countDown();
339 >                try {
340 >                    q.offer(new Object(), 2 * LONG_DELAY_MS, MILLISECONDS);
341 >                    shouldThrow();
342 >                } catch (InterruptedException success) {}
343 >            }});
344 >
345 >        await(pleaseInterrupt);
346 >        assertThreadStaysAlive(t);
347 >        t.interrupt();
348 >        awaitTermination(t);
349      }
350  
351      /**
352       * take retrieves elements in FIFO order
353       */
354 <    public void testTake() {
355 <        try {
356 <            LinkedBlockingQueue q = populatedQueue(SIZE);
357 <            for (int i = 0; i < SIZE; ++i) {
380 <                assertEquals(i, ((Integer)q.take()).intValue());
381 <            }
382 <        } catch (InterruptedException e){
383 <            unexpectedException();
384 <        }  
385 <    }
386 <
387 <    /**
388 <     * take blocks interruptibly when empty
389 <     */
390 <    public void testTakeFromEmpty() {
391 <        final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
392 <        Thread t = new Thread(new Runnable() {
393 <                public void run() {
394 <                    try {
395 <                        q.take();
396 <                        threadShouldThrow();
397 <                    } catch (InterruptedException success){ }                
398 <                }
399 <            });
400 <        try {
401 <            t.start();
402 <            Thread.sleep(SHORT_DELAY_MS);
403 <            t.interrupt();
404 <            t.join();
405 <        } catch (Exception e){
406 <            unexpectedException();
354 >    public void testTake() throws InterruptedException {
355 >        LinkedBlockingQueue q = populatedQueue(SIZE);
356 >        for (int i = 0; i < SIZE; ++i) {
357 >            assertEquals(i, q.take());
358          }
359      }
360  
361      /**
362       * Take removes existing elements until empty, then blocks interruptibly
363       */
364 <    public void testBlockingTake() {
365 <        Thread t = new Thread(new Runnable() {
366 <                public void run() {
367 <                    try {
368 <                        LinkedBlockingQueue q = populatedQueue(SIZE);
369 <                        for (int i = 0; i < SIZE; ++i) {
370 <                            assertEquals(i, ((Integer)q.take()).intValue());
371 <                        }
372 <                        q.take();
373 <                        threadShouldThrow();
374 <                    } catch (InterruptedException success){
375 <                    }  
376 <                }});
377 <        t.start();
378 <        try {
428 <           Thread.sleep(SHORT_DELAY_MS);
429 <           t.interrupt();
430 <           t.join();
431 <        }
432 <        catch (InterruptedException ie) {
433 <            unexpectedException();
434 <        }
435 <    }
364 >    public void testBlockingTake() throws InterruptedException {
365 >        final BlockingQueue q = populatedQueue(SIZE);
366 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
367 >        Thread t = newStartedThread(new CheckedRunnable() {
368 >            public void realRun() throws InterruptedException {
369 >                for (int i = 0; i < SIZE; ++i) {
370 >                    assertEquals(i, q.take());
371 >                }
372 >
373 >                Thread.currentThread().interrupt();
374 >                try {
375 >                    q.take();
376 >                    shouldThrow();
377 >                } catch (InterruptedException success) {}
378 >                assertFalse(Thread.interrupted());
379  
380 +                pleaseInterrupt.countDown();
381 +                try {
382 +                    q.take();
383 +                    shouldThrow();
384 +                } catch (InterruptedException success) {}
385 +                assertFalse(Thread.interrupted());
386 +            }});
387 +
388 +        await(pleaseInterrupt);
389 +        assertThreadStaysAlive(t);
390 +        t.interrupt();
391 +        awaitTermination(t);
392 +    }
393  
394      /**
395       * poll succeeds unless empty
# Line 441 | Line 397 | public class LinkedBlockingQueueTest ext
397      public void testPoll() {
398          LinkedBlockingQueue q = populatedQueue(SIZE);
399          for (int i = 0; i < SIZE; ++i) {
400 <            assertEquals(i, ((Integer)q.poll()).intValue());
400 >            assertEquals(i, q.poll());
401          }
402 <        assertNull(q.poll());
402 >        assertNull(q.poll());
403      }
404  
405      /**
406 <     * timed pool with zero timeout succeeds when non-empty, else times out
406 >     * timed poll with zero timeout succeeds when non-empty, else times out
407       */
408 <    public void testTimedPoll0() {
409 <        try {
410 <            LinkedBlockingQueue q = populatedQueue(SIZE);
411 <            for (int i = 0; i < SIZE; ++i) {
412 <                assertEquals(i, ((Integer)q.poll(0, TimeUnit.MILLISECONDS)).intValue());
413 <            }
458 <            assertNull(q.poll(0, TimeUnit.MILLISECONDS));
459 <        } catch (InterruptedException e){
460 <            unexpectedException();
461 <        }  
408 >    public void testTimedPoll0() throws InterruptedException {
409 >        LinkedBlockingQueue q = populatedQueue(SIZE);
410 >        for (int i = 0; i < SIZE; ++i) {
411 >            assertEquals(i, q.poll(0, MILLISECONDS));
412 >        }
413 >        assertNull(q.poll(0, MILLISECONDS));
414      }
415  
416      /**
417 <     * timed pool with nonzero timeout succeeds when non-empty, else times out
417 >     * timed poll with nonzero timeout succeeds when non-empty, else times out
418       */
419 <    public void testTimedPoll() {
420 <        try {
421 <            LinkedBlockingQueue q = populatedQueue(SIZE);
422 <            for (int i = 0; i < SIZE; ++i) {
423 <                assertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
424 <            }
425 <            assertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
426 <        } catch (InterruptedException e){
427 <            unexpectedException();
428 <        }  
419 >    public void testTimedPoll() throws InterruptedException {
420 >        LinkedBlockingQueue<Integer> q = populatedQueue(SIZE);
421 >        for (int i = 0; i < SIZE; ++i) {
422 >            long startTime = System.nanoTime();
423 >            assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
424 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
425 >        }
426 >        long startTime = System.nanoTime();
427 >        assertNull(q.poll(timeoutMillis(), MILLISECONDS));
428 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
429 >        checkEmpty(q);
430      }
431  
432      /**
433       * Interrupted timed poll throws InterruptedException instead of
434       * returning timeout status
435       */
436 <    public void testInterruptedTimedPoll() {
437 <        Thread t = new Thread(new Runnable() {
438 <                public void run() {
439 <                    try {
440 <                        LinkedBlockingQueue q = populatedQueue(SIZE);
441 <                        for (int i = 0; i < SIZE; ++i) {
442 <                            threadAssertEquals(i, ((Integer)q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS)).intValue());
443 <                        }
444 <                        threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
492 <                    } catch (InterruptedException success){
493 <                    }  
494 <                }});
495 <        t.start();
496 <        try {
497 <           Thread.sleep(SHORT_DELAY_MS);
498 <           t.interrupt();
499 <           t.join();
500 <        }
501 <        catch (InterruptedException ie) {
502 <            unexpectedException();
503 <        }
504 <    }
505 <
506 <    /**
507 <     *  timed poll before a delayed offer fails; after offer succeeds;
508 <     *  on interruption throws
509 <     */
510 <    public void testTimedPollWithOffer() {
511 <        final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
512 <        Thread t = new Thread(new Runnable() {
513 <                public void run() {
514 <                    try {
515 <                        threadAssertNull(q.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
516 <                        q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
517 <                        q.poll(LONG_DELAY_MS, TimeUnit.MILLISECONDS);
518 <                        threadShouldThrow();
519 <                    } catch (InterruptedException success) { }                
436 >    public void testInterruptedTimedPoll() throws InterruptedException {
437 >        final BlockingQueue<Integer> q = populatedQueue(SIZE);
438 >        final CountDownLatch aboutToWait = new CountDownLatch(1);
439 >        Thread t = newStartedThread(new CheckedRunnable() {
440 >            public void realRun() throws InterruptedException {
441 >                for (int i = 0; i < SIZE; ++i) {
442 >                    long t0 = System.nanoTime();
443 >                    assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
444 >                    assertTrue(millisElapsedSince(t0) < SMALL_DELAY_MS);
445                  }
446 <            });
447 <        try {
448 <            t.start();
449 <            Thread.sleep(SMALL_DELAY_MS);
450 <            assertTrue(q.offer(zero, SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
451 <            t.interrupt();
452 <            t.join();
453 <        } catch (Exception e){
454 <            unexpectedException();
455 <        }
456 <    }  
446 >                long t0 = System.nanoTime();
447 >                aboutToWait.countDown();
448 >                try {
449 >                    q.poll(MEDIUM_DELAY_MS, MILLISECONDS);
450 >                    shouldThrow();
451 >                } catch (InterruptedException success) {
452 >                    assertTrue(millisElapsedSince(t0) < MEDIUM_DELAY_MS);
453 >                }
454 >            }});
455 >
456 >        aboutToWait.await();
457 >        waitForThreadToEnterWaitState(t, SMALL_DELAY_MS);
458 >        t.interrupt();
459 >        awaitTermination(t, MEDIUM_DELAY_MS);
460 >        checkEmpty(q);
461 >    }
462  
463      /**
464       * peek returns next element, or null if empty
# Line 536 | Line 466 | public class LinkedBlockingQueueTest ext
466      public void testPeek() {
467          LinkedBlockingQueue q = populatedQueue(SIZE);
468          for (int i = 0; i < SIZE; ++i) {
469 <            assertEquals(i, ((Integer)q.peek()).intValue());
470 <            q.poll();
469 >            assertEquals(i, q.peek());
470 >            assertEquals(i, q.poll());
471              assertTrue(q.peek() == null ||
472 <                       i != ((Integer)q.peek()).intValue());
472 >                       !q.peek().equals(i));
473          }
474 <        assertNull(q.peek());
474 >        assertNull(q.peek());
475      }
476  
477      /**
# Line 550 | Line 480 | public class LinkedBlockingQueueTest ext
480      public void testElement() {
481          LinkedBlockingQueue q = populatedQueue(SIZE);
482          for (int i = 0; i < SIZE; ++i) {
483 <            assertEquals(i, ((Integer)q.element()).intValue());
484 <            q.poll();
483 >            assertEquals(i, q.element());
484 >            assertEquals(i, q.poll());
485          }
486          try {
487              q.element();
488              shouldThrow();
489 <        }
560 <        catch (NoSuchElementException success) {}
489 >        } catch (NoSuchElementException success) {}
490      }
491  
492      /**
# Line 566 | Line 495 | public class LinkedBlockingQueueTest ext
495      public void testRemove() {
496          LinkedBlockingQueue q = populatedQueue(SIZE);
497          for (int i = 0; i < SIZE; ++i) {
498 <            assertEquals(i, ((Integer)q.remove()).intValue());
498 >            assertEquals(i, q.remove());
499          }
500          try {
501              q.remove();
502              shouldThrow();
503 <        } catch (NoSuchElementException success){
575 <        }  
503 >        } catch (NoSuchElementException success) {}
504      }
505  
506      /**
# Line 581 | Line 509 | public class LinkedBlockingQueueTest ext
509      public void testRemoveElement() {
510          LinkedBlockingQueue q = populatedQueue(SIZE);
511          for (int i = 1; i < SIZE; i+=2) {
512 <            assertTrue(q.remove(new Integer(i)));
512 >            assertTrue(q.contains(i));
513 >            assertTrue(q.remove(i));
514 >            assertFalse(q.contains(i));
515 >            assertTrue(q.contains(i-1));
516          }
517          for (int i = 0; i < SIZE; i+=2) {
518 <            assertTrue(q.remove(new Integer(i)));
519 <            assertFalse(q.remove(new Integer(i+1)));
518 >            assertTrue(q.contains(i));
519 >            assertTrue(q.remove(i));
520 >            assertFalse(q.contains(i));
521 >            assertFalse(q.remove(i+1));
522 >            assertFalse(q.contains(i+1));
523          }
524          assertTrue(q.isEmpty());
525      }
526 <        
526 >
527 >    /**
528 >     * An add following remove(x) succeeds
529 >     */
530 >    public void testRemoveElementAndAdd() throws InterruptedException {
531 >        LinkedBlockingQueue q = new LinkedBlockingQueue();
532 >        assertTrue(q.add(new Integer(1)));
533 >        assertTrue(q.add(new Integer(2)));
534 >        assertTrue(q.remove(new Integer(1)));
535 >        assertTrue(q.remove(new Integer(2)));
536 >        assertTrue(q.add(new Integer(3)));
537 >        assertTrue(q.take() != null);
538 >    }
539 >
540      /**
541       * contains(x) reports true when elements added but not yet removed
542       */
# Line 613 | Line 560 | public class LinkedBlockingQueueTest ext
560          assertEquals(SIZE, q.remainingCapacity());
561          q.add(one);
562          assertFalse(q.isEmpty());
563 +        assertTrue(q.contains(one));
564          q.clear();
565          assertTrue(q.isEmpty());
566      }
# Line 667 | Line 615 | public class LinkedBlockingQueueTest ext
615      }
616  
617      /**
618 <     * toArray contains all elements
618 >     * toArray contains all elements in FIFO order
619       */
620      public void testToArray() {
621          LinkedBlockingQueue q = populatedQueue(SIZE);
622 <        Object[] o = q.toArray();
623 <        try {
624 <        for(int i = 0; i < o.length; i++)
625 <            assertEquals(o[i], q.take());
626 <        } catch (InterruptedException e){
627 <            unexpectedException();
628 <        }    
622 >        Object[] o = q.toArray();
623 >        for (int i = 0; i < o.length; i++)
624 >            assertSame(o[i], q.poll());
625 >    }
626 >
627 >    /**
628 >     * toArray(a) contains all elements in FIFO order
629 >     */
630 >    public void testToArray2() throws InterruptedException {
631 >        LinkedBlockingQueue<Integer> q = populatedQueue(SIZE);
632 >        Integer[] ints = new Integer[SIZE];
633 >        Integer[] array = q.toArray(ints);
634 >        assertSame(ints, array);
635 >        for (int i = 0; i < ints.length; i++)
636 >            assertSame(ints[i], q.poll());
637      }
638  
639      /**
640 <     * toArray(a) contains all elements
641 <     */
642 <    public void testToArray2() {
643 <        LinkedBlockingQueue q = populatedQueue(SIZE);
644 <        Integer[] ints = new Integer[SIZE];
645 <        ints = (Integer[])q.toArray(ints);
646 <        try {
647 <            for(int i = 0; i < ints.length; i++)
692 <                assertEquals(ints[i], q.take());
693 <        } catch (InterruptedException e){
694 <            unexpectedException();
695 <        }    
640 >     * toArray(incompatible array type) throws ArrayStoreException
641 >     */
642 >    public void testToArray1_BadArg() {
643 >        LinkedBlockingQueue q = populatedQueue(SIZE);
644 >        try {
645 >            q.toArray(new String[10]);
646 >            shouldThrow();
647 >        } catch (ArrayStoreException success) {}
648      }
649 <    
649 >
650      /**
651       * iterator iterates through all elements
652       */
653 <    public void testIterator() {
653 >    public void testIterator() throws InterruptedException {
654          LinkedBlockingQueue q = populatedQueue(SIZE);
655 <        Iterator it = q.iterator();
656 <        try {
657 <            while(it.hasNext()){
658 <                assertEquals(it.next(), q.take());
707 <            }
708 <        } catch (InterruptedException e){
709 <            unexpectedException();
710 <        }    
655 >        Iterator it = q.iterator();
656 >        while (it.hasNext()) {
657 >            assertEquals(it.next(), q.take());
658 >        }
659      }
660  
661      /**
662       * iterator.remove removes current element
663       */
664 <    public void testIteratorRemove () {
664 >    public void testIteratorRemove() {
665          final LinkedBlockingQueue q = new LinkedBlockingQueue(3);
666          q.add(two);
667          q.add(one);
# Line 722 | Line 670 | public class LinkedBlockingQueueTest ext
670          Iterator it = q.iterator();
671          it.next();
672          it.remove();
673 <        
673 >
674          it = q.iterator();
675 <        assertEquals(it.next(), one);
676 <        assertEquals(it.next(), three);
675 >        assertSame(it.next(), one);
676 >        assertSame(it.next(), three);
677          assertFalse(it.hasNext());
678      }
679  
732
680      /**
681       * iterator ordering is FIFO
682       */
# Line 741 | Line 688 | public class LinkedBlockingQueueTest ext
688          assertEquals(0, q.remainingCapacity());
689          int k = 0;
690          for (Iterator it = q.iterator(); it.hasNext();) {
691 <            int i = ((Integer)(it.next())).intValue();
745 <            assertEquals(++k, i);
691 >            assertEquals(++k, it.next());
692          }
693          assertEquals(3, k);
694      }
# Line 750 | Line 696 | public class LinkedBlockingQueueTest ext
696      /**
697       * Modifications do not cause iterators to fail
698       */
699 <    public void testWeaklyConsistentIteration () {
699 >    public void testWeaklyConsistentIteration() {
700          final LinkedBlockingQueue q = new LinkedBlockingQueue(3);
701          q.add(one);
702          q.add(two);
703          q.add(three);
704 <        try {
705 <            for (Iterator it = q.iterator(); it.hasNext();) {
706 <                q.remove();
761 <                it.next();
762 <            }
763 <        }
764 <        catch (ConcurrentModificationException e) {
765 <            unexpectedException();
704 >        for (Iterator it = q.iterator(); it.hasNext();) {
705 >            q.remove();
706 >            it.next();
707          }
708          assertEquals(0, q.size());
709      }
710  
770
711      /**
712       * toString contains toStrings of elements
713       */
# Line 775 | Line 715 | public class LinkedBlockingQueueTest ext
715          LinkedBlockingQueue q = populatedQueue(SIZE);
716          String s = q.toString();
717          for (int i = 0; i < SIZE; ++i) {
718 <            assertTrue(s.indexOf(String.valueOf(i)) >= 0);
718 >            assertTrue(s.contains(String.valueOf(i)));
719          }
720 <    }        
781 <
720 >    }
721  
722      /**
723       * offer transfers elements across Executor tasks
# Line 788 | Line 727 | public class LinkedBlockingQueueTest ext
727          q.add(one);
728          q.add(two);
729          ExecutorService executor = Executors.newFixedThreadPool(2);
730 <        executor.execute(new Runnable() {
731 <            public void run() {
732 <                threadAssertFalse(q.offer(three));
733 <                try {
734 <                    threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
735 <                    threadAssertEquals(0, q.remainingCapacity());
736 <                }
737 <                catch (InterruptedException e) {
738 <                    threadUnexpectedException();
739 <                }
740 <            }
741 <        });
730 >        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
731 >        executor.execute(new CheckedRunnable() {
732 >            public void realRun() throws InterruptedException {
733 >                assertFalse(q.offer(three));
734 >                threadsStarted.await();
735 >                assertTrue(q.offer(three, LONG_DELAY_MS, MILLISECONDS));
736 >                assertEquals(0, q.remainingCapacity());
737 >            }});
738 >
739 >        executor.execute(new CheckedRunnable() {
740 >            public void realRun() throws InterruptedException {
741 >                threadsStarted.await();
742 >                assertSame(one, q.take());
743 >            }});
744  
804        executor.execute(new Runnable() {
805            public void run() {
806                try {
807                    Thread.sleep(SMALL_DELAY_MS);
808                    threadAssertEquals(one, q.take());
809                }
810                catch (InterruptedException e) {
811                    threadUnexpectedException();
812                }
813            }
814        });
815        
745          joinPool(executor);
746      }
747  
748      /**
749 <     * poll retrieves elements across Executor threads
749 >     * timed poll retrieves elements across Executor threads
750       */
751      public void testPollInExecutor() {
752          final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
753 +        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
754          ExecutorService executor = Executors.newFixedThreadPool(2);
755 <        executor.execute(new Runnable() {
756 <            public void run() {
757 <                threadAssertNull(q.poll());
758 <                try {
759 <                    threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
760 <                    threadAssertTrue(q.isEmpty());
761 <                }
762 <                catch (InterruptedException e) {
763 <                    threadUnexpectedException();
764 <                }
765 <            }
766 <        });
755 >        executor.execute(new CheckedRunnable() {
756 >            public void realRun() throws InterruptedException {
757 >                assertNull(q.poll());
758 >                threadsStarted.await();
759 >                assertSame(one, q.poll(LONG_DELAY_MS, MILLISECONDS));
760 >                checkEmpty(q);
761 >            }});
762 >
763 >        executor.execute(new CheckedRunnable() {
764 >            public void realRun() throws InterruptedException {
765 >                threadsStarted.await();
766 >                q.put(one);
767 >            }});
768  
838        executor.execute(new Runnable() {
839            public void run() {
840                try {
841                    Thread.sleep(SMALL_DELAY_MS);
842                    q.put(one);
843                }
844                catch (InterruptedException e) {
845                    threadUnexpectedException();
846                }
847            }
848        });
849        
769          joinPool(executor);
770      }
771  
772      /**
773       * A deserialized serialized queue has same elements in same order
774       */
775 <    public void testSerialization() {
775 >    public void testSerialization() throws Exception {
776          LinkedBlockingQueue q = populatedQueue(SIZE);
777  
778 <        try {
779 <            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
780 <            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
781 <            out.writeObject(q);
782 <            out.close();
783 <
784 <            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
785 <            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
786 <            LinkedBlockingQueue r = (LinkedBlockingQueue)in.readObject();
787 <            assertEquals(q.size(), r.size());
788 <            while (!q.isEmpty())
789 <                assertEquals(q.remove(), r.remove());
790 <        } catch(Exception e){
791 <            unexpectedException();
778 >        ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
779 >        ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
780 >        out.writeObject(q);
781 >        out.close();
782 >
783 >        ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
784 >        ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
785 >        LinkedBlockingQueue r = (LinkedBlockingQueue)in.readObject();
786 >        assertEquals(q.size(), r.size());
787 >        while (!q.isEmpty())
788 >            assertEquals(q.remove(), r.remove());
789 >    }
790 >
791 >    /**
792 >     * drainTo(c) empties queue into another collection c
793 >     */
794 >    public void testDrainTo() {
795 >        LinkedBlockingQueue q = populatedQueue(SIZE);
796 >        ArrayList l = new ArrayList();
797 >        q.drainTo(l);
798 >        assertEquals(q.size(), 0);
799 >        assertEquals(l.size(), SIZE);
800 >        for (int i = 0; i < SIZE; ++i)
801 >            assertEquals(l.get(i), new Integer(i));
802 >        q.add(zero);
803 >        q.add(one);
804 >        assertFalse(q.isEmpty());
805 >        assertTrue(q.contains(zero));
806 >        assertTrue(q.contains(one));
807 >        l.clear();
808 >        q.drainTo(l);
809 >        assertEquals(q.size(), 0);
810 >        assertEquals(l.size(), 2);
811 >        for (int i = 0; i < 2; ++i)
812 >            assertEquals(l.get(i), new Integer(i));
813 >    }
814 >
815 >    /**
816 >     * drainTo empties full queue, unblocking a waiting put.
817 >     */
818 >    public void testDrainToWithActivePut() throws InterruptedException {
819 >        final LinkedBlockingQueue q = populatedQueue(SIZE);
820 >        Thread t = new Thread(new CheckedRunnable() {
821 >            public void realRun() throws InterruptedException {
822 >                q.put(new Integer(SIZE+1));
823 >            }});
824 >
825 >        t.start();
826 >        ArrayList l = new ArrayList();
827 >        q.drainTo(l);
828 >        assertTrue(l.size() >= SIZE);
829 >        for (int i = 0; i < SIZE; ++i)
830 >            assertEquals(l.get(i), new Integer(i));
831 >        t.join();
832 >        assertTrue(q.size() + l.size() >= SIZE);
833 >    }
834 >
835 >    /**
836 >     * drainTo(c, n) empties first min(n, size) elements of queue into c
837 >     */
838 >    public void testDrainToN() {
839 >        LinkedBlockingQueue q = new LinkedBlockingQueue();
840 >        for (int i = 0; i < SIZE + 2; ++i) {
841 >            for (int j = 0; j < SIZE; j++)
842 >                assertTrue(q.offer(new Integer(j)));
843 >            ArrayList l = new ArrayList();
844 >            q.drainTo(l, i);
845 >            int k = (i < SIZE) ? i : SIZE;
846 >            assertEquals(l.size(), k);
847 >            assertEquals(q.size(), SIZE-k);
848 >            for (int j = 0; j < k; ++j)
849 >                assertEquals(l.get(j), new Integer(j));
850 >            while (q.poll() != null) ;
851          }
852      }
853  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines