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.39 by dl, Fri May 6 11:22:07 2011 UTC vs.
Revision 1.88 by jsr166, Sun Mar 26 02:00:39 2017 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.NANOSECONDS;
11 > import static java.util.concurrent.TimeUnit.SECONDS;
12 >
13 > import java.util.ArrayList;
14 > import java.util.HashSet;
15 > import java.util.List;
16 > import java.util.concurrent.BlockingQueue;
17 > import java.util.concurrent.Callable;
18 > import java.util.concurrent.CancellationException;
19 > import java.util.concurrent.CountDownLatch;
20 > import java.util.concurrent.ExecutionException;
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.ThreadLocalRandom;
28 > import java.util.concurrent.ThreadPoolExecutor;
29 > import java.util.concurrent.atomic.AtomicBoolean;
30 > import java.util.concurrent.atomic.AtomicInteger;
31 > import java.util.concurrent.atomic.AtomicLong;
32 >
33 > import junit.framework.Test;
34 > import junit.framework.TestSuite;
35  
36   public class ScheduledExecutorTest 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(ScheduledExecutorTest.class);
42      }
43  
23
44      /**
45       * execute successfully executes a runnable
46       */
47      public void testExecute() throws InterruptedException {
48 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
49 <        final CountDownLatch done = new CountDownLatch(1);
50 <        final Runnable task = new CheckedRunnable() {
51 <            public void realRun() {
52 <                done.countDown();
33 <            }};
34 <        try {
48 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
49 >        try (PoolCleaner cleaner = cleaner(p)) {
50 >            final CountDownLatch done = new CountDownLatch(1);
51 >            final Runnable task = new CheckedRunnable() {
52 >                public void realRun() { done.countDown(); }};
53              p.execute(task);
54 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
37 <        } finally {
38 <            joinPool(p);
54 >            await(done);
55          }
56      }
57  
42
58      /**
59       * delayed schedule of callable successfully executes after delay
60       */
61      public void testSchedule1() throws Exception {
62 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
63 <        final long t0 = System.nanoTime();
64 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
65 <        final CountDownLatch done = new CountDownLatch(1);
51 <        try {
62 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
63 >        try (PoolCleaner cleaner = cleaner(p)) {
64 >            final long startTime = System.nanoTime();
65 >            final CountDownLatch done = new CountDownLatch(1);
66              Callable task = new CheckedCallable<Boolean>() {
67                  public Boolean realCall() {
68                      done.countDown();
69 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
69 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
70                      return Boolean.TRUE;
71                  }};
72 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
73 <            assertEquals(Boolean.TRUE, f.get());
74 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
72 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
73 >            assertSame(Boolean.TRUE, f.get());
74 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
75              assertTrue(done.await(0L, MILLISECONDS));
62        } finally {
63            joinPool(p);
76          }
77      }
78  
# Line 68 | Line 80 | public class ScheduledExecutorTest exten
80       * delayed schedule of runnable successfully executes after delay
81       */
82      public void testSchedule3() throws Exception {
83 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
84 <        final long t0 = System.nanoTime();
85 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
86 <        final CountDownLatch done = new CountDownLatch(1);
75 <        try {
83 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
84 >        try (PoolCleaner cleaner = cleaner(p)) {
85 >            final long startTime = System.nanoTime();
86 >            final CountDownLatch done = new CountDownLatch(1);
87              Runnable task = new CheckedRunnable() {
88                  public void realRun() {
89                      done.countDown();
90 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
90 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
91                  }};
92 <            Future f = p.schedule(task, SHORT_DELAY_MS, MILLISECONDS);
93 <            assertNull(f.get());
94 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
95 <            assertTrue(done.await(0L, MILLISECONDS));
85 <        } finally {
86 <            joinPool(p);
92 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
93 >            await(done);
94 >            assertNull(f.get(LONG_DELAY_MS, MILLISECONDS));
95 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
96          }
97      }
98  
# Line 91 | Line 100 | public class ScheduledExecutorTest exten
100       * scheduleAtFixedRate executes runnable after given initial delay
101       */
102      public void testSchedule4() throws Exception {
103 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
104 <        final long t0 = System.nanoTime();
105 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
106 <        final CountDownLatch done = new CountDownLatch(1);
98 <        try {
103 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
104 >        try (PoolCleaner cleaner = cleaner(p)) {
105 >            final long startTime = System.nanoTime();
106 >            final CountDownLatch done = new CountDownLatch(1);
107              Runnable task = new CheckedRunnable() {
108                  public void realRun() {
109                      done.countDown();
110 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
110 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
111                  }};
112              ScheduledFuture f =
113 <                p.scheduleAtFixedRate(task, SHORT_DELAY_MS,
114 <                                      SHORT_DELAY_MS, MILLISECONDS);
115 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
116 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
113 >                p.scheduleAtFixedRate(task, timeoutMillis(),
114 >                                      LONG_DELAY_MS, MILLISECONDS);
115 >            await(done);
116 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
117              f.cancel(true);
110        } finally {
111            joinPool(p);
118          }
119      }
120  
121      /**
122       * scheduleWithFixedDelay executes runnable after given initial delay
123       */
124 <    public void testSchedule5() throws InterruptedException {
125 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
126 <        final long t0 = System.nanoTime();
127 <        final long timeoutNanos = SHORT_DELAY_MS * 1000L * 1000L;
128 <        final CountDownLatch done = new CountDownLatch(1);
123 <        try {
124 >    public void testSchedule5() throws Exception {
125 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
126 >        try (PoolCleaner cleaner = cleaner(p)) {
127 >            final long startTime = System.nanoTime();
128 >            final CountDownLatch done = new CountDownLatch(1);
129              Runnable task = new CheckedRunnable() {
130                  public void realRun() {
131                      done.countDown();
132 <                    assertTrue(System.nanoTime() - t0 >= timeoutNanos);
132 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
133                  }};
134              ScheduledFuture f =
135 <                p.scheduleWithFixedDelay(task, SHORT_DELAY_MS,
136 <                                         SHORT_DELAY_MS, MILLISECONDS);
137 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
138 <            assertTrue(System.nanoTime() - t0 >= timeoutNanos);
135 >                p.scheduleWithFixedDelay(task, timeoutMillis(),
136 >                                         LONG_DELAY_MS, MILLISECONDS);
137 >            await(done);
138 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
139              f.cancel(true);
135        } finally {
136            joinPool(p);
140          }
141      }
142  
# Line 143 | Line 146 | public class ScheduledExecutorTest exten
146      }
147  
148      /**
149 <     * scheduleAtFixedRate executes series of tasks at given rate
149 >     * scheduleAtFixedRate executes series of tasks at given rate.
150 >     * Eventually, it must hold that:
151 >     *   cycles - 1 <= elapsedMillis/delay < cycles
152       */
153      public void testFixedRateSequence() throws InterruptedException {
154 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
155 <        RunnableCounter counter = new RunnableCounter();
156 <        ScheduledFuture h =
157 <            p.scheduleAtFixedRate(counter, 0, 1, MILLISECONDS);
158 <        delay(SMALL_DELAY_MS);
159 <        h.cancel(true);
160 <        int c = counter.count.get();
161 <        // By time scaling conventions, we must have at least
162 <        // an execution per SHORT delay, but no more than one SHORT more
163 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
164 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
165 <        joinPool(p);
154 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
155 >        try (PoolCleaner cleaner = cleaner(p)) {
156 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
157 >                final long startTime = System.nanoTime();
158 >                final int cycles = 8;
159 >                final CountDownLatch done = new CountDownLatch(cycles);
160 >                final Runnable task = new CheckedRunnable() {
161 >                    public void realRun() { done.countDown(); }};
162 >                final ScheduledFuture periodicTask =
163 >                    p.scheduleAtFixedRate(task, 0, delay, MILLISECONDS);
164 >                final int totalDelayMillis = (cycles - 1) * delay;
165 >                await(done, totalDelayMillis + LONG_DELAY_MS);
166 >                periodicTask.cancel(true);
167 >                final long elapsedMillis = millisElapsedSince(startTime);
168 >                assertTrue(elapsedMillis >= totalDelayMillis);
169 >                if (elapsedMillis <= cycles * delay)
170 >                    return;
171 >                // else retry with longer delay
172 >            }
173 >            fail("unexpected execution rate");
174 >        }
175      }
176  
177      /**
178 <     * scheduleWithFixedDelay executes series of tasks with given period
178 >     * scheduleWithFixedDelay executes series of tasks with given period.
179 >     * Eventually, it must hold that each task starts at least delay and at
180 >     * most 2 * delay after the termination of the previous task.
181       */
182      public void testFixedDelaySequence() throws InterruptedException {
183 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
184 <        RunnableCounter counter = new RunnableCounter();
185 <        ScheduledFuture h =
186 <            p.scheduleWithFixedDelay(counter, 0, 1, MILLISECONDS);
187 <        delay(SMALL_DELAY_MS);
188 <        h.cancel(true);
189 <        int c = counter.count.get();
190 <        assertTrue(c >= SMALL_DELAY_MS / SHORT_DELAY_MS);
191 <        assertTrue(c <= SMALL_DELAY_MS + SHORT_DELAY_MS);
192 <        joinPool(p);
183 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
184 >        try (PoolCleaner cleaner = cleaner(p)) {
185 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
186 >                final long startTime = System.nanoTime();
187 >                final AtomicLong previous = new AtomicLong(startTime);
188 >                final AtomicBoolean tryLongerDelay = new AtomicBoolean(false);
189 >                final int cycles = 8;
190 >                final CountDownLatch done = new CountDownLatch(cycles);
191 >                final int d = delay;
192 >                final Runnable task = new CheckedRunnable() {
193 >                    public void realRun() {
194 >                        long now = System.nanoTime();
195 >                        long elapsedMillis
196 >                            = NANOSECONDS.toMillis(now - previous.get());
197 >                        if (done.getCount() == cycles) { // first execution
198 >                            if (elapsedMillis >= d)
199 >                                tryLongerDelay.set(true);
200 >                        } else {
201 >                            assertTrue(elapsedMillis >= d);
202 >                            if (elapsedMillis >= 2 * d)
203 >                                tryLongerDelay.set(true);
204 >                        }
205 >                        previous.set(now);
206 >                        done.countDown();
207 >                    }};
208 >                final ScheduledFuture periodicTask =
209 >                    p.scheduleWithFixedDelay(task, 0, delay, MILLISECONDS);
210 >                final int totalDelayMillis = (cycles - 1) * delay;
211 >                await(done, totalDelayMillis + cycles * LONG_DELAY_MS);
212 >                periodicTask.cancel(true);
213 >                final long elapsedMillis = millisElapsedSince(startTime);
214 >                assertTrue(elapsedMillis >= totalDelayMillis);
215 >                if (!tryLongerDelay.get())
216 >                    return;
217 >                // else retry with longer delay
218 >            }
219 >            fail("unexpected execution rate");
220 >        }
221      }
222  
179
223      /**
224       * execute(null) throws NPE
225       */
226      public void testExecuteNull() throws InterruptedException {
227 <        ScheduledThreadPoolExecutor se = null;
228 <        try {
229 <            se = new ScheduledThreadPoolExecutor(1);
230 <            se.execute(null);
231 <            shouldThrow();
232 <        } catch (NullPointerException success) {}
233 <
191 <        joinPool(se);
227 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
228 >        try (PoolCleaner cleaner = cleaner(p)) {
229 >            try {
230 >                p.execute(null);
231 >                shouldThrow();
232 >            } catch (NullPointerException success) {}
233 >        }
234      }
235  
236      /**
237       * schedule(null) throws NPE
238       */
239      public void testScheduleNull() throws InterruptedException {
240 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
241 <        try {
242 <            TrackedCallable callable = null;
243 <            Future f = se.schedule(callable, SHORT_DELAY_MS, MILLISECONDS);
244 <            shouldThrow();
245 <        } catch (NullPointerException success) {}
246 <        joinPool(se);
240 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
241 >        try (PoolCleaner cleaner = cleaner(p)) {
242 >            try {
243 >                TrackedCallable callable = null;
244 >                Future f = p.schedule(callable, SHORT_DELAY_MS, MILLISECONDS);
245 >                shouldThrow();
246 >            } catch (NullPointerException success) {}
247 >        }
248      }
249  
250      /**
251       * execute throws RejectedExecutionException if shutdown
252       */
253      public void testSchedule1_RejectedExecutionException() throws InterruptedException {
254 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
255 <        try {
256 <            se.shutdown();
257 <            se.schedule(new NoOpRunnable(),
258 <                        MEDIUM_DELAY_MS, MILLISECONDS);
259 <            shouldThrow();
260 <        } catch (RejectedExecutionException success) {
261 <        } catch (SecurityException ok) {
254 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
255 >        try (PoolCleaner cleaner = cleaner(p)) {
256 >            try {
257 >                p.shutdown();
258 >                p.schedule(new NoOpRunnable(),
259 >                           MEDIUM_DELAY_MS, MILLISECONDS);
260 >                shouldThrow();
261 >            } catch (RejectedExecutionException success) {
262 >            } catch (SecurityException ok) {}
263          }
220
221        joinPool(se);
264      }
265  
266      /**
267       * schedule throws RejectedExecutionException if shutdown
268       */
269      public void testSchedule2_RejectedExecutionException() throws InterruptedException {
270 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
271 <        try {
272 <            se.shutdown();
273 <            se.schedule(new NoOpCallable(),
274 <                        MEDIUM_DELAY_MS, MILLISECONDS);
275 <            shouldThrow();
276 <        } catch (RejectedExecutionException success) {
277 <        } catch (SecurityException ok) {
270 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
271 >        try (PoolCleaner cleaner = cleaner(p)) {
272 >            try {
273 >                p.shutdown();
274 >                p.schedule(new NoOpCallable(),
275 >                           MEDIUM_DELAY_MS, MILLISECONDS);
276 >                shouldThrow();
277 >            } catch (RejectedExecutionException success) {
278 >            } catch (SecurityException ok) {}
279          }
237        joinPool(se);
280      }
281  
282      /**
283       * schedule callable throws RejectedExecutionException if shutdown
284       */
285      public void testSchedule3_RejectedExecutionException() throws InterruptedException {
286 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
287 <        try {
288 <            se.shutdown();
289 <            se.schedule(new NoOpCallable(),
290 <                        MEDIUM_DELAY_MS, MILLISECONDS);
291 <            shouldThrow();
292 <        } catch (RejectedExecutionException success) {
293 <        } catch (SecurityException ok) {
286 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
287 >        try (PoolCleaner cleaner = cleaner(p)) {
288 >            try {
289 >                p.shutdown();
290 >                p.schedule(new NoOpCallable(),
291 >                           MEDIUM_DELAY_MS, MILLISECONDS);
292 >                shouldThrow();
293 >            } catch (RejectedExecutionException success) {
294 >            } catch (SecurityException ok) {}
295          }
253        joinPool(se);
296      }
297  
298      /**
299       * scheduleAtFixedRate throws RejectedExecutionException if shutdown
300       */
301      public void testScheduleAtFixedRate1_RejectedExecutionException() throws InterruptedException {
302 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
303 <        try {
304 <            se.shutdown();
305 <            se.scheduleAtFixedRate(new NoOpRunnable(),
306 <                                   MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
307 <            shouldThrow();
308 <        } catch (RejectedExecutionException success) {
309 <        } catch (SecurityException ok) {
302 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
303 >        try (PoolCleaner cleaner = cleaner(p)) {
304 >            try {
305 >                p.shutdown();
306 >                p.scheduleAtFixedRate(new NoOpRunnable(),
307 >                                      MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
308 >                shouldThrow();
309 >            } catch (RejectedExecutionException success) {
310 >            } catch (SecurityException ok) {}
311          }
269        joinPool(se);
312      }
313  
314      /**
315       * scheduleWithFixedDelay throws RejectedExecutionException if shutdown
316       */
317      public void testScheduleWithFixedDelay1_RejectedExecutionException() throws InterruptedException {
318 <        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
319 <        try {
320 <            se.shutdown();
321 <            se.scheduleWithFixedDelay(new NoOpRunnable(),
322 <                                      MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
323 <            shouldThrow();
324 <        } catch (RejectedExecutionException success) {
325 <        } catch (SecurityException ok) {
318 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
319 >        try (PoolCleaner cleaner = cleaner(p)) {
320 >            try {
321 >                p.shutdown();
322 >                p.scheduleWithFixedDelay(new NoOpRunnable(),
323 >                                         MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
324 >                shouldThrow();
325 >            } catch (RejectedExecutionException success) {
326 >            } catch (SecurityException ok) {}
327          }
285        joinPool(se);
328      }
329  
330      /**
# Line 290 | Line 332 | public class ScheduledExecutorTest exten
332       * thread becomes active
333       */
334      public void testGetActiveCount() throws InterruptedException {
293        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
294        final CountDownLatch threadStarted = new CountDownLatch(1);
335          final CountDownLatch done = new CountDownLatch(1);
336 <        try {
336 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
337 >        try (PoolCleaner cleaner = cleaner(p, done)) {
338 >            final CountDownLatch threadStarted = new CountDownLatch(1);
339              assertEquals(0, p.getActiveCount());
340              p.execute(new CheckedRunnable() {
341                  public void realRun() throws InterruptedException {
342                      threadStarted.countDown();
343                      assertEquals(1, p.getActiveCount());
344 <                    done.await();
344 >                    await(done);
345                  }});
346 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
346 >            await(threadStarted);
347              assertEquals(1, p.getActiveCount());
306        } finally {
307            done.countDown();
308            joinPool(p);
348          }
349      }
350  
# Line 315 | Line 354 | public class ScheduledExecutorTest exten
354       */
355      public void testGetCompletedTaskCount() throws InterruptedException {
356          final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
357 <        final CountDownLatch threadStarted = new CountDownLatch(1);
358 <        final CountDownLatch threadProceed = new CountDownLatch(1);
359 <        final CountDownLatch threadDone = new CountDownLatch(1);
360 <        try {
357 >        try (PoolCleaner cleaner = cleaner(p)) {
358 >            final CountDownLatch threadStarted = new CountDownLatch(1);
359 >            final CountDownLatch threadProceed = new CountDownLatch(1);
360 >            final CountDownLatch threadDone = new CountDownLatch(1);
361              assertEquals(0, p.getCompletedTaskCount());
362              p.execute(new CheckedRunnable() {
363                  public void realRun() throws InterruptedException {
364                      threadStarted.countDown();
365                      assertEquals(0, p.getCompletedTaskCount());
366 <                    threadProceed.await();
366 >                    await(threadProceed);
367                      threadDone.countDown();
368                  }});
369 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
369 >            await(threadStarted);
370              assertEquals(0, p.getCompletedTaskCount());
371              threadProceed.countDown();
372 <            threadDone.await();
373 <            delay(SHORT_DELAY_MS);
374 <            assertEquals(1, p.getCompletedTaskCount());
375 <        } finally {
376 <            joinPool(p);
372 >            await(threadDone);
373 >            long startTime = System.nanoTime();
374 >            while (p.getCompletedTaskCount() != 1) {
375 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
376 >                    fail("timed out");
377 >                Thread.yield();
378 >            }
379          }
380      }
381  
# Line 343 | Line 384 | public class ScheduledExecutorTest exten
384       */
385      public void testGetCorePoolSize() throws InterruptedException {
386          ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
387 <        assertEquals(1, p.getCorePoolSize());
388 <        joinPool(p);
387 >        try (PoolCleaner cleaner = cleaner(p)) {
388 >            assertEquals(1, p.getCorePoolSize());
389 >        }
390      }
391  
392      /**
# Line 356 | Line 398 | public class ScheduledExecutorTest exten
398          final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(THREADS);
399          final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
400          final CountDownLatch done = new CountDownLatch(1);
401 <        try {
401 >        try (PoolCleaner cleaner = cleaner(p, done)) {
402              assertEquals(0, p.getLargestPoolSize());
403              for (int i = 0; i < THREADS; i++)
404                  p.execute(new CheckedRunnable() {
405                      public void realRun() throws InterruptedException {
406                          threadsStarted.countDown();
407 <                        done.await();
407 >                        await(done);
408                          assertEquals(THREADS, p.getLargestPoolSize());
409                      }});
410 <            assertTrue(threadsStarted.await(SMALL_DELAY_MS, MILLISECONDS));
369 <            assertEquals(THREADS, p.getLargestPoolSize());
370 <        } finally {
371 <            done.countDown();
372 <            joinPool(p);
410 >            await(threadsStarted);
411              assertEquals(THREADS, p.getLargestPoolSize());
412          }
413 +        assertEquals(THREADS, p.getLargestPoolSize());
414      }
415  
416      /**
# Line 382 | Line 421 | public class ScheduledExecutorTest exten
421          final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
422          final CountDownLatch threadStarted = new CountDownLatch(1);
423          final CountDownLatch done = new CountDownLatch(1);
424 <        try {
424 >        try (PoolCleaner cleaner = cleaner(p, done)) {
425              assertEquals(0, p.getPoolSize());
426              p.execute(new CheckedRunnable() {
427                  public void realRun() throws InterruptedException {
428                      threadStarted.countDown();
429                      assertEquals(1, p.getPoolSize());
430 <                    done.await();
430 >                    await(done);
431                  }});
432 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
432 >            await(threadStarted);
433              assertEquals(1, p.getPoolSize());
395        } finally {
396            done.countDown();
397            joinPool(p);
434          }
435      }
436  
# Line 403 | Line 439 | public class ScheduledExecutorTest exten
439       * submitted
440       */
441      public void testGetTaskCount() throws InterruptedException {
442 <        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
407 <        final CountDownLatch threadStarted = new CountDownLatch(1);
442 >        final int TASKS = 3;
443          final CountDownLatch done = new CountDownLatch(1);
444 <        final int TASKS = 5;
445 <        try {
444 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
445 >        try (PoolCleaner cleaner = cleaner(p, done)) {
446 >            final CountDownLatch threadStarted = new CountDownLatch(1);
447              assertEquals(0, p.getTaskCount());
448 <            for (int i = 0; i < TASKS; i++)
448 >            assertEquals(0, p.getCompletedTaskCount());
449 >            p.execute(new CheckedRunnable() {
450 >                public void realRun() throws InterruptedException {
451 >                    threadStarted.countDown();
452 >                    await(done);
453 >                }});
454 >            await(threadStarted);
455 >            assertEquals(1, p.getTaskCount());
456 >            assertEquals(0, p.getCompletedTaskCount());
457 >            for (int i = 0; i < TASKS; i++) {
458 >                assertEquals(1 + i, p.getTaskCount());
459                  p.execute(new CheckedRunnable() {
460                      public void realRun() throws InterruptedException {
461                          threadStarted.countDown();
462 <                        done.await();
462 >                        assertEquals(1 + TASKS, p.getTaskCount());
463 >                        await(done);
464                      }});
465 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
466 <            assertEquals(TASKS, p.getTaskCount());
467 <        } finally {
421 <            done.countDown();
422 <            joinPool(p);
465 >            }
466 >            assertEquals(1 + TASKS, p.getTaskCount());
467 >            assertEquals(0, p.getCompletedTaskCount());
468          }
469 +        assertEquals(1 + TASKS, p.getTaskCount());
470 +        assertEquals(1 + TASKS, p.getCompletedTaskCount());
471      }
472  
473      /**
474       * getThreadFactory returns factory in constructor if not set
475       */
476      public void testGetThreadFactory() throws InterruptedException {
477 <        ThreadFactory tf = new SimpleThreadFactory();
478 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1, tf);
479 <        assertSame(tf, p.getThreadFactory());
480 <        joinPool(p);
477 >        final ThreadFactory threadFactory = new SimpleThreadFactory();
478 >        final ScheduledThreadPoolExecutor p =
479 >            new ScheduledThreadPoolExecutor(1, threadFactory);
480 >        try (PoolCleaner cleaner = cleaner(p)) {
481 >            assertSame(threadFactory, p.getThreadFactory());
482 >        }
483      }
484  
485      /**
486       * setThreadFactory sets the thread factory returned by getThreadFactory
487       */
488      public void testSetThreadFactory() throws InterruptedException {
489 <        ThreadFactory tf = new SimpleThreadFactory();
490 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
491 <        p.setThreadFactory(tf);
492 <        assertSame(tf, p.getThreadFactory());
493 <        joinPool(p);
489 >        ThreadFactory threadFactory = new SimpleThreadFactory();
490 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
491 >        try (PoolCleaner cleaner = cleaner(p)) {
492 >            p.setThreadFactory(threadFactory);
493 >            assertSame(threadFactory, p.getThreadFactory());
494 >        }
495      }
496  
497      /**
498       * setThreadFactory(null) throws NPE
499       */
500      public void testSetThreadFactoryNull() throws InterruptedException {
501 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
502 <        try {
503 <            p.setThreadFactory(null);
504 <            shouldThrow();
505 <        } catch (NullPointerException success) {
506 <        } finally {
457 <            joinPool(p);
501 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
502 >        try (PoolCleaner cleaner = cleaner(p)) {
503 >            try {
504 >                p.setThreadFactory(null);
505 >                shouldThrow();
506 >            } catch (NullPointerException success) {}
507          }
508      }
509  
510      /**
511 <     * isShutDown is false before shutdown, true after
511 >     * The default rejected execution handler is AbortPolicy.
512       */
513 <    public void testIsShutdown() {
514 <
515 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
516 <        try {
517 <            assertFalse(p.isShutdown());
469 <        }
470 <        finally {
471 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
513 >    public void testDefaultRejectedExecutionHandler() {
514 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
515 >        try (PoolCleaner cleaner = cleaner(p)) {
516 >            assertTrue(p.getRejectedExecutionHandler()
517 >                       instanceof ThreadPoolExecutor.AbortPolicy);
518          }
473        assertTrue(p.isShutdown());
519      }
520  
521 +    /**
522 +     * isShutdown is false before shutdown, true after
523 +     */
524 +    public void testIsShutdown() {
525 +        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
526 +        assertFalse(p.isShutdown());
527 +        try (PoolCleaner cleaner = cleaner(p)) {
528 +            try {
529 +                p.shutdown();
530 +                assertTrue(p.isShutdown());
531 +            } catch (SecurityException ok) {}
532 +        }
533 +    }
534  
535      /**
536       * isTerminated is false before termination, true after
537       */
538      public void testIsTerminated() throws InterruptedException {
539          final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
540 <        final CountDownLatch threadStarted = new CountDownLatch(1);
541 <        final CountDownLatch done = new CountDownLatch(1);
542 <        assertFalse(p.isTerminated());
543 <        try {
540 >        try (PoolCleaner cleaner = cleaner(p)) {
541 >            final CountDownLatch threadStarted = new CountDownLatch(1);
542 >            final CountDownLatch done = new CountDownLatch(1);
543 >            assertFalse(p.isTerminated());
544              p.execute(new CheckedRunnable() {
545                  public void realRun() throws InterruptedException {
546                      assertFalse(p.isTerminated());
547                      threadStarted.countDown();
548 <                    done.await();
548 >                    await(done);
549                  }});
550 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
550 >            await(threadStarted);
551              assertFalse(p.isTerminating());
552              done.countDown();
495        } finally {
553              try { p.shutdown(); } catch (SecurityException ok) { return; }
554 +            assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
555 +            assertTrue(p.isTerminated());
556          }
498        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
499        assertTrue(p.isTerminated());
557      }
558  
559      /**
# Line 506 | Line 563 | public class ScheduledExecutorTest exten
563          final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
564          final CountDownLatch threadStarted = new CountDownLatch(1);
565          final CountDownLatch done = new CountDownLatch(1);
566 <        try {
566 >        try (PoolCleaner cleaner = cleaner(p)) {
567              assertFalse(p.isTerminating());
568              p.execute(new CheckedRunnable() {
569                  public void realRun() throws InterruptedException {
570                      assertFalse(p.isTerminating());
571                      threadStarted.countDown();
572 <                    done.await();
572 >                    await(done);
573                  }});
574 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
574 >            await(threadStarted);
575              assertFalse(p.isTerminating());
576              done.countDown();
520        } finally {
577              try { p.shutdown(); } catch (SecurityException ok) { return; }
578 +            assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
579 +            assertTrue(p.isTerminated());
580 +            assertFalse(p.isTerminating());
581          }
523        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
524        assertTrue(p.isTerminated());
525        assertFalse(p.isTerminating());
582      }
583  
584      /**
585       * getQueue returns the work queue, which contains queued tasks
586       */
587      public void testGetQueue() throws InterruptedException {
532        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
533        final CountDownLatch threadStarted = new CountDownLatch(1);
588          final CountDownLatch done = new CountDownLatch(1);
589 <        try {
589 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
590 >        try (PoolCleaner cleaner = cleaner(p, done)) {
591 >            final CountDownLatch threadStarted = new CountDownLatch(1);
592              ScheduledFuture[] tasks = new ScheduledFuture[5];
593              for (int i = 0; i < tasks.length; i++) {
594                  Runnable r = new CheckedRunnable() {
595                      public void realRun() throws InterruptedException {
596                          threadStarted.countDown();
597 <                        done.await();
597 >                        await(done);
598                      }};
599                  tasks[i] = p.schedule(r, 1, MILLISECONDS);
600              }
601 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
601 >            await(threadStarted);
602              BlockingQueue<Runnable> q = p.getQueue();
603              assertTrue(q.contains(tasks[tasks.length - 1]));
604              assertFalse(q.contains(tasks[0]));
549        } finally {
550            done.countDown();
551            joinPool(p);
605          }
606      }
607  
# Line 556 | Line 609 | public class ScheduledExecutorTest exten
609       * remove(task) removes queued task, and fails to remove active task
610       */
611      public void testRemove() throws InterruptedException {
559        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
560        ScheduledFuture[] tasks = new ScheduledFuture[5];
561        final CountDownLatch threadStarted = new CountDownLatch(1);
612          final CountDownLatch done = new CountDownLatch(1);
613 <        try {
613 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
614 >        try (PoolCleaner cleaner = cleaner(p, done)) {
615 >            ScheduledFuture[] tasks = new ScheduledFuture[5];
616 >            final CountDownLatch threadStarted = new CountDownLatch(1);
617              for (int i = 0; i < tasks.length; i++) {
618                  Runnable r = new CheckedRunnable() {
619                      public void realRun() throws InterruptedException {
620                          threadStarted.countDown();
621 <                        done.await();
621 >                        await(done);
622                      }};
623                  tasks[i] = p.schedule(r, 1, MILLISECONDS);
624              }
625 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
625 >            await(threadStarted);
626              BlockingQueue<Runnable> q = p.getQueue();
627              assertFalse(p.remove((Runnable)tasks[0]));
628              assertTrue(q.contains((Runnable)tasks[4]));
# Line 580 | Line 633 | public class ScheduledExecutorTest exten
633              assertTrue(q.contains((Runnable)tasks[3]));
634              assertTrue(p.remove((Runnable)tasks[3]));
635              assertFalse(q.contains((Runnable)tasks[3]));
583        } finally {
584            done.countDown();
585            joinPool(p);
636          }
637      }
638  
639      /**
640 <     * purge removes cancelled tasks from the queue
640 >     * purge eventually removes cancelled tasks from the queue
641       */
642      public void testPurge() throws InterruptedException {
643 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
644 <        ScheduledFuture[] tasks = new ScheduledFuture[5];
645 <        for (int i = 0; i < tasks.length; i++) {
646 <            tasks[i] = p.schedule(new SmallPossiblyInterruptedRunnable(), SHORT_DELAY_MS, MILLISECONDS);
647 <        }
648 <        try {
643 >        final ScheduledFuture[] tasks = new ScheduledFuture[5];
644 >        final Runnable releaser = new Runnable() { public void run() {
645 >            for (ScheduledFuture task : tasks)
646 >                if (task != null) task.cancel(true); }};
647 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
648 >        try (PoolCleaner cleaner = cleaner(p, releaser)) {
649 >            for (int i = 0; i < tasks.length; i++)
650 >                tasks[i] = p.schedule(new SmallPossiblyInterruptedRunnable(),
651 >                                      LONG_DELAY_MS, MILLISECONDS);
652              int max = tasks.length;
653              if (tasks[4].cancel(true)) --max;
654              if (tasks[3].cancel(true)) --max;
655              // There must eventually be an interference-free point at
656              // which purge will not fail. (At worst, when queue is empty.)
657 <            int k;
658 <            for (k = 0; k < SMALL_DELAY_MS; ++k) {
657 >            long startTime = System.nanoTime();
658 >            do {
659                  p.purge();
660                  long count = p.getTaskCount();
661 <                if (count >= 0 && count <= max)
662 <                    break;
663 <                delay(1);
664 <            }
612 <            assertTrue(k < SMALL_DELAY_MS);
613 <        } finally {
614 <            for (ScheduledFuture task : tasks)
615 <                task.cancel(true);
616 <            joinPool(p);
661 >                if (count == max)
662 >                    return;
663 >            } while (millisElapsedSince(startTime) < LONG_DELAY_MS);
664 >            fail("Purge failed to remove cancelled tasks");
665          }
666      }
667  
668      /**
669 <     * shutDownNow returns a list containing tasks that were not run
670 <     */
671 <    public void testShutDownNow() throws InterruptedException {
672 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
673 <        for (int i = 0; i < 5; i++)
674 <            p.schedule(new SmallPossiblyInterruptedRunnable(), SHORT_DELAY_MS, MILLISECONDS);
675 <        List l;
669 >     * shutdownNow returns a list containing tasks that were not run,
670 >     * and those tasks are drained from the queue
671 >     */
672 >    public void testShutdownNow() throws InterruptedException {
673 >        final int poolSize = 2;
674 >        final int count = 5;
675 >        final AtomicInteger ran = new AtomicInteger(0);
676 >        final ScheduledThreadPoolExecutor p =
677 >            new ScheduledThreadPoolExecutor(poolSize);
678 >        final CountDownLatch threadsStarted = new CountDownLatch(poolSize);
679 >        Runnable waiter = new CheckedRunnable() { public void realRun() {
680 >            threadsStarted.countDown();
681 >            try {
682 >                MILLISECONDS.sleep(2 * LONG_DELAY_MS);
683 >            } catch (InterruptedException success) {}
684 >            ran.getAndIncrement();
685 >        }};
686 >        for (int i = 0; i < count; i++)
687 >            p.execute(waiter);
688 >        await(threadsStarted);
689 >        assertEquals(poolSize, p.getActiveCount());
690 >        assertEquals(0, p.getCompletedTaskCount());
691 >        final List<Runnable> queuedTasks;
692          try {
693 <            l = p.shutdownNow();
693 >            queuedTasks = p.shutdownNow();
694          } catch (SecurityException ok) {
695 <            return;
695 >            return; // Allowed in case test doesn't have privs
696          }
697          assertTrue(p.isShutdown());
698 <        assertTrue(l.size() > 0 && l.size() <= 5);
699 <        joinPool(p);
698 >        assertTrue(p.getQueue().isEmpty());
699 >        assertEquals(count - poolSize, queuedTasks.size());
700 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
701 >        assertTrue(p.isTerminated());
702 >        assertEquals(poolSize, ran.get());
703 >        assertEquals(poolSize, p.getCompletedTaskCount());
704      }
705  
706      /**
707 <     * In default setting, shutdown cancels periodic but not delayed
708 <     * tasks at shutdown
707 >     * shutdownNow returns a list containing tasks that were not run,
708 >     * and those tasks are drained from the queue
709       */
710 <    public void testShutDown1() throws InterruptedException {
711 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
712 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
713 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
714 <
715 <        ScheduledFuture[] tasks = new ScheduledFuture[5];
716 <        for (int i = 0; i < tasks.length; i++)
717 <            tasks[i] = p.schedule(new NoOpRunnable(),
718 <                                  SHORT_DELAY_MS, MILLISECONDS);
719 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
720 <        BlockingQueue<Runnable> q = p.getQueue();
721 <        for (ScheduledFuture task : tasks) {
722 <            assertFalse(task.isDone());
723 <            assertFalse(task.isCancelled());
724 <            assertTrue(q.contains(task));
710 >    public void testShutdownNow_delayedTasks() throws InterruptedException {
711 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
712 >        List<ScheduledFuture> tasks = new ArrayList<>();
713 >        for (int i = 0; i < 3; i++) {
714 >            Runnable r = new NoOpRunnable();
715 >            tasks.add(p.schedule(r, 9, SECONDS));
716 >            tasks.add(p.scheduleAtFixedRate(r, 9, 9, SECONDS));
717 >            tasks.add(p.scheduleWithFixedDelay(r, 9, 9, SECONDS));
718 >        }
719 >        if (testImplementationDetails)
720 >            assertEquals(new HashSet(tasks), new HashSet(p.getQueue()));
721 >        final List<Runnable> queuedTasks;
722 >        try {
723 >            queuedTasks = p.shutdownNow();
724 >        } catch (SecurityException ok) {
725 >            return; // Allowed in case test doesn't have privs
726          }
727          assertTrue(p.isShutdown());
728 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
729 <        assertTrue(p.isTerminated());
728 >        assertTrue(p.getQueue().isEmpty());
729 >        if (testImplementationDetails)
730 >            assertEquals(new HashSet(tasks), new HashSet(queuedTasks));
731 >        assertEquals(tasks.size(), queuedTasks.size());
732          for (ScheduledFuture task : tasks) {
733 <            assertTrue(task.isDone());
733 >            assertFalse(task.isDone());
734              assertFalse(task.isCancelled());
735          }
736 <    }
666 <
667 <
668 <    /**
669 <     * If setExecuteExistingDelayedTasksAfterShutdownPolicy is false,
670 <     * delayed tasks are cancelled at shutdown
671 <     */
672 <    public void testShutDown2() throws InterruptedException {
673 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
674 <        p.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
675 <        assertFalse(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
676 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
677 <        ScheduledFuture[] tasks = new ScheduledFuture[5];
678 <        for (int i = 0; i < tasks.length; i++)
679 <            tasks[i] = p.schedule(new NoOpRunnable(),
680 <                                  SHORT_DELAY_MS, MILLISECONDS);
681 <        BlockingQueue q = p.getQueue();
682 <        assertEquals(tasks.length, q.size());
683 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
684 <        assertTrue(p.isShutdown());
685 <        assertTrue(q.isEmpty());
686 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
736 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
737          assertTrue(p.isTerminated());
688        for (ScheduledFuture task : tasks) {
689            assertTrue(task.isDone());
690            assertTrue(task.isCancelled());
691        }
738      }
739  
740      /**
741 <     * If setContinueExistingPeriodicTasksAfterShutdownPolicy is set false,
742 <     * periodic tasks are cancelled at shutdown
743 <     */
744 <    public void testShutDown3() throws InterruptedException {
745 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
746 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
747 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
748 <        p.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
749 <        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
750 <        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
751 <        ScheduledFuture task =
752 <            p.scheduleAtFixedRate(new NoOpRunnable(), 5, 5, MILLISECONDS);
741 >     * By default, periodic tasks are cancelled at shutdown.
742 >     * By default, delayed tasks keep running after shutdown.
743 >     * Check that changing the default values work:
744 >     * - setExecuteExistingDelayedTasksAfterShutdownPolicy
745 >     * - setContinueExistingPeriodicTasksAfterShutdownPolicy
746 >     */
747 >    public void testShutdown_cancellation() throws Exception {
748 >        final int poolSize = 2;
749 >        final ScheduledThreadPoolExecutor p
750 >            = new ScheduledThreadPoolExecutor(poolSize);
751 >        final ThreadLocalRandom rnd = ThreadLocalRandom.current();
752 >        final boolean effectiveDelayedPolicy;
753 >        final boolean effectivePeriodicPolicy;
754 >        final boolean effectiveRemovePolicy;
755 >
756 >        if (rnd.nextBoolean())
757 >            p.setExecuteExistingDelayedTasksAfterShutdownPolicy(
758 >                effectiveDelayedPolicy = rnd.nextBoolean());
759 >        else
760 >            effectiveDelayedPolicy = true;
761 >        assertEquals(effectiveDelayedPolicy,
762 >                     p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
763 >
764 >        if (rnd.nextBoolean())
765 >            p.setContinueExistingPeriodicTasksAfterShutdownPolicy(
766 >                effectivePeriodicPolicy = rnd.nextBoolean());
767 >        else
768 >            effectivePeriodicPolicy = false;
769 >        assertEquals(effectivePeriodicPolicy,
770 >                     p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
771 >
772 >        if (rnd.nextBoolean())
773 >            p.setRemoveOnCancelPolicy(
774 >                effectiveRemovePolicy = rnd.nextBoolean());
775 >        else
776 >            effectiveRemovePolicy = false;
777 >        assertEquals(effectiveRemovePolicy,
778 >                     p.getRemoveOnCancelPolicy());
779 >
780 >        // Strategy: Wedge the pool with poolSize "blocker" threads
781 >        final AtomicInteger ran = new AtomicInteger(0);
782 >        final CountDownLatch poolBlocked = new CountDownLatch(poolSize);
783 >        final CountDownLatch unblock = new CountDownLatch(1);
784 >        final CountDownLatch periodicLatch1 = new CountDownLatch(2);
785 >        final CountDownLatch periodicLatch2 = new CountDownLatch(2);
786 >        Runnable task = new CheckedRunnable() { public void realRun()
787 >                                                    throws InterruptedException {
788 >            poolBlocked.countDown();
789 >            await(unblock);
790 >            ran.getAndIncrement();
791 >        }};
792 >        List<Future<?>> blockers = new ArrayList<>();
793 >        List<Future<?>> periodics = new ArrayList<>();
794 >        List<Future<?>> delayeds = new ArrayList<>();
795 >        for (int i = 0; i < poolSize; i++)
796 >            blockers.add(p.submit(task));
797 >        await(poolBlocked);
798 >
799 >        periodics.add(p.scheduleAtFixedRate(
800 >                          countDowner(periodicLatch1), 1, 1, MILLISECONDS));
801 >        periodics.add(p.scheduleWithFixedDelay(
802 >                          countDowner(periodicLatch2), 1, 1, MILLISECONDS));
803 >        delayeds.add(p.schedule(task, 1, MILLISECONDS));
804 >
805 >        assertTrue(p.getQueue().containsAll(periodics));
806 >        assertTrue(p.getQueue().containsAll(delayeds));
807          try { p.shutdown(); } catch (SecurityException ok) { return; }
808          assertTrue(p.isShutdown());
809 <        BlockingQueue q = p.getQueue();
810 <        assertTrue(p.getQueue().isEmpty());
811 <        assertTrue(task.isDone());
812 <        assertTrue(task.isCancelled());
813 <        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
809 >        assertFalse(p.isTerminated());
810 >        for (Future<?> periodic : periodics) {
811 >            assertTrue(effectivePeriodicPolicy ^ periodic.isCancelled());
812 >            assertTrue(effectivePeriodicPolicy ^ periodic.isDone());
813 >        }
814 >        for (Future<?> delayed : delayeds) {
815 >            assertTrue(effectiveDelayedPolicy ^ delayed.isCancelled());
816 >            assertTrue(effectiveDelayedPolicy ^ delayed.isDone());
817 >        }
818 >        if (testImplementationDetails) {
819 >            assertEquals(effectivePeriodicPolicy,
820 >                         p.getQueue().containsAll(periodics));
821 >            assertEquals(effectiveDelayedPolicy,
822 >                         p.getQueue().containsAll(delayeds));
823 >        }
824 >        unblock.countDown();    // Release all pool threads
825 >
826 >        if (effectiveDelayedPolicy)
827 >            for (Future<?> delayed : delayeds) assertNull(delayed.get());
828 >        if (effectivePeriodicPolicy) {
829 >            await(periodicLatch1);
830 >            await(periodicLatch2);
831 >            for (Future<?> periodic : periodics) {
832 >                assertTrue(periodic.cancel(false));
833 >                assertTrue(periodic.isCancelled());
834 >                assertTrue(periodic.isDone());
835 >            }
836 >        }
837 >        for (Future<?> blocker : blockers) assertNull(blocker.get());
838 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
839          assertTrue(p.isTerminated());
715    }
840  
841 <    /**
842 <     * if setContinueExistingPeriodicTasksAfterShutdownPolicy is true,
843 <     * periodic tasks are not cancelled at shutdown
720 <     */
721 <    public void testShutDown4() throws InterruptedException {
722 <        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
723 <        final CountDownLatch counter = new CountDownLatch(2);
724 <        try {
725 <            p.setContinueExistingPeriodicTasksAfterShutdownPolicy(true);
726 <            assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
727 <            assertTrue(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
728 <            final Runnable r = new CheckedRunnable() {
729 <                public void realRun() {
730 <                    counter.countDown();
731 <                }};
732 <            ScheduledFuture task =
733 <                p.scheduleAtFixedRate(r, 1, 1, MILLISECONDS);
734 <            assertFalse(task.isDone());
735 <            assertFalse(task.isCancelled());
736 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
737 <            assertFalse(task.isCancelled());
738 <            assertFalse(p.isTerminated());
739 <            assertTrue(p.isShutdown());
740 <            assertTrue(counter.await(SMALL_DELAY_MS, MILLISECONDS));
741 <            assertFalse(task.isCancelled());
742 <            assertTrue(task.cancel(false));
743 <            assertTrue(task.isDone());
744 <            assertTrue(task.isCancelled());
745 <            assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
746 <            assertTrue(p.isTerminated());
747 <        }
748 <        finally {
749 <            joinPool(p);
841 >        for (Future<?> future : delayeds) {
842 >            assertTrue(effectiveDelayedPolicy ^ future.isCancelled());
843 >            assertTrue(future.isDone());
844          }
845 +        for (Future<?> future : periodics)
846 +            assertTrue(future.isCancelled());
847 +        for (Future<?> future : blockers)
848 +            assertNull(future.get());
849 +        assertEquals(2 + (effectiveDelayedPolicy ? 1 : 0), ran.get());
850      }
851  
852      /**
853       * completed submit of callable returns result
854       */
855      public void testSubmitCallable() throws Exception {
856 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
857 <        try {
856 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
857 >        try (PoolCleaner cleaner = cleaner(e)) {
858              Future<String> future = e.submit(new StringTask());
859              String result = future.get();
860              assertSame(TEST_STRING, result);
762        } finally {
763            joinPool(e);
861          }
862      }
863  
# Line 768 | Line 865 | public class ScheduledExecutorTest exten
865       * completed submit of runnable returns successfully
866       */
867      public void testSubmitRunnable() throws Exception {
868 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
869 <        try {
868 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
869 >        try (PoolCleaner cleaner = cleaner(e)) {
870              Future<?> future = e.submit(new NoOpRunnable());
871              future.get();
872              assertTrue(future.isDone());
776        } finally {
777            joinPool(e);
873          }
874      }
875  
# Line 782 | Line 877 | public class ScheduledExecutorTest exten
877       * completed submit of (runnable, result) returns result
878       */
879      public void testSubmitRunnable2() throws Exception {
880 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
881 <        try {
880 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
881 >        try (PoolCleaner cleaner = cleaner(e)) {
882              Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
883              String result = future.get();
884              assertSame(TEST_STRING, result);
790        } finally {
791            joinPool(e);
885          }
886      }
887  
# Line 796 | Line 889 | public class ScheduledExecutorTest exten
889       * invokeAny(null) throws NPE
890       */
891      public void testInvokeAny1() throws Exception {
892 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
893 <        try {
894 <            e.invokeAny(null);
895 <            shouldThrow();
896 <        } catch (NullPointerException success) {
897 <        } finally {
805 <            joinPool(e);
892 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
893 >        try (PoolCleaner cleaner = cleaner(e)) {
894 >            try {
895 >                e.invokeAny(null);
896 >                shouldThrow();
897 >            } catch (NullPointerException success) {}
898          }
899      }
900  
# Line 810 | Line 902 | public class ScheduledExecutorTest exten
902       * invokeAny(empty collection) throws IAE
903       */
904      public void testInvokeAny2() throws Exception {
905 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
906 <        try {
907 <            e.invokeAny(new ArrayList<Callable<String>>());
908 <            shouldThrow();
909 <        } catch (IllegalArgumentException success) {
910 <        } finally {
819 <            joinPool(e);
905 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
906 >        try (PoolCleaner cleaner = cleaner(e)) {
907 >            try {
908 >                e.invokeAny(new ArrayList<Callable<String>>());
909 >                shouldThrow();
910 >            } catch (IllegalArgumentException success) {}
911          }
912      }
913  
# Line 825 | Line 916 | public class ScheduledExecutorTest exten
916       */
917      public void testInvokeAny3() throws Exception {
918          CountDownLatch latch = new CountDownLatch(1);
919 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
920 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
921 <        l.add(latchAwaitingStringTask(latch));
922 <        l.add(null);
923 <        try {
924 <            e.invokeAny(l);
925 <            shouldThrow();
926 <        } catch (NullPointerException success) {
927 <        } finally {
919 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
920 >        try (PoolCleaner cleaner = cleaner(e)) {
921 >            List<Callable<String>> l = new ArrayList<>();
922 >            l.add(latchAwaitingStringTask(latch));
923 >            l.add(null);
924 >            try {
925 >                e.invokeAny(l);
926 >                shouldThrow();
927 >            } catch (NullPointerException success) {}
928              latch.countDown();
838            joinPool(e);
929          }
930      }
931  
# Line 843 | Line 933 | public class ScheduledExecutorTest exten
933       * invokeAny(c) throws ExecutionException if no task completes
934       */
935      public void testInvokeAny4() throws Exception {
936 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
937 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
938 <        l.add(new NPETask());
939 <        try {
940 <            e.invokeAny(l);
941 <            shouldThrow();
942 <        } catch (ExecutionException success) {
943 <            assertTrue(success.getCause() instanceof NullPointerException);
944 <        } finally {
945 <            joinPool(e);
936 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
937 >        try (PoolCleaner cleaner = cleaner(e)) {
938 >            List<Callable<String>> l = new ArrayList<>();
939 >            l.add(new NPETask());
940 >            try {
941 >                e.invokeAny(l);
942 >                shouldThrow();
943 >            } catch (ExecutionException success) {
944 >                assertTrue(success.getCause() instanceof NullPointerException);
945 >            }
946          }
947      }
948  
# Line 860 | Line 950 | public class ScheduledExecutorTest exten
950       * invokeAny(c) returns result of some task
951       */
952      public void testInvokeAny5() throws Exception {
953 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
954 <        try {
955 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
953 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
954 >        try (PoolCleaner cleaner = cleaner(e)) {
955 >            List<Callable<String>> l = new ArrayList<>();
956              l.add(new StringTask());
957              l.add(new StringTask());
958              String result = e.invokeAny(l);
959              assertSame(TEST_STRING, result);
870        } finally {
871            joinPool(e);
960          }
961      }
962  
# Line 876 | Line 964 | public class ScheduledExecutorTest exten
964       * invokeAll(null) throws NPE
965       */
966      public void testInvokeAll1() throws Exception {
967 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
968 <        try {
969 <            e.invokeAll(null);
970 <            shouldThrow();
971 <        } catch (NullPointerException success) {
972 <        } finally {
885 <            joinPool(e);
967 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
968 >        try (PoolCleaner cleaner = cleaner(e)) {
969 >            try {
970 >                e.invokeAll(null);
971 >                shouldThrow();
972 >            } catch (NullPointerException success) {}
973          }
974      }
975  
# Line 890 | Line 977 | public class ScheduledExecutorTest exten
977       * invokeAll(empty collection) returns empty collection
978       */
979      public void testInvokeAll2() throws Exception {
980 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
981 <        try {
980 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
981 >        try (PoolCleaner cleaner = cleaner(e)) {
982              List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
983              assertTrue(r.isEmpty());
897        } finally {
898            joinPool(e);
984          }
985      }
986  
# Line 903 | Line 988 | public class ScheduledExecutorTest exten
988       * invokeAll(c) throws NPE if c has null elements
989       */
990      public void testInvokeAll3() throws Exception {
991 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
992 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
993 <        l.add(new StringTask());
994 <        l.add(null);
995 <        try {
996 <            e.invokeAll(l);
997 <            shouldThrow();
998 <        } catch (NullPointerException success) {
999 <        } finally {
915 <            joinPool(e);
991 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
992 >        try (PoolCleaner cleaner = cleaner(e)) {
993 >            List<Callable<String>> l = new ArrayList<>();
994 >            l.add(new StringTask());
995 >            l.add(null);
996 >            try {
997 >                e.invokeAll(l);
998 >                shouldThrow();
999 >            } catch (NullPointerException success) {}
1000          }
1001      }
1002  
# Line 920 | Line 1004 | public class ScheduledExecutorTest exten
1004       * get of invokeAll(c) throws exception on failed task
1005       */
1006      public void testInvokeAll4() throws Exception {
1007 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1008 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1009 <        l.add(new NPETask());
1010 <        List<Future<String>> futures = e.invokeAll(l);
1011 <        assertEquals(1, futures.size());
1012 <        try {
1013 <            futures.get(0).get();
1014 <            shouldThrow();
1015 <        } catch (ExecutionException success) {
1016 <            assertTrue(success.getCause() instanceof NullPointerException);
1017 <        } finally {
1018 <            joinPool(e);
1007 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1008 >        try (PoolCleaner cleaner = cleaner(e)) {
1009 >            List<Callable<String>> l = new ArrayList<>();
1010 >            l.add(new NPETask());
1011 >            List<Future<String>> futures = e.invokeAll(l);
1012 >            assertEquals(1, futures.size());
1013 >            try {
1014 >                futures.get(0).get();
1015 >                shouldThrow();
1016 >            } catch (ExecutionException success) {
1017 >                assertTrue(success.getCause() instanceof NullPointerException);
1018 >            }
1019          }
1020      }
1021  
# Line 939 | Line 1023 | public class ScheduledExecutorTest exten
1023       * invokeAll(c) returns results of all completed tasks
1024       */
1025      public void testInvokeAll5() throws Exception {
1026 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1027 <        try {
1028 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1026 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1027 >        try (PoolCleaner cleaner = cleaner(e)) {
1028 >            List<Callable<String>> l = new ArrayList<>();
1029              l.add(new StringTask());
1030              l.add(new StringTask());
1031              List<Future<String>> futures = e.invokeAll(l);
1032              assertEquals(2, futures.size());
1033              for (Future<String> future : futures)
1034                  assertSame(TEST_STRING, future.get());
951        } finally {
952            joinPool(e);
1035          }
1036      }
1037  
# Line 957 | Line 1039 | public class ScheduledExecutorTest exten
1039       * timed invokeAny(null) throws NPE
1040       */
1041      public void testTimedInvokeAny1() throws Exception {
1042 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1043 <        try {
1044 <            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1045 <            shouldThrow();
1046 <        } catch (NullPointerException success) {
1047 <        } finally {
966 <            joinPool(e);
1042 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1043 >        try (PoolCleaner cleaner = cleaner(e)) {
1044 >            try {
1045 >                e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1046 >                shouldThrow();
1047 >            } catch (NullPointerException success) {}
1048          }
1049      }
1050  
# Line 971 | Line 1052 | public class ScheduledExecutorTest exten
1052       * timed invokeAny(,,null) throws NPE
1053       */
1054      public void testTimedInvokeAnyNullTimeUnit() throws Exception {
1055 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1056 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1057 <        l.add(new StringTask());
1058 <        try {
1059 <            e.invokeAny(l, MEDIUM_DELAY_MS, null);
1060 <            shouldThrow();
1061 <        } catch (NullPointerException success) {
1062 <        } finally {
982 <            joinPool(e);
1055 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1056 >        try (PoolCleaner cleaner = cleaner(e)) {
1057 >            List<Callable<String>> l = new ArrayList<>();
1058 >            l.add(new StringTask());
1059 >            try {
1060 >                e.invokeAny(l, MEDIUM_DELAY_MS, null);
1061 >                shouldThrow();
1062 >            } catch (NullPointerException success) {}
1063          }
1064      }
1065  
# Line 987 | Line 1067 | public class ScheduledExecutorTest exten
1067       * timed invokeAny(empty collection) throws IAE
1068       */
1069      public void testTimedInvokeAny2() throws Exception {
1070 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1071 <        try {
1072 <            e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1073 <            shouldThrow();
1074 <        } catch (IllegalArgumentException success) {
1075 <        } finally {
996 <            joinPool(e);
1070 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1071 >        try (PoolCleaner cleaner = cleaner(e)) {
1072 >            try {
1073 >                e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1074 >                shouldThrow();
1075 >            } catch (IllegalArgumentException success) {}
1076          }
1077      }
1078  
# Line 1002 | Line 1081 | public class ScheduledExecutorTest exten
1081       */
1082      public void testTimedInvokeAny3() throws Exception {
1083          CountDownLatch latch = new CountDownLatch(1);
1084 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1085 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1086 <        l.add(latchAwaitingStringTask(latch));
1087 <        l.add(null);
1088 <        try {
1089 <            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1090 <            shouldThrow();
1091 <        } catch (NullPointerException success) {
1092 <        } finally {
1084 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1085 >        try (PoolCleaner cleaner = cleaner(e)) {
1086 >            List<Callable<String>> l = new ArrayList<>();
1087 >            l.add(latchAwaitingStringTask(latch));
1088 >            l.add(null);
1089 >            try {
1090 >                e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1091 >                shouldThrow();
1092 >            } catch (NullPointerException success) {}
1093              latch.countDown();
1015            joinPool(e);
1094          }
1095      }
1096  
# Line 1020 | Line 1098 | public class ScheduledExecutorTest exten
1098       * timed invokeAny(c) throws ExecutionException if no task completes
1099       */
1100      public void testTimedInvokeAny4() throws Exception {
1101 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1102 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1103 <        l.add(new NPETask());
1104 <        try {
1105 <            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1106 <            shouldThrow();
1107 <        } catch (ExecutionException success) {
1108 <            assertTrue(success.getCause() instanceof NullPointerException);
1109 <        } finally {
1110 <            joinPool(e);
1101 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1102 >        try (PoolCleaner cleaner = cleaner(e)) {
1103 >            long startTime = System.nanoTime();
1104 >            List<Callable<String>> l = new ArrayList<>();
1105 >            l.add(new NPETask());
1106 >            try {
1107 >                e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1108 >                shouldThrow();
1109 >            } catch (ExecutionException success) {
1110 >                assertTrue(success.getCause() instanceof NullPointerException);
1111 >            }
1112 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1113          }
1114      }
1115  
# Line 1037 | Line 1117 | public class ScheduledExecutorTest exten
1117       * timed invokeAny(c) returns result of some task
1118       */
1119      public void testTimedInvokeAny5() throws Exception {
1120 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1121 <        try {
1122 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1120 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1121 >        try (PoolCleaner cleaner = cleaner(e)) {
1122 >            long startTime = System.nanoTime();
1123 >            List<Callable<String>> l = new ArrayList<>();
1124              l.add(new StringTask());
1125              l.add(new StringTask());
1126 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1126 >            String result = e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1127              assertSame(TEST_STRING, result);
1128 <        } finally {
1048 <            joinPool(e);
1128 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1129          }
1130      }
1131  
# Line 1053 | Line 1133 | public class ScheduledExecutorTest exten
1133       * timed invokeAll(null) throws NPE
1134       */
1135      public void testTimedInvokeAll1() throws Exception {
1136 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1137 <        try {
1138 <            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1139 <            shouldThrow();
1140 <        } catch (NullPointerException success) {
1141 <        } finally {
1062 <            joinPool(e);
1136 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1137 >        try (PoolCleaner cleaner = cleaner(e)) {
1138 >            try {
1139 >                e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1140 >                shouldThrow();
1141 >            } catch (NullPointerException success) {}
1142          }
1143      }
1144  
# Line 1067 | Line 1146 | public class ScheduledExecutorTest exten
1146       * timed invokeAll(,,null) throws NPE
1147       */
1148      public void testTimedInvokeAllNullTimeUnit() throws Exception {
1149 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1150 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1151 <        l.add(new StringTask());
1152 <        try {
1153 <            e.invokeAll(l, MEDIUM_DELAY_MS, null);
1154 <            shouldThrow();
1155 <        } catch (NullPointerException success) {
1156 <        } finally {
1078 <            joinPool(e);
1149 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1150 >        try (PoolCleaner cleaner = cleaner(e)) {
1151 >            List<Callable<String>> l = new ArrayList<>();
1152 >            l.add(new StringTask());
1153 >            try {
1154 >                e.invokeAll(l, MEDIUM_DELAY_MS, null);
1155 >                shouldThrow();
1156 >            } catch (NullPointerException success) {}
1157          }
1158      }
1159  
# Line 1083 | Line 1161 | public class ScheduledExecutorTest exten
1161       * timed invokeAll(empty collection) returns empty collection
1162       */
1163      public void testTimedInvokeAll2() throws Exception {
1164 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1165 <        try {
1166 <            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1164 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1165 >        try (PoolCleaner cleaner = cleaner(e)) {
1166 >            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(),
1167 >                                                 MEDIUM_DELAY_MS, MILLISECONDS);
1168              assertTrue(r.isEmpty());
1090        } finally {
1091            joinPool(e);
1169          }
1170      }
1171  
# Line 1096 | Line 1173 | public class ScheduledExecutorTest exten
1173       * timed invokeAll(c) throws NPE if c has null elements
1174       */
1175      public void testTimedInvokeAll3() throws Exception {
1176 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1177 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1178 <        l.add(new StringTask());
1179 <        l.add(null);
1180 <        try {
1181 <            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1182 <            shouldThrow();
1183 <        } catch (NullPointerException success) {
1184 <        } finally {
1108 <            joinPool(e);
1176 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1177 >        try (PoolCleaner cleaner = cleaner(e)) {
1178 >            List<Callable<String>> l = new ArrayList<>();
1179 >            l.add(new StringTask());
1180 >            l.add(null);
1181 >            try {
1182 >                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1183 >                shouldThrow();
1184 >            } catch (NullPointerException success) {}
1185          }
1186      }
1187  
# Line 1113 | Line 1189 | public class ScheduledExecutorTest exten
1189       * get of element of invokeAll(c) throws exception on failed task
1190       */
1191      public void testTimedInvokeAll4() throws Exception {
1192 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1193 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1194 <        l.add(new NPETask());
1195 <        List<Future<String>> futures =
1196 <            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1197 <        assertEquals(1, futures.size());
1198 <        try {
1199 <            futures.get(0).get();
1200 <            shouldThrow();
1201 <        } catch (ExecutionException success) {
1202 <            assertTrue(success.getCause() instanceof NullPointerException);
1203 <        } finally {
1204 <            joinPool(e);
1192 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1193 >        try (PoolCleaner cleaner = cleaner(e)) {
1194 >            List<Callable<String>> l = new ArrayList<>();
1195 >            l.add(new NPETask());
1196 >            List<Future<String>> futures =
1197 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1198 >            assertEquals(1, futures.size());
1199 >            try {
1200 >                futures.get(0).get();
1201 >                shouldThrow();
1202 >            } catch (ExecutionException success) {
1203 >                assertTrue(success.getCause() instanceof NullPointerException);
1204 >            }
1205          }
1206      }
1207  
# Line 1133 | Line 1209 | public class ScheduledExecutorTest exten
1209       * timed invokeAll(c) returns results of all completed tasks
1210       */
1211      public void testTimedInvokeAll5() throws Exception {
1212 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1213 <        try {
1214 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1212 >        final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1213 >        try (PoolCleaner cleaner = cleaner(e)) {
1214 >            List<Callable<String>> l = new ArrayList<>();
1215              l.add(new StringTask());
1216              l.add(new StringTask());
1217              List<Future<String>> futures =
1218 <                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1218 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1219              assertEquals(2, futures.size());
1220              for (Future<String> future : futures)
1221                  assertSame(TEST_STRING, future.get());
1146        } finally {
1147            joinPool(e);
1222          }
1223      }
1224  
# Line 1152 | Line 1226 | public class ScheduledExecutorTest exten
1226       * timed invokeAll(c) cancels tasks not completed by timeout
1227       */
1228      public void testTimedInvokeAll6() throws Exception {
1229 <        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1230 <        try {
1231 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1232 <            l.add(new StringTask());
1233 <            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
1234 <            l.add(new StringTask());
1235 <            List<Future<String>> futures =
1236 <                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
1237 <            assertEquals(3, futures.size());
1238 <            Iterator<Future<String>> it = futures.iterator();
1239 <            Future<String> f1 = it.next();
1240 <            Future<String> f2 = it.next();
1241 <            Future<String> f3 = it.next();
1242 <            assertTrue(f1.isDone());
1243 <            assertTrue(f2.isDone());
1244 <            assertTrue(f3.isDone());
1245 <            assertFalse(f1.isCancelled());
1246 <            assertTrue(f2.isCancelled());
1247 <        } finally {
1248 <            joinPool(e);
1229 >        for (long timeout = timeoutMillis();;) {
1230 >            final CountDownLatch done = new CountDownLatch(1);
1231 >            final Callable<String> waiter = new CheckedCallable<String>() {
1232 >                public String realCall() {
1233 >                    try { done.await(LONG_DELAY_MS, MILLISECONDS); }
1234 >                    catch (InterruptedException ok) {}
1235 >                    return "1"; }};
1236 >            final ExecutorService p = new ScheduledThreadPoolExecutor(2);
1237 >            try (PoolCleaner cleaner = cleaner(p, done)) {
1238 >                List<Callable<String>> tasks = new ArrayList<>();
1239 >                tasks.add(new StringTask("0"));
1240 >                tasks.add(waiter);
1241 >                tasks.add(new StringTask("2"));
1242 >                long startTime = System.nanoTime();
1243 >                List<Future<String>> futures =
1244 >                    p.invokeAll(tasks, timeout, MILLISECONDS);
1245 >                assertEquals(tasks.size(), futures.size());
1246 >                assertTrue(millisElapsedSince(startTime) >= timeout);
1247 >                for (Future future : futures)
1248 >                    assertTrue(future.isDone());
1249 >                assertTrue(futures.get(1).isCancelled());
1250 >                try {
1251 >                    assertEquals("0", futures.get(0).get());
1252 >                    assertEquals("2", futures.get(2).get());
1253 >                    break;
1254 >                } catch (CancellationException retryWithLongerTimeout) {
1255 >                    timeout *= 2;
1256 >                    if (timeout >= LONG_DELAY_MS / 2)
1257 >                        fail("expected exactly one task to be cancelled");
1258 >                }
1259 >            }
1260          }
1261      }
1262  
1263 +    /**
1264 +     * A fixed delay task with overflowing period should not prevent a
1265 +     * one-shot task from executing.
1266 +     * https://bugs.openjdk.java.net/browse/JDK-8051859
1267 +     */
1268 +    public void testScheduleWithFixedDelay_overflow() throws Exception {
1269 +        final CountDownLatch delayedDone = new CountDownLatch(1);
1270 +        final CountDownLatch immediateDone = new CountDownLatch(1);
1271 +        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
1272 +        try (PoolCleaner cleaner = cleaner(p)) {
1273 +            final Runnable immediate = new Runnable() { public void run() {
1274 +                immediateDone.countDown();
1275 +            }};
1276 +            final Runnable delayed = new Runnable() { public void run() {
1277 +                delayedDone.countDown();
1278 +                p.submit(immediate);
1279 +            }};
1280 +            p.scheduleWithFixedDelay(delayed, 0L, Long.MAX_VALUE, SECONDS);
1281 +            await(delayedDone);
1282 +            await(immediateDone);
1283 +        }
1284 +    }
1285  
1286   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines