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

Comparing jsr166/src/test/tck/DelayQueueTest.java (file contents):
Revision 1.31 by jsr166, Wed Aug 25 00:07:03 2010 UTC vs.
Revision 1.53 by jsr166, Mon May 30 22:43:20 2011 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 Andrew Wright, Jeffrey Hayes,
6   * Pat Fisher, Mike Judd.
7   */
8  
9   import junit.framework.*;
10 < import java.util.*;
10 > import java.util.Arrays;
11 > import java.util.ArrayList;
12 > import java.util.Collection;
13 > import java.util.Iterator;
14 > import java.util.NoSuchElementException;
15 > import java.util.concurrent.BlockingQueue;
16 > import java.util.concurrent.CountDownLatch;
17 > import java.util.concurrent.Delayed;
18 > import java.util.concurrent.DelayQueue;
19 > import java.util.concurrent.Executors;
20 > import java.util.concurrent.ExecutorService;
21 > import java.util.concurrent.TimeUnit;
22   import static java.util.concurrent.TimeUnit.MILLISECONDS;
12 import java.util.concurrent.*;
23  
24   public class DelayQueueTest extends JSR166TestCase {
25 +
26 +    public static class Generic extends BlockingQueueTest {
27 +        protected BlockingQueue emptyCollection() {
28 +            return new DelayQueue();
29 +        }
30 +        protected PDelay makeElement(int i) {
31 +            return new PDelay(i);
32 +        }
33 +    }
34 +
35      public static void main(String[] args) {
36          junit.textui.TestRunner.run(suite());
37      }
38  
39      public static Test suite() {
40 <        return new TestSuite(DelayQueueTest.class);
40 >        return newTestSuite(DelayQueueTest.class,
41 >                            new Generic().testSuite());
42      }
43  
44      private static final int NOCAP = Integer.MAX_VALUE;
# Line 49 | Line 70 | public class DelayQueueTest extends JSR1
70              return other.pseudodelay == pseudodelay;
71          }
72  
52
73          public long getDelay(TimeUnit ignore) {
74              return pseudodelay;
75          }
# Line 62 | Line 82 | public class DelayQueueTest extends JSR1
82          }
83      }
84  
65
85      /**
86       * Delayed implementation that actually delays
87       */
# Line 104 | Line 123 | public class DelayQueueTest extends JSR1
123          }
124      }
125  
107
126      /**
127       * Create a queue of given size containing consecutive
128       * PDelays 0 ... n.
129       */
130 <    private DelayQueue populatedQueue(int n) {
131 <        DelayQueue q = new DelayQueue();
130 >    private DelayQueue<PDelay> populatedQueue(int n) {
131 >        DelayQueue<PDelay> q = new DelayQueue<PDelay>();
132          assertTrue(q.isEmpty());
133          for (int i = n-1; i >= 0; i-=2)
134              assertTrue(q.offer(new PDelay(i)));
# Line 209 | Line 227 | public class DelayQueueTest extends JSR1
227      }
228  
229      /**
212     * offer(null) throws NPE
213     */
214    public void testOfferNull() {
215        try {
216            DelayQueue q = new DelayQueue();
217            q.offer(null);
218            shouldThrow();
219        } catch (NullPointerException success) {}
220    }
221
222    /**
223     * add(null) throws NPE
224     */
225    public void testAddNull() {
226        try {
227            DelayQueue q = new DelayQueue();
228            q.add(null);
229            shouldThrow();
230        } catch (NullPointerException success) {}
231    }
232
233    /**
230       * offer non-null succeeds
231       */
232      public void testOffer() {
# Line 251 | Line 247 | public class DelayQueueTest extends JSR1
247      }
248  
249      /**
254     * addAll(null) throws NPE
255     */
256    public void testAddAll1() {
257        try {
258            DelayQueue q = new DelayQueue();
259            q.addAll(null);
260            shouldThrow();
261        } catch (NullPointerException success) {}
262    }
263
264
265    /**
250       * addAll(this) throws IAE
251       */
252      public void testAddAllSelf() {
# Line 274 | Line 258 | public class DelayQueueTest extends JSR1
258      }
259  
260      /**
277     * addAll of a collection with null elements throws NPE
278     */
279    public void testAddAll2() {
280        try {
281            DelayQueue q = new DelayQueue();
282            PDelay[] ints = new PDelay[SIZE];
283            q.addAll(Arrays.asList(ints));
284            shouldThrow();
285        } catch (NullPointerException success) {}
286    }
287    /**
261       * addAll of a collection with any null elements throws NPE after
262       * possibly adding some elements
263       */
# Line 315 | Line 288 | public class DelayQueueTest extends JSR1
288      }
289  
290      /**
318     * put(null) throws NPE
319     */
320     public void testPutNull() {
321        try {
322            DelayQueue q = new DelayQueue();
323            q.put(null);
324            shouldThrow();
325        } catch (NullPointerException success) {}
326     }
327
328    /**
291       * all elements successfully put are contained
292       */
293 <     public void testPut() {
294 <         DelayQueue q = new DelayQueue();
295 <         for (int i = 0; i < SIZE; ++i) {
296 <             PDelay I = new PDelay(i);
297 <             q.put(I);
298 <             assertTrue(q.contains(I));
299 <         }
300 <         assertEquals(SIZE, q.size());
293 >    public void testPut() {
294 >        DelayQueue q = new DelayQueue();
295 >        for (int i = 0; i < SIZE; ++i) {
296 >            PDelay I = new PDelay(i);
297 >            q.put(I);
298 >            assertTrue(q.contains(I));
299 >        }
300 >        assertEquals(SIZE, q.size());
301      }
302  
303      /**
# Line 343 | Line 305 | public class DelayQueueTest extends JSR1
305       */
306      public void testPutWithTake() throws InterruptedException {
307          final DelayQueue q = new DelayQueue();
308 <        Thread t = new Thread(new CheckedRunnable() {
308 >        Thread t = newStartedThread(new CheckedRunnable() {
309              public void realRun() {
310                  q.put(new PDelay(0));
311                  q.put(new PDelay(0));
# Line 351 | Line 313 | public class DelayQueueTest extends JSR1
313                  q.put(new PDelay(0));
314              }});
315  
316 <        t.start();
317 <        Thread.sleep(SHORT_DELAY_MS);
356 <        q.take();
357 <        t.interrupt();
358 <        t.join();
316 >        awaitTermination(t);
317 >        assertEquals(4, q.size());
318      }
319  
320      /**
# Line 363 | Line 322 | public class DelayQueueTest extends JSR1
322       */
323      public void testTimedOffer() throws InterruptedException {
324          final DelayQueue q = new DelayQueue();
325 <        Thread t = new Thread(new CheckedRunnable() {
325 >        Thread t = newStartedThread(new CheckedRunnable() {
326              public void realRun() throws InterruptedException {
327                  q.put(new PDelay(0));
328                  q.put(new PDelay(0));
# Line 371 | Line 330 | public class DelayQueueTest extends JSR1
330                  assertTrue(q.offer(new PDelay(0), LONG_DELAY_MS, MILLISECONDS));
331              }});
332  
333 <        t.start();
375 <        Thread.sleep(SMALL_DELAY_MS);
376 <        t.interrupt();
377 <        t.join();
333 >        awaitTermination(t);
334      }
335  
336      /**
# Line 388 | Line 344 | public class DelayQueueTest extends JSR1
344      }
345  
346      /**
391     * take blocks interruptibly when empty
392     */
393    public void testTakeFromEmpty() throws InterruptedException {
394        final DelayQueue q = new DelayQueue();
395        Thread t = new ThreadShouldThrow(InterruptedException.class) {
396            public void realRun() throws InterruptedException {
397                q.take();
398            }};
399
400        t.start();
401        Thread.sleep(SHORT_DELAY_MS);
402        t.interrupt();
403        t.join();
404    }
405
406    /**
347       * Take removes existing elements until empty, then blocks interruptibly
348       */
349      public void testBlockingTake() throws InterruptedException {
350          final DelayQueue q = populatedQueue(SIZE);
351 <        Thread t = new Thread(new CheckedRunnable() {
351 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
352 >        Thread t = newStartedThread(new CheckedRunnable() {
353              public void realRun() throws InterruptedException {
354                  for (int i = 0; i < SIZE; ++i) {
355                      assertEquals(new PDelay(i), ((PDelay)q.take()));
356                  }
357 +
358 +                Thread.currentThread().interrupt();
359 +                try {
360 +                    q.take();
361 +                    shouldThrow();
362 +                } catch (InterruptedException success) {}
363 +                assertFalse(Thread.interrupted());
364 +
365 +                pleaseInterrupt.countDown();
366                  try {
367                      q.take();
368                      shouldThrow();
369                  } catch (InterruptedException success) {}
370 +                assertFalse(Thread.interrupted());
371              }});
372  
373 <        t.start();
374 <        Thread.sleep(SHORT_DELAY_MS);
373 >        await(pleaseInterrupt);
374 >        assertThreadStaysAlive(t);
375          t.interrupt();
376 <        t.join();
376 >        awaitTermination(t);
377      }
378  
428
379      /**
380       * poll succeeds unless empty
381       */
# Line 438 | Line 388 | public class DelayQueueTest extends JSR1
388      }
389  
390      /**
391 <     * timed pool with zero timeout succeeds when non-empty, else times out
391 >     * timed poll with zero timeout succeeds when non-empty, else times out
392       */
393      public void testTimedPoll0() throws InterruptedException {
394          DelayQueue q = populatedQueue(SIZE);
# Line 449 | Line 399 | public class DelayQueueTest extends JSR1
399      }
400  
401      /**
402 <     * timed pool with nonzero timeout succeeds when non-empty, else times out
402 >     * timed poll with nonzero timeout succeeds when non-empty, else times out
403       */
404      public void testTimedPoll() throws InterruptedException {
405          DelayQueue q = populatedQueue(SIZE);
406          for (int i = 0; i < SIZE; ++i) {
407 <            assertEquals(new PDelay(i), ((PDelay)q.poll(SHORT_DELAY_MS, MILLISECONDS)));
408 <        }
409 <        assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
407 >            long startTime = System.nanoTime();
408 >            assertEquals(new PDelay(i), ((PDelay)q.poll(LONG_DELAY_MS, MILLISECONDS)));
409 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
410 >        }
411 >        long startTime = System.nanoTime();
412 >        assertNull(q.poll(timeoutMillis(), MILLISECONDS));
413 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
414 >        checkEmpty(q);
415      }
416  
417      /**
# Line 464 | Line 419 | public class DelayQueueTest extends JSR1
419       * returning timeout status
420       */
421      public void testInterruptedTimedPoll() throws InterruptedException {
422 <        Thread t = new Thread(new CheckedRunnable() {
422 >        final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
423 >        Thread t = newStartedThread(new CheckedRunnable() {
424              public void realRun() throws InterruptedException {
425                  DelayQueue q = populatedQueue(SIZE);
426                  for (int i = 0; i < SIZE; ++i) {
427                      assertEquals(new PDelay(i), ((PDelay)q.poll(SHORT_DELAY_MS, MILLISECONDS)));
428                  }
429 +
430 +                Thread.currentThread().interrupt();
431                  try {
432 <                    q.poll(SMALL_DELAY_MS, MILLISECONDS);
432 >                    q.poll(LONG_DELAY_MS, MILLISECONDS);
433                      shouldThrow();
434                  } catch (InterruptedException success) {}
435 <            }});
435 >                assertFalse(Thread.interrupted());
436  
437 <        t.start();
480 <        Thread.sleep(SHORT_DELAY_MS);
481 <        t.interrupt();
482 <        t.join();
483 <    }
484 <
485 <    /**
486 <     *  timed poll before a delayed offer fails; after offer succeeds;
487 <     *  on interruption throws
488 <     */
489 <    public void testTimedPollWithOffer() throws InterruptedException {
490 <        final DelayQueue q = new DelayQueue();
491 <        final PDelay pdelay = new PDelay(0);
492 <        Thread t = new Thread(new CheckedRunnable() {
493 <            public void realRun() throws InterruptedException {
494 <                assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
495 <                assertSame(pdelay, q.poll(LONG_DELAY_MS, MILLISECONDS));
437 >                pleaseInterrupt.countDown();
438                  try {
439                      q.poll(LONG_DELAY_MS, MILLISECONDS);
440                      shouldThrow();
441                  } catch (InterruptedException success) {}
442 +                assertFalse(Thread.interrupted());
443              }});
444  
445 <        t.start();
446 <        Thread.sleep(SMALL_DELAY_MS);
504 <        assertTrue(q.offer(pdelay, SHORT_DELAY_MS, MILLISECONDS));
445 >        await(pleaseInterrupt);
446 >        assertThreadStaysAlive(t);
447          t.interrupt();
448 <        t.join();
448 >        awaitTermination(t);
449      }
450  
509
451      /**
452       * peek returns next element, or null if empty
453       */
# Line 653 | Line 594 | public class DelayQueueTest extends JSR1
594          Object[] o = q.toArray();
595          Arrays.sort(o);
596          for (int i = 0; i < o.length; i++)
597 <            assertEquals(o[i], q.take());
597 >            assertSame(o[i], q.take());
598      }
599  
600      /**
601       * toArray(a) contains all elements
602       */
603 <    public void testToArray2() throws InterruptedException {
604 <        DelayQueue q = populatedQueue(SIZE);
603 >    public void testToArray2() {
604 >        DelayQueue<PDelay> q = populatedQueue(SIZE);
605          PDelay[] ints = new PDelay[SIZE];
606 <        ints = (PDelay[])q.toArray(ints);
606 >        PDelay[] array = q.toArray(ints);
607 >        assertSame(ints, array);
608          Arrays.sort(ints);
609          for (int i = 0; i < ints.length; i++)
610 <            assertEquals(ints[i], q.take());
610 >            assertSame(ints[i], q.remove());
611      }
612  
671
613      /**
614 <     * toArray(null) throws NPE
674 <     */
675 <    public void testToArray_BadArg() {
676 <        DelayQueue q = populatedQueue(SIZE);
677 <        try {
678 <            Object o[] = q.toArray(null);
679 <            shouldThrow();
680 <        } catch (NullPointerException success) {}
681 <    }
682 <
683 <    /**
684 <     * toArray with incompatible array type throws CCE
614 >     * toArray(incompatible array type) throws ArrayStoreException
615       */
616      public void testToArray1_BadArg() {
617          DelayQueue q = populatedQueue(SIZE);
618          try {
619 <            Object o[] = q.toArray(new String[10]);
619 >            q.toArray(new String[10]);
620              shouldThrow();
621          } catch (ArrayStoreException success) {}
622      }
# Line 722 | Line 652 | public class DelayQueueTest extends JSR1
652          assertFalse(it.hasNext());
653      }
654  
725
655      /**
656       * toString contains toStrings of elements
657       */
# Line 730 | Line 659 | public class DelayQueueTest extends JSR1
659          DelayQueue q = populatedQueue(SIZE);
660          String s = q.toString();
661          for (int i = 0; i < SIZE; ++i) {
662 <            assertTrue(s.indexOf(String.valueOf(Integer.MIN_VALUE+i)) >= 0);
662 >            assertTrue(s.contains(String.valueOf(Integer.MIN_VALUE+i)));
663          }
664      }
665  
666      /**
667 <     * offer transfers elements across Executor tasks
667 >     * timed poll transfers elements across Executor tasks
668       */
669      public void testPollInExecutor() {
670          final DelayQueue q = new DelayQueue();
671 +        final CheckedBarrier threadsStarted = new CheckedBarrier(2);
672          ExecutorService executor = Executors.newFixedThreadPool(2);
673          executor.execute(new CheckedRunnable() {
674              public void realRun() throws InterruptedException {
675                  assertNull(q.poll());
676 <                assertTrue(null != q.poll(MEDIUM_DELAY_MS, MILLISECONDS));
677 <                assertTrue(q.isEmpty());
676 >                threadsStarted.await();
677 >                assertTrue(null != q.poll(LONG_DELAY_MS, MILLISECONDS));
678 >                checkEmpty(q);
679              }});
680  
681          executor.execute(new CheckedRunnable() {
682              public void realRun() throws InterruptedException {
683 <                Thread.sleep(SHORT_DELAY_MS);
683 >                threadsStarted.await();
684                  q.put(new PDelay(1));
685              }});
686  
687          joinPool(executor);
688      }
689  
759
690      /**
691       * Delayed actions do not occur until their delay elapses
692       */
693      public void testDelay() throws InterruptedException {
694 <        DelayQueue q = new DelayQueue();
695 <        NanoDelay[] elements = new NanoDelay[SIZE];
696 <        for (int i = 0; i < SIZE; ++i) {
767 <            elements[i] = new NanoDelay(1000000000L + 1000000L * (SIZE - i));
768 <        }
769 <        for (int i = 0; i < SIZE; ++i) {
770 <            q.add(elements[i]);
771 <        }
694 >        DelayQueue<NanoDelay> q = new DelayQueue<NanoDelay>();
695 >        for (int i = 0; i < SIZE; ++i)
696 >            q.add(new NanoDelay(1000000L * (SIZE - i)));
697  
698          long last = 0;
699          for (int i = 0; i < SIZE; ++i) {
700 <            NanoDelay e = (NanoDelay)(q.take());
700 >            NanoDelay e = q.take();
701              long tt = e.getTriggerTime();
702 <            assertTrue(tt <= System.nanoTime());
702 >            assertTrue(System.nanoTime() - tt >= 0);
703              if (i != 0)
704                  assertTrue(tt >= last);
705              last = tt;
706          }
707 +        assertTrue(q.isEmpty());
708      }
709  
710      /**
# Line 787 | Line 713 | public class DelayQueueTest extends JSR1
713      public void testPeekDelayed() {
714          DelayQueue q = new DelayQueue();
715          q.add(new NanoDelay(Long.MAX_VALUE));
716 <        assert(q.peek() != null);
716 >        assertNotNull(q.peek());
717      }
718  
793
719      /**
720       * poll of a non-empty queue returns null if no expired elements.
721       */
# Line 806 | Line 731 | public class DelayQueueTest extends JSR1
731      public void testTimedPollDelayed() throws InterruptedException {
732          DelayQueue q = new DelayQueue();
733          q.add(new NanoDelay(LONG_DELAY_MS * 1000000L));
734 <        assertNull(q.poll(SHORT_DELAY_MS, MILLISECONDS));
810 <    }
811 <
812 <    /**
813 <     * drainTo(null) throws NPE
814 <     */
815 <    public void testDrainToNull() {
816 <        DelayQueue q = populatedQueue(SIZE);
817 <        try {
818 <            q.drainTo(null);
819 <            shouldThrow();
820 <        } catch (NullPointerException success) {}
821 <    }
822 <
823 <    /**
824 <     * drainTo(this) throws IAE
825 <     */
826 <    public void testDrainToSelf() {
827 <        DelayQueue q = populatedQueue(SIZE);
828 <        try {
829 <            q.drainTo(q);
830 <            shouldThrow();
831 <        } catch (IllegalArgumentException success) {}
734 >        assertNull(q.poll(timeoutMillis(), MILLISECONDS));
735      }
736  
737      /**
# Line 878 | Line 781 | public class DelayQueueTest extends JSR1
781      }
782  
783      /**
784 <     * drainTo(null, n) throws NPE
882 <     */
883 <    public void testDrainToNullN() {
884 <        DelayQueue q = populatedQueue(SIZE);
885 <        try {
886 <            q.drainTo(null, 0);
887 <            shouldThrow();
888 <        } catch (NullPointerException success) {}
889 <    }
890 <
891 <    /**
892 <     * drainTo(this, n) throws IAE
893 <     */
894 <    public void testDrainToSelfN() {
895 <        DelayQueue q = populatedQueue(SIZE);
896 <        try {
897 <            q.drainTo(q, 0);
898 <            shouldThrow();
899 <        } catch (IllegalArgumentException success) {}
900 <    }
901 <
902 <    /**
903 <     * drainTo(c, n) empties first max {n, size} elements of queue into c
784 >     * drainTo(c, n) empties first min(n, size) elements of queue into c
785       */
786      public void testDrainToN() {
787          for (int i = 0; i < SIZE + 2; ++i) {
788              DelayQueue q = populatedQueue(SIZE);
789              ArrayList l = new ArrayList();
790              q.drainTo(l, i);
791 <            int k = (i < SIZE)? i : SIZE;
791 >            int k = (i < SIZE) ? i : SIZE;
792              assertEquals(q.size(), SIZE-k);
793              assertEquals(l.size(), k);
794          }
795      }
796  
916
797   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines