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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines