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

Comparing jsr166/src/test/tck/ScheduledExecutorTest.java (file contents):
Revision 1.53 by jsr166, Sun Sep 27 18:50:50 2015 UTC vs.
Revision 1.61 by jsr166, Sun Oct 4 02:15:08 2015 UTC

# Line 7 | Line 7
7   */
8  
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 + import static java.util.concurrent.TimeUnit.SECONDS;
11  
12   import java.util.ArrayList;
13 + import java.util.HashSet;
14   import java.util.List;
15   import java.util.concurrent.BlockingQueue;
16   import java.util.concurrent.Callable;
# Line 332 | Line 334 | public class ScheduledExecutorTest exten
334                      assertEquals(1, p.getActiveCount());
335                      done.await();
336                  }});
337 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
337 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
338              assertEquals(1, p.getActiveCount());
339          } finally {
340              done.countDown();
# Line 400 | Line 402 | public class ScheduledExecutorTest exten
402                          done.await();
403                          assertEquals(THREADS, p.getLargestPoolSize());
404                      }});
405 <            assertTrue(threadsStarted.await(SMALL_DELAY_MS, MILLISECONDS));
405 >            assertTrue(threadsStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
406              assertEquals(THREADS, p.getLargestPoolSize());
407          } finally {
408              done.countDown();
# Line 425 | Line 427 | public class ScheduledExecutorTest exten
427                      assertEquals(1, p.getPoolSize());
428                      done.await();
429                  }});
430 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
430 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
431              assertEquals(1, p.getPoolSize());
432          } finally {
433              done.countDown();
# Line 450 | Line 452 | public class ScheduledExecutorTest exten
452                          threadStarted.countDown();
453                          done.await();
454                      }});
455 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
455 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
456              assertEquals(TASKS, p.getTaskCount());
457          } finally {
458              done.countDown();
# Line 523 | Line 525 | public class ScheduledExecutorTest exten
525                      threadStarted.countDown();
526                      done.await();
527                  }});
528 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
528 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
529              assertFalse(p.isTerminating());
530              done.countDown();
531          } finally {
# Line 548 | Line 550 | public class ScheduledExecutorTest exten
550                      threadStarted.countDown();
551                      done.await();
552                  }});
553 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
553 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
554              assertFalse(p.isTerminating());
555              done.countDown();
556          } finally {
# Line 576 | Line 578 | public class ScheduledExecutorTest exten
578                      }};
579                  tasks[i] = p.schedule(r, 1, MILLISECONDS);
580              }
581 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
581 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
582              BlockingQueue<Runnable> q = p.getQueue();
583              assertTrue(q.contains(tasks[tasks.length - 1]));
584              assertFalse(q.contains(tasks[0]));
# Line 603 | Line 605 | public class ScheduledExecutorTest exten
605                      }};
606                  tasks[i] = p.schedule(r, 1, MILLISECONDS);
607              }
608 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
608 >            assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
609              BlockingQueue<Runnable> q = p.getQueue();
610              assertFalse(p.remove((Runnable)tasks[0]));
611              assertTrue(q.contains((Runnable)tasks[4]));
# Line 654 | Line 656 | public class ScheduledExecutorTest exten
656       * shutdownNow returns a list containing tasks that were not run,
657       * and those tasks are drained from the queue
658       */
659 <    public void testShutdownNow() {
660 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
661 <        for (int i = 0; i < 5; i++)
662 <            p.schedule(new SmallPossiblyInterruptedRunnable(),
663 <                       LONG_DELAY_MS, MILLISECONDS);
664 <        try {
665 <            List<Runnable> l = p.shutdownNow();
666 <            assertTrue(p.isShutdown());
667 <            assertTrue(p.getQueue().isEmpty());
668 <            assertEquals(5, l.size());
659 >    public void testShutdownNow() throws InterruptedException {
660 >        final int poolSize = 2;
661 >        final int count = 5;
662 >        final AtomicInteger ran = new AtomicInteger(0);
663 >        final ScheduledThreadPoolExecutor p =
664 >            new ScheduledThreadPoolExecutor(poolSize);
665 >        CountDownLatch threadsStarted = new CountDownLatch(poolSize);
666 >        Runnable waiter = new CheckedRunnable() { public void realRun() {
667 >            threadsStarted.countDown();
668 >            try {
669 >                MILLISECONDS.sleep(2 * LONG_DELAY_MS);
670 >            } catch (InterruptedException success) {}
671 >            ran.getAndIncrement();
672 >        }};
673 >        for (int i = 0; i < count; i++)
674 >            p.execute(waiter);
675 >        assertTrue(threadsStarted.await(LONG_DELAY_MS, MILLISECONDS));
676 >        assertEquals(poolSize, p.getActiveCount());
677 >        assertEquals(0, p.getCompletedTaskCount());
678 >        final List<Runnable> queuedTasks;
679 >        try {
680 >            queuedTasks = p.shutdownNow();
681          } catch (SecurityException ok) {
682 <            // Allowed in case test doesn't have privs
669 <        } finally {
670 <            joinPool(p);
682 >            return; // Allowed in case test doesn't have privs
683          }
684 +        assertTrue(p.isShutdown());
685 +        assertTrue(p.getQueue().isEmpty());
686 +        assertEquals(count - poolSize, queuedTasks.size());
687 +        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
688 +        assertTrue(p.isTerminated());
689 +        assertEquals(poolSize, ran.get());
690 +        assertEquals(poolSize, p.getCompletedTaskCount());
691      }
692  
693      /**
694 <     * In default setting, shutdown cancels periodic but not delayed
695 <     * tasks at shutdown
694 >     * shutdownNow returns a list containing tasks that were not run,
695 >     * and those tasks are drained from the queue
696       */
697 <    public void testShutdown1() throws InterruptedException {
697 >    public void testShutdownNow_delayedTasks() throws InterruptedException {
698          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
699 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
700 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
701 <
702 <        ScheduledFuture[] tasks = new ScheduledFuture[5];
703 <        for (int i = 0; i < tasks.length; i++)
704 <            tasks[i] = p.schedule(new NoOpRunnable(),
705 <                                  SHORT_DELAY_MS, MILLISECONDS);
706 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
707 <        BlockingQueue<Runnable> q = p.getQueue();
708 <        for (ScheduledFuture task : tasks) {
709 <            assertFalse(task.isDone());
710 <            assertFalse(task.isCancelled());
711 <            assertTrue(q.contains(task));
699 >        List<ScheduledFuture> tasks = new ArrayList<>();
700 >        for (int i = 0; i < 3; i++) {
701 >            Runnable r = new NoOpRunnable();
702 >            tasks.add(p.schedule(r, 9, SECONDS));
703 >            tasks.add(p.scheduleAtFixedRate(r, 9, 9, SECONDS));
704 >            tasks.add(p.scheduleWithFixedDelay(r, 9, 9, SECONDS));
705 >        }
706 >        if (testImplementationDetails)
707 >            assertEquals(new HashSet(tasks), new HashSet(p.getQueue()));
708 >        final List<Runnable> queuedTasks;
709 >        try {
710 >            queuedTasks = p.shutdownNow();
711 >        } catch (SecurityException ok) {
712 >            return; // Allowed in case test doesn't have privs
713          }
714          assertTrue(p.isShutdown());
715 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
716 <        assertTrue(p.isTerminated());
715 >        assertTrue(p.getQueue().isEmpty());
716 >        if (testImplementationDetails)
717 >            assertEquals(new HashSet(tasks), new HashSet(queuedTasks));
718 >        assertEquals(tasks.size(), queuedTasks.size());
719          for (ScheduledFuture task : tasks) {
720 <            assertTrue(task.isDone());
720 >            assertFalse(task.isDone());
721              assertFalse(task.isCancelled());
722          }
723 <    }
702 <
703 <    /**
704 <     * If setExecuteExistingDelayedTasksAfterShutdownPolicy is false,
705 <     * delayed tasks are cancelled at shutdown
706 <     */
707 <    public void testShutdown2() throws InterruptedException {
708 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
709 <        p.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
710 <        assertFalse(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
711 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
712 <        ScheduledFuture[] tasks = new ScheduledFuture[5];
713 <        for (int i = 0; i < tasks.length; i++)
714 <            tasks[i] = p.schedule(new NoOpRunnable(),
715 <                                  SHORT_DELAY_MS, MILLISECONDS);
716 <        BlockingQueue q = p.getQueue();
717 <        assertEquals(tasks.length, q.size());
718 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
719 <        assertTrue(p.isShutdown());
720 <        assertTrue(q.isEmpty());
721 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
723 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
724          assertTrue(p.isTerminated());
723        for (ScheduledFuture task : tasks) {
724            assertTrue(task.isDone());
725            assertTrue(task.isCancelled());
726        }
725      }
726  
727      /**
728 <     * If setContinueExistingPeriodicTasksAfterShutdownPolicy is set false,
729 <     * periodic tasks are cancelled at shutdown
730 <     */
731 <    public void testShutdown3() throws InterruptedException {
732 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
733 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
734 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
735 <        p.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
736 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
737 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
738 <        long initialDelay = LONG_DELAY_MS;
739 <        ScheduledFuture task =
740 <            p.scheduleAtFixedRate(new NoOpRunnable(), initialDelay,
741 <                                  5, MILLISECONDS);
728 >     * By default, periodic tasks are cancelled at shutdown.
729 >     * By default, delayed tasks keep running after shutdown.
730 >     * Check that changing the default values work:
731 >     * - setExecuteExistingDelayedTasksAfterShutdownPolicy
732 >     * - setContinueExistingPeriodicTasksAfterShutdownPolicy
733 >     */
734 >    public void testShutdown_cancellation() throws Exception {
735 >        Boolean[] allBooleans = { null, Boolean.FALSE, Boolean.TRUE };
736 >        for (Boolean policy : allBooleans)
737 >    {
738 >        final int poolSize = 2;
739 >        final ScheduledThreadPoolExecutor p
740 >            = new ScheduledThreadPoolExecutor(poolSize);
741 >        final boolean effectiveDelayedPolicy = (policy != Boolean.FALSE);
742 >        final boolean effectivePeriodicPolicy = (policy == Boolean.TRUE);
743 >        final boolean effectiveRemovePolicy = (policy == Boolean.TRUE);
744 >        if (policy != null) {
745 >            p.setExecuteExistingDelayedTasksAfterShutdownPolicy(policy);
746 >            p.setContinueExistingPeriodicTasksAfterShutdownPolicy(policy);
747 >            p.setRemoveOnCancelPolicy(policy);
748 >        }
749 >        assertEquals(effectiveDelayedPolicy,
750 >                     p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
751 >        assertEquals(effectivePeriodicPolicy,
752 >                     p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
753 >        assertEquals(effectiveRemovePolicy,
754 >                     p.getRemoveOnCancelPolicy());
755 >        // Strategy: Wedge the pool with poolSize "blocker" threads
756 >        final AtomicInteger ran = new AtomicInteger(0);
757 >        final CountDownLatch poolBlocked = new CountDownLatch(poolSize);
758 >        final CountDownLatch unblock = new CountDownLatch(1);
759 >        final CountDownLatch periodicLatch1 = new CountDownLatch(2);
760 >        final CountDownLatch periodicLatch2 = new CountDownLatch(2);
761 >        Runnable task = new CheckedRunnable() { public void realRun()
762 >                                                    throws InterruptedException {
763 >            poolBlocked.countDown();
764 >            assertTrue(unblock.await(LONG_DELAY_MS, MILLISECONDS));
765 >            ran.getAndIncrement();
766 >        }};
767 >        List<Future<?>> blockers = new ArrayList<>();
768 >        List<Future<?>> periodics = new ArrayList<>();
769 >        List<Future<?>> delayeds = new ArrayList<>();
770 >        for (int i = 0; i < poolSize; i++)
771 >            blockers.add(p.submit(task));
772 >        assertTrue(poolBlocked.await(LONG_DELAY_MS, MILLISECONDS));
773 >
774 >        periodics.add(p.scheduleAtFixedRate(countDowner(periodicLatch1),
775 >                                            1, 1, MILLISECONDS));
776 >        periodics.add(p.scheduleWithFixedDelay(countDowner(periodicLatch2),
777 >                                               1, 1, MILLISECONDS));
778 >        delayeds.add(p.schedule(task, 1, MILLISECONDS));
779 >
780 >        assertTrue(p.getQueue().containsAll(periodics));
781 >        assertTrue(p.getQueue().containsAll(delayeds));
782          try { p.shutdown(); } catch (SecurityException ok) { return; }
783          assertTrue(p.isShutdown());
784 <        assertTrue(p.getQueue().isEmpty());
785 <        assertTrue(task.isDone());
786 <        assertTrue(task.isCancelled());
787 <        joinPool(p);
788 <    }
789 <
790 <    /**
791 <     * if setContinueExistingPeriodicTasksAfterShutdownPolicy is true,
792 <     * periodic tasks are not cancelled at shutdown
793 <     */
794 <    public void testShutdown4() throws InterruptedException {
795 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
796 <        final CountDownLatch counter = new CountDownLatch(2);
797 <        try {
798 <            p.setContinueExistingPeriodicTasksAfterShutdownPolicy(true);
799 <            assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
800 <            assertTrue(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
801 <            final Runnable r = new CheckedRunnable() {
802 <                public void realRun() {
803 <                    counter.countDown();
804 <                }};
805 <            ScheduledFuture task =
768 <                p.scheduleAtFixedRate(r, 1, 1, MILLISECONDS);
769 <            assertFalse(task.isDone());
770 <            assertFalse(task.isCancelled());
771 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
772 <            assertFalse(task.isCancelled());
773 <            assertFalse(p.isTerminated());
774 <            assertTrue(p.isShutdown());
775 <            assertTrue(counter.await(SMALL_DELAY_MS, MILLISECONDS));
776 <            assertFalse(task.isCancelled());
777 <            assertTrue(task.cancel(false));
778 <            assertTrue(task.isDone());
779 <            assertTrue(task.isCancelled());
780 <            assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
781 <            assertTrue(p.isTerminated());
784 >        assertFalse(p.isTerminated());
785 >        for (Future<?> periodic : periodics) {
786 >            assertTrue(effectivePeriodicPolicy ^ periodic.isCancelled());
787 >            assertTrue(effectivePeriodicPolicy ^ periodic.isDone());
788 >        }
789 >        for (Future<?> delayed : delayeds) {
790 >            assertTrue(effectiveDelayedPolicy ^ delayed.isCancelled());
791 >            assertTrue(effectiveDelayedPolicy ^ delayed.isDone());
792 >        }
793 >        if (testImplementationDetails) {
794 >            assertEquals(effectivePeriodicPolicy,
795 >                         p.getQueue().containsAll(periodics));
796 >            assertEquals(effectiveDelayedPolicy,
797 >                         p.getQueue().containsAll(delayeds));
798 >        }
799 >        // Release all pool threads
800 >        unblock.countDown();
801 >
802 >        for (Future<?> delayed : delayeds) {
803 >            if (effectiveDelayedPolicy) {
804 >                assertNull(delayed.get());
805 >            }
806          }
807 <        finally {
808 <            joinPool(p);
807 >        if (effectivePeriodicPolicy) {
808 >            assertTrue(periodicLatch1.await(LONG_DELAY_MS, MILLISECONDS));
809 >            assertTrue(periodicLatch2.await(LONG_DELAY_MS, MILLISECONDS));
810 >            for (Future<?> periodic : periodics) {
811 >                assertTrue(periodic.cancel(false));
812 >                assertTrue(periodic.isCancelled());
813 >                assertTrue(periodic.isDone());
814 >            }
815          }
816 <    }
816 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
817 >        assertTrue(p.isTerminated());
818 >        assertEquals(2 + (effectiveDelayedPolicy ? 1 : 0), ran.get());
819 >    }}
820  
821      /**
822       * completed submit of callable returns result
# Line 1174 | Line 1207 | public class ScheduledExecutorTest exten
1207              l.add(new StringTask());
1208              l.add(new StringTask());
1209              List<Future<String>> futures =
1210 <                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1210 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1211              assertEquals(2, futures.size());
1212              for (Future<String> future : futures)
1213                  assertSame(TEST_STRING, future.get());

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines