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.40 by jsr166, Sat May 7 19:34:51 2011 UTC vs.
Revision 1.56 by jsr166, Mon Sep 28 02:41:29 2015 UTC

# Line 6 | Line 6
6   * Pat Fisher, Mike Judd.
7   */
8  
9 import junit.framework.*;
10 import java.util.*;
11 import java.util.concurrent.*;
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 < import java.util.concurrent.atomic.*;
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;
17 > import java.util.concurrent.CancellationException;
18 > import java.util.concurrent.CountDownLatch;
19 > import java.util.concurrent.ExecutionException;
20 > import java.util.concurrent.Executors;
21 > import java.util.concurrent.ExecutorService;
22 > import java.util.concurrent.Future;
23 > import java.util.concurrent.RejectedExecutionException;
24 > import java.util.concurrent.ScheduledFuture;
25 > import java.util.concurrent.ScheduledThreadPoolExecutor;
26 > import java.util.concurrent.ThreadFactory;
27 > import java.util.concurrent.ThreadPoolExecutor;
28 > import java.util.concurrent.atomic.AtomicInteger;
29 >
30 > import junit.framework.Test;
31 > import junit.framework.TestSuite;
32  
33   public class ScheduledExecutorTest extends JSR166TestCase {
34      public static void main(String[] args) {
35 <        junit.textui.TestRunner.run(suite());
35 >        main(suite(), args);
36      }
37      public static Test suite() {
38          return new TestSuite(ScheduledExecutorTest.class);
39      }
40  
23
41      /**
42       * execute successfully executes a runnable
43       */
# Line 39 | Line 56 | public class ScheduledExecutorTest exten
56          }
57      }
58  
42
59      /**
60       * delayed schedule of callable successfully executes after delay
61       */
62      public void testSchedule1() throws Exception {
63          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
64 <        final long t0 = System.nanoTime();
49 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
64 >        final long startTime = System.nanoTime();
65          final CountDownLatch done = new CountDownLatch(1);
66          try {
67              Callable task = new CheckedCallable<Boolean>() {
68                  public Boolean realCall() {
69                      done.countDown();
70 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
70 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
71                      return Boolean.TRUE;
72                  }};
73 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
74 <            assertEquals(Boolean.TRUE, f.get());
75 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
73 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
74 >            assertSame(Boolean.TRUE, f.get());
75 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
76              assertTrue(done.await(0L, MILLISECONDS));
77          } finally {
78              joinPool(p);
# Line 69 | Line 84 | public class ScheduledExecutorTest exten
84       */
85      public void testSchedule3() throws Exception {
86          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
87 <        final long t0 = System.nanoTime();
73 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
87 >        final long startTime = System.nanoTime();
88          final CountDownLatch done = new CountDownLatch(1);
89          try {
90              Runnable task = new CheckedRunnable() {
91                  public void realRun() {
92                      done.countDown();
93 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
93 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
94                  }};
95 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
96 <            assertNull(f.get());
97 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
98 <            assertTrue(done.await(0L, MILLISECONDS));
95 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
96 >            await(done);
97 >            assertNull(f.get(LONG_DELAY_MS, MILLISECONDS));
98 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
99          } finally {
100              joinPool(p);
101          }
# Line 92 | Line 106 | public class ScheduledExecutorTest exten
106       */
107      public void testSchedule4() throws Exception {
108          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
109 <        final long t0 = System.nanoTime();
96 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
109 >        final long startTime = System.nanoTime();
110          final CountDownLatch done = new CountDownLatch(1);
111          try {
112              Runnable task = new CheckedRunnable() {
113                  public void realRun() {
114                      done.countDown();
115 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
115 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
116                  }};
117              ScheduledFuture f =
118 <                p.scheduleAtFixedRate(task, SHORT_DELAY_MS,
119 <                                      SHORT_DELAY_MS, MILLISECONDS);
120 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
121 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
118 >                p.scheduleAtFixedRate(task, timeoutMillis(),
119 >                                      LONG_DELAY_MS, MILLISECONDS);
120 >            await(done);
121 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
122              f.cancel(true);
123          } finally {
124              joinPool(p);
# Line 115 | Line 128 | public class ScheduledExecutorTest exten
128      /**
129       * scheduleWithFixedDelay executes runnable after given initial delay
130       */
131 <    public void testSchedule5() throws InterruptedException {
131 >    public void testSchedule5() throws Exception {
132          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
133 <        final long t0 = System.nanoTime();
121 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
133 >        final long startTime = System.nanoTime();
134          final CountDownLatch done = new CountDownLatch(1);
135          try {
136              Runnable task = new CheckedRunnable() {
137                  public void realRun() {
138                      done.countDown();
139 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
139 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
140                  }};
141              ScheduledFuture f =
142 <                p.scheduleWithFixedDelay(task, SHORT_DELAY_MS,
143 <                                         SHORT_DELAY_MS, MILLISECONDS);
144 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
145 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
142 >                p.scheduleWithFixedDelay(task, timeoutMillis(),
143 >                                         LONG_DELAY_MS, MILLISECONDS);
144 >            await(done);
145 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
146              f.cancel(true);
147          } finally {
148              joinPool(p);
# Line 147 | Line 159 | public class ScheduledExecutorTest exten
159       */
160      public void testFixedRateSequence() throws InterruptedException {
161          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
162 <        RunnableCounter counter = new RunnableCounter();
163 <        ScheduledFuture h =
164 <            p.scheduleAtFixedRate(counter, 0, 1, MILLISECONDS);
165 <        delay(SMALL_DELAY_MS);
166 <        h.cancel(true);
167 <        int c = counter.count.get();
168 <        // By time scaling conventions, we must have at least
169 <        // an execution per SHORT delay, but no more than one SHORT more
170 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
171 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
172 <        joinPool(p);
162 >        try {
163 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
164 >                long startTime = System.nanoTime();
165 >                int cycles = 10;
166 >                final CountDownLatch done = new CountDownLatch(cycles);
167 >                Runnable task = new CheckedRunnable() {
168 >                    public void realRun() { done.countDown(); }};
169 >                ScheduledFuture h =
170 >                    p.scheduleAtFixedRate(task, 0, delay, MILLISECONDS);
171 >                done.await();
172 >                h.cancel(true);
173 >                double normalizedTime =
174 >                    (double) millisElapsedSince(startTime) / delay;
175 >                if (normalizedTime >= cycles - 1 &&
176 >                    normalizedTime <= cycles)
177 >                    return;
178 >            }
179 >            throw new AssertionError("unexpected execution rate");
180 >        } finally {
181 >            joinPool(p);
182 >        }
183      }
184  
185      /**
# Line 165 | Line 187 | public class ScheduledExecutorTest exten
187       */
188      public void testFixedDelaySequence() throws InterruptedException {
189          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
190 <        RunnableCounter counter = new RunnableCounter();
191 <        ScheduledFuture h =
192 <            p.scheduleWithFixedDelay(counter, 0, 1, MILLISECONDS);
193 <        delay(SMALL_DELAY_MS);
194 <        h.cancel(true);
195 <        int c = counter.count.get();
196 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
197 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
198 <        joinPool(p);
190 >        try {
191 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
192 >                long startTime = System.nanoTime();
193 >                int cycles = 10;
194 >                final CountDownLatch done = new CountDownLatch(cycles);
195 >                Runnable task = new CheckedRunnable() {
196 >                    public void realRun() { done.countDown(); }};
197 >                ScheduledFuture h =
198 >                    p.scheduleWithFixedDelay(task, 0, delay, MILLISECONDS);
199 >                done.await();
200 >                h.cancel(true);
201 >                double normalizedTime =
202 >                    (double) millisElapsedSince(startTime) / delay;
203 >                if (normalizedTime >= cycles - 1 &&
204 >                    normalizedTime <= cycles)
205 >                    return;
206 >            }
207 >            throw new AssertionError("unexpected execution rate");
208 >        } finally {
209 >            joinPool(p);
210 >        }
211      }
212  
179
213      /**
214       * execute(null) throws NPE
215       */
# Line 327 | Line 360 | public class ScheduledExecutorTest exten
360                      threadProceed.await();
361                      threadDone.countDown();
362                  }});
363 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
363 >            await(threadStarted);
364              assertEquals(0, p.getCompletedTaskCount());
365              threadProceed.countDown();
366              threadDone.await();
367 <            delay(SHORT_DELAY_MS);
368 <            assertEquals(1, p.getCompletedTaskCount());
367 >            long startTime = System.nanoTime();
368 >            while (p.getCompletedTaskCount() != 1) {
369 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
370 >                    fail("timed out");
371 >                Thread.yield();
372 >            }
373          } finally {
374              joinPool(p);
375          }
# Line 459 | Line 496 | public class ScheduledExecutorTest exten
496      }
497  
498      /**
499 <     * isShutDown is false before shutdown, true after
499 >     * isShutdown is false before shutdown, true after
500       */
501      public void testIsShutdown() {
502  
# Line 473 | Line 510 | public class ScheduledExecutorTest exten
510          assertTrue(p.isShutdown());
511      }
512  
476
513      /**
514       * isTerminated is false before termination, true after
515       */
# Line 617 | Line 653 | public class ScheduledExecutorTest exten
653      }
654  
655      /**
656 <     * shutDownNow returns a list containing tasks that were not run
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() {
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 >        CheckedRunnable 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 >        final List<Runnable> queuedTasks;
677 >        try {
678 >            queuedTasks = p.shutdownNow();
679 >        } catch (SecurityException ok) {
680 >            return; // Allowed in case test doesn't have privs
681 >        }
682 >        assertTrue(p.isShutdown());
683 >        assertTrue(p.getQueue().isEmpty());
684 >        assertEquals(count - poolSize, queuedTasks.size());
685 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
686 >        assertTrue(p.isTerminated());
687 >        assertEquals(poolSize, ran.get());
688 >    }
689 >
690 >    /**
691 >     * shutdownNow returns a list containing tasks that were not run,
692 >     * and those tasks are drained from the queue
693 >     */
694 >    public void testShutdownNow_delayedTasks() throws InterruptedException {
695          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
696 <        for (int i = 0; i < 5; i++)
697 <            p.schedule(new SmallPossiblyInterruptedRunnable(),
698 <                       LONG_DELAY_MS, MILLISECONDS);
696 >        List<ScheduledFuture> tasks = new ArrayList<>();
697 >        for (int i = 0; i < 3; i++) {
698 >            Runnable r = new NoOpRunnable();
699 >            tasks.add(p.schedule(r, 9, SECONDS));
700 >            tasks.add(p.scheduleAtFixedRate(r, 9, 9, SECONDS));
701 >            tasks.add(p.scheduleWithFixedDelay(r, 9, 9, SECONDS));
702 >        }
703 >        if (testImplementationDetails)
704 >            assertEquals(new HashSet(tasks), new HashSet(p.getQueue()));
705 >        final List<Runnable> queuedTasks;
706          try {
707 <            List<Runnable> l = p.shutdownNow();
629 <            assertTrue(p.isShutdown());
630 <            assertEquals(5, l.size());
707 >            queuedTasks = p.shutdownNow();
708          } catch (SecurityException ok) {
709 <            // Allowed in case test doesn't have privs
633 <        } finally {
634 <            joinPool(p);
709 >            return; // Allowed in case test doesn't have privs
710          }
711 +        assertTrue(p.isShutdown());
712 +        assertTrue(p.getQueue().isEmpty());
713 +        if (testImplementationDetails)
714 +            assertEquals(new HashSet(tasks), new HashSet(queuedTasks));
715 +        assertEquals(tasks.size(), queuedTasks.size());
716 +        for (ScheduledFuture task : tasks) {
717 +            assertFalse(task.isDone());
718 +            assertFalse(task.isCancelled());
719 +        }
720 +        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
721 +        assertTrue(p.isTerminated());
722      }
723  
724      /**
725       * In default setting, shutdown cancels periodic but not delayed
726       * tasks at shutdown
727       */
728 <    public void testShutDown1() throws InterruptedException {
728 >    public void testShutdown1() throws InterruptedException {
729          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
730          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
731          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
# Line 664 | Line 750 | public class ScheduledExecutorTest exten
750          }
751      }
752  
667
753      /**
754       * If setExecuteExistingDelayedTasksAfterShutdownPolicy is false,
755       * delayed tasks are cancelled at shutdown
756       */
757 <    public void testShutDown2() throws InterruptedException {
757 >    public void testShutdown2() throws InterruptedException {
758          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
759          p.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
760          assertFalse(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
# Line 695 | Line 780 | public class ScheduledExecutorTest exten
780       * If setContinueExistingPeriodicTasksAfterShutdownPolicy is set false,
781       * periodic tasks are cancelled at shutdown
782       */
783 <    public void testShutDown3() throws InterruptedException {
783 >    public void testShutdown3() throws InterruptedException {
784          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
785          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
786          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
787          p.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
788          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
789          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
790 +        long initialDelay = LONG_DELAY_MS;
791          ScheduledFuture task =
792 <            p.scheduleAtFixedRate(new NoOpRunnable(), 5, 5, MILLISECONDS);
792 >            p.scheduleAtFixedRate(new NoOpRunnable(), initialDelay,
793 >                                  5, MILLISECONDS);
794          try { p.shutdown(); } catch (SecurityException ok) { return; }
795          assertTrue(p.isShutdown());
709        BlockingQueue q = p.getQueue();
796          assertTrue(p.getQueue().isEmpty());
797          assertTrue(task.isDone());
798          assertTrue(task.isCancelled());
799 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
714 <        assertTrue(p.isTerminated());
799 >        joinPool(p);
800      }
801  
802      /**
803       * if setContinueExistingPeriodicTasksAfterShutdownPolicy is true,
804       * periodic tasks are not cancelled at shutdown
805       */
806 <    public void testShutDown4() throws InterruptedException {
806 >    public void testShutdown4() throws InterruptedException {
807          ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
808          final CountDownLatch counter = new CountDownLatch(2);
809          try {
# Line 1154 | Line 1239 | public class ScheduledExecutorTest exten
1239      public void testTimedInvokeAll6() throws Exception {
1240          ExecutorService e = new ScheduledThreadPoolExecutor(2);
1241          try {
1242 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1243 <            l.add(new StringTask());
1244 <            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
1245 <            l.add(new StringTask());
1246 <            List<Future<String>> futures =
1247 <                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
1248 <            assertEquals(3, futures.size());
1249 <            Iterator<Future<String>> it = futures.iterator();
1250 <            Future<String> f1 = it.next();
1251 <            Future<String> f2 = it.next();
1252 <            Future<String> f3 = it.next();
1253 <            assertTrue(f1.isDone());
1254 <            assertTrue(f2.isDone());
1255 <            assertTrue(f3.isDone());
1256 <            assertFalse(f1.isCancelled());
1257 <            assertTrue(f2.isCancelled());
1242 >            for (long timeout = timeoutMillis();;) {
1243 >                List<Callable<String>> tasks = new ArrayList<>();
1244 >                tasks.add(new StringTask("0"));
1245 >                tasks.add(Executors.callable(new LongPossiblyInterruptedRunnable(), TEST_STRING));
1246 >                tasks.add(new StringTask("2"));
1247 >                long startTime = System.nanoTime();
1248 >                List<Future<String>> futures =
1249 >                    e.invokeAll(tasks, timeout, MILLISECONDS);
1250 >                assertEquals(tasks.size(), futures.size());
1251 >                assertTrue(millisElapsedSince(startTime) >= timeout);
1252 >                for (Future future : futures)
1253 >                    assertTrue(future.isDone());
1254 >                assertTrue(futures.get(1).isCancelled());
1255 >                try {
1256 >                    assertEquals("0", futures.get(0).get());
1257 >                    assertEquals("2", futures.get(2).get());
1258 >                    break;
1259 >                } catch (CancellationException retryWithLongerTimeout) {
1260 >                    timeout *= 2;
1261 >                    if (timeout >= LONG_DELAY_MS / 2)
1262 >                        fail("expected exactly one task to be cancelled");
1263 >                }
1264 >            }
1265          } finally {
1266              joinPool(e);
1267          }
1268      }
1269  
1178
1270   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines