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

Comparing jsr166/src/test/tck/ThreadPoolExecutorTest.java (file contents):
Revision 1.51 by jsr166, Sat Apr 25 04:55:31 2015 UTC vs.
Revision 1.112 by jsr166, Thu Oct 8 03:08:38 2015 UTC

# Line 8 | Line 8
8  
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
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.List;
15   import java.util.concurrent.ArrayBlockingQueue;
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.Executors;
# Line 27 | Line 29 | import java.util.concurrent.SynchronousQ
29   import java.util.concurrent.ThreadFactory;
30   import java.util.concurrent.ThreadPoolExecutor;
31   import java.util.concurrent.TimeUnit;
32 + import java.util.concurrent.atomic.AtomicInteger;
33  
34   import junit.framework.Test;
35   import junit.framework.TestSuite;
# Line 84 | Line 87 | public class ThreadPoolExecutorTest exte
87              new ThreadPoolExecutor(1, 1,
88                                     LONG_DELAY_MS, MILLISECONDS,
89                                     new ArrayBlockingQueue<Runnable>(10));
90 <        final CountDownLatch done = new CountDownLatch(1);
91 <        final Runnable task = new CheckedRunnable() {
92 <            public void realRun() {
93 <                done.countDown();
91 <            }};
92 <        try {
90 >        try (PoolCleaner cleaner = cleaner(p)) {
91 >            final CountDownLatch done = new CountDownLatch(1);
92 >            final Runnable task = new CheckedRunnable() {
93 >                public void realRun() { done.countDown(); }};
94              p.execute(task);
95 <            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
95 <        } finally {
96 <            joinPool(p);
95 >            assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS));
96          }
97      }
98  
# Line 102 | Line 101 | public class ThreadPoolExecutorTest exte
101       * thread becomes active
102       */
103      public void testGetActiveCount() throws InterruptedException {
104 +        final CountDownLatch done = new CountDownLatch(1);
105          final ThreadPoolExecutor p =
106              new ThreadPoolExecutor(2, 2,
107                                     LONG_DELAY_MS, MILLISECONDS,
108                                     new ArrayBlockingQueue<Runnable>(10));
109 <        final CountDownLatch threadStarted = new CountDownLatch(1);
110 <        final CountDownLatch done = new CountDownLatch(1);
111 <        try {
109 >        try (PoolCleaner cleaner = cleaner(p, done)) {
110 >            final CountDownLatch threadStarted = new CountDownLatch(1);
111              assertEquals(0, p.getActiveCount());
112              p.execute(new CheckedRunnable() {
113                  public void realRun() throws InterruptedException {
114                      threadStarted.countDown();
115                      assertEquals(1, p.getActiveCount());
116 <                    done.await();
116 >                    await(done);
117                  }});
118 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
118 >            await(threadStarted);
119              assertEquals(1, p.getActiveCount());
121        } finally {
122            done.countDown();
123            joinPool(p);
120          }
121      }
122  
# Line 129 | Line 125 | public class ThreadPoolExecutorTest exte
125       */
126      public void testPrestartCoreThread() {
127          final ThreadPoolExecutor p =
128 <            new ThreadPoolExecutor(2, 2,
128 >            new ThreadPoolExecutor(2, 6,
129                                     LONG_DELAY_MS, MILLISECONDS,
130                                     new ArrayBlockingQueue<Runnable>(10));
131 <        assertEquals(0, p.getPoolSize());
132 <        assertTrue(p.prestartCoreThread());
133 <        assertEquals(1, p.getPoolSize());
134 <        assertTrue(p.prestartCoreThread());
135 <        assertEquals(2, p.getPoolSize());
136 <        assertFalse(p.prestartCoreThread());
137 <        assertEquals(2, p.getPoolSize());
138 <        joinPool(p);
131 >        try (PoolCleaner cleaner = cleaner(p)) {
132 >            assertEquals(0, p.getPoolSize());
133 >            assertTrue(p.prestartCoreThread());
134 >            assertEquals(1, p.getPoolSize());
135 >            assertTrue(p.prestartCoreThread());
136 >            assertEquals(2, p.getPoolSize());
137 >            assertFalse(p.prestartCoreThread());
138 >            assertEquals(2, p.getPoolSize());
139 >            p.setCorePoolSize(4);
140 >            assertTrue(p.prestartCoreThread());
141 >            assertEquals(3, p.getPoolSize());
142 >            assertTrue(p.prestartCoreThread());
143 >            assertEquals(4, p.getPoolSize());
144 >            assertFalse(p.prestartCoreThread());
145 >            assertEquals(4, p.getPoolSize());
146 >        }
147      }
148  
149      /**
# Line 147 | Line 151 | public class ThreadPoolExecutorTest exte
151       */
152      public void testPrestartAllCoreThreads() {
153          final ThreadPoolExecutor p =
154 <            new ThreadPoolExecutor(2, 2,
154 >            new ThreadPoolExecutor(2, 6,
155                                     LONG_DELAY_MS, MILLISECONDS,
156                                     new ArrayBlockingQueue<Runnable>(10));
157 <        assertEquals(0, p.getPoolSize());
158 <        p.prestartAllCoreThreads();
159 <        assertEquals(2, p.getPoolSize());
160 <        p.prestartAllCoreThreads();
161 <        assertEquals(2, p.getPoolSize());
162 <        joinPool(p);
157 >        try (PoolCleaner cleaner = cleaner(p)) {
158 >            assertEquals(0, p.getPoolSize());
159 >            p.prestartAllCoreThreads();
160 >            assertEquals(2, p.getPoolSize());
161 >            p.prestartAllCoreThreads();
162 >            assertEquals(2, p.getPoolSize());
163 >            p.setCorePoolSize(4);
164 >            p.prestartAllCoreThreads();
165 >            assertEquals(4, p.getPoolSize());
166 >            p.prestartAllCoreThreads();
167 >            assertEquals(4, p.getPoolSize());
168 >        }
169      }
170  
171      /**
# Line 167 | Line 177 | public class ThreadPoolExecutorTest exte
177              new ThreadPoolExecutor(2, 2,
178                                     LONG_DELAY_MS, MILLISECONDS,
179                                     new ArrayBlockingQueue<Runnable>(10));
180 <        final CountDownLatch threadStarted = new CountDownLatch(1);
181 <        final CountDownLatch threadProceed = new CountDownLatch(1);
182 <        final CountDownLatch threadDone = new CountDownLatch(1);
183 <        try {
180 >        try (PoolCleaner cleaner = cleaner(p)) {
181 >            final CountDownLatch threadStarted = new CountDownLatch(1);
182 >            final CountDownLatch threadProceed = new CountDownLatch(1);
183 >            final CountDownLatch threadDone = new CountDownLatch(1);
184              assertEquals(0, p.getCompletedTaskCount());
185              p.execute(new CheckedRunnable() {
186                  public void realRun() throws InterruptedException {
# Line 189 | Line 199 | public class ThreadPoolExecutorTest exte
199                      fail("timed out");
200                  Thread.yield();
201              }
192        } finally {
193            joinPool(p);
202          }
203      }
204  
# Line 202 | Line 210 | public class ThreadPoolExecutorTest exte
210              new ThreadPoolExecutor(1, 1,
211                                     LONG_DELAY_MS, MILLISECONDS,
212                                     new ArrayBlockingQueue<Runnable>(10));
213 <        assertEquals(1, p.getCorePoolSize());
214 <        joinPool(p);
213 >        try (PoolCleaner cleaner = cleaner(p)) {
214 >            assertEquals(1, p.getCorePoolSize());
215 >        }
216      }
217  
218      /**
# Line 214 | Line 223 | public class ThreadPoolExecutorTest exte
223              new ThreadPoolExecutor(2, 2,
224                                     1000, MILLISECONDS,
225                                     new ArrayBlockingQueue<Runnable>(10));
226 <        assertEquals(1, p.getKeepAliveTime(TimeUnit.SECONDS));
227 <        joinPool(p);
226 >        try (PoolCleaner cleaner = cleaner(p)) {
227 >            assertEquals(1, p.getKeepAliveTime(SECONDS));
228 >        }
229      }
230  
231      /**
232       * getThreadFactory returns factory in constructor if not set
233       */
234      public void testGetThreadFactory() {
235 <        ThreadFactory tf = new SimpleThreadFactory();
235 >        ThreadFactory threadFactory = new SimpleThreadFactory();
236          final ThreadPoolExecutor p =
237              new ThreadPoolExecutor(1, 2,
238                                     LONG_DELAY_MS, MILLISECONDS,
239                                     new ArrayBlockingQueue<Runnable>(10),
240 <                                   tf,
240 >                                   threadFactory,
241                                     new NoOpREHandler());
242 <        assertSame(tf, p.getThreadFactory());
243 <        joinPool(p);
242 >        try (PoolCleaner cleaner = cleaner(p)) {
243 >            assertSame(threadFactory, p.getThreadFactory());
244 >        }
245      }
246  
247      /**
# Line 241 | Line 252 | public class ThreadPoolExecutorTest exte
252              new ThreadPoolExecutor(1, 2,
253                                     LONG_DELAY_MS, MILLISECONDS,
254                                     new ArrayBlockingQueue<Runnable>(10));
255 <        ThreadFactory tf = new SimpleThreadFactory();
256 <        p.setThreadFactory(tf);
257 <        assertSame(tf, p.getThreadFactory());
258 <        joinPool(p);
255 >        try (PoolCleaner cleaner = cleaner(p)) {
256 >            ThreadFactory threadFactory = new SimpleThreadFactory();
257 >            p.setThreadFactory(threadFactory);
258 >            assertSame(threadFactory, p.getThreadFactory());
259 >        }
260      }
261  
262      /**
# Line 255 | Line 267 | public class ThreadPoolExecutorTest exte
267              new ThreadPoolExecutor(1, 2,
268                                     LONG_DELAY_MS, MILLISECONDS,
269                                     new ArrayBlockingQueue<Runnable>(10));
270 <        try {
271 <            p.setThreadFactory(null);
272 <            shouldThrow();
273 <        } catch (NullPointerException success) {
274 <        } finally {
263 <            joinPool(p);
270 >        try (PoolCleaner cleaner = cleaner(p)) {
271 >            try {
272 >                p.setThreadFactory(null);
273 >                shouldThrow();
274 >            } catch (NullPointerException success) {}
275          }
276      }
277  
# Line 268 | Line 279 | public class ThreadPoolExecutorTest exte
279       * getRejectedExecutionHandler returns handler in constructor if not set
280       */
281      public void testGetRejectedExecutionHandler() {
282 <        final RejectedExecutionHandler h = new NoOpREHandler();
282 >        final RejectedExecutionHandler handler = new NoOpREHandler();
283          final ThreadPoolExecutor p =
284              new ThreadPoolExecutor(1, 2,
285                                     LONG_DELAY_MS, MILLISECONDS,
286                                     new ArrayBlockingQueue<Runnable>(10),
287 <                                   h);
288 <        assertSame(h, p.getRejectedExecutionHandler());
289 <        joinPool(p);
287 >                                   handler);
288 >        try (PoolCleaner cleaner = cleaner(p)) {
289 >            assertSame(handler, p.getRejectedExecutionHandler());
290 >        }
291      }
292  
293      /**
# Line 287 | Line 299 | public class ThreadPoolExecutorTest exte
299              new ThreadPoolExecutor(1, 2,
300                                     LONG_DELAY_MS, MILLISECONDS,
301                                     new ArrayBlockingQueue<Runnable>(10));
302 <        RejectedExecutionHandler h = new NoOpREHandler();
303 <        p.setRejectedExecutionHandler(h);
304 <        assertSame(h, p.getRejectedExecutionHandler());
305 <        joinPool(p);
302 >        try (PoolCleaner cleaner = cleaner(p)) {
303 >            RejectedExecutionHandler handler = new NoOpREHandler();
304 >            p.setRejectedExecutionHandler(handler);
305 >            assertSame(handler, p.getRejectedExecutionHandler());
306 >        }
307      }
308  
309      /**
# Line 301 | Line 314 | public class ThreadPoolExecutorTest exte
314              new ThreadPoolExecutor(1, 2,
315                                     LONG_DELAY_MS, MILLISECONDS,
316                                     new ArrayBlockingQueue<Runnable>(10));
317 <        try {
318 <            p.setRejectedExecutionHandler(null);
319 <            shouldThrow();
320 <        } catch (NullPointerException success) {
321 <        } finally {
309 <            joinPool(p);
317 >        try (PoolCleaner cleaner = cleaner(p)) {
318 >            try {
319 >                p.setRejectedExecutionHandler(null);
320 >                shouldThrow();
321 >            } catch (NullPointerException success) {}
322          }
323      }
324  
# Line 316 | Line 328 | public class ThreadPoolExecutorTest exte
328       */
329      public void testGetLargestPoolSize() throws InterruptedException {
330          final int THREADS = 3;
331 +        final CountDownLatch done = new CountDownLatch(1);
332          final ThreadPoolExecutor p =
333              new ThreadPoolExecutor(THREADS, THREADS,
334                                     LONG_DELAY_MS, MILLISECONDS,
335                                     new ArrayBlockingQueue<Runnable>(10));
336 <        final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
324 <        final CountDownLatch done = new CountDownLatch(1);
325 <        try {
336 >        try (PoolCleaner cleaner = cleaner(p, done)) {
337              assertEquals(0, p.getLargestPoolSize());
338 +            final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
339              for (int i = 0; i < THREADS; i++)
340                  p.execute(new CheckedRunnable() {
341                      public void realRun() throws InterruptedException {
342                          threadsStarted.countDown();
343 <                        done.await();
343 >                        await(done);
344                          assertEquals(THREADS, p.getLargestPoolSize());
345                      }});
346 <            assertTrue(threadsStarted.await(SMALL_DELAY_MS, MILLISECONDS));
335 <            assertEquals(THREADS, p.getLargestPoolSize());
336 <        } finally {
337 <            done.countDown();
338 <            joinPool(p);
346 >            await(threadsStarted);
347              assertEquals(THREADS, p.getLargestPoolSize());
348          }
349 +        assertEquals(THREADS, p.getLargestPoolSize());
350      }
351  
352      /**
# Line 349 | Line 358 | public class ThreadPoolExecutorTest exte
358              new ThreadPoolExecutor(2, 3,
359                                     LONG_DELAY_MS, MILLISECONDS,
360                                     new ArrayBlockingQueue<Runnable>(10));
361 <        assertEquals(3, p.getMaximumPoolSize());
362 <        joinPool(p);
361 >        try (PoolCleaner cleaner = cleaner(p)) {
362 >            assertEquals(3, p.getMaximumPoolSize());
363 >            p.setMaximumPoolSize(5);
364 >            assertEquals(5, p.getMaximumPoolSize());
365 >            p.setMaximumPoolSize(4);
366 >            assertEquals(4, p.getMaximumPoolSize());
367 >        }
368      }
369  
370      /**
# Line 358 | Line 372 | public class ThreadPoolExecutorTest exte
372       * become active
373       */
374      public void testGetPoolSize() throws InterruptedException {
375 +        final CountDownLatch done = new CountDownLatch(1);
376          final ThreadPoolExecutor p =
377              new ThreadPoolExecutor(1, 1,
378                                     LONG_DELAY_MS, MILLISECONDS,
379                                     new ArrayBlockingQueue<Runnable>(10));
380 <        final CountDownLatch threadStarted = new CountDownLatch(1);
366 <        final CountDownLatch done = new CountDownLatch(1);
367 <        try {
380 >        try (PoolCleaner cleaner = cleaner(p, done)) {
381              assertEquals(0, p.getPoolSize());
382 +            final CountDownLatch threadStarted = new CountDownLatch(1);
383              p.execute(new CheckedRunnable() {
384                  public void realRun() throws InterruptedException {
385                      threadStarted.countDown();
386                      assertEquals(1, p.getPoolSize());
387 <                    done.await();
387 >                    await(done);
388                  }});
389 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
389 >            await(threadStarted);
390              assertEquals(1, p.getPoolSize());
377        } finally {
378            done.countDown();
379            joinPool(p);
391          }
392      }
393  
# Line 384 | Line 395 | public class ThreadPoolExecutorTest exte
395       * getTaskCount increases, but doesn't overestimate, when tasks submitted
396       */
397      public void testGetTaskCount() throws InterruptedException {
398 +        final int TASKS = 3;
399 +        final CountDownLatch done = new CountDownLatch(1);
400          final ThreadPoolExecutor p =
401              new ThreadPoolExecutor(1, 1,
402                                     LONG_DELAY_MS, MILLISECONDS,
403                                     new ArrayBlockingQueue<Runnable>(10));
404 <        final CountDownLatch threadStarted = new CountDownLatch(1);
405 <        final CountDownLatch done = new CountDownLatch(1);
393 <        try {
404 >        try (PoolCleaner cleaner = cleaner(p, done)) {
405 >            final CountDownLatch threadStarted = new CountDownLatch(1);
406              assertEquals(0, p.getTaskCount());
407 +            assertEquals(0, p.getCompletedTaskCount());
408              p.execute(new CheckedRunnable() {
409                  public void realRun() throws InterruptedException {
410                      threadStarted.countDown();
411 <                    assertEquals(1, p.getTaskCount());
399 <                    done.await();
411 >                    await(done);
412                  }});
413 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
413 >            await(threadStarted);
414              assertEquals(1, p.getTaskCount());
415 <        } finally {
416 <            done.countDown();
417 <            joinPool(p);
415 >            assertEquals(0, p.getCompletedTaskCount());
416 >            for (int i = 0; i < TASKS; i++) {
417 >                assertEquals(1 + i, p.getTaskCount());
418 >                p.execute(new CheckedRunnable() {
419 >                    public void realRun() throws InterruptedException {
420 >                        threadStarted.countDown();
421 >                        assertEquals(1 + TASKS, p.getTaskCount());
422 >                        await(done);
423 >                    }});
424 >            }
425 >            assertEquals(1 + TASKS, p.getTaskCount());
426 >            assertEquals(0, p.getCompletedTaskCount());
427          }
428 +        assertEquals(1 + TASKS, p.getTaskCount());
429 +        assertEquals(1 + TASKS, p.getCompletedTaskCount());
430      }
431  
432      /**
# Line 414 | Line 437 | public class ThreadPoolExecutorTest exte
437              new ThreadPoolExecutor(1, 1,
438                                     LONG_DELAY_MS, MILLISECONDS,
439                                     new ArrayBlockingQueue<Runnable>(10));
440 <        assertFalse(p.isShutdown());
441 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
442 <        assertTrue(p.isShutdown());
443 <        joinPool(p);
440 >        try (PoolCleaner cleaner = cleaner(p)) {
441 >            assertFalse(p.isShutdown());
442 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
443 >            assertTrue(p.isShutdown());
444 >        }
445      }
446  
447      /**
# Line 428 | Line 452 | public class ThreadPoolExecutorTest exte
452              new ThreadPoolExecutor(1, 1,
453                                     LONG_DELAY_MS, MILLISECONDS,
454                                     new ArrayBlockingQueue<Runnable>(10));
455 <        assertFalse(p.isTerminated());
456 <        assertFalse(p.awaitTermination(Long.MIN_VALUE, NANOSECONDS));
457 <        assertFalse(p.awaitTermination(Long.MIN_VALUE, MILLISECONDS));
458 <        assertFalse(p.awaitTermination(-1L, NANOSECONDS));
459 <        assertFalse(p.awaitTermination(-1L, MILLISECONDS));
460 <        assertFalse(p.awaitTermination(0L, NANOSECONDS));
461 <        assertFalse(p.awaitTermination(0L, MILLISECONDS));
462 <        long timeoutNanos = 999999L;
463 <        long startTime = System.nanoTime();
464 <        assertFalse(p.awaitTermination(timeoutNanos, NANOSECONDS));
465 <        assertTrue(System.nanoTime() - startTime >= timeoutNanos);
466 <        assertFalse(p.isTerminated());
467 <        startTime = System.nanoTime();
468 <        long timeoutMillis = timeoutMillis();
469 <        assertFalse(p.awaitTermination(timeoutMillis, MILLISECONDS));
470 <        assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
471 <        assertFalse(p.isTerminated());
472 <        p.shutdown();
473 <        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
474 <        assertTrue(p.isTerminated());
455 >        try (PoolCleaner cleaner = cleaner(p)) {
456 >            assertFalse(p.isTerminated());
457 >            assertFalse(p.awaitTermination(Long.MIN_VALUE, NANOSECONDS));
458 >            assertFalse(p.awaitTermination(Long.MIN_VALUE, MILLISECONDS));
459 >            assertFalse(p.awaitTermination(-1L, NANOSECONDS));
460 >            assertFalse(p.awaitTermination(-1L, MILLISECONDS));
461 >            assertFalse(p.awaitTermination(0L, NANOSECONDS));
462 >            assertFalse(p.awaitTermination(0L, MILLISECONDS));
463 >            long timeoutNanos = 999999L;
464 >            long startTime = System.nanoTime();
465 >            assertFalse(p.awaitTermination(timeoutNanos, NANOSECONDS));
466 >            assertTrue(System.nanoTime() - startTime >= timeoutNanos);
467 >            assertFalse(p.isTerminated());
468 >            startTime = System.nanoTime();
469 >            long timeoutMillis = timeoutMillis();
470 >            assertFalse(p.awaitTermination(timeoutMillis, MILLISECONDS));
471 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
472 >            assertFalse(p.isTerminated());
473 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
474 >            assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
475 >            assertTrue(p.isTerminated());
476 >        }
477      }
478  
479      /**
# Line 458 | Line 484 | public class ThreadPoolExecutorTest exte
484              new ThreadPoolExecutor(1, 1,
485                                     LONG_DELAY_MS, MILLISECONDS,
486                                     new ArrayBlockingQueue<Runnable>(10));
487 <        final CountDownLatch threadStarted = new CountDownLatch(1);
488 <        final CountDownLatch done = new CountDownLatch(1);
489 <        assertFalse(p.isTerminated());
490 <        try {
487 >        try (PoolCleaner cleaner = cleaner(p)) {
488 >            final CountDownLatch threadStarted = new CountDownLatch(1);
489 >            final CountDownLatch done = new CountDownLatch(1);
490 >            assertFalse(p.isTerminating());
491              p.execute(new CheckedRunnable() {
492                  public void realRun() throws InterruptedException {
493 <                    assertFalse(p.isTerminated());
493 >                    assertFalse(p.isTerminating());
494                      threadStarted.countDown();
495 <                    done.await();
495 >                    await(done);
496                  }});
497 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
497 >            await(threadStarted);
498              assertFalse(p.isTerminating());
499              done.countDown();
474        } finally {
500              try { p.shutdown(); } catch (SecurityException ok) { return; }
501 +            assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
502 +            assertTrue(p.isTerminated());
503 +            assertFalse(p.isTerminating());
504          }
477        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
478        assertTrue(p.isTerminated());
505      }
506  
507      /**
# Line 486 | Line 512 | public class ThreadPoolExecutorTest exte
512              new ThreadPoolExecutor(1, 1,
513                                     LONG_DELAY_MS, MILLISECONDS,
514                                     new ArrayBlockingQueue<Runnable>(10));
515 <        final CountDownLatch threadStarted = new CountDownLatch(1);
516 <        final CountDownLatch done = new CountDownLatch(1);
517 <        try {
515 >        try (PoolCleaner cleaner = cleaner(p)) {
516 >            final CountDownLatch threadStarted = new CountDownLatch(1);
517 >            final CountDownLatch done = new CountDownLatch(1);
518              assertFalse(p.isTerminating());
519              p.execute(new CheckedRunnable() {
520                  public void realRun() throws InterruptedException {
521                      assertFalse(p.isTerminating());
522                      threadStarted.countDown();
523 <                    done.await();
523 >                    await(done);
524                  }});
525 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
525 >            await(threadStarted);
526              assertFalse(p.isTerminating());
527              done.countDown();
502        } finally {
528              try { p.shutdown(); } catch (SecurityException ok) { return; }
529 +            assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
530 +            assertTrue(p.isTerminated());
531 +            assertFalse(p.isTerminating());
532          }
505        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
506        assertTrue(p.isTerminated());
507        assertFalse(p.isTerminating());
533      }
534  
535      /**
536       * getQueue returns the work queue, which contains queued tasks
537       */
538      public void testGetQueue() throws InterruptedException {
539 +        final CountDownLatch done = new CountDownLatch(1);
540          final BlockingQueue<Runnable> q = new ArrayBlockingQueue<Runnable>(10);
541          final ThreadPoolExecutor p =
542              new ThreadPoolExecutor(1, 1,
543                                     LONG_DELAY_MS, MILLISECONDS,
544                                     q);
545 <        final CountDownLatch threadStarted = new CountDownLatch(1);
546 <        final CountDownLatch done = new CountDownLatch(1);
521 <        try {
545 >        try (PoolCleaner cleaner = cleaner(p, done)) {
546 >            final CountDownLatch threadStarted = new CountDownLatch(1);
547              FutureTask[] tasks = new FutureTask[5];
548              for (int i = 0; i < tasks.length; i++) {
549                  Callable task = new CheckedCallable<Boolean>() {
550                      public Boolean realCall() throws InterruptedException {
551                          threadStarted.countDown();
552                          assertSame(q, p.getQueue());
553 <                        done.await();
553 >                        await(done);
554                          return Boolean.TRUE;
555                      }};
556                  tasks[i] = new FutureTask(task);
557                  p.execute(tasks[i]);
558              }
559 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
559 >            await(threadStarted);
560              assertSame(q, p.getQueue());
561              assertFalse(q.contains(tasks[0]));
562              assertTrue(q.contains(tasks[tasks.length - 1]));
563              assertEquals(tasks.length - 1, q.size());
539        } finally {
540            done.countDown();
541            joinPool(p);
564          }
565      }
566  
# Line 546 | Line 568 | public class ThreadPoolExecutorTest exte
568       * remove(task) removes queued task, and fails to remove active task
569       */
570      public void testRemove() throws InterruptedException {
571 +        final CountDownLatch done = new CountDownLatch(1);
572          BlockingQueue<Runnable> q = new ArrayBlockingQueue<Runnable>(10);
573          final ThreadPoolExecutor p =
574              new ThreadPoolExecutor(1, 1,
575                                     LONG_DELAY_MS, MILLISECONDS,
576                                     q);
577 <        Runnable[] tasks = new Runnable[5];
578 <        final CountDownLatch threadStarted = new CountDownLatch(1);
579 <        final CountDownLatch done = new CountDownLatch(1);
557 <        try {
577 >        try (PoolCleaner cleaner = cleaner(p, done)) {
578 >            Runnable[] tasks = new Runnable[6];
579 >            final CountDownLatch threadStarted = new CountDownLatch(1);
580              for (int i = 0; i < tasks.length; i++) {
581                  tasks[i] = new CheckedRunnable() {
582                      public void realRun() throws InterruptedException {
583                          threadStarted.countDown();
584 <                        done.await();
584 >                        await(done);
585                      }};
586                  p.execute(tasks[i]);
587              }
588 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
588 >            await(threadStarted);
589              assertFalse(p.remove(tasks[0]));
590              assertTrue(q.contains(tasks[4]));
591              assertTrue(q.contains(tasks[3]));
# Line 573 | Line 595 | public class ThreadPoolExecutorTest exte
595              assertTrue(q.contains(tasks[3]));
596              assertTrue(p.remove(tasks[3]));
597              assertFalse(q.contains(tasks[3]));
576        } finally {
577            done.countDown();
578            joinPool(p);
598          }
599      }
600  
# Line 590 | Line 609 | public class ThreadPoolExecutorTest exte
609              new ThreadPoolExecutor(1, 1,
610                                     LONG_DELAY_MS, MILLISECONDS,
611                                     q);
612 <        FutureTask[] tasks = new FutureTask[5];
613 <        try {
612 >        try (PoolCleaner cleaner = cleaner(p, done)) {
613 >            FutureTask[] tasks = new FutureTask[5];
614              for (int i = 0; i < tasks.length; i++) {
615                  Callable task = new CheckedCallable<Boolean>() {
616                      public Boolean realCall() throws InterruptedException {
617                          threadStarted.countDown();
618 <                        done.await();
618 >                        await(done);
619                          return Boolean.TRUE;
620                      }};
621                  tasks[i] = new FutureTask(task);
622                  p.execute(tasks[i]);
623              }
624 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
624 >            await(threadStarted);
625              assertEquals(tasks.length, p.getTaskCount());
626              assertEquals(tasks.length - 1, q.size());
627              assertEquals(1L, p.getActiveCount());
# Line 615 | Line 634 | public class ThreadPoolExecutorTest exte
634              p.purge();         // Nothing to do
635              assertEquals(tasks.length - 3, q.size());
636              assertEquals(tasks.length - 2, p.getTaskCount());
618        } finally {
619            done.countDown();
620            joinPool(p);
637          }
638      }
639  
640      /**
641 <     * shutdownNow returns a list containing tasks that were not run
641 >     * shutdownNow returns a list containing tasks that were not run,
642 >     * and those tasks are drained from the queue
643       */
644 <    public void testShutdownNow() {
644 >    public void testShutdownNow() throws InterruptedException {
645 >        final int poolSize = 2;
646 >        final int count = 5;
647 >        final AtomicInteger ran = new AtomicInteger(0);
648          final ThreadPoolExecutor p =
649 <            new ThreadPoolExecutor(1, 1,
649 >            new ThreadPoolExecutor(poolSize, poolSize,
650                                     LONG_DELAY_MS, MILLISECONDS,
651                                     new ArrayBlockingQueue<Runnable>(10));
652 <        List l;
653 <        try {
654 <            for (int i = 0; i < 5; i++)
635 <                p.execute(new MediumPossiblyInterruptedRunnable());
636 <        }
637 <        finally {
652 >        final CountDownLatch threadsStarted = new CountDownLatch(poolSize);
653 >        Runnable waiter = new CheckedRunnable() { public void realRun() {
654 >            threadsStarted.countDown();
655              try {
656 <                l = p.shutdownNow();
657 <            } catch (SecurityException ok) { return; }
656 >                MILLISECONDS.sleep(2 * LONG_DELAY_MS);
657 >            } catch (InterruptedException success) {}
658 >            ran.getAndIncrement();
659 >        }};
660 >        for (int i = 0; i < count; i++)
661 >            p.execute(waiter);
662 >        await(threadsStarted);
663 >        assertEquals(poolSize, p.getActiveCount());
664 >        assertEquals(0, p.getCompletedTaskCount());
665 >        final List<Runnable> queuedTasks;
666 >        try {
667 >            queuedTasks = p.shutdownNow();
668 >        } catch (SecurityException ok) {
669 >            return; // Allowed in case test doesn't have privs
670          }
671          assertTrue(p.isShutdown());
672 <        assertTrue(l.size() <= 4);
672 >        assertTrue(p.getQueue().isEmpty());
673 >        assertEquals(count - poolSize, queuedTasks.size());
674 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
675 >        assertTrue(p.isTerminated());
676 >        assertEquals(poolSize, ran.get());
677 >        assertEquals(poolSize, p.getCompletedTaskCount());
678      }
679  
680      // Exception Tests
# Line 650 | Line 684 | public class ThreadPoolExecutorTest exte
684       */
685      public void testConstructor1() {
686          try {
687 <            new ThreadPoolExecutor(-1, 1,
654 <                                   LONG_DELAY_MS, MILLISECONDS,
687 >            new ThreadPoolExecutor(-1, 1, 1L, SECONDS,
688                                     new ArrayBlockingQueue<Runnable>(10));
689              shouldThrow();
690          } catch (IllegalArgumentException success) {}
# Line 662 | Line 695 | public class ThreadPoolExecutorTest exte
695       */
696      public void testConstructor2() {
697          try {
698 <            new ThreadPoolExecutor(1, -1,
666 <                                   LONG_DELAY_MS, MILLISECONDS,
698 >            new ThreadPoolExecutor(1, -1, 1L, SECONDS,
699                                     new ArrayBlockingQueue<Runnable>(10));
700              shouldThrow();
701          } catch (IllegalArgumentException success) {}
# Line 674 | Line 706 | public class ThreadPoolExecutorTest exte
706       */
707      public void testConstructor3() {
708          try {
709 <            new ThreadPoolExecutor(1, 0,
678 <                                   LONG_DELAY_MS, MILLISECONDS,
709 >            new ThreadPoolExecutor(1, 0, 1L, SECONDS,
710                                     new ArrayBlockingQueue<Runnable>(10));
711              shouldThrow();
712          } catch (IllegalArgumentException success) {}
# Line 686 | Line 717 | public class ThreadPoolExecutorTest exte
717       */
718      public void testConstructor4() {
719          try {
720 <            new ThreadPoolExecutor(1, 2,
690 <                                   -1L, MILLISECONDS,
720 >            new ThreadPoolExecutor(1, 2, -1L, SECONDS,
721                                     new ArrayBlockingQueue<Runnable>(10));
722              shouldThrow();
723          } catch (IllegalArgumentException success) {}
# Line 698 | Line 728 | public class ThreadPoolExecutorTest exte
728       */
729      public void testConstructor5() {
730          try {
731 <            new ThreadPoolExecutor(2, 1,
702 <                                   LONG_DELAY_MS, MILLISECONDS,
731 >            new ThreadPoolExecutor(2, 1, 1L, SECONDS,
732                                     new ArrayBlockingQueue<Runnable>(10));
733              shouldThrow();
734          } catch (IllegalArgumentException success) {}
# Line 710 | Line 739 | public class ThreadPoolExecutorTest exte
739       */
740      public void testConstructorNullPointerException() {
741          try {
742 <            new ThreadPoolExecutor(1, 2,
714 <                                   LONG_DELAY_MS, MILLISECONDS,
742 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
743                                     (BlockingQueue) null);
744              shouldThrow();
745          } catch (NullPointerException success) {}
# Line 722 | Line 750 | public class ThreadPoolExecutorTest exte
750       */
751      public void testConstructor6() {
752          try {
753 <            new ThreadPoolExecutor(-1, 1,
726 <                                   LONG_DELAY_MS, MILLISECONDS,
753 >            new ThreadPoolExecutor(-1, 1, 1L, SECONDS,
754                                     new ArrayBlockingQueue<Runnable>(10),
755                                     new SimpleThreadFactory());
756              shouldThrow();
# Line 735 | Line 762 | public class ThreadPoolExecutorTest exte
762       */
763      public void testConstructor7() {
764          try {
765 <            new ThreadPoolExecutor(1, -1,
739 <                                   LONG_DELAY_MS, MILLISECONDS,
765 >            new ThreadPoolExecutor(1, -1, 1L, SECONDS,
766                                     new ArrayBlockingQueue<Runnable>(10),
767                                     new SimpleThreadFactory());
768              shouldThrow();
# Line 748 | Line 774 | public class ThreadPoolExecutorTest exte
774       */
775      public void testConstructor8() {
776          try {
777 <            new ThreadPoolExecutor(1, 0,
752 <                                   LONG_DELAY_MS, MILLISECONDS,
777 >            new ThreadPoolExecutor(1, 0, 1L, SECONDS,
778                                     new ArrayBlockingQueue<Runnable>(10),
779                                     new SimpleThreadFactory());
780              shouldThrow();
# Line 761 | Line 786 | public class ThreadPoolExecutorTest exte
786       */
787      public void testConstructor9() {
788          try {
789 <            new ThreadPoolExecutor(1, 2,
765 <                                   -1L, MILLISECONDS,
789 >            new ThreadPoolExecutor(1, 2, -1L, SECONDS,
790                                     new ArrayBlockingQueue<Runnable>(10),
791                                     new SimpleThreadFactory());
792              shouldThrow();
# Line 774 | Line 798 | public class ThreadPoolExecutorTest exte
798       */
799      public void testConstructor10() {
800          try {
801 <            new ThreadPoolExecutor(2, 1,
778 <                                   LONG_DELAY_MS, MILLISECONDS,
801 >            new ThreadPoolExecutor(2, 1, 1L, SECONDS,
802                                     new ArrayBlockingQueue<Runnable>(10),
803                                     new SimpleThreadFactory());
804              shouldThrow();
# Line 787 | Line 810 | public class ThreadPoolExecutorTest exte
810       */
811      public void testConstructorNullPointerException2() {
812          try {
813 <            new ThreadPoolExecutor(1, 2,
791 <                                   LONG_DELAY_MS, MILLISECONDS,
813 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
814                                     (BlockingQueue) null,
815                                     new SimpleThreadFactory());
816              shouldThrow();
# Line 800 | Line 822 | public class ThreadPoolExecutorTest exte
822       */
823      public void testConstructorNullPointerException3() {
824          try {
825 <            new ThreadPoolExecutor(1, 2,
804 <                                   LONG_DELAY_MS, MILLISECONDS,
825 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
826                                     new ArrayBlockingQueue<Runnable>(10),
827                                     (ThreadFactory) null);
828              shouldThrow();
# Line 813 | Line 834 | public class ThreadPoolExecutorTest exte
834       */
835      public void testConstructor11() {
836          try {
837 <            new ThreadPoolExecutor(-1, 1,
817 <                                   LONG_DELAY_MS, MILLISECONDS,
837 >            new ThreadPoolExecutor(-1, 1, 1L, SECONDS,
838                                     new ArrayBlockingQueue<Runnable>(10),
839                                     new NoOpREHandler());
840              shouldThrow();
# Line 826 | Line 846 | public class ThreadPoolExecutorTest exte
846       */
847      public void testConstructor12() {
848          try {
849 <            new ThreadPoolExecutor(1, -1,
830 <                                   LONG_DELAY_MS, MILLISECONDS,
849 >            new ThreadPoolExecutor(1, -1, 1L, SECONDS,
850                                     new ArrayBlockingQueue<Runnable>(10),
851                                     new NoOpREHandler());
852              shouldThrow();
# Line 839 | Line 858 | public class ThreadPoolExecutorTest exte
858       */
859      public void testConstructor13() {
860          try {
861 <            new ThreadPoolExecutor(1, 0,
843 <                                   LONG_DELAY_MS, MILLISECONDS,
861 >            new ThreadPoolExecutor(1, 0, 1L, SECONDS,
862                                     new ArrayBlockingQueue<Runnable>(10),
863                                     new NoOpREHandler());
864              shouldThrow();
# Line 852 | Line 870 | public class ThreadPoolExecutorTest exte
870       */
871      public void testConstructor14() {
872          try {
873 <            new ThreadPoolExecutor(1, 2,
856 <                                   -1L, MILLISECONDS,
873 >            new ThreadPoolExecutor(1, 2, -1L, SECONDS,
874                                     new ArrayBlockingQueue<Runnable>(10),
875                                     new NoOpREHandler());
876              shouldThrow();
# Line 865 | Line 882 | public class ThreadPoolExecutorTest exte
882       */
883      public void testConstructor15() {
884          try {
885 <            new ThreadPoolExecutor(2, 1,
869 <                                   LONG_DELAY_MS, MILLISECONDS,
885 >            new ThreadPoolExecutor(2, 1, 1L, SECONDS,
886                                     new ArrayBlockingQueue<Runnable>(10),
887                                     new NoOpREHandler());
888              shouldThrow();
# Line 878 | Line 894 | public class ThreadPoolExecutorTest exte
894       */
895      public void testConstructorNullPointerException4() {
896          try {
897 <            new ThreadPoolExecutor(1, 2,
882 <                                   LONG_DELAY_MS, MILLISECONDS,
897 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
898                                     (BlockingQueue) null,
899                                     new NoOpREHandler());
900              shouldThrow();
# Line 891 | Line 906 | public class ThreadPoolExecutorTest exte
906       */
907      public void testConstructorNullPointerException5() {
908          try {
909 <            new ThreadPoolExecutor(1, 2,
895 <                                   LONG_DELAY_MS, MILLISECONDS,
909 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
910                                     new ArrayBlockingQueue<Runnable>(10),
911                                     (RejectedExecutionHandler) null);
912              shouldThrow();
# Line 904 | Line 918 | public class ThreadPoolExecutorTest exte
918       */
919      public void testConstructor16() {
920          try {
921 <            new ThreadPoolExecutor(-1, 1,
908 <                                   LONG_DELAY_MS, MILLISECONDS,
921 >            new ThreadPoolExecutor(-1, 1, 1L, SECONDS,
922                                     new ArrayBlockingQueue<Runnable>(10),
923                                     new SimpleThreadFactory(),
924                                     new NoOpREHandler());
# Line 918 | Line 931 | public class ThreadPoolExecutorTest exte
931       */
932      public void testConstructor17() {
933          try {
934 <            new ThreadPoolExecutor(1, -1,
922 <                                   LONG_DELAY_MS, MILLISECONDS,
934 >            new ThreadPoolExecutor(1, -1, 1L, SECONDS,
935                                     new ArrayBlockingQueue<Runnable>(10),
936                                     new SimpleThreadFactory(),
937                                     new NoOpREHandler());
# Line 932 | Line 944 | public class ThreadPoolExecutorTest exte
944       */
945      public void testConstructor18() {
946          try {
947 <            new ThreadPoolExecutor(1, 0,
936 <                                   LONG_DELAY_MS, MILLISECONDS,
947 >            new ThreadPoolExecutor(1, 0, 1L, SECONDS,
948                                     new ArrayBlockingQueue<Runnable>(10),
949                                     new SimpleThreadFactory(),
950                                     new NoOpREHandler());
# Line 946 | Line 957 | public class ThreadPoolExecutorTest exte
957       */
958      public void testConstructor19() {
959          try {
960 <            new ThreadPoolExecutor(1, 2,
950 <                                   -1L, MILLISECONDS,
960 >            new ThreadPoolExecutor(1, 2, -1L, SECONDS,
961                                     new ArrayBlockingQueue<Runnable>(10),
962                                     new SimpleThreadFactory(),
963                                     new NoOpREHandler());
# Line 960 | Line 970 | public class ThreadPoolExecutorTest exte
970       */
971      public void testConstructor20() {
972          try {
973 <            new ThreadPoolExecutor(2, 1,
964 <                                   LONG_DELAY_MS, MILLISECONDS,
973 >            new ThreadPoolExecutor(2, 1, 1L, SECONDS,
974                                     new ArrayBlockingQueue<Runnable>(10),
975                                     new SimpleThreadFactory(),
976                                     new NoOpREHandler());
# Line 974 | Line 983 | public class ThreadPoolExecutorTest exte
983       */
984      public void testConstructorNullPointerException6() {
985          try {
986 <            new ThreadPoolExecutor(1, 2,
978 <                                   LONG_DELAY_MS, MILLISECONDS,
986 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
987                                     (BlockingQueue) null,
988                                     new SimpleThreadFactory(),
989                                     new NoOpREHandler());
# Line 988 | Line 996 | public class ThreadPoolExecutorTest exte
996       */
997      public void testConstructorNullPointerException7() {
998          try {
999 <            new ThreadPoolExecutor(1, 2,
992 <                                   LONG_DELAY_MS, MILLISECONDS,
999 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
1000                                     new ArrayBlockingQueue<Runnable>(10),
1001                                     new SimpleThreadFactory(),
1002                                     (RejectedExecutionHandler) null);
# Line 1002 | Line 1009 | public class ThreadPoolExecutorTest exte
1009       */
1010      public void testConstructorNullPointerException8() {
1011          try {
1012 <            new ThreadPoolExecutor(1, 2,
1006 <                                   LONG_DELAY_MS, MILLISECONDS,
1012 >            new ThreadPoolExecutor(1, 2, 1L, SECONDS,
1013                                     new ArrayBlockingQueue<Runnable>(10),
1014                                     (ThreadFactory) null,
1015                                     new NoOpREHandler());
# Line 1015 | Line 1021 | public class ThreadPoolExecutorTest exte
1021       * get of submitted callable throws InterruptedException if interrupted
1022       */
1023      public void testInterruptedSubmit() throws InterruptedException {
1024 +        final CountDownLatch done = new CountDownLatch(1);
1025          final ThreadPoolExecutor p =
1026              new ThreadPoolExecutor(1, 1,
1027 <                                   60, TimeUnit.SECONDS,
1027 >                                   60, SECONDS,
1028                                     new ArrayBlockingQueue<Runnable>(10));
1029  
1030 <        final CountDownLatch threadStarted = new CountDownLatch(1);
1031 <        final CountDownLatch done = new CountDownLatch(1);
1025 <        try {
1030 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1031 >            final CountDownLatch threadStarted = new CountDownLatch(1);
1032              Thread t = newStartedThread(new CheckedInterruptedRunnable() {
1033                  public void realRun() throws Exception {
1034                      Callable task = new CheckedCallable<Boolean>() {
1035                          public Boolean realCall() throws InterruptedException {
1036                              threadStarted.countDown();
1037 <                            done.await();
1037 >                            await(done);
1038                              return Boolean.TRUE;
1039                          }};
1040                      p.submit(task).get();
1041                  }});
1042  
1043 <            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
1043 >            await(threadStarted);
1044              t.interrupt();
1045 <            awaitTermination(t, MEDIUM_DELAY_MS);
1040 <        } finally {
1041 <            done.countDown();
1042 <            joinPool(p);
1045 >            awaitTermination(t);
1046          }
1047      }
1048  
# Line 1047 | Line 1050 | public class ThreadPoolExecutorTest exte
1050       * execute throws RejectedExecutionException if saturated.
1051       */
1052      public void testSaturatedExecute() {
1053 <        ThreadPoolExecutor p =
1053 >        final CountDownLatch done = new CountDownLatch(1);
1054 >        final ThreadPoolExecutor p =
1055              new ThreadPoolExecutor(1, 1,
1056                                     LONG_DELAY_MS, MILLISECONDS,
1057                                     new ArrayBlockingQueue<Runnable>(1));
1058 <        final CountDownLatch done = new CountDownLatch(1);
1055 <        try {
1058 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1059              Runnable task = new CheckedRunnable() {
1060                  public void realRun() throws InterruptedException {
1061 <                    done.await();
1061 >                    await(done);
1062                  }};
1063              for (int i = 0; i < 2; ++i)
1064                  p.execute(task);
# Line 1066 | Line 1069 | public class ThreadPoolExecutorTest exte
1069                  } catch (RejectedExecutionException success) {}
1070                  assertTrue(p.getTaskCount() <= 2);
1071              }
1069        } finally {
1070            done.countDown();
1071            joinPool(p);
1072          }
1073      }
1074  
# Line 1076 | Line 1076 | public class ThreadPoolExecutorTest exte
1076       * submit(runnable) throws RejectedExecutionException if saturated.
1077       */
1078      public void testSaturatedSubmitRunnable() {
1079 <        ThreadPoolExecutor p =
1079 >        final CountDownLatch done = new CountDownLatch(1);
1080 >        final ThreadPoolExecutor p =
1081              new ThreadPoolExecutor(1, 1,
1082                                     LONG_DELAY_MS, MILLISECONDS,
1083                                     new ArrayBlockingQueue<Runnable>(1));
1084 <        final CountDownLatch done = new CountDownLatch(1);
1084 <        try {
1084 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1085              Runnable task = new CheckedRunnable() {
1086                  public void realRun() throws InterruptedException {
1087 <                    done.await();
1087 >                    await(done);
1088                  }};
1089              for (int i = 0; i < 2; ++i)
1090                  p.submit(task);
# Line 1095 | Line 1095 | public class ThreadPoolExecutorTest exte
1095                  } catch (RejectedExecutionException success) {}
1096                  assertTrue(p.getTaskCount() <= 2);
1097              }
1098        } finally {
1099            done.countDown();
1100            joinPool(p);
1098          }
1099      }
1100  
# Line 1105 | Line 1102 | public class ThreadPoolExecutorTest exte
1102       * submit(callable) throws RejectedExecutionException if saturated.
1103       */
1104      public void testSaturatedSubmitCallable() {
1105 <        ThreadPoolExecutor p =
1105 >        final CountDownLatch done = new CountDownLatch(1);
1106 >        final ThreadPoolExecutor p =
1107              new ThreadPoolExecutor(1, 1,
1108                                     LONG_DELAY_MS, MILLISECONDS,
1109                                     new ArrayBlockingQueue<Runnable>(1));
1110 <        final CountDownLatch done = new CountDownLatch(1);
1113 <        try {
1110 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1111              Runnable task = new CheckedRunnable() {
1112                  public void realRun() throws InterruptedException {
1113 <                    done.await();
1113 >                    await(done);
1114                  }};
1115              for (int i = 0; i < 2; ++i)
1116                  p.submit(Executors.callable(task));
# Line 1124 | Line 1121 | public class ThreadPoolExecutorTest exte
1121                  } catch (RejectedExecutionException success) {}
1122                  assertTrue(p.getTaskCount() <= 2);
1123              }
1127        } finally {
1128            done.countDown();
1129            joinPool(p);
1124          }
1125      }
1126  
# Line 1134 | Line 1128 | public class ThreadPoolExecutorTest exte
1128       * executor using CallerRunsPolicy runs task if saturated.
1129       */
1130      public void testSaturatedExecute2() {
1137        RejectedExecutionHandler h = new ThreadPoolExecutor.CallerRunsPolicy();
1131          final ThreadPoolExecutor p =
1132              new ThreadPoolExecutor(1, 1,
1133                                     LONG_DELAY_MS,
1134                                     MILLISECONDS,
1135                                     new ArrayBlockingQueue<Runnable>(1),
1136 <                                   h);
1137 <        try {
1136 >                                   new ThreadPoolExecutor.CallerRunsPolicy());
1137 >        try (PoolCleaner cleaner = cleaner(p)) {
1138 >            final CountDownLatch done = new CountDownLatch(1);
1139 >            Runnable blocker = new CheckedRunnable() {
1140 >                public void realRun() throws InterruptedException {
1141 >                    await(done);
1142 >                }};
1143 >            p.execute(blocker);
1144              TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1145 <            for (int i = 0; i < tasks.length; ++i)
1145 >            for (int i = 0; i < tasks.length; i++)
1146                  tasks[i] = new TrackedNoOpRunnable();
1147 <            TrackedLongRunnable mr = new TrackedLongRunnable();
1149 <            p.execute(mr);
1150 <            for (int i = 0; i < tasks.length; ++i)
1147 >            for (int i = 0; i < tasks.length; i++)
1148                  p.execute(tasks[i]);
1149 <            for (int i = 1; i < tasks.length; ++i)
1149 >            for (int i = 1; i < tasks.length; i++)
1150                  assertTrue(tasks[i].done);
1151 <            try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1152 <        } finally {
1156 <            joinPool(p);
1151 >            assertFalse(tasks[0].done); // waiting in queue
1152 >            done.countDown();
1153          }
1154      }
1155  
# Line 1161 | Line 1157 | public class ThreadPoolExecutorTest exte
1157       * executor using DiscardPolicy drops task if saturated.
1158       */
1159      public void testSaturatedExecute3() {
1160 <        RejectedExecutionHandler h = new ThreadPoolExecutor.DiscardPolicy();
1160 >        final CountDownLatch done = new CountDownLatch(1);
1161 >        final TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1162 >        for (int i = 0; i < tasks.length; ++i)
1163 >            tasks[i] = new TrackedNoOpRunnable();
1164          final ThreadPoolExecutor p =
1165              new ThreadPoolExecutor(1, 1,
1166 <                                   LONG_DELAY_MS, MILLISECONDS,
1167 <                                   new ArrayBlockingQueue<Runnable>(1),
1168 <                                   h);
1169 <        try {
1170 <            TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1171 <            for (int i = 0; i < tasks.length; ++i)
1173 <                tasks[i] = new TrackedNoOpRunnable();
1174 <            p.execute(new TrackedLongRunnable());
1166 >                          LONG_DELAY_MS, MILLISECONDS,
1167 >                          new ArrayBlockingQueue<Runnable>(1),
1168 >                          new ThreadPoolExecutor.DiscardPolicy());
1169 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1170 >            p.execute(awaiter(done));
1171 >
1172              for (TrackedNoOpRunnable task : tasks)
1173                  p.execute(task);
1174 <            for (TrackedNoOpRunnable task : tasks)
1175 <                assertFalse(task.done);
1179 <            try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1180 <        } finally {
1181 <            joinPool(p);
1174 >            for (int i = 1; i < tasks.length; i++)
1175 >                assertFalse(tasks[i].done);
1176          }
1177 +        for (int i = 1; i < tasks.length; i++)
1178 +            assertFalse(tasks[i].done);
1179 +        assertTrue(tasks[0].done); // was waiting in queue
1180      }
1181  
1182      /**
1183       * executor using DiscardOldestPolicy drops oldest task if saturated.
1184       */
1185      public void testSaturatedExecute4() {
1186 <        RejectedExecutionHandler h = new ThreadPoolExecutor.DiscardOldestPolicy();
1186 >        final CountDownLatch done = new CountDownLatch(1);
1187 >        LatchAwaiter r1 = awaiter(done);
1188 >        LatchAwaiter r2 = awaiter(done);
1189 >        LatchAwaiter r3 = awaiter(done);
1190          final ThreadPoolExecutor p =
1191              new ThreadPoolExecutor(1, 1,
1192                                     LONG_DELAY_MS, MILLISECONDS,
1193                                     new ArrayBlockingQueue<Runnable>(1),
1194 <                                   h);
1195 <        try {
1196 <            p.execute(new TrackedLongRunnable());
1197 <            TrackedLongRunnable r2 = new TrackedLongRunnable();
1194 >                                   new ThreadPoolExecutor.DiscardOldestPolicy());
1195 >        try (PoolCleaner cleaner = cleaner(p, done)) {
1196 >            assertEquals(LatchAwaiter.NEW, r1.state);
1197 >            assertEquals(LatchAwaiter.NEW, r2.state);
1198 >            assertEquals(LatchAwaiter.NEW, r3.state);
1199 >            p.execute(r1);
1200              p.execute(r2);
1201              assertTrue(p.getQueue().contains(r2));
1200            TrackedNoOpRunnable r3 = new TrackedNoOpRunnable();
1202              p.execute(r3);
1203              assertFalse(p.getQueue().contains(r2));
1204              assertTrue(p.getQueue().contains(r3));
1204            try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1205        } finally {
1206            joinPool(p);
1205          }
1206 +        assertEquals(LatchAwaiter.DONE, r1.state);
1207 +        assertEquals(LatchAwaiter.NEW, r2.state);
1208 +        assertEquals(LatchAwaiter.DONE, r3.state);
1209      }
1210  
1211      /**
1212       * execute throws RejectedExecutionException if shutdown
1213       */
1214      public void testRejectedExecutionExceptionOnShutdown() {
1215 <        ThreadPoolExecutor p =
1215 >        final ThreadPoolExecutor p =
1216              new ThreadPoolExecutor(1, 1,
1217                                     LONG_DELAY_MS, MILLISECONDS,
1218                                     new ArrayBlockingQueue<Runnable>(1));
1219          try { p.shutdown(); } catch (SecurityException ok) { return; }
1220 <        try {
1221 <            p.execute(new NoOpRunnable());
1222 <            shouldThrow();
1223 <        } catch (RejectedExecutionException success) {}
1224 <
1225 <        joinPool(p);
1220 >        try (PoolCleaner cleaner = cleaner(p)) {
1221 >            try {
1222 >                p.execute(new NoOpRunnable());
1223 >                shouldThrow();
1224 >            } catch (RejectedExecutionException success) {}
1225 >        }
1226      }
1227  
1228      /**
# Line 1235 | Line 1236 | public class ThreadPoolExecutorTest exte
1236                                     new ArrayBlockingQueue<Runnable>(1), h);
1237  
1238          try { p.shutdown(); } catch (SecurityException ok) { return; }
1239 <        try {
1239 >        try (PoolCleaner cleaner = cleaner(p)) {
1240              TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1241              p.execute(r);
1242              assertFalse(r.done);
1242        } finally {
1243            joinPool(p);
1243          }
1244      }
1245  
# Line 1248 | Line 1247 | public class ThreadPoolExecutorTest exte
1247       * execute using DiscardPolicy drops task on shutdown
1248       */
1249      public void testDiscardOnShutdown() {
1250 <        RejectedExecutionHandler h = new ThreadPoolExecutor.DiscardPolicy();
1252 <        ThreadPoolExecutor p =
1250 >        final ThreadPoolExecutor p =
1251              new ThreadPoolExecutor(1, 1,
1252                                     LONG_DELAY_MS, MILLISECONDS,
1253                                     new ArrayBlockingQueue<Runnable>(1),
1254 <                                   h);
1254 >                                   new ThreadPoolExecutor.DiscardPolicy());
1255  
1256          try { p.shutdown(); } catch (SecurityException ok) { return; }
1257 <        try {
1257 >        try (PoolCleaner cleaner = cleaner(p)) {
1258              TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1259              p.execute(r);
1260              assertFalse(r.done);
1263        } finally {
1264            joinPool(p);
1261          }
1262      }
1263  
# Line 1269 | Line 1265 | public class ThreadPoolExecutorTest exte
1265       * execute using DiscardOldestPolicy drops task on shutdown
1266       */
1267      public void testDiscardOldestOnShutdown() {
1268 <        RejectedExecutionHandler h = new ThreadPoolExecutor.DiscardOldestPolicy();
1273 <        ThreadPoolExecutor p =
1268 >        final ThreadPoolExecutor p =
1269              new ThreadPoolExecutor(1, 1,
1270                                     LONG_DELAY_MS, MILLISECONDS,
1271                                     new ArrayBlockingQueue<Runnable>(1),
1272 <                                   h);
1272 >                                   new ThreadPoolExecutor.DiscardOldestPolicy());
1273  
1274          try { p.shutdown(); } catch (SecurityException ok) { return; }
1275 <        try {
1275 >        try (PoolCleaner cleaner = cleaner(p)) {
1276              TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1277              p.execute(r);
1278              assertFalse(r.done);
1284        } finally {
1285            joinPool(p);
1279          }
1280      }
1281  
# Line 1290 | Line 1283 | public class ThreadPoolExecutorTest exte
1283       * execute(null) throws NPE
1284       */
1285      public void testExecuteNull() {
1286 <        ThreadPoolExecutor p =
1286 >        final ThreadPoolExecutor p =
1287              new ThreadPoolExecutor(1, 2,
1288 <                                   LONG_DELAY_MS, MILLISECONDS,
1288 >                                   1L, SECONDS,
1289                                     new ArrayBlockingQueue<Runnable>(10));
1290 <        try {
1291 <            p.execute(null);
1292 <            shouldThrow();
1293 <        } catch (NullPointerException success) {}
1294 <
1295 <        joinPool(p);
1290 >        try (PoolCleaner cleaner = cleaner(p)) {
1291 >            try {
1292 >                p.execute(null);
1293 >                shouldThrow();
1294 >            } catch (NullPointerException success) {}
1295 >        }
1296      }
1297  
1298      /**
1299       * setCorePoolSize of negative value throws IllegalArgumentException
1300       */
1301      public void testCorePoolSizeIllegalArgumentException() {
1302 <        ThreadPoolExecutor p =
1302 >        final ThreadPoolExecutor p =
1303              new ThreadPoolExecutor(1, 2,
1304                                     LONG_DELAY_MS, MILLISECONDS,
1305                                     new ArrayBlockingQueue<Runnable>(10));
1306 <        try {
1307 <            p.setCorePoolSize(-1);
1308 <            shouldThrow();
1309 <        } catch (IllegalArgumentException success) {
1310 <        } finally {
1318 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
1306 >        try (PoolCleaner cleaner = cleaner(p)) {
1307 >            try {
1308 >                p.setCorePoolSize(-1);
1309 >                shouldThrow();
1310 >            } catch (IllegalArgumentException success) {}
1311          }
1320        joinPool(p);
1312      }
1313  
1314      /**
# Line 1325 | Line 1316 | public class ThreadPoolExecutorTest exte
1316       * given a value less the core pool size
1317       */
1318      public void testMaximumPoolSizeIllegalArgumentException() {
1319 <        ThreadPoolExecutor p =
1319 >        final ThreadPoolExecutor p =
1320              new ThreadPoolExecutor(2, 3,
1321                                     LONG_DELAY_MS, MILLISECONDS,
1322                                     new ArrayBlockingQueue<Runnable>(10));
1323 <        try {
1324 <            p.setMaximumPoolSize(1);
1325 <            shouldThrow();
1326 <        } catch (IllegalArgumentException success) {
1327 <        } finally {
1337 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
1323 >        try (PoolCleaner cleaner = cleaner(p)) {
1324 >            try {
1325 >                p.setMaximumPoolSize(1);
1326 >                shouldThrow();
1327 >            } catch (IllegalArgumentException success) {}
1328          }
1339        joinPool(p);
1329      }
1330  
1331      /**
# Line 1344 | Line 1333 | public class ThreadPoolExecutorTest exte
1333       * if given a negative value
1334       */
1335      public void testMaximumPoolSizeIllegalArgumentException2() {
1336 <        ThreadPoolExecutor p =
1336 >        final ThreadPoolExecutor p =
1337              new ThreadPoolExecutor(2, 3,
1338                                     LONG_DELAY_MS, MILLISECONDS,
1339                                     new ArrayBlockingQueue<Runnable>(10));
1340 <        try {
1341 <            p.setMaximumPoolSize(-1);
1342 <            shouldThrow();
1343 <        } catch (IllegalArgumentException success) {
1344 <        } finally {
1345 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
1340 >        try (PoolCleaner cleaner = cleaner(p)) {
1341 >            try {
1342 >                p.setMaximumPoolSize(-1);
1343 >                shouldThrow();
1344 >            } catch (IllegalArgumentException success) {}
1345 >        }
1346 >    }
1347 >
1348 >    /**
1349 >     * Configuration changes that allow core pool size greater than
1350 >     * max pool size result in IllegalArgumentException.
1351 >     */
1352 >    public void testPoolSizeInvariants() {
1353 >        final ThreadPoolExecutor p =
1354 >            new ThreadPoolExecutor(1, 1,
1355 >                                   LONG_DELAY_MS, MILLISECONDS,
1356 >                                   new ArrayBlockingQueue<Runnable>(10));
1357 >        try (PoolCleaner cleaner = cleaner(p)) {
1358 >            for (int s = 1; s < 5; s++) {
1359 >                p.setMaximumPoolSize(s);
1360 >                p.setCorePoolSize(s);
1361 >                try {
1362 >                    p.setMaximumPoolSize(s - 1);
1363 >                    shouldThrow();
1364 >                } catch (IllegalArgumentException success) {}
1365 >                assertEquals(s, p.getCorePoolSize());
1366 >                assertEquals(s, p.getMaximumPoolSize());
1367 >                try {
1368 >                    p.setCorePoolSize(s + 1);
1369 >                    shouldThrow();
1370 >                } catch (IllegalArgumentException success) {}
1371 >                assertEquals(s, p.getCorePoolSize());
1372 >                assertEquals(s, p.getMaximumPoolSize());
1373 >            }
1374          }
1358        joinPool(p);
1375      }
1376  
1377      /**
# Line 1363 | Line 1379 | public class ThreadPoolExecutorTest exte
1379       * when given a negative value
1380       */
1381      public void testKeepAliveTimeIllegalArgumentException() {
1382 <        ThreadPoolExecutor p =
1382 >        final ThreadPoolExecutor p =
1383              new ThreadPoolExecutor(2, 3,
1384                                     LONG_DELAY_MS, MILLISECONDS,
1385                                     new ArrayBlockingQueue<Runnable>(10));
1386 <        try {
1387 <            p.setKeepAliveTime(-1,MILLISECONDS);
1388 <            shouldThrow();
1389 <        } catch (IllegalArgumentException success) {
1390 <        } finally {
1375 <            try { p.shutdown(); } catch (SecurityException ok) { return; }
1386 >        try (PoolCleaner cleaner = cleaner(p)) {
1387 >            try {
1388 >                p.setKeepAliveTime(-1, MILLISECONDS);
1389 >                shouldThrow();
1390 >            } catch (IllegalArgumentException success) {}
1391          }
1377        joinPool(p);
1392      }
1393  
1394      /**
# Line 1382 | Line 1396 | public class ThreadPoolExecutorTest exte
1396       */
1397      public void testTerminated() {
1398          ExtendedTPE p = new ExtendedTPE();
1399 <        try { p.shutdown(); } catch (SecurityException ok) { return; }
1400 <        assertTrue(p.terminatedCalled());
1401 <        joinPool(p);
1399 >        try (PoolCleaner cleaner = cleaner(p)) {
1400 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
1401 >            assertTrue(p.terminatedCalled());
1402 >            assertTrue(p.isShutdown());
1403 >        }
1404      }
1405  
1406      /**
# Line 1392 | Line 1408 | public class ThreadPoolExecutorTest exte
1408       */
1409      public void testBeforeAfter() throws InterruptedException {
1410          ExtendedTPE p = new ExtendedTPE();
1411 <        try {
1411 >        try (PoolCleaner cleaner = cleaner(p)) {
1412              final CountDownLatch done = new CountDownLatch(1);
1413              p.execute(new CheckedRunnable() {
1414                  public void realRun() {
# Line 1402 | Line 1418 | public class ThreadPoolExecutorTest exte
1418              assertEquals(0, done.getCount());
1419              assertTrue(p.afterCalled());
1420              assertTrue(p.beforeCalled());
1405            try { p.shutdown(); } catch (SecurityException ok) { return; }
1406        } finally {
1407            joinPool(p);
1421          }
1422      }
1423  
# Line 1412 | Line 1425 | public class ThreadPoolExecutorTest exte
1425       * completed submit of callable returns result
1426       */
1427      public void testSubmitCallable() throws Exception {
1428 <        ExecutorService e =
1428 >        final ExecutorService e =
1429              new ThreadPoolExecutor(2, 2,
1430                                     LONG_DELAY_MS, MILLISECONDS,
1431                                     new ArrayBlockingQueue<Runnable>(10));
1432 <        try {
1432 >        try (PoolCleaner cleaner = cleaner(e)) {
1433              Future<String> future = e.submit(new StringTask());
1434              String result = future.get();
1435              assertSame(TEST_STRING, result);
1423        } finally {
1424            joinPool(e);
1436          }
1437      }
1438  
# Line 1429 | Line 1440 | public class ThreadPoolExecutorTest exte
1440       * completed submit of runnable returns successfully
1441       */
1442      public void testSubmitRunnable() throws Exception {
1443 <        ExecutorService e =
1443 >        final ExecutorService e =
1444              new ThreadPoolExecutor(2, 2,
1445                                     LONG_DELAY_MS, MILLISECONDS,
1446                                     new ArrayBlockingQueue<Runnable>(10));
1447 <        try {
1447 >        try (PoolCleaner cleaner = cleaner(e)) {
1448              Future<?> future = e.submit(new NoOpRunnable());
1449              future.get();
1450              assertTrue(future.isDone());
1440        } finally {
1441            joinPool(e);
1451          }
1452      }
1453  
# Line 1446 | Line 1455 | public class ThreadPoolExecutorTest exte
1455       * completed submit of (runnable, result) returns result
1456       */
1457      public void testSubmitRunnable2() throws Exception {
1458 <        ExecutorService e =
1458 >        final ExecutorService e =
1459              new ThreadPoolExecutor(2, 2,
1460                                     LONG_DELAY_MS, MILLISECONDS,
1461                                     new ArrayBlockingQueue<Runnable>(10));
1462 <        try {
1462 >        try (PoolCleaner cleaner = cleaner(e)) {
1463              Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
1464              String result = future.get();
1465              assertSame(TEST_STRING, result);
1457        } finally {
1458            joinPool(e);
1466          }
1467      }
1468  
# Line 1463 | Line 1470 | public class ThreadPoolExecutorTest exte
1470       * invokeAny(null) throws NPE
1471       */
1472      public void testInvokeAny1() throws Exception {
1473 <        ExecutorService e =
1473 >        final ExecutorService e =
1474              new ThreadPoolExecutor(2, 2,
1475                                     LONG_DELAY_MS, MILLISECONDS,
1476                                     new ArrayBlockingQueue<Runnable>(10));
1477 <        try {
1478 <            e.invokeAny(null);
1479 <            shouldThrow();
1480 <        } catch (NullPointerException success) {
1481 <        } finally {
1475 <            joinPool(e);
1477 >        try (PoolCleaner cleaner = cleaner(e)) {
1478 >            try {
1479 >                e.invokeAny(null);
1480 >                shouldThrow();
1481 >            } catch (NullPointerException success) {}
1482          }
1483      }
1484  
# Line 1480 | Line 1486 | public class ThreadPoolExecutorTest exte
1486       * invokeAny(empty collection) throws IAE
1487       */
1488      public void testInvokeAny2() throws Exception {
1489 <        ExecutorService e =
1489 >        final ExecutorService e =
1490              new ThreadPoolExecutor(2, 2,
1491                                     LONG_DELAY_MS, MILLISECONDS,
1492                                     new ArrayBlockingQueue<Runnable>(10));
1493 <        try {
1494 <            e.invokeAny(new ArrayList<Callable<String>>());
1495 <            shouldThrow();
1496 <        } catch (IllegalArgumentException success) {
1497 <        } finally {
1492 <            joinPool(e);
1493 >        try (PoolCleaner cleaner = cleaner(e)) {
1494 >            try {
1495 >                e.invokeAny(new ArrayList<Callable<String>>());
1496 >                shouldThrow();
1497 >            } catch (IllegalArgumentException success) {}
1498          }
1499      }
1500  
# Line 1502 | Line 1507 | public class ThreadPoolExecutorTest exte
1507              new ThreadPoolExecutor(2, 2,
1508                                     LONG_DELAY_MS, MILLISECONDS,
1509                                     new ArrayBlockingQueue<Runnable>(10));
1510 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1511 <        l.add(latchAwaitingStringTask(latch));
1512 <        l.add(null);
1513 <        try {
1514 <            e.invokeAny(l);
1515 <            shouldThrow();
1516 <        } catch (NullPointerException success) {
1517 <        } finally {
1510 >        try (PoolCleaner cleaner = cleaner(e)) {
1511 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1512 >            l.add(latchAwaitingStringTask(latch));
1513 >            l.add(null);
1514 >            try {
1515 >                e.invokeAny(l);
1516 >                shouldThrow();
1517 >            } catch (NullPointerException success) {}
1518              latch.countDown();
1514            joinPool(e);
1519          }
1520      }
1521  
# Line 1519 | Line 1523 | public class ThreadPoolExecutorTest exte
1523       * invokeAny(c) throws ExecutionException if no task completes
1524       */
1525      public void testInvokeAny4() throws Exception {
1526 <        ExecutorService e =
1526 >        final ExecutorService e =
1527              new ThreadPoolExecutor(2, 2,
1528                                     LONG_DELAY_MS, MILLISECONDS,
1529                                     new ArrayBlockingQueue<Runnable>(10));
1530 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1531 <        l.add(new NPETask());
1532 <        try {
1533 <            e.invokeAny(l);
1534 <            shouldThrow();
1535 <        } catch (ExecutionException success) {
1536 <            assertTrue(success.getCause() instanceof NullPointerException);
1537 <        } finally {
1538 <            joinPool(e);
1530 >        try (PoolCleaner cleaner = cleaner(e)) {
1531 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1532 >            l.add(new NPETask());
1533 >            try {
1534 >                e.invokeAny(l);
1535 >                shouldThrow();
1536 >            } catch (ExecutionException success) {
1537 >                assertTrue(success.getCause() instanceof NullPointerException);
1538 >            }
1539          }
1540      }
1541  
# Line 1539 | Line 1543 | public class ThreadPoolExecutorTest exte
1543       * invokeAny(c) returns result of some task
1544       */
1545      public void testInvokeAny5() throws Exception {
1546 <        ExecutorService e =
1546 >        final ExecutorService e =
1547              new ThreadPoolExecutor(2, 2,
1548                                     LONG_DELAY_MS, MILLISECONDS,
1549                                     new ArrayBlockingQueue<Runnable>(10));
1550 <        try {
1550 >        try (PoolCleaner cleaner = cleaner(e)) {
1551              List<Callable<String>> l = new ArrayList<Callable<String>>();
1552              l.add(new StringTask());
1553              l.add(new StringTask());
1554              String result = e.invokeAny(l);
1555              assertSame(TEST_STRING, result);
1552        } finally {
1553            joinPool(e);
1556          }
1557      }
1558  
# Line 1558 | Line 1560 | public class ThreadPoolExecutorTest exte
1560       * invokeAll(null) throws NPE
1561       */
1562      public void testInvokeAll1() throws Exception {
1563 <        ExecutorService e =
1563 >        final ExecutorService e =
1564              new ThreadPoolExecutor(2, 2,
1565                                     LONG_DELAY_MS, MILLISECONDS,
1566                                     new ArrayBlockingQueue<Runnable>(10));
1567 <        try {
1568 <            e.invokeAll(null);
1569 <            shouldThrow();
1570 <        } catch (NullPointerException success) {
1571 <        } finally {
1570 <            joinPool(e);
1567 >        try (PoolCleaner cleaner = cleaner(e)) {
1568 >            try {
1569 >                e.invokeAll(null);
1570 >                shouldThrow();
1571 >            } catch (NullPointerException success) {}
1572          }
1573      }
1574  
# Line 1575 | Line 1576 | public class ThreadPoolExecutorTest exte
1576       * invokeAll(empty collection) returns empty collection
1577       */
1578      public void testInvokeAll2() throws InterruptedException {
1579 <        ExecutorService e =
1579 >        final ExecutorService e =
1580              new ThreadPoolExecutor(2, 2,
1581                                     LONG_DELAY_MS, MILLISECONDS,
1582                                     new ArrayBlockingQueue<Runnable>(10));
1583 <        try {
1583 >        try (PoolCleaner cleaner = cleaner(e)) {
1584              List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
1585              assertTrue(r.isEmpty());
1585        } finally {
1586            joinPool(e);
1586          }
1587      }
1588  
# Line 1591 | Line 1590 | public class ThreadPoolExecutorTest exte
1590       * invokeAll(c) throws NPE if c has null elements
1591       */
1592      public void testInvokeAll3() throws Exception {
1593 <        ExecutorService e =
1593 >        final ExecutorService e =
1594              new ThreadPoolExecutor(2, 2,
1595                                     LONG_DELAY_MS, MILLISECONDS,
1596                                     new ArrayBlockingQueue<Runnable>(10));
1597 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1598 <        l.add(new StringTask());
1599 <        l.add(null);
1600 <        try {
1601 <            e.invokeAll(l);
1602 <            shouldThrow();
1603 <        } catch (NullPointerException success) {
1604 <        } finally {
1606 <            joinPool(e);
1597 >        try (PoolCleaner cleaner = cleaner(e)) {
1598 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1599 >            l.add(new StringTask());
1600 >            l.add(null);
1601 >            try {
1602 >                e.invokeAll(l);
1603 >                shouldThrow();
1604 >            } catch (NullPointerException success) {}
1605          }
1606      }
1607  
# Line 1611 | Line 1609 | public class ThreadPoolExecutorTest exte
1609       * get of element of invokeAll(c) throws exception on failed task
1610       */
1611      public void testInvokeAll4() throws Exception {
1612 <        ExecutorService e =
1612 >        final ExecutorService e =
1613              new ThreadPoolExecutor(2, 2,
1614                                     LONG_DELAY_MS, MILLISECONDS,
1615                                     new ArrayBlockingQueue<Runnable>(10));
1616 <        try {
1616 >        try (PoolCleaner cleaner = cleaner(e)) {
1617              List<Callable<String>> l = new ArrayList<Callable<String>>();
1618              l.add(new NPETask());
1619              List<Future<String>> futures = e.invokeAll(l);
# Line 1626 | Line 1624 | public class ThreadPoolExecutorTest exte
1624              } catch (ExecutionException success) {
1625                  assertTrue(success.getCause() instanceof NullPointerException);
1626              }
1629        } finally {
1630            joinPool(e);
1627          }
1628      }
1629  
# Line 1635 | Line 1631 | public class ThreadPoolExecutorTest exte
1631       * invokeAll(c) returns results of all completed tasks
1632       */
1633      public void testInvokeAll5() throws Exception {
1634 <        ExecutorService e =
1634 >        final ExecutorService e =
1635              new ThreadPoolExecutor(2, 2,
1636                                     LONG_DELAY_MS, MILLISECONDS,
1637                                     new ArrayBlockingQueue<Runnable>(10));
1638 <        try {
1638 >        try (PoolCleaner cleaner = cleaner(e)) {
1639              List<Callable<String>> l = new ArrayList<Callable<String>>();
1640              l.add(new StringTask());
1641              l.add(new StringTask());
# Line 1647 | Line 1643 | public class ThreadPoolExecutorTest exte
1643              assertEquals(2, futures.size());
1644              for (Future<String> future : futures)
1645                  assertSame(TEST_STRING, future.get());
1650        } finally {
1651            joinPool(e);
1646          }
1647      }
1648  
# Line 1656 | Line 1650 | public class ThreadPoolExecutorTest exte
1650       * timed invokeAny(null) throws NPE
1651       */
1652      public void testTimedInvokeAny1() throws Exception {
1653 <        ExecutorService e =
1653 >        final ExecutorService e =
1654              new ThreadPoolExecutor(2, 2,
1655                                     LONG_DELAY_MS, MILLISECONDS,
1656                                     new ArrayBlockingQueue<Runnable>(10));
1657 <        try {
1658 <            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1659 <            shouldThrow();
1660 <        } catch (NullPointerException success) {
1661 <        } finally {
1668 <            joinPool(e);
1657 >        try (PoolCleaner cleaner = cleaner(e)) {
1658 >            try {
1659 >                e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1660 >                shouldThrow();
1661 >            } catch (NullPointerException success) {}
1662          }
1663      }
1664  
# Line 1673 | Line 1666 | public class ThreadPoolExecutorTest exte
1666       * timed invokeAny(,,null) throws NPE
1667       */
1668      public void testTimedInvokeAnyNullTimeUnit() throws Exception {
1669 <        ExecutorService e =
1669 >        final ExecutorService e =
1670              new ThreadPoolExecutor(2, 2,
1671                                     LONG_DELAY_MS, MILLISECONDS,
1672                                     new ArrayBlockingQueue<Runnable>(10));
1673 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1674 <        l.add(new StringTask());
1675 <        try {
1676 <            e.invokeAny(l, MEDIUM_DELAY_MS, null);
1677 <            shouldThrow();
1678 <        } catch (NullPointerException success) {
1679 <        } finally {
1687 <            joinPool(e);
1673 >        try (PoolCleaner cleaner = cleaner(e)) {
1674 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1675 >            l.add(new StringTask());
1676 >            try {
1677 >                e.invokeAny(l, MEDIUM_DELAY_MS, null);
1678 >                shouldThrow();
1679 >            } catch (NullPointerException success) {}
1680          }
1681      }
1682  
# Line 1692 | Line 1684 | public class ThreadPoolExecutorTest exte
1684       * timed invokeAny(empty collection) throws IAE
1685       */
1686      public void testTimedInvokeAny2() throws Exception {
1687 <        ExecutorService e =
1687 >        final ExecutorService e =
1688              new ThreadPoolExecutor(2, 2,
1689                                     LONG_DELAY_MS, MILLISECONDS,
1690                                     new ArrayBlockingQueue<Runnable>(10));
1691 <        try {
1692 <            e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1693 <            shouldThrow();
1694 <        } catch (IllegalArgumentException success) {
1695 <        } finally {
1696 <            joinPool(e);
1691 >        try (PoolCleaner cleaner = cleaner(e)) {
1692 >            try {
1693 >                e.invokeAny(new ArrayList<Callable<String>>(),
1694 >                            MEDIUM_DELAY_MS, MILLISECONDS);
1695 >                shouldThrow();
1696 >            } catch (IllegalArgumentException success) {}
1697          }
1698      }
1699  
# Line 1714 | Line 1706 | public class ThreadPoolExecutorTest exte
1706              new ThreadPoolExecutor(2, 2,
1707                                     LONG_DELAY_MS, MILLISECONDS,
1708                                     new ArrayBlockingQueue<Runnable>(10));
1709 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1710 <        l.add(latchAwaitingStringTask(latch));
1711 <        l.add(null);
1712 <        try {
1713 <            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1714 <            shouldThrow();
1715 <        } catch (NullPointerException success) {
1716 <        } finally {
1709 >        try (PoolCleaner cleaner = cleaner(e)) {
1710 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1711 >            l.add(latchAwaitingStringTask(latch));
1712 >            l.add(null);
1713 >            try {
1714 >                e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1715 >                shouldThrow();
1716 >            } catch (NullPointerException success) {}
1717              latch.countDown();
1726            joinPool(e);
1718          }
1719      }
1720  
# Line 1731 | Line 1722 | public class ThreadPoolExecutorTest exte
1722       * timed invokeAny(c) throws ExecutionException if no task completes
1723       */
1724      public void testTimedInvokeAny4() throws Exception {
1725 <        ExecutorService e =
1725 >        final ExecutorService e =
1726              new ThreadPoolExecutor(2, 2,
1727                                     LONG_DELAY_MS, MILLISECONDS,
1728                                     new ArrayBlockingQueue<Runnable>(10));
1729 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1730 <        l.add(new NPETask());
1731 <        try {
1732 <            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1733 <            shouldThrow();
1734 <        } catch (ExecutionException success) {
1735 <            assertTrue(success.getCause() instanceof NullPointerException);
1736 <        } finally {
1737 <            joinPool(e);
1729 >        try (PoolCleaner cleaner = cleaner(e)) {
1730 >            long startTime = System.nanoTime();
1731 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1732 >            l.add(new NPETask());
1733 >            try {
1734 >                e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1735 >                shouldThrow();
1736 >            } catch (ExecutionException success) {
1737 >                assertTrue(success.getCause() instanceof NullPointerException);
1738 >            }
1739 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1740          }
1741      }
1742  
# Line 1751 | Line 1744 | public class ThreadPoolExecutorTest exte
1744       * timed invokeAny(c) returns result of some task
1745       */
1746      public void testTimedInvokeAny5() throws Exception {
1747 <        ExecutorService e =
1747 >        final ExecutorService e =
1748              new ThreadPoolExecutor(2, 2,
1749                                     LONG_DELAY_MS, MILLISECONDS,
1750                                     new ArrayBlockingQueue<Runnable>(10));
1751 <        try {
1751 >        try (PoolCleaner cleaner = cleaner(e)) {
1752 >            long startTime = System.nanoTime();
1753              List<Callable<String>> l = new ArrayList<Callable<String>>();
1754              l.add(new StringTask());
1755              l.add(new StringTask());
1756 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1756 >            String result = e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1757              assertSame(TEST_STRING, result);
1758 <        } finally {
1765 <            joinPool(e);
1758 >            assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1759          }
1760      }
1761  
# Line 1770 | Line 1763 | public class ThreadPoolExecutorTest exte
1763       * timed invokeAll(null) throws NPE
1764       */
1765      public void testTimedInvokeAll1() throws Exception {
1766 <        ExecutorService e =
1766 >        final ExecutorService e =
1767              new ThreadPoolExecutor(2, 2,
1768                                     LONG_DELAY_MS, MILLISECONDS,
1769                                     new ArrayBlockingQueue<Runnable>(10));
1770 <        try {
1771 <            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1772 <            shouldThrow();
1773 <        } catch (NullPointerException success) {
1774 <        } finally {
1782 <            joinPool(e);
1770 >        try (PoolCleaner cleaner = cleaner(e)) {
1771 >            try {
1772 >                e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1773 >                shouldThrow();
1774 >            } catch (NullPointerException success) {}
1775          }
1776      }
1777  
# Line 1787 | Line 1779 | public class ThreadPoolExecutorTest exte
1779       * timed invokeAll(,,null) throws NPE
1780       */
1781      public void testTimedInvokeAllNullTimeUnit() throws Exception {
1782 <        ExecutorService e =
1782 >        final ExecutorService e =
1783              new ThreadPoolExecutor(2, 2,
1784                                     LONG_DELAY_MS, MILLISECONDS,
1785                                     new ArrayBlockingQueue<Runnable>(10));
1786 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1787 <        l.add(new StringTask());
1788 <        try {
1789 <            e.invokeAll(l, MEDIUM_DELAY_MS, null);
1790 <            shouldThrow();
1791 <        } catch (NullPointerException success) {
1792 <        } finally {
1801 <            joinPool(e);
1786 >        try (PoolCleaner cleaner = cleaner(e)) {
1787 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1788 >            l.add(new StringTask());
1789 >            try {
1790 >                e.invokeAll(l, MEDIUM_DELAY_MS, null);
1791 >                shouldThrow();
1792 >            } catch (NullPointerException success) {}
1793          }
1794      }
1795  
# Line 1806 | Line 1797 | public class ThreadPoolExecutorTest exte
1797       * timed invokeAll(empty collection) returns empty collection
1798       */
1799      public void testTimedInvokeAll2() throws InterruptedException {
1800 <        ExecutorService e =
1800 >        final ExecutorService e =
1801              new ThreadPoolExecutor(2, 2,
1802                                     LONG_DELAY_MS, MILLISECONDS,
1803                                     new ArrayBlockingQueue<Runnable>(10));
1804 <        try {
1805 <            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1804 >        try (PoolCleaner cleaner = cleaner(e)) {
1805 >            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(),
1806 >                                                 MEDIUM_DELAY_MS, MILLISECONDS);
1807              assertTrue(r.isEmpty());
1816        } finally {
1817            joinPool(e);
1808          }
1809      }
1810  
# Line 1822 | Line 1812 | public class ThreadPoolExecutorTest exte
1812       * timed invokeAll(c) throws NPE if c has null elements
1813       */
1814      public void testTimedInvokeAll3() throws Exception {
1815 <        ExecutorService e =
1815 >        final ExecutorService e =
1816              new ThreadPoolExecutor(2, 2,
1817                                     LONG_DELAY_MS, MILLISECONDS,
1818                                     new ArrayBlockingQueue<Runnable>(10));
1819 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1820 <        l.add(new StringTask());
1821 <        l.add(null);
1822 <        try {
1823 <            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1824 <            shouldThrow();
1825 <        } catch (NullPointerException success) {
1826 <        } finally {
1837 <            joinPool(e);
1819 >        try (PoolCleaner cleaner = cleaner(e)) {
1820 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1821 >            l.add(new StringTask());
1822 >            l.add(null);
1823 >            try {
1824 >                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1825 >                shouldThrow();
1826 >            } catch (NullPointerException success) {}
1827          }
1828      }
1829  
# Line 1842 | Line 1831 | public class ThreadPoolExecutorTest exte
1831       * get of element of invokeAll(c) throws exception on failed task
1832       */
1833      public void testTimedInvokeAll4() throws Exception {
1834 <        ExecutorService e =
1834 >        final ExecutorService e =
1835              new ThreadPoolExecutor(2, 2,
1836                                     LONG_DELAY_MS, MILLISECONDS,
1837                                     new ArrayBlockingQueue<Runnable>(10));
1838 <        List<Callable<String>> l = new ArrayList<Callable<String>>();
1839 <        l.add(new NPETask());
1840 <        List<Future<String>> futures =
1841 <            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1842 <        assertEquals(1, futures.size());
1843 <        try {
1844 <            futures.get(0).get();
1845 <            shouldThrow();
1846 <        } catch (ExecutionException success) {
1847 <            assertTrue(success.getCause() instanceof NullPointerException);
1848 <        } finally {
1849 <            joinPool(e);
1838 >        try (PoolCleaner cleaner = cleaner(e)) {
1839 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1840 >            l.add(new NPETask());
1841 >            List<Future<String>> futures =
1842 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1843 >            assertEquals(1, futures.size());
1844 >            try {
1845 >                futures.get(0).get();
1846 >                shouldThrow();
1847 >            } catch (ExecutionException success) {
1848 >                assertTrue(success.getCause() instanceof NullPointerException);
1849 >            }
1850          }
1851      }
1852  
# Line 1865 | Line 1854 | public class ThreadPoolExecutorTest exte
1854       * timed invokeAll(c) returns results of all completed tasks
1855       */
1856      public void testTimedInvokeAll5() throws Exception {
1857 <        ExecutorService e =
1857 >        final ExecutorService e =
1858              new ThreadPoolExecutor(2, 2,
1859                                     LONG_DELAY_MS, MILLISECONDS,
1860                                     new ArrayBlockingQueue<Runnable>(10));
1861 <        try {
1861 >        try (PoolCleaner cleaner = cleaner(e)) {
1862              List<Callable<String>> l = new ArrayList<Callable<String>>();
1863              l.add(new StringTask());
1864              l.add(new StringTask());
1865              List<Future<String>> futures =
1866 <                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1866 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1867              assertEquals(2, futures.size());
1868              for (Future<String> future : futures)
1869                  assertSame(TEST_STRING, future.get());
1881        } finally {
1882            joinPool(e);
1870          }
1871      }
1872  
# Line 1887 | Line 1874 | public class ThreadPoolExecutorTest exte
1874       * timed invokeAll(c) cancels tasks not completed by timeout
1875       */
1876      public void testTimedInvokeAll6() throws Exception {
1877 <        ExecutorService e =
1878 <            new ThreadPoolExecutor(2, 2,
1879 <                                   LONG_DELAY_MS, MILLISECONDS,
1880 <                                   new ArrayBlockingQueue<Runnable>(10));
1881 <        try {
1882 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
1883 <            l.add(new StringTask());
1884 <            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
1885 <            l.add(new StringTask());
1886 <            List<Future<String>> futures =
1887 <                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
1888 <            assertEquals(l.size(), futures.size());
1889 <            for (Future future : futures)
1890 <                assertTrue(future.isDone());
1891 <            assertFalse(futures.get(0).isCancelled());
1892 <            assertTrue(futures.get(1).isCancelled());
1893 <        } finally {
1894 <            joinPool(e);
1877 >        for (long timeout = timeoutMillis();;) {
1878 >            final CountDownLatch done = new CountDownLatch(1);
1879 >            final Callable<String> waiter = new CheckedCallable<String>() {
1880 >                public String realCall() {
1881 >                    try { done.await(LONG_DELAY_MS, MILLISECONDS); }
1882 >                    catch (InterruptedException ok) {}
1883 >                    return "1"; }};
1884 >            final ExecutorService p =
1885 >                new ThreadPoolExecutor(2, 2,
1886 >                                       LONG_DELAY_MS, MILLISECONDS,
1887 >                                       new ArrayBlockingQueue<Runnable>(10));
1888 >            try (PoolCleaner cleaner = cleaner(p, done)) {
1889 >                List<Callable<String>> tasks = new ArrayList<>();
1890 >                tasks.add(new StringTask("0"));
1891 >                tasks.add(waiter);
1892 >                tasks.add(new StringTask("2"));
1893 >                long startTime = System.nanoTime();
1894 >                List<Future<String>> futures =
1895 >                    p.invokeAll(tasks, timeout, MILLISECONDS);
1896 >                assertEquals(tasks.size(), futures.size());
1897 >                assertTrue(millisElapsedSince(startTime) >= timeout);
1898 >                for (Future future : futures)
1899 >                    assertTrue(future.isDone());
1900 >                assertTrue(futures.get(1).isCancelled());
1901 >                try {
1902 >                    assertEquals("0", futures.get(0).get());
1903 >                    assertEquals("2", futures.get(2).get());
1904 >                    break;
1905 >                } catch (CancellationException retryWithLongerTimeout) {
1906 >                    timeout *= 2;
1907 >                    if (timeout >= LONG_DELAY_MS / 2)
1908 >                        fail("expected exactly one task to be cancelled");
1909 >                }
1910 >            }
1911          }
1912      }
1913  
# Line 1918 | Line 1921 | public class ThreadPoolExecutorTest exte
1921                                     LONG_DELAY_MS, MILLISECONDS,
1922                                     new LinkedBlockingQueue<Runnable>(),
1923                                     new FailingThreadFactory());
1924 <        try {
1924 >        try (PoolCleaner cleaner = cleaner(e)) {
1925              final int TASKS = 100;
1926              final CountDownLatch done = new CountDownLatch(TASKS);
1927              for (int k = 0; k < TASKS; ++k)
# Line 1927 | Line 1930 | public class ThreadPoolExecutorTest exte
1930                          done.countDown();
1931                      }});
1932              assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS));
1930        } finally {
1931            joinPool(e);
1933          }
1934      }
1935  
# Line 1940 | Line 1941 | public class ThreadPoolExecutorTest exte
1941              new ThreadPoolExecutor(2, 2,
1942                                     1000, MILLISECONDS,
1943                                     new ArrayBlockingQueue<Runnable>(10));
1944 <        assertFalse(p.allowsCoreThreadTimeOut());
1945 <        joinPool(p);
1944 >        try (PoolCleaner cleaner = cleaner(p)) {
1945 >            assertFalse(p.allowsCoreThreadTimeOut());
1946 >        }
1947      }
1948  
1949      /**
1950       * allowCoreThreadTimeOut(true) causes idle threads to time out
1951       */
1952      public void testAllowCoreThreadTimeOut_true() throws Exception {
1953 <        long coreThreadTimeOut = SHORT_DELAY_MS;
1953 >        long keepAliveTime = timeoutMillis();
1954          final ThreadPoolExecutor p =
1955              new ThreadPoolExecutor(2, 10,
1956 <                                   coreThreadTimeOut, MILLISECONDS,
1956 >                                   keepAliveTime, MILLISECONDS,
1957                                     new ArrayBlockingQueue<Runnable>(10));
1958 <        final CountDownLatch threadStarted = new CountDownLatch(1);
1959 <        try {
1958 >        try (PoolCleaner cleaner = cleaner(p)) {
1959 >            final CountDownLatch threadStarted = new CountDownLatch(1);
1960              p.allowCoreThreadTimeOut(true);
1961              p.execute(new CheckedRunnable() {
1962                  public void realRun() {
# Line 1962 | Line 1964 | public class ThreadPoolExecutorTest exte
1964                      assertEquals(1, p.getPoolSize());
1965                  }});
1966              await(threadStarted);
1967 <            delay(coreThreadTimeOut);
1967 >            delay(keepAliveTime);
1968              long startTime = System.nanoTime();
1969              while (p.getPoolSize() > 0
1970                     && millisElapsedSince(startTime) < LONG_DELAY_MS)
1971                  Thread.yield();
1972              assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1973              assertEquals(0, p.getPoolSize());
1972        } finally {
1973            joinPool(p);
1974          }
1975      }
1976  
# Line 1978 | Line 1978 | public class ThreadPoolExecutorTest exte
1978       * allowCoreThreadTimeOut(false) causes idle threads not to time out
1979       */
1980      public void testAllowCoreThreadTimeOut_false() throws Exception {
1981 <        long coreThreadTimeOut = SHORT_DELAY_MS;
1981 >        long keepAliveTime = timeoutMillis();
1982          final ThreadPoolExecutor p =
1983              new ThreadPoolExecutor(2, 10,
1984 <                                   coreThreadTimeOut, MILLISECONDS,
1984 >                                   keepAliveTime, MILLISECONDS,
1985                                     new ArrayBlockingQueue<Runnable>(10));
1986 <        final CountDownLatch threadStarted = new CountDownLatch(1);
1987 <        try {
1986 >        try (PoolCleaner cleaner = cleaner(p)) {
1987 >            final CountDownLatch threadStarted = new CountDownLatch(1);
1988              p.allowCoreThreadTimeOut(false);
1989              p.execute(new CheckedRunnable() {
1990                  public void realRun() throws InterruptedException {
1991                      threadStarted.countDown();
1992                      assertTrue(p.getPoolSize() >= 1);
1993                  }});
1994 <            delay(2 * coreThreadTimeOut);
1994 >            delay(2 * keepAliveTime);
1995              assertTrue(p.getPoolSize() >= 1);
1996        } finally {
1997            joinPool(p);
1996          }
1997      }
1998  
# Line 2010 | Line 2008 | public class ThreadPoolExecutorTest exte
2008                  done.countDown();
2009              }};
2010          final ThreadPoolExecutor p =
2011 <            new ThreadPoolExecutor(1, 30, 60, TimeUnit.SECONDS,
2011 >            new ThreadPoolExecutor(1, 30,
2012 >                                   60, SECONDS,
2013                                     new ArrayBlockingQueue(30));
2014 <        try {
2014 >        try (PoolCleaner cleaner = cleaner(p)) {
2015              for (int i = 0; i < nTasks; ++i) {
2016                  for (;;) {
2017                      try {
# Line 2024 | Line 2023 | public class ThreadPoolExecutorTest exte
2023              }
2024              // enough time to run all tasks
2025              assertTrue(done.await(nTasks * SHORT_DELAY_MS, MILLISECONDS));
2026 <        } finally {
2027 <            joinPool(p);
2026 >        }
2027 >    }
2028 >
2029 >    /**
2030 >     * get(cancelled task) throws CancellationException
2031 >     */
2032 >    public void testGet_cancelled() throws Exception {
2033 >        final CountDownLatch done = new CountDownLatch(1);
2034 >        final ExecutorService e =
2035 >            new ThreadPoolExecutor(1, 1,
2036 >                                   LONG_DELAY_MS, MILLISECONDS,
2037 >                                   new LinkedBlockingQueue<Runnable>());
2038 >        try (PoolCleaner cleaner = cleaner(e, done)) {
2039 >            final CountDownLatch blockerStarted = new CountDownLatch(1);
2040 >            final List<Future<?>> futures = new ArrayList<>();
2041 >            for (int i = 0; i < 2; i++) {
2042 >                Runnable r = new CheckedRunnable() { public void realRun()
2043 >                                                         throws Throwable {
2044 >                    blockerStarted.countDown();
2045 >                    assertTrue(done.await(2 * LONG_DELAY_MS, MILLISECONDS));
2046 >                }};
2047 >                futures.add(e.submit(r));
2048 >            }
2049 >            await(blockerStarted);
2050 >            for (Future<?> future : futures) future.cancel(false);
2051 >            for (Future<?> future : futures) {
2052 >                try {
2053 >                    future.get();
2054 >                    shouldThrow();
2055 >                } catch (CancellationException success) {}
2056 >                try {
2057 >                    future.get(LONG_DELAY_MS, MILLISECONDS);
2058 >                    shouldThrow();
2059 >                } catch (CancellationException success) {}
2060 >                assertTrue(future.isCancelled());
2061 >                assertTrue(future.isDone());
2062 >            }
2063          }
2064      }
2065  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines