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.54 by jsr166, Wed Dec 31 19:05:42 2014 UTC

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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines