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

Comparing jsr166/src/test/tck/LinkedTransferQueueTest.java (file contents):
Revision 1.15 by jsr166, Sat Nov 21 17:38:05 2009 UTC vs.
Revision 1.54 by jsr166, Wed Dec 31 16:44:02 2014 UTC

# Line 1 | Line 1
1   /*
2   * Written by Doug Lea with assistance from members of JCP JSR-166
3   * Expert Group and released to the public domain, as explained at
4 < * http://creativecommons.org/licenses/publicdomain
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5   * Other contributors include John Vint
6   */
7  
8 < import java.io.BufferedInputStream;
9 < import java.io.BufferedOutputStream;
10 < import java.io.ByteArrayInputStream;
11 < import java.io.ByteArrayOutputStream;
12 < import java.io.ObjectInputStream;
13 < import java.io.ObjectOutputStream;
14 < import java.util.ArrayList;
8 > import junit.framework.*;
9   import java.util.Arrays;
10 + import java.util.ArrayList;
11 + import java.util.Collection;
12   import java.util.Iterator;
13   import java.util.List;
14   import java.util.NoSuchElementException;
15 < import java.util.concurrent.*;
15 > import java.util.Queue;
16 > import java.util.concurrent.BlockingQueue;
17 > import java.util.concurrent.CountDownLatch;
18 > import java.util.concurrent.Executors;
19 > import java.util.concurrent.ExecutorService;
20 > import java.util.concurrent.LinkedTransferQueue;
21   import static java.util.concurrent.TimeUnit.MILLISECONDS;
21 import junit.framework.Test;
22 import junit.framework.TestSuite;
22  
23   @SuppressWarnings({"unchecked", "rawtypes"})
24   public class LinkedTransferQueueTest extends JSR166TestCase {
25  
26 +    public static class Generic extends BlockingQueueTest {
27 +        protected BlockingQueue emptyCollection() {
28 +            return new LinkedTransferQueue();
29 +        }
30 +    }
31 +
32      public static void main(String[] args) {
33          junit.textui.TestRunner.run(suite());
34      }
35  
36      public static Test suite() {
37 <        return new TestSuite(LinkedTransferQueueTest.class);
38 <    }
34 <
35 <    void checkEmpty(LinkedTransferQueue q) throws InterruptedException {
36 <        assertTrue(q.isEmpty());
37 <        assertEquals(0, q.size());
38 <        assertNull(q.peek());
39 <        assertNull(q.poll());
40 <        assertNull(q.poll(0, MILLISECONDS));
41 <        assertEquals(q.toString(), "[]");
42 <        assertTrue(Arrays.equals(q.toArray(), new Object[0]));
43 <        assertFalse(q.iterator().hasNext());
44 <        try {
45 <            q.element();
46 <            shouldThrow();
47 <        } catch (NoSuchElementException success) {
48 <        }
49 <        try {
50 <            q.iterator().next();
51 <            shouldThrow();
52 <        } catch (NoSuchElementException success) {
53 <        }
54 <        try {
55 <            q.remove();
56 <            shouldThrow();
57 <        } catch (NoSuchElementException success) {
58 <        }
37 >        return newTestSuite(LinkedTransferQueueTest.class,
38 >                            new Generic().testSuite());
39      }
40  
41      /**
# Line 75 | Line 55 | public class LinkedTransferQueueTest ext
55          try {
56              new LinkedTransferQueue(null);
57              shouldThrow();
58 <        } catch (NullPointerException success) {
79 <        }
58 >        } catch (NullPointerException success) {}
59      }
60  
61      /**
# Line 84 | Line 63 | public class LinkedTransferQueueTest ext
63       * NullPointerException
64       */
65      public void testConstructor3() {
66 +        Collection<Integer> elements = Arrays.asList(new Integer[SIZE]);
67          try {
68 <            Integer[] ints = new Integer[SIZE];
89 <            new LinkedTransferQueue(Arrays.asList(ints));
68 >            new LinkedTransferQueue(elements);
69              shouldThrow();
70 <        } catch (NullPointerException success) {
92 <        }
70 >        } catch (NullPointerException success) {}
71      }
72  
73      /**
# Line 97 | Line 75 | public class LinkedTransferQueueTest ext
75       * throws NullPointerException
76       */
77      public void testConstructor4() {
78 +        Integer[] ints = new Integer[SIZE];
79 +        for (int i = 0; i < SIZE-1; ++i)
80 +            ints[i] = i;
81 +        Collection<Integer> elements = Arrays.asList(ints);
82          try {
83 <            Integer[] ints = new Integer[SIZE];
102 <            for (int i = 0; i < SIZE - 1; ++i) {
103 <                ints[i] = i;
104 <            }
105 <            new LinkedTransferQueue(Arrays.asList(ints));
83 >            new LinkedTransferQueue(elements);
84              shouldThrow();
85 <        } catch (NullPointerException success) {
108 <        }
85 >        } catch (NullPointerException success) {}
86      }
87  
88      /**
# Line 150 | Line 127 | public class LinkedTransferQueueTest ext
127      }
128  
129      /**
153     * offer(null) throws NullPointerException
154     */
155    public void testOfferNull() {
156        try {
157            LinkedTransferQueue q = new LinkedTransferQueue();
158            q.offer(null);
159            shouldThrow();
160        } catch (NullPointerException success) {
161        }
162    }
163
164    /**
165     * add(null) throws NullPointerException
166     */
167    public void testAddNull() {
168        try {
169            LinkedTransferQueue q = new LinkedTransferQueue();
170            q.add(null);
171            shouldThrow();
172        } catch (NullPointerException success) {
173        }
174    }
175
176    /**
177     * addAll(null) throws NullPointerException
178     */
179    public void testAddAll1() {
180        try {
181            LinkedTransferQueue q = new LinkedTransferQueue();
182            q.addAll(null);
183            shouldThrow();
184        } catch (NullPointerException success) {
185        }
186    }
187
188    /**
130       * addAll(this) throws IllegalArgumentException
131       */
132      public void testAddAllSelf() {
# Line 193 | Line 134 | public class LinkedTransferQueueTest ext
134              LinkedTransferQueue q = populatedQueue(SIZE);
135              q.addAll(q);
136              shouldThrow();
137 <        } catch (IllegalArgumentException success) {
197 <        }
198 <    }
199 <
200 <    /**
201 <     * addAll of a collection with null elements throws NullPointerException
202 <     */
203 <    public void testAddAll2() {
204 <        try {
205 <            LinkedTransferQueue q = new LinkedTransferQueue();
206 <            Integer[] ints = new Integer[SIZE];
207 <            q.addAll(Arrays.asList(ints));
208 <            shouldThrow();
209 <        } catch (NullPointerException success) {
210 <        }
137 >        } catch (IllegalArgumentException success) {}
138      }
139  
140      /**
# Line 223 | Line 150 | public class LinkedTransferQueueTest ext
150              }
151              q.addAll(Arrays.asList(ints));
152              shouldThrow();
153 <        } catch (NullPointerException success) {
227 <        }
153 >        } catch (NullPointerException success) {}
154      }
155  
156      /**
# Line 245 | Line 171 | public class LinkedTransferQueueTest ext
171      }
172  
173      /**
248     * put(null) throws NullPointerException
249     */
250    public void testPutNull() throws InterruptedException {
251        try {
252            LinkedTransferQueue q = new LinkedTransferQueue();
253            q.put(null);
254            shouldThrow();
255        } catch (NullPointerException success) {}
256    }
257
258    /**
174       * all elements successfully put are contained
175       */
176      public void testPut() {
177          LinkedTransferQueue<Integer> q = new LinkedTransferQueue<Integer>();
178          for (int i = 0; i < SIZE; ++i) {
179 <            assertEquals(q.size(), i);
179 >            assertEquals(i, q.size());
180              q.put(i);
181              assertTrue(q.contains(i));
182          }
# Line 278 | Line 193 | public class LinkedTransferQueueTest ext
193      }
194  
195      /**
196 <     * take blocks interruptibly when empty
282 <     */
283 <    public void testTakeFromEmpty() throws InterruptedException {
284 <        final LinkedTransferQueue q = new LinkedTransferQueue();
285 <        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
286 <            void realRun() throws InterruptedException {
287 <                q.take();
288 <            }});
289 <        Thread.sleep(SHORT_DELAY_MS);
290 <        t.interrupt();
291 <        t.join();
292 <    }
293 <
294 <    /**
295 <     * Take removes existing elements until empty, then blocks interruptibly
196 >     * take removes existing elements until empty, then blocks interruptibly
197       */
198      public void testBlockingTake() throws InterruptedException {
199 <        final LinkedTransferQueue<Integer> q = populatedQueue(SIZE);
200 <        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
201 <            void realRun() throws InterruptedException {
199 >        final BlockingQueue q = populatedQueue(SIZE);
200 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
201 >        Thread t = newStartedThread(new CheckedRunnable() {
202 >            public void realRun() throws InterruptedException {
203                  for (int i = 0; i < SIZE; ++i) {
204 <                    threadAssertEquals(i, (int) q.take());
204 >                    assertEquals(i, q.take());
205                  }
206 <                q.take();
206 >
207 >                Thread.currentThread().interrupt();
208 >                try {
209 >                    q.take();
210 >                    shouldThrow();
211 >                } catch (InterruptedException success) {}
212 >                assertFalse(Thread.interrupted());
213 >
214 >                pleaseInterrupt.countDown();
215 >                try {
216 >                    q.take();
217 >                    shouldThrow();
218 >                } catch (InterruptedException success) {}
219 >                assertFalse(Thread.interrupted());
220              }});
221 <        Thread.sleep(SMALL_DELAY_MS);
221 >
222 >        await(pleaseInterrupt);
223 >        assertThreadStaysAlive(t);
224          t.interrupt();
225 <        t.join();
309 <        checkEmpty(q);
225 >        awaitTermination(t);
226      }
227  
228      /**
# Line 322 | Line 238 | public class LinkedTransferQueueTest ext
238      }
239  
240      /**
241 <     * timed pool with zero timeout succeeds when non-empty, else times out
241 >     * timed poll with zero timeout succeeds when non-empty, else times out
242       */
243      public void testTimedPoll0() throws InterruptedException {
244          LinkedTransferQueue<Integer> q = populatedQueue(SIZE);
# Line 334 | Line 250 | public class LinkedTransferQueueTest ext
250      }
251  
252      /**
253 <     * timed pool with nonzero timeout succeeds when non-empty, else times out
253 >     * timed poll with nonzero timeout succeeds when non-empty, else times out
254       */
255      public void testTimedPoll() throws InterruptedException {
256          LinkedTransferQueue<Integer> q = populatedQueue(SIZE);
257          for (int i = 0; i < SIZE; ++i) {
258 <            long t0 = System.nanoTime();
258 >            long startTime = System.nanoTime();
259              assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
260 <            long millisElapsed = (System.nanoTime() - t0)/(1024 * 1024);
345 <            assertTrue(millisElapsed < SMALL_DELAY_MS);
260 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
261          }
262 <        assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
262 >        long startTime = System.nanoTime();
263 >        assertNull(q.poll(timeoutMillis(), MILLISECONDS));
264 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
265          checkEmpty(q);
266      }
267  
# Line 353 | Line 270 | public class LinkedTransferQueueTest ext
270       * returning timeout status
271       */
272      public void testInterruptedTimedPoll() throws InterruptedException {
273 <        final LinkedTransferQueue<Integer> q = populatedQueue(SIZE);
274 <        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
275 <            void realRun() throws InterruptedException {
273 >        final BlockingQueue<Integer> q = populatedQueue(SIZE);
274 >        final CountDownLatch aboutToWait = new CountDownLatch(1);
275 >        Thread t = newStartedThread(new CheckedRunnable() {
276 >            public void realRun() throws InterruptedException {
277                  for (int i = 0; i < SIZE; ++i) {
278                      long t0 = System.nanoTime();
279 <                    threadAssertEquals(i, (int) q.poll(LONG_DELAY_MS,
280 <                                                       MILLISECONDS));
281 <                    long millisElapsed = (System.nanoTime() - t0)/(1024 * 1024);
282 <                    assertTrue(millisElapsed < SMALL_DELAY_MS);
279 >                    assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
280 >                    assertTrue(millisElapsedSince(t0) < SMALL_DELAY_MS);
281 >                }
282 >                long t0 = System.nanoTime();
283 >                aboutToWait.countDown();
284 >                try {
285 >                    q.poll(MEDIUM_DELAY_MS, MILLISECONDS);
286 >                    shouldThrow();
287 >                } catch (InterruptedException success) {
288 >                    assertTrue(millisElapsedSince(t0) < MEDIUM_DELAY_MS);
289                  }
366                q.poll(LONG_DELAY_MS, MILLISECONDS);
290              }});
291 <        Thread.sleep(SMALL_DELAY_MS);
291 >
292 >        aboutToWait.await();
293 >        waitForThreadToEnterWaitState(t, SMALL_DELAY_MS);
294          t.interrupt();
295 <        t.join();
295 >        awaitTermination(t, MEDIUM_DELAY_MS);
296          checkEmpty(q);
297      }
298  
299      /**
300 <     * timed poll before a delayed offer fails; after offer succeeds;
301 <     * on interruption throws
300 >     * timed poll after thread interrupted throws InterruptedException
301 >     * instead of returning timeout status
302       */
303 <    public void testTimedPollWithOffer() throws InterruptedException {
304 <        final LinkedTransferQueue q = new LinkedTransferQueue();
305 <        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
306 <            void realRun() throws InterruptedException {
307 <                threadAssertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
308 <                q.poll(LONG_DELAY_MS, MILLISECONDS);
309 <                q.poll(LONG_DELAY_MS, MILLISECONDS);
303 >    public void testTimedPollAfterInterrupt() throws InterruptedException {
304 >        final BlockingQueue<Integer> q = populatedQueue(SIZE);
305 >        Thread t = newStartedThread(new CheckedRunnable() {
306 >            public void realRun() throws InterruptedException {
307 >                Thread.currentThread().interrupt();
308 >                for (int i = 0; i < SIZE; ++i) {
309 >                    long t0 = System.nanoTime();
310 >                    assertEquals(i, (int) q.poll(LONG_DELAY_MS, MILLISECONDS));
311 >                    assertTrue(millisElapsedSince(t0) < SMALL_DELAY_MS);
312 >                }
313 >                try {
314 >                    q.poll(MEDIUM_DELAY_MS, MILLISECONDS);
315 >                    shouldThrow();
316 >                } catch (InterruptedException success) {}
317              }});
318 <        Thread.sleep(SMALL_DELAY_MS);
319 <        assertTrue(q.offer(zero, SHORT_DELAY_MS, MILLISECONDS));
320 <        t.interrupt();
389 <        t.join();
318 >
319 >        awaitTermination(t, MEDIUM_DELAY_MS);
320 >        checkEmpty(q);
321      }
322  
323      /**
# Line 416 | Line 347 | public class LinkedTransferQueueTest ext
347          try {
348              q.element();
349              shouldThrow();
350 <        } catch (NoSuchElementException success) {
420 <        }
350 >        } catch (NoSuchElementException success) {}
351          checkEmpty(q);
352      }
353  
# Line 432 | Line 362 | public class LinkedTransferQueueTest ext
362          try {
363              q.remove();
364              shouldThrow();
365 <        } catch (NoSuchElementException success) {
436 <        }
437 <        checkEmpty(q);
438 <    }
439 <
440 <    /**
441 <     * remove(x) removes x and returns true if present
442 <     */
443 <    public void testRemoveElement() throws InterruptedException {
444 <        LinkedTransferQueue q = populatedQueue(SIZE);
445 <        for (int i = 1; i < SIZE; i += 2) {
446 <            assertTrue(q.remove(i));
447 <        }
448 <        for (int i = 0; i < SIZE; i += 2) {
449 <            assertTrue(q.remove(i));
450 <            assertFalse(q.remove(i + 1));
451 <        }
365 >        } catch (NoSuchElementException success) {}
366          checkEmpty(q);
367      }
368  
# Line 462 | Line 376 | public class LinkedTransferQueueTest ext
376          assertTrue(q.remove(one));
377          assertTrue(q.remove(two));
378          assertTrue(q.add(three));
379 <        assertTrue(q.take() == three);
379 >        assertSame(q.take(), three);
380      }
381  
382      /**
# Line 544 | Line 458 | public class LinkedTransferQueueTest ext
458      }
459  
460      /**
461 <     * toArray() contains all elements
461 >     * toArray() contains all elements in FIFO order
462       */
463 <    public void testToArray() throws InterruptedException {
463 >    public void testToArray() {
464          LinkedTransferQueue q = populatedQueue(SIZE);
465          Object[] o = q.toArray();
466          for (int i = 0; i < o.length; i++) {
467 <            assertEquals(o[i], q.take());
467 >            assertSame(o[i], q.poll());
468          }
469      }
470  
471      /**
472 <     * toArray(a) contains all elements
472 >     * toArray(a) contains all elements in FIFO order
473       */
474 <    public void testToArray2() throws InterruptedException {
474 >    public void testToArray2() {
475          LinkedTransferQueue<Integer> q = populatedQueue(SIZE);
476          Integer[] ints = new Integer[SIZE];
477 <        ints = q.toArray(ints);
477 >        Integer[] array = q.toArray(ints);
478 >        assertSame(ints, array);
479          for (int i = 0; i < ints.length; i++) {
480 <            assertEquals(ints[i], q.take());
480 >            assertSame(ints[i], q.poll());
481          }
482      }
483  
484      /**
485 <     * toArray(null) throws NullPointerException
571 <     */
572 <    public void testToArray_BadArg() {
573 <        try {
574 <            LinkedTransferQueue q = populatedQueue(SIZE);
575 <            Object o[] = q.toArray(null);
576 <            shouldThrow();
577 <        } catch (NullPointerException success) {
578 <        }
579 <    }
580 <
581 <    /**
582 <     * toArray(incompatible array type) throws CCE
485 >     * toArray(incompatible array type) throws ArrayStoreException
486       */
487      public void testToArray1_BadArg() {
488 +        LinkedTransferQueue q = populatedQueue(SIZE);
489          try {
490 <            LinkedTransferQueue q = populatedQueue(SIZE);
587 <            Object o[] = q.toArray(new String[10]);
490 >            q.toArray(new String[10]);
491              shouldThrow();
492 <        } catch (ArrayStoreException success) {
590 <        }
492 >        } catch (ArrayStoreException success) {}
493      }
494  
495      /**
# Line 617 | Line 519 | public class LinkedTransferQueueTest ext
519          it.remove();
520  
521          it = q.iterator();
522 <        assertEquals(it.next(), one);
523 <        assertEquals(it.next(), three);
522 >        assertSame(it.next(), one);
523 >        assertSame(it.next(), three);
524          assertFalse(it.hasNext());
525      }
526  
# Line 662 | Line 564 | public class LinkedTransferQueueTest ext
564          LinkedTransferQueue q = populatedQueue(SIZE);
565          String s = q.toString();
566          for (int i = 0; i < SIZE; ++i) {
567 <            assertTrue(s.indexOf(String.valueOf(i)) >= 0);
567 >            assertTrue(s.contains(String.valueOf(i)));
568          }
569      }
570  
# Line 671 | Line 573 | public class LinkedTransferQueueTest ext
573       */
574      public void testOfferInExecutor() {
575          final LinkedTransferQueue q = new LinkedTransferQueue();
576 <        q.add(one);
675 <        q.add(two);
576 >        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
577          ExecutorService executor = Executors.newFixedThreadPool(2);
578  
579          executor.execute(new CheckedRunnable() {
580 <            void realRun() {
581 <                threadAssertTrue(q.offer(three, MEDIUM_DELAY_MS,
582 <                                         MILLISECONDS));
580 >            public void realRun() throws InterruptedException {
581 >                threadsStarted.await();
582 >                assertTrue(q.offer(one, LONG_DELAY_MS, MILLISECONDS));
583              }});
584  
585          executor.execute(new CheckedRunnable() {
586 <            void realRun() throws InterruptedException {
587 <                Thread.sleep(SMALL_DELAY_MS);
588 <                threadAssertEquals(one, q.take());
586 >            public void realRun() throws InterruptedException {
587 >                threadsStarted.await();
588 >                assertSame(one, q.take());
589 >                checkEmpty(q);
590              }});
591  
592          joinPool(executor);
# Line 695 | Line 597 | public class LinkedTransferQueueTest ext
597       */
598      public void testPollInExecutor() {
599          final LinkedTransferQueue q = new LinkedTransferQueue();
600 +        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
601          ExecutorService executor = Executors.newFixedThreadPool(2);
602  
603          executor.execute(new CheckedRunnable() {
604 <            void realRun() throws InterruptedException {
605 <                threadAssertNull(q.poll());
606 <                threadAssertTrue(null != q.poll(MEDIUM_DELAY_MS,
607 <                                                MILLISECONDS));
608 <                threadAssertTrue(q.isEmpty());
604 >            public void realRun() throws InterruptedException {
605 >                assertNull(q.poll());
606 >                threadsStarted.await();
607 >                assertSame(one, q.poll(LONG_DELAY_MS, MILLISECONDS));
608 >                checkEmpty(q);
609              }});
610  
611          executor.execute(new CheckedRunnable() {
612 <            void realRun() throws InterruptedException {
613 <                Thread.sleep(SMALL_DELAY_MS);
612 >            public void realRun() throws InterruptedException {
613 >                threadsStarted.await();
614                  q.put(one);
615              }});
616  
# Line 718 | Line 621 | public class LinkedTransferQueueTest ext
621       * A deserialized serialized queue has same elements in same order
622       */
623      public void testSerialization() throws Exception {
624 <        LinkedTransferQueue q = populatedQueue(SIZE);
624 >        Queue x = populatedQueue(SIZE);
625 >        Queue y = serialClone(x);
626  
627 <        ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
628 <        ObjectOutputStream out
629 <            = new ObjectOutputStream(new BufferedOutputStream(bout));
630 <        out.writeObject(q);
631 <        out.close();
632 <
633 <        ByteArrayInputStream bin
730 <            = new ByteArrayInputStream(bout.toByteArray());
731 <        ObjectInputStream in
732 <            = new ObjectInputStream(new BufferedInputStream(bin));
733 <        LinkedTransferQueue r = (LinkedTransferQueue) in.readObject();
734 <
735 <        assertEquals(q.size(), r.size());
736 <        while (!q.isEmpty()) {
737 <            assertEquals(q.remove(), r.remove());
738 <        }
739 <    }
740 <
741 <    /**
742 <     * drainTo(null) throws NullPointerException
743 <     */
744 <    public void testDrainToNull() {
745 <        LinkedTransferQueue q = populatedQueue(SIZE);
746 <        try {
747 <            q.drainTo(null);
748 <            shouldThrow();
749 <        } catch (NullPointerException success) {
750 <        }
751 <    }
752 <
753 <    /**
754 <     * drainTo(this) throws IllegalArgumentException
755 <     */
756 <    public void testDrainToSelf() {
757 <        LinkedTransferQueue q = populatedQueue(SIZE);
758 <        try {
759 <            q.drainTo(q);
760 <            shouldThrow();
761 <        } catch (IllegalArgumentException success) {
627 >        assertNotSame(y, x);
628 >        assertEquals(x.size(), y.size());
629 >        assertEquals(x.toString(), y.toString());
630 >        assertTrue(Arrays.equals(x.toArray(), y.toArray()));
631 >        while (!x.isEmpty()) {
632 >            assertFalse(y.isEmpty());
633 >            assertEquals(x.remove(), y.remove());
634          }
635 +        assertTrue(y.isEmpty());
636      }
637  
638      /**
# Line 769 | Line 642 | public class LinkedTransferQueueTest ext
642          LinkedTransferQueue q = populatedQueue(SIZE);
643          ArrayList l = new ArrayList();
644          q.drainTo(l);
645 <        assertEquals(q.size(), 0);
646 <        assertEquals(l.size(), SIZE);
645 >        assertEquals(0, q.size());
646 >        assertEquals(SIZE, l.size());
647          for (int i = 0; i < SIZE; ++i) {
648 <            assertEquals(l.get(i), i);
648 >            assertEquals(i, l.get(i));
649          }
650          q.add(zero);
651          q.add(one);
# Line 781 | Line 654 | public class LinkedTransferQueueTest ext
654          assertTrue(q.contains(one));
655          l.clear();
656          q.drainTo(l);
657 <        assertEquals(q.size(), 0);
658 <        assertEquals(l.size(), 2);
657 >        assertEquals(0, q.size());
658 >        assertEquals(2, l.size());
659          for (int i = 0; i < 2; ++i) {
660 <            assertEquals(l.get(i), i);
660 >            assertEquals(i, l.get(i));
661          }
662      }
663  
# Line 794 | Line 667 | public class LinkedTransferQueueTest ext
667      public void testDrainToWithActivePut() throws InterruptedException {
668          final LinkedTransferQueue q = populatedQueue(SIZE);
669          Thread t = newStartedThread(new CheckedRunnable() {
670 <            void realRun() {
670 >            public void realRun() {
671                  q.put(SIZE + 1);
672              }});
673          ArrayList l = new ArrayList();
674          q.drainTo(l);
675          assertTrue(l.size() >= SIZE);
676 <        for (int i = 0; i < SIZE; ++i) {
677 <            assertEquals(l.get(i), i);
678 <        }
806 <        t.join();
676 >        for (int i = 0; i < SIZE; ++i)
677 >            assertEquals(i, l.get(i));
678 >        awaitTermination(t, MEDIUM_DELAY_MS);
679          assertTrue(q.size() + l.size() >= SIZE);
680      }
681  
682      /**
683 <     * drainTo(null, n) throws NullPointerException
812 <     */
813 <    public void testDrainToNullN() {
814 <        LinkedTransferQueue q = populatedQueue(SIZE);
815 <        try {
816 <            q.drainTo(null, SIZE);
817 <            shouldThrow();
818 <        } catch (NullPointerException success) {
819 <        }
820 <    }
821 <
822 <    /**
823 <     * drainTo(this, n) throws IllegalArgumentException
824 <     */
825 <    public void testDrainToSelfN() {
826 <        LinkedTransferQueue q = populatedQueue(SIZE);
827 <        try {
828 <            q.drainTo(q, SIZE);
829 <            shouldThrow();
830 <        } catch (IllegalArgumentException success) {
831 <        }
832 <    }
833 <
834 <    /**
835 <     * drainTo(c, n) empties first max {n, size} elements of queue into c
683 >     * drainTo(c, n) empties first min(n, size) elements of queue into c
684       */
685      public void testDrainToN() {
686          LinkedTransferQueue q = new LinkedTransferQueue();
# Line 843 | Line 691 | public class LinkedTransferQueueTest ext
691              ArrayList l = new ArrayList();
692              q.drainTo(l, i);
693              int k = (i < SIZE) ? i : SIZE;
694 <            assertEquals(l.size(), k);
695 <            assertEquals(q.size(), SIZE - k);
696 <            for (int j = 0; j < k; ++j) {
697 <                assertEquals(l.get(j), j);
850 <            }
694 >            assertEquals(k, l.size());
695 >            assertEquals(SIZE - k, q.size());
696 >            for (int j = 0; j < k; ++j)
697 >                assertEquals(j, l.get(j));
698              while (q.poll() != null)
699                  ;
700          }
# Line 859 | Line 706 | public class LinkedTransferQueueTest ext
706       */
707      public void testWaitingConsumer() throws InterruptedException {
708          final LinkedTransferQueue q = new LinkedTransferQueue();
709 <        assertEquals(q.getWaitingConsumerCount(), 0);
709 >        assertEquals(0, q.getWaitingConsumerCount());
710          assertFalse(q.hasWaitingConsumer());
711 +        final CountDownLatch threadStarted = new CountDownLatch(1);
712  
713          Thread t = newStartedThread(new CheckedRunnable() {
714 <            void realRun() throws InterruptedException {
715 <                Thread.sleep(SMALL_DELAY_MS);
716 <                threadAssertTrue(q.hasWaitingConsumer());
717 <                threadAssertEquals(q.getWaitingConsumerCount(), 1);
718 <                threadAssertTrue(q.offer(new Object()));
871 <                threadAssertFalse(q.hasWaitingConsumer());
872 <                threadAssertEquals(q.getWaitingConsumerCount(), 0);
714 >            public void realRun() throws InterruptedException {
715 >                threadStarted.countDown();
716 >                assertSame(one, q.poll(LONG_DELAY_MS, MILLISECONDS));
717 >                assertEquals(0, q.getWaitingConsumerCount());
718 >                assertFalse(q.hasWaitingConsumer());
719              }});
720  
721 <        assertTrue(q.poll(LONG_DELAY_MS, MILLISECONDS) != null);
722 <        assertEquals(q.getWaitingConsumerCount(), 0);
721 >        threadStarted.await();
722 >        waitForThreadToEnterWaitState(t, SMALL_DELAY_MS);
723 >        assertEquals(1, q.getWaitingConsumerCount());
724 >        assertTrue(q.hasWaitingConsumer());
725 >
726 >        assertTrue(q.offer(one));
727 >        assertEquals(0, q.getWaitingConsumerCount());
728          assertFalse(q.hasWaitingConsumer());
729 <        t.join();
729 >
730 >        awaitTermination(t, MEDIUM_DELAY_MS);
731      }
732  
733      /**
# Line 896 | Line 748 | public class LinkedTransferQueueTest ext
748      public void testTransfer2() throws InterruptedException {
749          final LinkedTransferQueue<Integer> q
750              = new LinkedTransferQueue<Integer>();
751 +        final CountDownLatch threadStarted = new CountDownLatch(1);
752  
753          Thread t = newStartedThread(new CheckedRunnable() {
754 <            void realRun() throws InterruptedException {
755 <                q.transfer(SIZE);
756 <                threadAssertTrue(q.isEmpty());
754 >            public void realRun() throws InterruptedException {
755 >                threadStarted.countDown();
756 >                q.transfer(five);
757 >                checkEmpty(q);
758              }});
759  
760 <        Thread.sleep(SHORT_DELAY_MS);
760 >        threadStarted.await();
761 >        waitForThreadToEnterWaitState(t, SMALL_DELAY_MS);
762          assertEquals(1, q.size());
763 <        assertEquals(SIZE, (int) q.poll());
764 <        assertTrue(q.isEmpty());
765 <        t.join();
763 >        assertSame(five, q.poll());
764 >        checkEmpty(q);
765 >        awaitTermination(t, MEDIUM_DELAY_MS);
766      }
767  
768      /**
# Line 918 | Line 773 | public class LinkedTransferQueueTest ext
773              = new LinkedTransferQueue<Integer>();
774  
775          Thread first = newStartedThread(new CheckedRunnable() {
776 <            void realRun() throws InterruptedException {
777 <                Integer i = SIZE + 1;
778 <                q.transfer(i);
779 <                threadAssertTrue(!q.contains(i));
925 <                threadAssertEquals(1, q.size());
776 >            public void realRun() throws InterruptedException {
777 >                q.transfer(four);
778 >                assertTrue(!q.contains(four));
779 >                assertEquals(1, q.size());
780              }});
781  
782          Thread interruptedThread = newStartedThread(
783              new CheckedInterruptedRunnable() {
784 <                void realRun() throws InterruptedException {
785 <                    while (q.size() == 0)
784 >                public void realRun() throws InterruptedException {
785 >                    while (q.isEmpty())
786                          Thread.yield();
787 <                    q.transfer(SIZE);
787 >                    q.transfer(five);
788                  }});
789  
790          while (q.size() < 2)
791              Thread.yield();
792          assertEquals(2, q.size());
793 <        assertEquals(SIZE + 1, (int) q.poll());
793 >        assertSame(four, q.poll());
794          first.join();
795          assertEquals(1, q.size());
796          interruptedThread.interrupt();
797          interruptedThread.join();
798 <        assertEquals(0, q.size());
945 <        assertTrue(q.isEmpty());
798 >        checkEmpty(q);
799      }
800  
801      /**
# Line 953 | Line 806 | public class LinkedTransferQueueTest ext
806          final LinkedTransferQueue q = new LinkedTransferQueue();
807  
808          Thread t = newStartedThread(new CheckedRunnable() {
809 <            void realRun() throws InterruptedException {
809 >            public void realRun() throws InterruptedException {
810                  q.transfer(four);
811 <                threadAssertFalse(q.contains(four));
812 <                threadAssertEquals(three, q.poll());
811 >                assertFalse(q.contains(four));
812 >                assertSame(three, q.poll());
813              }});
814  
815 <        Thread.sleep(SHORT_DELAY_MS);
815 >        while (q.isEmpty())
816 >            Thread.yield();
817 >        assertFalse(q.isEmpty());
818 >        assertEquals(1, q.size());
819          assertTrue(q.offer(three));
820 <        assertEquals(four, q.poll());
821 <        t.join();
820 >        assertSame(four, q.poll());
821 >        awaitTermination(t, MEDIUM_DELAY_MS);
822      }
823  
824      /**
# Line 974 | Line 830 | public class LinkedTransferQueueTest ext
830              = new LinkedTransferQueue<Integer>();
831  
832          Thread t = newStartedThread(new CheckedRunnable() {
833 <            void realRun() throws InterruptedException {
834 <                q.transfer(SIZE);
833 >            public void realRun() throws InterruptedException {
834 >                q.transfer(four);
835                  checkEmpty(q);
836              }});
837  
838 <        Thread.sleep(SHORT_DELAY_MS);
839 <        assertEquals(SIZE, (int) q.take());
838 >        while (q.isEmpty())
839 >            Thread.yield();
840 >        assertFalse(q.isEmpty());
841 >        assertEquals(1, q.size());
842 >        assertSame(four, q.take());
843          checkEmpty(q);
844 <        t.join();
844 >        awaitTermination(t, MEDIUM_DELAY_MS);
845      }
846  
847      /**
# Line 1016 | Line 875 | public class LinkedTransferQueueTest ext
875          final LinkedTransferQueue q = new LinkedTransferQueue();
876  
877          Thread t = newStartedThread(new CheckedRunnable() {
878 <            void realRun() {
878 >            public void realRun() {
879                  while (! q.hasWaitingConsumer())
880                      Thread.yield();
881 <                threadAssertTrue(q.hasWaitingConsumer());
882 <                threadAssertTrue(q.isEmpty());
883 <                threadAssertTrue(q.size() == 0);
1025 <                threadAssertTrue(q.tryTransfer(hotPotato));
881 >                assertTrue(q.hasWaitingConsumer());
882 >                checkEmpty(q);
883 >                assertTrue(q.tryTransfer(hotPotato));
884              }});
885  
886 <        assertTrue(q.poll(MEDIUM_DELAY_MS, MILLISECONDS) == hotPotato);
886 >        assertSame(hotPotato, q.poll(MEDIUM_DELAY_MS, MILLISECONDS));
887          checkEmpty(q);
888 <        t.join();
888 >        awaitTermination(t, MEDIUM_DELAY_MS);
889      }
890  
891      /**
# Line 1039 | Line 897 | public class LinkedTransferQueueTest ext
897          final LinkedTransferQueue q = new LinkedTransferQueue();
898  
899          Thread t = newStartedThread(new CheckedRunnable() {
900 <            void realRun() {
900 >            public void realRun() {
901                  while (! q.hasWaitingConsumer())
902                      Thread.yield();
903 <                threadAssertTrue(q.hasWaitingConsumer());
904 <                threadAssertTrue(q.isEmpty());
905 <                threadAssertTrue(q.size() == 0);
1048 <                threadAssertTrue(q.tryTransfer(hotPotato));
903 >                assertTrue(q.hasWaitingConsumer());
904 >                checkEmpty(q);
905 >                assertTrue(q.tryTransfer(hotPotato));
906              }});
907  
908 <        assertTrue(q.take() == hotPotato);
908 >        assertSame(q.take(), hotPotato);
909          checkEmpty(q);
910 <        t.join();
910 >        awaitTermination(t, MEDIUM_DELAY_MS);
911      }
912  
913      /**
914 <     * tryTransfer waits the amount given if interrupted, and
1058 <     * throws interrupted exception
914 >     * tryTransfer blocks interruptibly if no takers
915       */
916      public void testTryTransfer5() throws InterruptedException {
917          final LinkedTransferQueue q = new LinkedTransferQueue();
918 +        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
919 +        assertTrue(q.isEmpty());
920  
921 <        Thread toInterrupt = newStartedThread(new CheckedInterruptedRunnable() {
922 <            void realRun() throws InterruptedException {
923 <                q.tryTransfer(new Object(), LONG_DELAY_MS, MILLISECONDS);
921 >        Thread t = newStartedThread(new CheckedRunnable() {
922 >            public void realRun() throws InterruptedException {
923 >                Thread.currentThread().interrupt();
924 >                try {
925 >                    q.tryTransfer(new Object(), LONG_DELAY_MS, MILLISECONDS);
926 >                    shouldThrow();
927 >                } catch (InterruptedException success) {}
928 >                assertFalse(Thread.interrupted());
929 >
930 >                pleaseInterrupt.countDown();
931 >                try {
932 >                    q.tryTransfer(new Object(), LONG_DELAY_MS, MILLISECONDS);
933 >                    shouldThrow();
934 >                } catch (InterruptedException success) {}
935 >                assertFalse(Thread.interrupted());
936              }});
937  
938 <        Thread.sleep(SMALL_DELAY_MS);
939 <        toInterrupt.interrupt();
940 <        toInterrupt.join();
938 >        await(pleaseInterrupt);
939 >        assertThreadStaysAlive(t);
940 >        t.interrupt();
941 >        awaitTermination(t);
942 >        checkEmpty(q);
943      }
944  
945      /**
946 <     * tryTransfer gives up after the timeout and return false
946 >     * tryTransfer gives up after the timeout and returns false
947       */
948      public void testTryTransfer6() throws InterruptedException {
949          final LinkedTransferQueue q = new LinkedTransferQueue();
950  
951          Thread t = newStartedThread(new CheckedRunnable() {
952 <            void realRun() throws InterruptedException {
953 <                threadAssertFalse
954 <                    (q.tryTransfer(new Object(),
955 <                                   SHORT_DELAY_MS, MILLISECONDS));
952 >            public void realRun() throws InterruptedException {
953 >                long t0 = System.nanoTime();
954 >                assertFalse(q.tryTransfer(new Object(),
955 >                                          timeoutMillis(), MILLISECONDS));
956 >                assertTrue(millisElapsedSince(t0) >= timeoutMillis());
957 >                checkEmpty(q);
958              }});
959  
960 <        Thread.sleep(SMALL_DELAY_MS);
960 >        awaitTermination(t);
961          checkEmpty(q);
1088        t.join();
962      }
963  
964      /**
# Line 1097 | Line 970 | public class LinkedTransferQueueTest ext
970          assertTrue(q.offer(four));
971  
972          Thread t = newStartedThread(new CheckedRunnable() {
973 <            void realRun() throws InterruptedException {
974 <                threadAssertTrue(q.tryTransfer(five,
975 <                                               MEDIUM_DELAY_MS, MILLISECONDS));
1103 <                threadAssertTrue(q.isEmpty());
973 >            public void realRun() throws InterruptedException {
974 >                assertTrue(q.tryTransfer(five, MEDIUM_DELAY_MS, MILLISECONDS));
975 >                checkEmpty(q);
976              }});
977  
978 <        Thread.sleep(SHORT_DELAY_MS);
978 >        while (q.size() != 2)
979 >            Thread.yield();
980          assertEquals(2, q.size());
981 <        assertEquals(four, q.poll());
982 <        assertEquals(five, q.poll());
981 >        assertSame(four, q.poll());
982 >        assertSame(five, q.poll());
983          checkEmpty(q);
984 <        t.join();
984 >        awaitTermination(t, MEDIUM_DELAY_MS);
985      }
986  
987      /**
988 <     * tryTransfer attempts to enqueue into the q and fails returning
989 <     * false not enqueueing and the successive poll is null
988 >     * tryTransfer attempts to enqueue into the queue and fails
989 >     * returning false not enqueueing and the successive poll is null
990       */
991      public void testTryTransfer8() throws InterruptedException {
992          final LinkedTransferQueue q = new LinkedTransferQueue();
993          assertTrue(q.offer(four));
994          assertEquals(1, q.size());
995 <        assertFalse(q.tryTransfer(five, SHORT_DELAY_MS, MILLISECONDS));
995 >        long t0 = System.nanoTime();
996 >        assertFalse(q.tryTransfer(five, timeoutMillis(), MILLISECONDS));
997 >        assertTrue(millisElapsedSince(t0) >= timeoutMillis());
998          assertEquals(1, q.size());
999 <        assertEquals(four, q.poll());
999 >        assertSame(four, q.poll());
1000          assertNull(q.poll());
1001          checkEmpty(q);
1002      }
1003  
1004      private LinkedTransferQueue<Integer> populatedQueue(int n) {
1005          LinkedTransferQueue<Integer> q = new LinkedTransferQueue<Integer>();
1006 <        assertTrue(q.isEmpty());
1006 >        checkEmpty(q);
1007          for (int i = 0; i < n; i++) {
1008              assertEquals(i, q.size());
1009              assertTrue(q.offer(i));
# Line 1137 | Line 1012 | public class LinkedTransferQueueTest ext
1012          assertFalse(q.isEmpty());
1013          return q;
1014      }
1015 +
1016 +    /**
1017 +     * remove(null), contains(null) always return false
1018 +     */
1019 +    public void testNeverContainsNull() {
1020 +        Collection<?>[] qs = {
1021 +            new LinkedTransferQueue<Object>(),
1022 +            populatedQueue(2),
1023 +        };
1024 +
1025 +        for (Collection<?> q : qs) {
1026 +            assertFalse(q.contains(null));
1027 +            assertFalse(q.remove(null));
1028 +        }
1029 +    }
1030   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines