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.60 by jsr166, Sat May 23 00:53:08 2015 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 >        main(suite(), args);
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 127 | Line 148 | public class LinkedBlockingQueueTest ext
148       * remainingCapacity decreases on add, increases on remove
149       */
150      public void testRemainingCapacity() {
151 <        LinkedBlockingQueue q = populatedQueue(SIZE);
151 >        BlockingQueue q = populatedQueue(SIZE);
152          for (int i = 0; i < SIZE; ++i) {
153              assertEquals(i, q.remainingCapacity());
154 <            assertEquals(SIZE-i, q.size());
155 <            q.remove();
154 >            assertEquals(SIZE, q.size() + q.remainingCapacity());
155 >            assertEquals(i, q.remove());
156          }
157          for (int i = 0; i < SIZE; ++i) {
158 <            assertEquals(SIZE-i, q.remainingCapacity());
159 <            assertEquals(i, q.size());
160 <            q.add(new Integer(i));
158 >            assertEquals(SIZE - i, q.remainingCapacity());
159 >            assertEquals(SIZE, q.size() + q.remainingCapacity());
160 >            assertTrue(q.add(i));
161          }
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 x = new Integer(i);
251 >            q.put(x);
252 >            assertTrue(q.contains(x));
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 645 | Line 574 | public class LinkedBlockingQueueTest ext
574                  assertTrue(changed);
575  
576              assertTrue(q.containsAll(p));
577 <            assertEquals(SIZE-i, q.size());
577 >            assertEquals(SIZE - i, q.size());
578              p.remove();
579          }
580      }
# Line 658 | Line 587 | public class LinkedBlockingQueueTest ext
587              LinkedBlockingQueue q = populatedQueue(SIZE);
588              LinkedBlockingQueue p = populatedQueue(i);
589              assertTrue(q.removeAll(p));
590 <            assertEquals(SIZE-i, q.size());
590 >            assertEquals(SIZE - i, q.size());
591              for (int j = 0; j < i; ++j) {
592 <                Integer I = (Integer)(p.remove());
593 <                assertFalse(q.contains(I));
592 >                Integer x = (Integer)(p.remove());
593 >                assertFalse(q.contains(x));
594              }
595          }
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++)
606 <            assertEquals(o[i], q.take());
607 <        } catch (InterruptedException e){
608 <            unexpectedException();
609 <        }    
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 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(a) contains all elements
622 <     */
623 <    public void testToArray2() {
624 <        LinkedBlockingQueue q = populatedQueue(SIZE);
625 <        Integer[] ints = new Integer[SIZE];
626 <        ints = (Integer[])q.toArray(ints);
627 <        try {
628 <            for(int i = 0; i < ints.length; i++)
692 <                assertEquals(ints[i], q.take());
693 <        } catch (InterruptedException e){
694 <            unexpectedException();
695 <        }    
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());
640 <            }
641 <        } catch (InterruptedException e){
642 <            unexpectedException();
643 <        }    
636 >        Iterator it = q.iterator();
637 >        int i;
638 >        for (i = 0; it.hasNext(); i++)
639 >            assertTrue(q.contains(it.next()));
640 >        assertEquals(i, SIZE);
641 >        assertIteratorExhausted(it);
642 >
643 >        it = q.iterator();
644 >        for (i = 0; it.hasNext(); i++)
645 >            assertEquals(it.next(), q.take());
646 >        assertEquals(i, SIZE);
647 >        assertIteratorExhausted(it);
648 >    }
649 >
650 >    /**
651 >     * iterator of empty collection has no elements
652 >     */
653 >    public void testEmptyIterator() {
654 >        assertIteratorExhausted(new LinkedBlockingQueue().iterator());
655      }
656  
657      /**
658       * iterator.remove removes current element
659       */
660 <    public void testIteratorRemove () {
660 >    public void testIteratorRemove() {
661          final LinkedBlockingQueue q = new LinkedBlockingQueue(3);
662          q.add(two);
663          q.add(one);
# Line 722 | Line 666 | public class LinkedBlockingQueueTest ext
666          Iterator it = q.iterator();
667          it.next();
668          it.remove();
669 <        
669 >
670          it = q.iterator();
671 <        assertEquals(it.next(), one);
672 <        assertEquals(it.next(), three);
671 >        assertSame(it.next(), one);
672 >        assertSame(it.next(), three);
673          assertFalse(it.hasNext());
674      }
675  
732
676      /**
677       * iterator ordering is FIFO
678       */
# Line 741 | Line 684 | public class LinkedBlockingQueueTest ext
684          assertEquals(0, q.remainingCapacity());
685          int k = 0;
686          for (Iterator it = q.iterator(); it.hasNext();) {
687 <            int i = ((Integer)(it.next())).intValue();
745 <            assertEquals(++k, i);
687 >            assertEquals(++k, it.next());
688          }
689          assertEquals(3, k);
690      }
# Line 750 | Line 692 | public class LinkedBlockingQueueTest ext
692      /**
693       * Modifications do not cause iterators to fail
694       */
695 <    public void testWeaklyConsistentIteration () {
695 >    public void testWeaklyConsistentIteration() {
696          final LinkedBlockingQueue q = new LinkedBlockingQueue(3);
697          q.add(one);
698          q.add(two);
699          q.add(three);
700 <        try {
701 <            for (Iterator it = q.iterator(); it.hasNext();) {
702 <                q.remove();
761 <                it.next();
762 <            }
763 <        }
764 <        catch (ConcurrentModificationException e) {
765 <            unexpectedException();
700 >        for (Iterator it = q.iterator(); it.hasNext();) {
701 >            q.remove();
702 >            it.next();
703          }
704          assertEquals(0, q.size());
705      }
706  
770
707      /**
708       * toString contains toStrings of elements
709       */
# Line 775 | Line 711 | public class LinkedBlockingQueueTest ext
711          LinkedBlockingQueue q = populatedQueue(SIZE);
712          String s = q.toString();
713          for (int i = 0; i < SIZE; ++i) {
714 <            assertTrue(s.indexOf(String.valueOf(i)) >= 0);
714 >            assertTrue(s.contains(String.valueOf(i)));
715          }
716 <    }        
781 <
716 >    }
717  
718      /**
719       * offer transfers elements across Executor tasks
# Line 788 | Line 723 | public class LinkedBlockingQueueTest ext
723          q.add(one);
724          q.add(two);
725          ExecutorService executor = Executors.newFixedThreadPool(2);
726 <        executor.execute(new Runnable() {
727 <            public void run() {
728 <                threadAssertFalse(q.offer(three));
729 <                try {
730 <                    threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
731 <                    threadAssertEquals(0, q.remainingCapacity());
732 <                }
733 <                catch (InterruptedException e) {
734 <                    threadUnexpectedException();
735 <                }
736 <            }
737 <        });
726 >        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
727 >        executor.execute(new CheckedRunnable() {
728 >            public void realRun() throws InterruptedException {
729 >                assertFalse(q.offer(three));
730 >                threadsStarted.await();
731 >                assertTrue(q.offer(three, LONG_DELAY_MS, MILLISECONDS));
732 >                assertEquals(0, q.remainingCapacity());
733 >            }});
734 >
735 >        executor.execute(new CheckedRunnable() {
736 >            public void realRun() throws InterruptedException {
737 >                threadsStarted.await();
738 >                assertSame(one, q.take());
739 >            }});
740  
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        
741          joinPool(executor);
742      }
743  
744      /**
745 <     * poll retrieves elements across Executor threads
745 >     * timed poll retrieves elements across Executor threads
746       */
747      public void testPollInExecutor() {
748          final LinkedBlockingQueue q = new LinkedBlockingQueue(2);
749 +        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
750          ExecutorService executor = Executors.newFixedThreadPool(2);
751 <        executor.execute(new Runnable() {
752 <            public void run() {
753 <                threadAssertNull(q.poll());
754 <                try {
755 <                    threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS));
756 <                    threadAssertTrue(q.isEmpty());
757 <                }
758 <                catch (InterruptedException e) {
759 <                    threadUnexpectedException();
760 <                }
761 <            }
762 <        });
751 >        executor.execute(new CheckedRunnable() {
752 >            public void realRun() throws InterruptedException {
753 >                assertNull(q.poll());
754 >                threadsStarted.await();
755 >                assertSame(one, q.poll(LONG_DELAY_MS, MILLISECONDS));
756 >                checkEmpty(q);
757 >            }});
758 >
759 >        executor.execute(new CheckedRunnable() {
760 >            public void realRun() throws InterruptedException {
761 >                threadsStarted.await();
762 >                q.put(one);
763 >            }});
764  
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        
765          joinPool(executor);
766      }
767  
768      /**
769       * A deserialized serialized queue has same elements in same order
770       */
771 <    public void testSerialization() {
771 >    public void testSerialization() throws Exception {
772 >        Queue x = populatedQueue(SIZE);
773 >        Queue y = serialClone(x);
774 >
775 >        assertNotSame(x, y);
776 >        assertEquals(x.size(), y.size());
777 >        assertEquals(x.toString(), y.toString());
778 >        assertTrue(Arrays.equals(x.toArray(), y.toArray()));
779 >        while (!x.isEmpty()) {
780 >            assertFalse(y.isEmpty());
781 >            assertEquals(x.remove(), y.remove());
782 >        }
783 >        assertTrue(y.isEmpty());
784 >    }
785 >
786 >    /**
787 >     * drainTo(c) empties queue into another collection c
788 >     */
789 >    public void testDrainTo() {
790          LinkedBlockingQueue q = populatedQueue(SIZE);
791 +        ArrayList l = new ArrayList();
792 +        q.drainTo(l);
793 +        assertEquals(0, q.size());
794 +        assertEquals(SIZE, l.size());
795 +        for (int i = 0; i < SIZE; ++i)
796 +            assertEquals(l.get(i), new Integer(i));
797 +        q.add(zero);
798 +        q.add(one);
799 +        assertFalse(q.isEmpty());
800 +        assertTrue(q.contains(zero));
801 +        assertTrue(q.contains(one));
802 +        l.clear();
803 +        q.drainTo(l);
804 +        assertEquals(0, q.size());
805 +        assertEquals(2, l.size());
806 +        for (int i = 0; i < 2; ++i)
807 +            assertEquals(l.get(i), new Integer(i));
808 +    }
809  
810 <        try {
811 <            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
812 <            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
813 <            out.writeObject(q);
814 <            out.close();
815 <
816 <            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
817 <            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
818 <            LinkedBlockingQueue r = (LinkedBlockingQueue)in.readObject();
819 <            assertEquals(q.size(), r.size());
820 <            while (!q.isEmpty())
821 <                assertEquals(q.remove(), r.remove());
822 <        } catch(Exception e){
823 <            unexpectedException();
810 >    /**
811 >     * drainTo empties full queue, unblocking a waiting put.
812 >     */
813 >    public void testDrainToWithActivePut() throws InterruptedException {
814 >        final LinkedBlockingQueue q = populatedQueue(SIZE);
815 >        Thread t = new Thread(new CheckedRunnable() {
816 >            public void realRun() throws InterruptedException {
817 >                q.put(new Integer(SIZE + 1));
818 >            }});
819 >
820 >        t.start();
821 >        ArrayList l = new ArrayList();
822 >        q.drainTo(l);
823 >        assertTrue(l.size() >= SIZE);
824 >        for (int i = 0; i < SIZE; ++i)
825 >            assertEquals(l.get(i), new Integer(i));
826 >        t.join();
827 >        assertTrue(q.size() + l.size() >= SIZE);
828 >    }
829 >
830 >    /**
831 >     * drainTo(c, n) empties first min(n, size) elements of queue into c
832 >     */
833 >    public void testDrainToN() {
834 >        LinkedBlockingQueue q = new LinkedBlockingQueue();
835 >        for (int i = 0; i < SIZE + 2; ++i) {
836 >            for (int j = 0; j < SIZE; j++)
837 >                assertTrue(q.offer(new Integer(j)));
838 >            ArrayList l = new ArrayList();
839 >            q.drainTo(l, i);
840 >            int k = (i < SIZE) ? i : SIZE;
841 >            assertEquals(k, l.size());
842 >            assertEquals(SIZE - k, q.size());
843 >            for (int j = 0; j < k; ++j)
844 >                assertEquals(l.get(j), new Integer(j));
845 >            do {} while (q.poll() != null);
846 >        }
847 >    }
848 >
849 >    /**
850 >     * remove(null), contains(null) always return false
851 >     */
852 >    public void testNeverContainsNull() {
853 >        Collection<?>[] qs = {
854 >            new LinkedBlockingQueue<Object>(),
855 >            populatedQueue(2),
856 >        };
857 >
858 >        for (Collection<?> q : qs) {
859 >            assertFalse(q.contains(null));
860 >            assertFalse(q.remove(null));
861          }
862      }
863  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines