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

Comparing jsr166/src/test/tck/ScheduledExecutorSubclassTest.java (file contents):
Revision 1.22 by jsr166, Sat May 7 19:34:51 2011 UTC vs.
Revision 1.40 by jsr166, Mon Sep 28 02:41:29 2015 UTC

# Line 4 | Line 4
4   * http://creativecommons.org/publicdomain/zero/1.0/
5   */
6  
7 import junit.framework.*;
8 import java.util.*;
9 import java.util.concurrent.*;
7   import static java.util.concurrent.TimeUnit.MILLISECONDS;
8 < import java.util.concurrent.atomic.*;
8 > import static java.util.concurrent.TimeUnit.SECONDS;
9 >
10 > import java.util.ArrayList;
11 > import java.util.HashSet;
12 > import java.util.List;
13 > import java.util.concurrent.BlockingQueue;
14 > import java.util.concurrent.Callable;
15 > import java.util.concurrent.CancellationException;
16 > import java.util.concurrent.CountDownLatch;
17 > import java.util.concurrent.Delayed;
18 > import java.util.concurrent.ExecutionException;
19 > import java.util.concurrent.Executors;
20 > import java.util.concurrent.ExecutorService;
21 > import java.util.concurrent.Future;
22 > import java.util.concurrent.RejectedExecutionException;
23 > import java.util.concurrent.RejectedExecutionHandler;
24 > import java.util.concurrent.RunnableScheduledFuture;
25 > import java.util.concurrent.ScheduledFuture;
26 > import java.util.concurrent.ScheduledThreadPoolExecutor;
27 > import java.util.concurrent.ThreadFactory;
28 > import java.util.concurrent.ThreadPoolExecutor;
29 > import java.util.concurrent.TimeoutException;
30 > import java.util.concurrent.TimeUnit;
31 > import java.util.concurrent.atomic.AtomicInteger;
32 >
33 > import junit.framework.Test;
34 > import junit.framework.TestSuite;
35  
36   public class ScheduledExecutorSubclassTest extends JSR166TestCase {
37      public static void main(String[] args) {
38 <        junit.textui.TestRunner.run(suite());
38 >        main(suite(), args);
39      }
40      public static Test suite() {
41          return new TestSuite(ScheduledExecutorSubclassTest.class);
# Line 36 | Line 59 | public class ScheduledExecutorSubclassTe
59          }
60          public boolean isCancelled() { return task.isCancelled(); }
61          public boolean isDone() { return task.isDone(); }
62 <        public V get() throws InterruptedException,  ExecutionException {
62 >        public V get() throws InterruptedException, ExecutionException {
63              V v = task.get();
64              assertTrue(ran);
65              return v;
66          }
67 <        public V get(long time, TimeUnit unit) throws InterruptedException,  ExecutionException, TimeoutException {
67 >        public V get(long time, TimeUnit unit) throws InterruptedException, ExecutionException, TimeoutException {
68              V v = task.get(time, unit);
69              assertTrue(ran);
70              return v;
71          }
72      }
73  
51
74      public class CustomExecutor extends ScheduledThreadPoolExecutor {
75  
76          protected <V> RunnableScheduledFuture<V> decorateTask(Runnable r, RunnableScheduledFuture<V> task) {
# Line 58 | Line 80 | public class ScheduledExecutorSubclassTe
80          protected <V> RunnableScheduledFuture<V> decorateTask(Callable<V> c, RunnableScheduledFuture<V> task) {
81              return new CustomTask<V>(task);
82          }
83 <        CustomExecutor(int corePoolSize) { super(corePoolSize);}
83 >        CustomExecutor(int corePoolSize) { super(corePoolSize); }
84          CustomExecutor(int corePoolSize, RejectedExecutionHandler handler) {
85              super(corePoolSize, handler);
86          }
# Line 73 | Line 95 | public class ScheduledExecutorSubclassTe
95  
96      }
97  
76
98      /**
99       * execute successfully executes a runnable
100       */
# Line 92 | Line 113 | public class ScheduledExecutorSubclassTe
113          }
114      }
115  
95
116      /**
117       * delayed schedule of callable successfully executes after delay
118       */
119      public void testSchedule1() throws Exception {
120          CustomExecutor p = new CustomExecutor(1);
121 <        final long t0 = System.nanoTime();
102 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
121 >        final long startTime = System.nanoTime();
122          final CountDownLatch done = new CountDownLatch(1);
123          try {
124              Callable task = new CheckedCallable<Boolean>() {
125                  public Boolean realCall() {
126                      done.countDown();
127 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
127 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
128                      return Boolean.TRUE;
129                  }};
130 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
131 <            assertEquals(Boolean.TRUE, f.get());
132 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
130 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
131 >            assertSame(Boolean.TRUE, f.get());
132 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
133              assertTrue(done.await(0L, MILLISECONDS));
134          } finally {
135              joinPool(p);
# Line 122 | Line 141 | public class ScheduledExecutorSubclassTe
141       */
142      public void testSchedule3() throws Exception {
143          CustomExecutor p = new CustomExecutor(1);
144 <        final long t0 = System.nanoTime();
126 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
144 >        final long startTime = System.nanoTime();
145          final CountDownLatch done = new CountDownLatch(1);
146          try {
147              Runnable task = new CheckedRunnable() {
148                  public void realRun() {
149                      done.countDown();
150 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
150 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
151                  }};
152 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
153 <            assertNull(f.get());
154 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
155 <            assertTrue(done.await(0L, MILLISECONDS));
152 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
153 >            await(done);
154 >            assertNull(f.get(LONG_DELAY_MS, MILLISECONDS));
155 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
156          } finally {
157              joinPool(p);
158          }
# Line 145 | Line 163 | public class ScheduledExecutorSubclassTe
163       */
164      public void testSchedule4() throws InterruptedException {
165          CustomExecutor p = new CustomExecutor(1);
166 <        final long t0 = System.nanoTime();
149 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
166 >        final long startTime = System.nanoTime();
167          final CountDownLatch done = new CountDownLatch(1);
168          try {
169              Runnable task = new CheckedRunnable() {
170                  public void realRun() {
171                      done.countDown();
172 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
172 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
173                  }};
174              ScheduledFuture f =
175 <                p.scheduleAtFixedRate(task, SHORT_DELAY_MS,
176 <                                      SHORT_DELAY_MS, MILLISECONDS);
177 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
178 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
175 >                p.scheduleAtFixedRate(task, timeoutMillis(),
176 >                                      LONG_DELAY_MS, MILLISECONDS);
177 >            await(done);
178 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
179              f.cancel(true);
180          } finally {
181              joinPool(p);
# Line 170 | Line 187 | public class ScheduledExecutorSubclassTe
187       */
188      public void testSchedule5() throws InterruptedException {
189          CustomExecutor p = new CustomExecutor(1);
190 <        final long t0 = System.nanoTime();
174 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
190 >        final long startTime = System.nanoTime();
191          final CountDownLatch done = new CountDownLatch(1);
192          try {
193              Runnable task = new CheckedRunnable() {
194                  public void realRun() {
195                      done.countDown();
196 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
196 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
197                  }};
198              ScheduledFuture f =
199 <                p.scheduleWithFixedDelay(task, SHORT_DELAY_MS,
200 <                                         SHORT_DELAY_MS, MILLISECONDS);
201 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
202 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
199 >                p.scheduleWithFixedDelay(task, timeoutMillis(),
200 >                                         LONG_DELAY_MS, MILLISECONDS);
201 >            await(done);
202 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
203              f.cancel(true);
204          } finally {
205              joinPool(p);
# Line 200 | Line 216 | public class ScheduledExecutorSubclassTe
216       */
217      public void testFixedRateSequence() throws InterruptedException {
218          CustomExecutor p = new CustomExecutor(1);
219 <        RunnableCounter counter = new RunnableCounter();
220 <        ScheduledFuture h =
221 <            p.scheduleAtFixedRate(counter, 0, 1, MILLISECONDS);
222 <        delay(SMALL_DELAY_MS);
223 <        h.cancel(true);
224 <        int c = counter.count.get();
225 <        // By time scaling conventions, we must have at least
226 <        // an execution per SHORT delay, but no more than one SHORT more
227 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
228 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
229 <        joinPool(p);
219 >        try {
220 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
221 >                long startTime = System.nanoTime();
222 >                int cycles = 10;
223 >                final CountDownLatch done = new CountDownLatch(cycles);
224 >                Runnable task = new CheckedRunnable() {
225 >                    public void realRun() { done.countDown(); }};
226 >                ScheduledFuture h =
227 >                    p.scheduleAtFixedRate(task, 0, delay, MILLISECONDS);
228 >                done.await();
229 >                h.cancel(true);
230 >                double normalizedTime =
231 >                    (double) millisElapsedSince(startTime) / delay;
232 >                if (normalizedTime >= cycles - 1 &&
233 >                    normalizedTime <= cycles)
234 >                    return;
235 >            }
236 >            throw new AssertionError("unexpected execution rate");
237 >        } finally {
238 >            joinPool(p);
239 >        }
240      }
241  
242      /**
# Line 218 | Line 244 | public class ScheduledExecutorSubclassTe
244       */
245      public void testFixedDelaySequence() throws InterruptedException {
246          CustomExecutor p = new CustomExecutor(1);
247 <        RunnableCounter counter = new RunnableCounter();
248 <        ScheduledFuture h =
249 <            p.scheduleWithFixedDelay(counter, 0, 1, MILLISECONDS);
250 <        delay(SMALL_DELAY_MS);
251 <        h.cancel(true);
252 <        int c = counter.count.get();
253 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
254 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
255 <        joinPool(p);
247 >        try {
248 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
249 >                long startTime = System.nanoTime();
250 >                int cycles = 10;
251 >                final CountDownLatch done = new CountDownLatch(cycles);
252 >                Runnable task = new CheckedRunnable() {
253 >                    public void realRun() { done.countDown(); }};
254 >                ScheduledFuture h =
255 >                    p.scheduleWithFixedDelay(task, 0, delay, MILLISECONDS);
256 >                done.await();
257 >                h.cancel(true);
258 >                double normalizedTime =
259 >                    (double) millisElapsedSince(startTime) / delay;
260 >                if (normalizedTime >= cycles - 1 &&
261 >                    normalizedTime <= cycles)
262 >                    return;
263 >            }
264 >            throw new AssertionError("unexpected execution rate");
265 >        } finally {
266 >            joinPool(p);
267 >        }
268      }
269  
232
270      /**
271       * execute(null) throws NPE
272       */
# Line 378 | Line 415 | public class ScheduledExecutorSubclassTe
415                      threadProceed.await();
416                      threadDone.countDown();
417                  }});
418 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
418 >            await(threadStarted);
419              assertEquals(0, p.getCompletedTaskCount());
420              threadProceed.countDown();
421              threadDone.await();
422 <            delay(SHORT_DELAY_MS);
423 <            assertEquals(1, p.getCompletedTaskCount());
422 >            long startTime = System.nanoTime();
423 >            while (p.getCompletedTaskCount() != 1) {
424 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
425 >                    fail("timed out");
426 >                Thread.yield();
427 >            }
428          } finally {
429              joinPool(p);
430          }
# Line 510 | Line 551 | public class ScheduledExecutorSubclassTe
551      }
552  
553      /**
554 <     * isShutDown is false before shutdown, true after
554 >     * isShutdown is false before shutdown, true after
555       */
556      public void testIsShutdown() {
557          CustomExecutor p = new CustomExecutor(1);
# Line 523 | Line 564 | public class ScheduledExecutorSubclassTe
564          assertTrue(p.isShutdown());
565      }
566  
526
567      /**
568       * isTerminated is false before termination, true after
569       */
# Line 667 | Line 707 | public class ScheduledExecutorSubclassTe
707      }
708  
709      /**
710 <     * shutDownNow returns a list containing tasks that were not run
710 >     * shutdownNow returns a list containing tasks that were not run,
711 >     * and those tasks are drained from the queue
712 >     */
713 >    public void testShutdownNow() throws InterruptedException {
714 >        final int poolSize = 2;
715 >        final int count = 5;
716 >        final AtomicInteger ran = new AtomicInteger(0);
717 >        final CustomExecutor p = new CustomExecutor(poolSize);
718 >        CountDownLatch threadsStarted = new CountDownLatch(poolSize);
719 >        CheckedRunnable waiter = new CheckedRunnable() { public void realRun() {
720 >            threadsStarted.countDown();
721 >            try {
722 >                MILLISECONDS.sleep(2 * LONG_DELAY_MS);
723 >            } catch (InterruptedException success) {}
724 >            ran.getAndIncrement();
725 >        }};
726 >        for (int i = 0; i < count; i++)
727 >            p.execute(waiter);
728 >        assertTrue(threadsStarted.await(LONG_DELAY_MS, MILLISECONDS));
729 >        final List<Runnable> queuedTasks;
730 >        try {
731 >            queuedTasks = p.shutdownNow();
732 >        } catch (SecurityException ok) {
733 >            return; // Allowed in case test doesn't have privs
734 >        }
735 >        assertTrue(p.isShutdown());
736 >        assertTrue(p.getQueue().isEmpty());
737 >        assertEquals(count - poolSize, queuedTasks.size());
738 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
739 >        assertTrue(p.isTerminated());
740 >        assertEquals(poolSize, ran.get());
741 >    }
742 >
743 >    /**
744 >     * shutdownNow returns a list containing tasks that were not run,
745 >     * and those tasks are drained from the queue
746       */
747 <    public void testShutDownNow() {
747 >    public void testShutdownNow_delayedTasks() throws InterruptedException {
748          CustomExecutor p = new CustomExecutor(1);
749 <        for (int i = 0; i < 5; i++)
750 <            p.schedule(new SmallPossiblyInterruptedRunnable(),
751 <                       LONG_DELAY_MS, MILLISECONDS);
749 >        List<ScheduledFuture> tasks = new ArrayList<>();
750 >        for (int i = 0; i < 3; i++) {
751 >            Runnable r = new NoOpRunnable();
752 >            tasks.add(p.schedule(r, 9, SECONDS));
753 >            tasks.add(p.scheduleAtFixedRate(r, 9, 9, SECONDS));
754 >            tasks.add(p.scheduleWithFixedDelay(r, 9, 9, SECONDS));
755 >        }
756 >        if (testImplementationDetails)
757 >            assertEquals(new HashSet(tasks), new HashSet(p.getQueue()));
758 >        final List<Runnable> queuedTasks;
759          try {
760 <            List<Runnable> l = p.shutdownNow();
679 <            assertTrue(p.isShutdown());
680 <            assertEquals(5, l.size());
760 >            queuedTasks = p.shutdownNow();
761          } catch (SecurityException ok) {
762 <            // Allowed in case test doesn't have privs
763 <        } finally {
764 <            joinPool(p);
762 >            return; // Allowed in case test doesn't have privs
763 >        }
764 >        assertTrue(p.isShutdown());
765 >        assertTrue(p.getQueue().isEmpty());
766 >        if (testImplementationDetails)
767 >            assertEquals(new HashSet(tasks), new HashSet(queuedTasks));
768 >        assertEquals(tasks.size(), queuedTasks.size());
769 >        for (ScheduledFuture task : tasks) {
770 >            assertFalse(((CustomTask)task).ran);
771 >            assertFalse(task.isDone());
772 >            assertFalse(task.isCancelled());
773          }
774 +        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
775 +        assertTrue(p.isTerminated());
776      }
777  
778      /**
779       * In default setting, shutdown cancels periodic but not delayed
780       * tasks at shutdown
781       */
782 <    public void testShutDown1() throws InterruptedException {
782 >    public void testShutdown1() throws InterruptedException {
783          CustomExecutor p = new CustomExecutor(1);
784          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
785          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
# Line 714 | Line 804 | public class ScheduledExecutorSubclassTe
804          }
805      }
806  
717
807      /**
808       * If setExecuteExistingDelayedTasksAfterShutdownPolicy is false,
809       * delayed tasks are cancelled at shutdown
810       */
811 <    public void testShutDown2() throws InterruptedException {
811 >    public void testShutdown2() throws InterruptedException {
812          CustomExecutor p = new CustomExecutor(1);
813          p.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
814          assertFalse(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
# Line 741 | Line 830 | public class ScheduledExecutorSubclassTe
830          }
831      }
832  
744
833      /**
834       * If setContinueExistingPeriodicTasksAfterShutdownPolicy is set false,
835       * periodic tasks are cancelled at shutdown
836       */
837 <    public void testShutDown3() throws InterruptedException {
837 >    public void testShutdown3() throws InterruptedException {
838          CustomExecutor p = new CustomExecutor(1);
839          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
840          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
841          p.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
842          assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
843          assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
844 +        long initialDelay = LONG_DELAY_MS;
845          ScheduledFuture task =
846 <            p.scheduleAtFixedRate(new NoOpRunnable(), 5, 5, MILLISECONDS);
846 >            p.scheduleAtFixedRate(new NoOpRunnable(), initialDelay,
847 >                                  5, MILLISECONDS);
848          try { p.shutdown(); } catch (SecurityException ok) { return; }
849          assertTrue(p.isShutdown());
760        BlockingQueue q = p.getQueue();
850          assertTrue(p.getQueue().isEmpty());
851          assertTrue(task.isDone());
852          assertTrue(task.isCancelled());
853 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
765 <        assertTrue(p.isTerminated());
853 >        joinPool(p);
854      }
855  
856      /**
857       * if setContinueExistingPeriodicTasksAfterShutdownPolicy is true,
858       * periodic tasks are not cancelled at shutdown
859       */
860 <    public void testShutDown4() throws InterruptedException {
860 >    public void testShutdown4() throws InterruptedException {
861          CustomExecutor p = new CustomExecutor(1);
862          final CountDownLatch counter = new CountDownLatch(2);
863          try {
# Line 1205 | Line 1293 | public class ScheduledExecutorSubclassTe
1293      public void testTimedInvokeAll6() throws Exception {
1294          ExecutorService e = new CustomExecutor(2);
1295          try {
1296 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1297 <            l.add(new StringTask());
1298 <            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
1299 <            l.add(new StringTask());
1300 <            List<Future<String>> futures =
1301 <                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
1302 <            assertEquals(3, futures.size());
1303 <            Iterator<Future<String>> it = futures.iterator();
1304 <            Future<String> f1 = it.next();
1305 <            Future<String> f2 = it.next();
1306 <            Future<String> f3 = it.next();
1307 <            assertTrue(f1.isDone());
1308 <            assertTrue(f2.isDone());
1309 <            assertTrue(f3.isDone());
1310 <            assertFalse(f1.isCancelled());
1311 <            assertTrue(f2.isCancelled());
1296 >            for (long timeout = timeoutMillis();;) {
1297 >                List<Callable<String>> tasks = new ArrayList<>();
1298 >                tasks.add(new StringTask("0"));
1299 >                tasks.add(Executors.callable(new LongPossiblyInterruptedRunnable(), TEST_STRING));
1300 >                tasks.add(new StringTask("2"));
1301 >                long startTime = System.nanoTime();
1302 >                List<Future<String>> futures =
1303 >                    e.invokeAll(tasks, timeout, MILLISECONDS);
1304 >                assertEquals(tasks.size(), futures.size());
1305 >                assertTrue(millisElapsedSince(startTime) >= timeout);
1306 >                for (Future future : futures)
1307 >                    assertTrue(future.isDone());
1308 >                assertTrue(futures.get(1).isCancelled());
1309 >                try {
1310 >                    assertEquals("0", futures.get(0).get());
1311 >                    assertEquals("2", futures.get(2).get());
1312 >                    break;
1313 >                } catch (CancellationException retryWithLongerTimeout) {
1314 >                    timeout *= 2;
1315 >                    if (timeout >= LONG_DELAY_MS / 2)
1316 >                        fail("expected exactly one task to be cancelled");
1317 >                }
1318 >            }
1319          } finally {
1320              joinPool(e);
1321          }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines