[cvs] / jsr166 / src / test / tck / ThreadPoolExecutorSubclassTest.java Repository:
ViewVC logotype

Annotation of /jsr166/src/test/tck/ThreadPoolExecutorSubclassTest.java

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.66 - (view) (download)

1 : dl 1.1 /*
2 :     * Written by Doug Lea with assistance from members of JCP JSR-166
3 :     * Expert Group and released to the public domain, as explained at
4 : jsr166 1.25 * http://creativecommons.org/publicdomain/zero/1.0/
5 : jsr166 1.2 * Other contributors include Andrew Wright, Jeffrey Hayes,
6 :     * Pat Fisher, Mike Judd.
7 : dl 1.1 */
8 :    
9 : jsr166 1.9 import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 : jsr166 1.36 import static java.util.concurrent.TimeUnit.SECONDS;
11 : jsr166 1.33
12 :     import java.util.ArrayList;
13 :     import java.util.List;
14 :     import java.util.concurrent.ArrayBlockingQueue;
15 :     import java.util.concurrent.BlockingQueue;
16 :     import java.util.concurrent.Callable;
17 : jsr166 1.39 import java.util.concurrent.CancellationException;
18 : jsr166 1.33 import java.util.concurrent.CountDownLatch;
19 :     import java.util.concurrent.ExecutionException;
20 :     import java.util.concurrent.Executors;
21 :     import java.util.concurrent.ExecutorService;
22 :     import java.util.concurrent.Future;
23 :     import java.util.concurrent.FutureTask;
24 :     import java.util.concurrent.LinkedBlockingQueue;
25 :     import java.util.concurrent.RejectedExecutionException;
26 :     import java.util.concurrent.RejectedExecutionHandler;
27 :     import java.util.concurrent.RunnableFuture;
28 :     import java.util.concurrent.SynchronousQueue;
29 :     import java.util.concurrent.ThreadFactory;
30 :     import java.util.concurrent.ThreadPoolExecutor;
31 :     import java.util.concurrent.TimeoutException;
32 :     import java.util.concurrent.TimeUnit;
33 : jsr166 1.41 import java.util.concurrent.atomic.AtomicInteger;
34 : jsr166 1.31 import java.util.concurrent.locks.Condition;
35 :     import java.util.concurrent.locks.ReentrantLock;
36 : jsr166 1.33
37 :     import junit.framework.Test;
38 :     import junit.framework.TestSuite;
39 : dl 1.1
40 :     public class ThreadPoolExecutorSubclassTest extends JSR166TestCase {
41 :     public static void main(String[] args) {
42 : jsr166 1.35 main(suite(), args);
43 : dl 1.1 }
44 :     public static Test suite() {
45 : jsr166 1.6 return new TestSuite(ThreadPoolExecutorSubclassTest.class);
46 : dl 1.1 }
47 :    
48 :     static class CustomTask<V> implements RunnableFuture<V> {
49 :     final Callable<V> callable;
50 :     final ReentrantLock lock = new ReentrantLock();
51 :     final Condition cond = lock.newCondition();
52 :     boolean done;
53 :     boolean cancelled;
54 :     V result;
55 :     Thread thread;
56 :     Exception exception;
57 : jsr166 1.6 CustomTask(Callable<V> c) {
58 :     if (c == null) throw new NullPointerException();
59 :     callable = c;
60 :     }
61 :     CustomTask(final Runnable r, final V res) {
62 :     if (r == null) throw new NullPointerException();
63 :     callable = new Callable<V>() {
64 : jsr166 1.34 public V call() throws Exception { r.run(); return res; }};
65 : dl 1.1 }
66 :     public boolean isDone() {
67 :     lock.lock(); try { return done; } finally { lock.unlock() ; }
68 :     }
69 :     public boolean isCancelled() {
70 :     lock.lock(); try { return cancelled; } finally { lock.unlock() ; }
71 :     }
72 :     public boolean cancel(boolean mayInterrupt) {
73 :     lock.lock();
74 :     try {
75 :     if (!done) {
76 :     cancelled = true;
77 :     done = true;
78 : jsr166 1.2 if (mayInterrupt && thread != null)
79 : dl 1.1 thread.interrupt();
80 :     return true;
81 :     }
82 :     return false;
83 :     }
84 :     finally { lock.unlock() ; }
85 :     }
86 :     public void run() {
87 :     lock.lock();
88 :     try {
89 : jsr166 1.22 if (done)
90 :     return;
91 :     thread = Thread.currentThread();
92 : dl 1.1 }
93 :     finally { lock.unlock() ; }
94 :     V v = null;
95 :     Exception e = null;
96 :     try {
97 :     v = callable.call();
98 :     }
99 : jsr166 1.3 catch (Exception ex) {
100 : dl 1.1 e = ex;
101 :     }
102 :     lock.lock();
103 :     try {
104 : jsr166 1.44 if (!done) {
105 :     result = v;
106 :     exception = e;
107 :     done = true;
108 :     thread = null;
109 :     cond.signalAll();
110 :     }
111 : dl 1.1 }
112 :     finally { lock.unlock(); }
113 :     }
114 :     public V get() throws InterruptedException, ExecutionException {
115 :     lock.lock();
116 :     try {
117 : jsr166 1.2 while (!done)
118 : dl 1.1 cond.await();
119 : jsr166 1.44 if (cancelled)
120 :     throw new CancellationException();
121 : dl 1.1 if (exception != null)
122 :     throw new ExecutionException(exception);
123 :     return result;
124 :     }
125 :     finally { lock.unlock(); }
126 :     }
127 :     public V get(long timeout, TimeUnit unit)
128 : jsr166 1.4 throws InterruptedException, ExecutionException, TimeoutException {
129 : dl 1.1 long nanos = unit.toNanos(timeout);
130 :     lock.lock();
131 :     try {
132 : jsr166 1.44 while (!done) {
133 : jsr166 1.47 if (nanos <= 0L)
134 : dl 1.1 throw new TimeoutException();
135 :     nanos = cond.awaitNanos(nanos);
136 :     }
137 : jsr166 1.44 if (cancelled)
138 :     throw new CancellationException();
139 : dl 1.1 if (exception != null)
140 :     throw new ExecutionException(exception);
141 :     return result;
142 :     }
143 :     finally { lock.unlock(); }
144 :     }
145 : jsr166 1.2 }
146 :    
147 : dl 1.1 static class CustomTPE extends ThreadPoolExecutor {
148 :     protected <V> RunnableFuture<V> newTaskFor(Callable<V> c) {
149 :     return new CustomTask<V>(c);
150 :     }
151 :     protected <V> RunnableFuture<V> newTaskFor(Runnable r, V v) {
152 :     return new CustomTask<V>(r, v);
153 : jsr166 1.2 }
154 :    
155 : dl 1.1 CustomTPE(int corePoolSize,
156 :     int maximumPoolSize,
157 :     long keepAliveTime,
158 :     TimeUnit unit,
159 :     BlockingQueue<Runnable> workQueue) {
160 : jsr166 1.2 super(corePoolSize, maximumPoolSize, keepAliveTime, unit,
161 : dl 1.1 workQueue);
162 :     }
163 :     CustomTPE(int corePoolSize,
164 :     int maximumPoolSize,
165 :     long keepAliveTime,
166 :     TimeUnit unit,
167 :     BlockingQueue<Runnable> workQueue,
168 :     ThreadFactory threadFactory) {
169 :     super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue,
170 :     threadFactory);
171 :     }
172 :    
173 :     CustomTPE(int corePoolSize,
174 :     int maximumPoolSize,
175 :     long keepAliveTime,
176 :     TimeUnit unit,
177 :     BlockingQueue<Runnable> workQueue,
178 :     RejectedExecutionHandler handler) {
179 :     super(corePoolSize, maximumPoolSize, keepAliveTime, unit, workQueue,
180 :     handler);
181 :     }
182 :     CustomTPE(int corePoolSize,
183 :     int maximumPoolSize,
184 :     long keepAliveTime,
185 :     TimeUnit unit,
186 :     BlockingQueue<Runnable> workQueue,
187 :     ThreadFactory threadFactory,
188 :     RejectedExecutionHandler handler) {
189 : jsr166 1.2 super(corePoolSize, maximumPoolSize, keepAliveTime, unit,
190 : dl 1.1 workQueue, threadFactory, handler);
191 :     }
192 :    
193 : jsr166 1.30 final CountDownLatch beforeCalled = new CountDownLatch(1);
194 :     final CountDownLatch afterCalled = new CountDownLatch(1);
195 :     final CountDownLatch terminatedCalled = new CountDownLatch(1);
196 :    
197 : dl 1.1 public CustomTPE() {
198 : jsr166 1.9 super(1, 1, LONG_DELAY_MS, MILLISECONDS, new SynchronousQueue<Runnable>());
199 : dl 1.1 }
200 :     protected void beforeExecute(Thread t, Runnable r) {
201 : jsr166 1.30 beforeCalled.countDown();
202 : dl 1.1 }
203 :     protected void afterExecute(Runnable r, Throwable t) {
204 : jsr166 1.30 afterCalled.countDown();
205 : dl 1.1 }
206 :     protected void terminated() {
207 : jsr166 1.30 terminatedCalled.countDown();
208 : dl 1.1 }
209 : jsr166 1.2
210 : jsr166 1.30 public boolean beforeCalled() {
211 :     return beforeCalled.getCount() == 0;
212 :     }
213 :     public boolean afterCalled() {
214 :     return afterCalled.getCount() == 0;
215 :     }
216 :     public boolean terminatedCalled() {
217 :     return terminatedCalled.getCount() == 0;
218 :     }
219 : dl 1.1 }
220 :    
221 : jsr166 1.4 static class FailingThreadFactory implements ThreadFactory {
222 : dl 1.1 int calls = 0;
223 : jsr166 1.4 public Thread newThread(Runnable r) {
224 : dl 1.1 if (++calls > 1) return null;
225 :     return new Thread(r);
226 : jsr166 1.2 }
227 : dl 1.1 }
228 : jsr166 1.2
229 : dl 1.1 /**
230 : jsr166 1.19 * execute successfully executes a runnable
231 : dl 1.1 */
232 : jsr166 1.5 public void testExecute() throws InterruptedException {
233 : jsr166 1.50 final ThreadPoolExecutor p =
234 :     new CustomTPE(1, 1,
235 :     2 * LONG_DELAY_MS, MILLISECONDS,
236 :     new ArrayBlockingQueue<Runnable>(10));
237 : jsr166 1.51 try (PoolCleaner cleaner = cleaner(p)) {
238 : jsr166 1.48 final CountDownLatch done = new CountDownLatch(1);
239 :     final Runnable task = new CheckedRunnable() {
240 : jsr166 1.50 public void realRun() { done.countDown(); }};
241 : jsr166 1.21 p.execute(task);
242 : jsr166 1.48 assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS));
243 : jsr166 1.2 }
244 : dl 1.1 }
245 :    
246 :     /**
247 : jsr166 1.19 * getActiveCount increases but doesn't overestimate, when a
248 :     * thread becomes active
249 : dl 1.1 */
250 : jsr166 1.5 public void testGetActiveCount() throws InterruptedException {
251 : jsr166 1.21 final ThreadPoolExecutor p =
252 :     new CustomTPE(2, 2,
253 :     LONG_DELAY_MS, MILLISECONDS,
254 :     new ArrayBlockingQueue<Runnable>(10));
255 : jsr166 1.51 try (PoolCleaner cleaner = cleaner(p)) {
256 : jsr166 1.66 final CountDownLatch threadStarted = new CountDownLatch(1);
257 :     final CountDownLatch done = new CountDownLatch(1);
258 : jsr166 1.21 assertEquals(0, p.getActiveCount());
259 :     p.execute(new CheckedRunnable() {
260 :     public void realRun() throws InterruptedException {
261 :     threadStarted.countDown();
262 :     assertEquals(1, p.getActiveCount());
263 :     done.await();
264 :     }});
265 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
266 : jsr166 1.21 assertEquals(1, p.getActiveCount());
267 :     done.countDown();
268 :     }
269 : dl 1.1 }
270 :    
271 :     /**
272 : jsr166 1.19 * prestartCoreThread starts a thread if under corePoolSize, else doesn't
273 : dl 1.1 */
274 :     public void testPrestartCoreThread() {
275 : jsr166 1.52 ThreadPoolExecutor p =
276 :     new CustomTPE(2, 6,
277 :     LONG_DELAY_MS, MILLISECONDS,
278 :     new ArrayBlockingQueue<Runnable>(10));
279 :     try (PoolCleaner cleaner = cleaner(p)) {
280 :     assertEquals(0, p.getPoolSize());
281 :     assertTrue(p.prestartCoreThread());
282 :     assertEquals(1, p.getPoolSize());
283 :     assertTrue(p.prestartCoreThread());
284 :     assertEquals(2, p.getPoolSize());
285 :     assertFalse(p.prestartCoreThread());
286 :     assertEquals(2, p.getPoolSize());
287 :     p.setCorePoolSize(4);
288 :     assertTrue(p.prestartCoreThread());
289 :     assertEquals(3, p.getPoolSize());
290 :     assertTrue(p.prestartCoreThread());
291 :     assertEquals(4, p.getPoolSize());
292 :     assertFalse(p.prestartCoreThread());
293 :     assertEquals(4, p.getPoolSize());
294 :     }
295 : dl 1.1 }
296 :    
297 :     /**
298 : jsr166 1.19 * prestartAllCoreThreads starts all corePoolSize threads
299 : dl 1.1 */
300 :     public void testPrestartAllCoreThreads() {
301 : jsr166 1.53 ThreadPoolExecutor p =
302 :     new CustomTPE(2, 6,
303 :     LONG_DELAY_MS, MILLISECONDS,
304 :     new ArrayBlockingQueue<Runnable>(10));
305 :     try (PoolCleaner cleaner = cleaner(p)) {
306 :     assertEquals(0, p.getPoolSize());
307 :     p.prestartAllCoreThreads();
308 :     assertEquals(2, p.getPoolSize());
309 :     p.prestartAllCoreThreads();
310 :     assertEquals(2, p.getPoolSize());
311 :     p.setCorePoolSize(4);
312 :     p.prestartAllCoreThreads();
313 :     assertEquals(4, p.getPoolSize());
314 :     p.prestartAllCoreThreads();
315 :     assertEquals(4, p.getPoolSize());
316 :     }
317 : dl 1.1 }
318 : jsr166 1.2
319 : dl 1.1 /**
320 : jsr166 1.19 * getCompletedTaskCount increases, but doesn't overestimate,
321 :     * when tasks complete
322 : dl 1.1 */
323 : jsr166 1.5 public void testGetCompletedTaskCount() throws InterruptedException {
324 : jsr166 1.21 final ThreadPoolExecutor p =
325 :     new CustomTPE(2, 2,
326 :     LONG_DELAY_MS, MILLISECONDS,
327 :     new ArrayBlockingQueue<Runnable>(10));
328 : jsr166 1.54 try (PoolCleaner cleaner = cleaner(p)) {
329 :     final CountDownLatch threadStarted = new CountDownLatch(1);
330 :     final CountDownLatch threadProceed = new CountDownLatch(1);
331 :     final CountDownLatch threadDone = new CountDownLatch(1);
332 : jsr166 1.21 assertEquals(0, p.getCompletedTaskCount());
333 :     p.execute(new CheckedRunnable() {
334 :     public void realRun() throws InterruptedException {
335 :     threadStarted.countDown();
336 :     assertEquals(0, p.getCompletedTaskCount());
337 :     threadProceed.await();
338 :     threadDone.countDown();
339 :     }});
340 : jsr166 1.30 await(threadStarted);
341 : jsr166 1.21 assertEquals(0, p.getCompletedTaskCount());
342 :     threadProceed.countDown();
343 :     threadDone.await();
344 : jsr166 1.30 long startTime = System.nanoTime();
345 :     while (p.getCompletedTaskCount() != 1) {
346 :     if (millisElapsedSince(startTime) > LONG_DELAY_MS)
347 :     fail("timed out");
348 :     Thread.yield();
349 :     }
350 : jsr166 1.21 }
351 : dl 1.1 }
352 : jsr166 1.2
353 : dl 1.1 /**
354 : jsr166 1.19 * getCorePoolSize returns size given in constructor if not otherwise set
355 : dl 1.1 */
356 :     public void testGetCorePoolSize() {
357 : jsr166 1.59 final ThreadPoolExecutor p =
358 : jsr166 1.55 new CustomTPE(1, 1,
359 :     LONG_DELAY_MS, MILLISECONDS,
360 :     new ArrayBlockingQueue<Runnable>(10));
361 :     try (PoolCleaner cleaner = cleaner(p)) {
362 :     assertEquals(1, p.getCorePoolSize());
363 :     }
364 : dl 1.1 }
365 : jsr166 1.2
366 : dl 1.1 /**
367 : jsr166 1.19 * getKeepAliveTime returns value given in constructor if not otherwise set
368 : dl 1.1 */
369 :     public void testGetKeepAliveTime() {
370 : jsr166 1.59 final ThreadPoolExecutor p =
371 : jsr166 1.56 new CustomTPE(2, 2,
372 :     1000, MILLISECONDS,
373 :     new ArrayBlockingQueue<Runnable>(10));
374 :     try (PoolCleaner cleaner = cleaner(p)) {
375 :     assertEquals(1, p.getKeepAliveTime(SECONDS));
376 :     }
377 : dl 1.1 }
378 :    
379 : jsr166 1.2 /**
380 : dl 1.1 * getThreadFactory returns factory in constructor if not set
381 :     */
382 :     public void testGetThreadFactory() {
383 : jsr166 1.59 final ThreadFactory threadFactory = new SimpleThreadFactory();
384 :     final ThreadPoolExecutor p =
385 : jsr166 1.57 new CustomTPE(1, 2,
386 :     LONG_DELAY_MS, MILLISECONDS,
387 :     new ArrayBlockingQueue<Runnable>(10),
388 :     threadFactory,
389 :     new NoOpREHandler());
390 :     try (PoolCleaner cleaner = cleaner(p)) {
391 :     assertSame(threadFactory, p.getThreadFactory());
392 :     }
393 : dl 1.1 }
394 :    
395 : jsr166 1.2 /**
396 : dl 1.1 * setThreadFactory sets the thread factory returned by getThreadFactory
397 :     */
398 :     public void testSetThreadFactory() {
399 : jsr166 1.59 final ThreadPoolExecutor p =
400 : jsr166 1.57 new CustomTPE(1, 2,
401 :     LONG_DELAY_MS, MILLISECONDS,
402 :     new ArrayBlockingQueue<Runnable>(10));
403 :     try (PoolCleaner cleaner = cleaner(p)) {
404 :     ThreadFactory threadFactory = new SimpleThreadFactory();
405 :     p.setThreadFactory(threadFactory);
406 :     assertSame(threadFactory, p.getThreadFactory());
407 :     }
408 : dl 1.1 }
409 :    
410 : jsr166 1.2 /**
411 : dl 1.1 * setThreadFactory(null) throws NPE
412 :     */
413 :     public void testSetThreadFactoryNull() {
414 : jsr166 1.59 final ThreadPoolExecutor p =
415 : jsr166 1.58 new CustomTPE(1, 2,
416 :     LONG_DELAY_MS, MILLISECONDS,
417 :     new ArrayBlockingQueue<Runnable>(10));
418 :     try (PoolCleaner cleaner = cleaner(p)) {
419 :     try {
420 :     p.setThreadFactory(null);
421 :     shouldThrow();
422 :     } catch (NullPointerException success) {}
423 : dl 1.1 }
424 :     }
425 :    
426 : jsr166 1.2 /**
427 : dl 1.1 * getRejectedExecutionHandler returns handler in constructor if not set
428 :     */
429 :     public void testGetRejectedExecutionHandler() {
430 : jsr166 1.60 final RejectedExecutionHandler handler = new NoOpREHandler();
431 : jsr166 1.59 final ThreadPoolExecutor p =
432 :     new CustomTPE(1, 2,
433 :     LONG_DELAY_MS, MILLISECONDS,
434 :     new ArrayBlockingQueue<Runnable>(10),
435 : jsr166 1.60 handler);
436 : jsr166 1.59 try (PoolCleaner cleaner = cleaner(p)) {
437 : jsr166 1.60 assertSame(handler, p.getRejectedExecutionHandler());
438 : jsr166 1.59 }
439 : dl 1.1 }
440 :    
441 : jsr166 1.2 /**
442 : dl 1.1 * setRejectedExecutionHandler sets the handler returned by
443 :     * getRejectedExecutionHandler
444 :     */
445 :     public void testSetRejectedExecutionHandler() {
446 : jsr166 1.60 final ThreadPoolExecutor p =
447 :     new CustomTPE(1, 2,
448 :     LONG_DELAY_MS, MILLISECONDS,
449 :     new ArrayBlockingQueue<Runnable>(10));
450 :     try (PoolCleaner cleaner = cleaner(p)) {
451 :     RejectedExecutionHandler handler = new NoOpREHandler();
452 :     p.setRejectedExecutionHandler(handler);
453 :     assertSame(handler, p.getRejectedExecutionHandler());
454 :     }
455 : dl 1.1 }
456 :    
457 : jsr166 1.2 /**
458 : dl 1.1 * setRejectedExecutionHandler(null) throws NPE
459 :     */
460 :     public void testSetRejectedExecutionHandlerNull() {
461 : jsr166 1.61 final ThreadPoolExecutor p =
462 :     new CustomTPE(1, 2,
463 :     LONG_DELAY_MS, MILLISECONDS,
464 :     new ArrayBlockingQueue<Runnable>(10));
465 :     try (PoolCleaner cleaner = cleaner(p)) {
466 :     try {
467 :     p.setRejectedExecutionHandler(null);
468 :     shouldThrow();
469 :     } catch (NullPointerException success) {}
470 : dl 1.1 }
471 :     }
472 :    
473 :     /**
474 : jsr166 1.19 * getLargestPoolSize increases, but doesn't overestimate, when
475 :     * multiple threads active
476 : dl 1.1 */
477 : jsr166 1.5 public void testGetLargestPoolSize() throws InterruptedException {
478 : jsr166 1.21 final int THREADS = 3;
479 :     final ThreadPoolExecutor p =
480 :     new CustomTPE(THREADS, THREADS,
481 :     LONG_DELAY_MS, MILLISECONDS,
482 :     new ArrayBlockingQueue<Runnable>(10));
483 : jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
484 :     final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
485 :     final CountDownLatch done = new CountDownLatch(1);
486 : jsr166 1.21 assertEquals(0, p.getLargestPoolSize());
487 :     for (int i = 0; i < THREADS; i++)
488 :     p.execute(new CheckedRunnable() {
489 :     public void realRun() throws InterruptedException {
490 :     threadsStarted.countDown();
491 :     done.await();
492 :     assertEquals(THREADS, p.getLargestPoolSize());
493 :     }});
494 : jsr166 1.62 assertTrue(threadsStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
495 : jsr166 1.21 assertEquals(THREADS, p.getLargestPoolSize());
496 : jsr166 1.62 done.countDown(); // release pool
497 : jsr166 1.21 }
498 : jsr166 1.62 assertEquals(THREADS, p.getLargestPoolSize());
499 : dl 1.1 }
500 : jsr166 1.2
501 : dl 1.1 /**
502 : jsr166 1.19 * getMaximumPoolSize returns value given in constructor if not
503 :     * otherwise set
504 : dl 1.1 */
505 :     public void testGetMaximumPoolSize() {
506 : jsr166 1.63 final ThreadPoolExecutor p =
507 :     new CustomTPE(2, 3,
508 :     LONG_DELAY_MS, MILLISECONDS,
509 :     new ArrayBlockingQueue<Runnable>(10));
510 :     try (PoolCleaner cleaner = cleaner(p)) {
511 :     assertEquals(3, p.getMaximumPoolSize());
512 :     p.setMaximumPoolSize(5);
513 :     assertEquals(5, p.getMaximumPoolSize());
514 :     p.setMaximumPoolSize(4);
515 :     assertEquals(4, p.getMaximumPoolSize());
516 :     }
517 : dl 1.1 }
518 : jsr166 1.2
519 : dl 1.1 /**
520 : jsr166 1.19 * getPoolSize increases, but doesn't overestimate, when threads
521 :     * become active
522 : dl 1.1 */
523 : jsr166 1.21 public void testGetPoolSize() throws InterruptedException {
524 :     final ThreadPoolExecutor p =
525 :     new CustomTPE(1, 1,
526 :     LONG_DELAY_MS, MILLISECONDS,
527 :     new ArrayBlockingQueue<Runnable>(10));
528 : jsr166 1.64 try (PoolCleaner cleaner = cleaner(p)) {
529 :     final CountDownLatch threadStarted = new CountDownLatch(1);
530 :     final CountDownLatch done = new CountDownLatch(1);
531 : jsr166 1.21 assertEquals(0, p.getPoolSize());
532 :     p.execute(new CheckedRunnable() {
533 :     public void realRun() throws InterruptedException {
534 :     threadStarted.countDown();
535 :     assertEquals(1, p.getPoolSize());
536 :     done.await();
537 :     }});
538 : jsr166 1.64 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
539 : jsr166 1.21 assertEquals(1, p.getPoolSize());
540 : jsr166 1.64 done.countDown(); // release pool
541 : jsr166 1.21 }
542 : dl 1.1 }
543 : jsr166 1.2
544 : dl 1.1 /**
545 : jsr166 1.19 * getTaskCount increases, but doesn't overestimate, when tasks submitted
546 : dl 1.1 */
547 : jsr166 1.5 public void testGetTaskCount() throws InterruptedException {
548 : jsr166 1.21 final ThreadPoolExecutor p =
549 :     new CustomTPE(1, 1,
550 :     LONG_DELAY_MS, MILLISECONDS,
551 :     new ArrayBlockingQueue<Runnable>(10));
552 :     final CountDownLatch threadStarted = new CountDownLatch(1);
553 :     final CountDownLatch done = new CountDownLatch(1);
554 :     try {
555 :     assertEquals(0, p.getTaskCount());
556 :     p.execute(new CheckedRunnable() {
557 :     public void realRun() throws InterruptedException {
558 :     threadStarted.countDown();
559 :     assertEquals(1, p.getTaskCount());
560 :     done.await();
561 :     }});
562 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
563 : jsr166 1.21 assertEquals(1, p.getTaskCount());
564 :     } finally {
565 :     done.countDown();
566 :     joinPool(p);
567 :     }
568 : dl 1.1 }
569 : jsr166 1.2
570 : dl 1.1 /**
571 : jsr166 1.28 * isShutdown is false before shutdown, true after
572 : dl 1.1 */
573 :     public void testIsShutdown() {
574 : jsr166 1.2
575 : jsr166 1.21 ThreadPoolExecutor p = new CustomTPE(1, 1, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
576 :     assertFalse(p.isShutdown());
577 :     try { p.shutdown(); } catch (SecurityException ok) { return; }
578 :     assertTrue(p.isShutdown());
579 :     joinPool(p);
580 : dl 1.1 }
581 :    
582 :     /**
583 : jsr166 1.19 * isTerminated is false before termination, true after
584 : dl 1.1 */
585 : jsr166 1.5 public void testIsTerminated() throws InterruptedException {
586 : jsr166 1.21 final ThreadPoolExecutor p =
587 :     new CustomTPE(1, 1,
588 :     LONG_DELAY_MS, MILLISECONDS,
589 :     new ArrayBlockingQueue<Runnable>(10));
590 :     final CountDownLatch threadStarted = new CountDownLatch(1);
591 :     final CountDownLatch done = new CountDownLatch(1);
592 :     try {
593 :     assertFalse(p.isTerminating());
594 :     p.execute(new CheckedRunnable() {
595 :     public void realRun() throws InterruptedException {
596 : jsr166 1.24 assertFalse(p.isTerminating());
597 : jsr166 1.21 threadStarted.countDown();
598 :     done.await();
599 :     }});
600 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
601 : jsr166 1.21 assertFalse(p.isTerminating());
602 :     done.countDown();
603 :     } finally {
604 :     try { p.shutdown(); } catch (SecurityException ok) { return; }
605 :     }
606 :     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
607 :     assertTrue(p.isTerminated());
608 :     assertFalse(p.isTerminating());
609 : dl 1.1 }
610 :    
611 :     /**
612 : jsr166 1.19 * isTerminating is not true when running or when terminated
613 : dl 1.1 */
614 : jsr166 1.5 public void testIsTerminating() throws InterruptedException {
615 : jsr166 1.21 final ThreadPoolExecutor p =
616 :     new CustomTPE(1, 1,
617 :     LONG_DELAY_MS, MILLISECONDS,
618 :     new ArrayBlockingQueue<Runnable>(10));
619 :     final CountDownLatch threadStarted = new CountDownLatch(1);
620 :     final CountDownLatch done = new CountDownLatch(1);
621 :     try {
622 :     assertFalse(p.isTerminating());
623 :     p.execute(new CheckedRunnable() {
624 :     public void realRun() throws InterruptedException {
625 : jsr166 1.23 assertFalse(p.isTerminating());
626 : jsr166 1.21 threadStarted.countDown();
627 :     done.await();
628 :     }});
629 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
630 : jsr166 1.21 assertFalse(p.isTerminating());
631 :     done.countDown();
632 :     } finally {
633 :     try { p.shutdown(); } catch (SecurityException ok) { return; }
634 :     }
635 :     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
636 :     assertTrue(p.isTerminated());
637 :     assertFalse(p.isTerminating());
638 : dl 1.1 }
639 :    
640 :     /**
641 :     * getQueue returns the work queue, which contains queued tasks
642 :     */
643 : jsr166 1.5 public void testGetQueue() throws InterruptedException {
644 : jsr166 1.21 final BlockingQueue<Runnable> q = new ArrayBlockingQueue<Runnable>(10);
645 :     final ThreadPoolExecutor p =
646 :     new CustomTPE(1, 1,
647 :     LONG_DELAY_MS, MILLISECONDS,
648 :     q);
649 :     final CountDownLatch threadStarted = new CountDownLatch(1);
650 :     final CountDownLatch done = new CountDownLatch(1);
651 :     try {
652 :     FutureTask[] tasks = new FutureTask[5];
653 :     for (int i = 0; i < tasks.length; i++) {
654 :     Callable task = new CheckedCallable<Boolean>() {
655 :     public Boolean realCall() throws InterruptedException {
656 :     threadStarted.countDown();
657 :     assertSame(q, p.getQueue());
658 :     done.await();
659 :     return Boolean.TRUE;
660 :     }};
661 :     tasks[i] = new FutureTask(task);
662 :     p.execute(tasks[i]);
663 :     }
664 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
665 : jsr166 1.21 assertSame(q, p.getQueue());
666 :     assertFalse(q.contains(tasks[0]));
667 :     assertTrue(q.contains(tasks[tasks.length - 1]));
668 :     assertEquals(tasks.length - 1, q.size());
669 : dl 1.1 } finally {
670 : jsr166 1.21 done.countDown();
671 :     joinPool(p);
672 : dl 1.1 }
673 :     }
674 :    
675 :     /**
676 :     * remove(task) removes queued task, and fails to remove active task
677 :     */
678 : jsr166 1.5 public void testRemove() throws InterruptedException {
679 : dl 1.1 BlockingQueue<Runnable> q = new ArrayBlockingQueue<Runnable>(10);
680 : jsr166 1.21 final ThreadPoolExecutor p =
681 :     new CustomTPE(1, 1,
682 :     LONG_DELAY_MS, MILLISECONDS,
683 :     q);
684 :     Runnable[] tasks = new Runnable[6];
685 :     final CountDownLatch threadStarted = new CountDownLatch(1);
686 :     final CountDownLatch done = new CountDownLatch(1);
687 :     try {
688 :     for (int i = 0; i < tasks.length; i++) {
689 :     tasks[i] = new CheckedRunnable() {
690 :     public void realRun() throws InterruptedException {
691 :     threadStarted.countDown();
692 :     done.await();
693 :     }};
694 :     p.execute(tasks[i]);
695 :     }
696 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
697 : jsr166 1.21 assertFalse(p.remove(tasks[0]));
698 : dl 1.1 assertTrue(q.contains(tasks[4]));
699 :     assertTrue(q.contains(tasks[3]));
700 : jsr166 1.21 assertTrue(p.remove(tasks[4]));
701 :     assertFalse(p.remove(tasks[4]));
702 : dl 1.1 assertFalse(q.contains(tasks[4]));
703 :     assertTrue(q.contains(tasks[3]));
704 : jsr166 1.21 assertTrue(p.remove(tasks[3]));
705 : dl 1.1 assertFalse(q.contains(tasks[3]));
706 :     } finally {
707 : jsr166 1.21 done.countDown();
708 :     joinPool(p);
709 : dl 1.1 }
710 :     }
711 :    
712 :     /**
713 : jsr166 1.19 * purge removes cancelled tasks from the queue
714 : dl 1.1 */
715 : jsr166 1.21 public void testPurge() throws InterruptedException {
716 :     final CountDownLatch threadStarted = new CountDownLatch(1);
717 :     final CountDownLatch done = new CountDownLatch(1);
718 :     final BlockingQueue<Runnable> q = new ArrayBlockingQueue<Runnable>(10);
719 :     final ThreadPoolExecutor p =
720 :     new CustomTPE(1, 1,
721 :     LONG_DELAY_MS, MILLISECONDS,
722 :     q);
723 : dl 1.1 FutureTask[] tasks = new FutureTask[5];
724 : jsr166 1.21 try {
725 :     for (int i = 0; i < tasks.length; i++) {
726 :     Callable task = new CheckedCallable<Boolean>() {
727 :     public Boolean realCall() throws InterruptedException {
728 :     threadStarted.countDown();
729 :     done.await();
730 :     return Boolean.TRUE;
731 :     }};
732 :     tasks[i] = new FutureTask(task);
733 :     p.execute(tasks[i]);
734 :     }
735 : jsr166 1.65 assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS));
736 : jsr166 1.21 assertEquals(tasks.length, p.getTaskCount());
737 :     assertEquals(tasks.length - 1, q.size());
738 :     assertEquals(1L, p.getActiveCount());
739 :     assertEquals(0L, p.getCompletedTaskCount());
740 :     tasks[4].cancel(true);
741 :     tasks[3].cancel(false);
742 :     p.purge();
743 :     assertEquals(tasks.length - 3, q.size());
744 :     assertEquals(tasks.length - 2, p.getTaskCount());
745 :     p.purge(); // Nothing to do
746 :     assertEquals(tasks.length - 3, q.size());
747 :     assertEquals(tasks.length - 2, p.getTaskCount());
748 :     } finally {
749 :     done.countDown();
750 :     joinPool(p);
751 :     }
752 : dl 1.1 }
753 :    
754 :     /**
755 : jsr166 1.40 * shutdownNow returns a list containing tasks that were not run,
756 :     * and those tasks are drained from the queue
757 : dl 1.1 */
758 : jsr166 1.41 public void testShutdownNow() throws InterruptedException {
759 :     final int poolSize = 2;
760 :     final int count = 5;
761 :     final AtomicInteger ran = new AtomicInteger(0);
762 :     ThreadPoolExecutor p =
763 :     new CustomTPE(poolSize, poolSize, LONG_DELAY_MS, MILLISECONDS,
764 :     new ArrayBlockingQueue<Runnable>(10));
765 :     CountDownLatch threadsStarted = new CountDownLatch(poolSize);
766 : jsr166 1.43 Runnable waiter = new CheckedRunnable() { public void realRun() {
767 : jsr166 1.41 threadsStarted.countDown();
768 : dl 1.1 try {
769 : jsr166 1.41 MILLISECONDS.sleep(2 * LONG_DELAY_MS);
770 :     } catch (InterruptedException success) {}
771 :     ran.getAndIncrement();
772 :     }};
773 :     for (int i = 0; i < count; i++)
774 :     p.execute(waiter);
775 :     assertTrue(threadsStarted.await(LONG_DELAY_MS, MILLISECONDS));
776 : jsr166 1.42 assertEquals(poolSize, p.getActiveCount());
777 :     assertEquals(0, p.getCompletedTaskCount());
778 : jsr166 1.41 final List<Runnable> queuedTasks;
779 :     try {
780 :     queuedTasks = p.shutdownNow();
781 :     } catch (SecurityException ok) {
782 :     return; // Allowed in case test doesn't have privs
783 : dl 1.1 }
784 : jsr166 1.21 assertTrue(p.isShutdown());
785 : jsr166 1.40 assertTrue(p.getQueue().isEmpty());
786 : jsr166 1.41 assertEquals(count - poolSize, queuedTasks.size());
787 :     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
788 :     assertTrue(p.isTerminated());
789 :     assertEquals(poolSize, ran.get());
790 : jsr166 1.42 assertEquals(poolSize, p.getCompletedTaskCount());
791 : dl 1.1 }
792 :    
793 :     // Exception Tests
794 :    
795 : jsr166 1.2 /**
796 :     * Constructor throws if corePoolSize argument is less than zero
797 : dl 1.1 */
798 :     public void testConstructor1() {
799 :     try {
800 : jsr166 1.36 new CustomTPE(-1, 1, 1L, SECONDS,
801 :     new ArrayBlockingQueue<Runnable>(10));
802 : dl 1.1 shouldThrow();
803 : jsr166 1.5 } catch (IllegalArgumentException success) {}
804 : dl 1.1 }
805 : jsr166 1.2
806 :     /**
807 :     * Constructor throws if maximumPoolSize is less than zero
808 : dl 1.1 */
809 :     public void testConstructor2() {
810 :     try {
811 : jsr166 1.36 new CustomTPE(1, -1, 1L, SECONDS,
812 :     new ArrayBlockingQueue<Runnable>(10));
813 : dl 1.1 shouldThrow();
814 : jsr166 1.5 } catch (IllegalArgumentException success) {}
815 : dl 1.1 }
816 : jsr166 1.2
817 :     /**
818 :     * Constructor throws if maximumPoolSize is equal to zero
819 : dl 1.1 */
820 :     public void testConstructor3() {
821 :     try {
822 : jsr166 1.36 new CustomTPE(1, 0, 1L, SECONDS,
823 :     new ArrayBlockingQueue<Runnable>(10));
824 : dl 1.1 shouldThrow();
825 : jsr166 1.5 } catch (IllegalArgumentException success) {}
826 : dl 1.1 }
827 :    
828 : jsr166 1.2 /**
829 :     * Constructor throws if keepAliveTime is less than zero
830 : dl 1.1 */
831 :     public void testConstructor4() {
832 :     try {
833 : jsr166 1.36 new CustomTPE(1, 2, -1L, SECONDS,
834 :     new ArrayBlockingQueue<Runnable>(10));
835 : dl 1.1 shouldThrow();
836 : jsr166 1.5 } catch (IllegalArgumentException success) {}
837 : dl 1.1 }
838 :    
839 : jsr166 1.2 /**
840 :     * Constructor throws if corePoolSize is greater than the maximumPoolSize
841 : dl 1.1 */
842 :     public void testConstructor5() {
843 :     try {
844 : jsr166 1.36 new CustomTPE(2, 1, 1L, SECONDS,
845 :     new ArrayBlockingQueue<Runnable>(10));
846 : dl 1.1 shouldThrow();
847 : jsr166 1.5 } catch (IllegalArgumentException success) {}
848 : dl 1.1 }
849 : jsr166 1.2
850 :     /**
851 :     * Constructor throws if workQueue is set to null
852 : dl 1.1 */
853 :     public void testConstructorNullPointerException() {
854 :     try {
855 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS, null);
856 : dl 1.1 shouldThrow();
857 : jsr166 1.5 } catch (NullPointerException success) {}
858 : dl 1.1 }
859 :    
860 : jsr166 1.2 /**
861 :     * Constructor throws if corePoolSize argument is less than zero
862 : dl 1.1 */
863 :     public void testConstructor6() {
864 :     try {
865 : jsr166 1.36 new CustomTPE(-1, 1, 1L, SECONDS,
866 :     new ArrayBlockingQueue<Runnable>(10),
867 :     new SimpleThreadFactory());
868 : dl 1.1 shouldThrow();
869 : jsr166 1.4 } catch (IllegalArgumentException success) {}
870 : dl 1.1 }
871 : jsr166 1.2
872 :     /**
873 :     * Constructor throws if maximumPoolSize is less than zero
874 : dl 1.1 */
875 :     public void testConstructor7() {
876 :     try {
877 : jsr166 1.36 new CustomTPE(1,-1, 1L, SECONDS,
878 :     new ArrayBlockingQueue<Runnable>(10),
879 :     new SimpleThreadFactory());
880 : dl 1.1 shouldThrow();
881 : jsr166 1.5 } catch (IllegalArgumentException success) {}
882 : dl 1.1 }
883 :    
884 : jsr166 1.2 /**
885 :     * Constructor throws if maximumPoolSize is equal to zero
886 : dl 1.1 */
887 :     public void testConstructor8() {
888 :     try {
889 : jsr166 1.36 new CustomTPE(1, 0, 1L, SECONDS,
890 :     new ArrayBlockingQueue<Runnable>(10),
891 :     new SimpleThreadFactory());
892 : dl 1.1 shouldThrow();
893 : jsr166 1.5 } catch (IllegalArgumentException success) {}
894 : dl 1.1 }
895 :    
896 : jsr166 1.2 /**
897 :     * Constructor throws if keepAliveTime is less than zero
898 : dl 1.1 */
899 :     public void testConstructor9() {
900 :     try {
901 : jsr166 1.36 new CustomTPE(1, 2, -1L, SECONDS,
902 :     new ArrayBlockingQueue<Runnable>(10),
903 :     new SimpleThreadFactory());
904 : dl 1.1 shouldThrow();
905 : jsr166 1.5 } catch (IllegalArgumentException success) {}
906 : dl 1.1 }
907 :    
908 : jsr166 1.2 /**
909 :     * Constructor throws if corePoolSize is greater than the maximumPoolSize
910 : dl 1.1 */
911 :     public void testConstructor10() {
912 :     try {
913 : jsr166 1.36 new CustomTPE(2, 1, 1L, SECONDS,
914 :     new ArrayBlockingQueue<Runnable>(10),
915 :     new SimpleThreadFactory());
916 : dl 1.1 shouldThrow();
917 : jsr166 1.5 } catch (IllegalArgumentException success) {}
918 : dl 1.1 }
919 :    
920 : jsr166 1.2 /**
921 :     * Constructor throws if workQueue is set to null
922 : dl 1.1 */
923 :     public void testConstructorNullPointerException2() {
924 :     try {
925 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS, null, new SimpleThreadFactory());
926 : dl 1.1 shouldThrow();
927 : jsr166 1.5 } catch (NullPointerException success) {}
928 : dl 1.1 }
929 :    
930 : jsr166 1.2 /**
931 :     * Constructor throws if threadFactory is set to null
932 : dl 1.1 */
933 :     public void testConstructorNullPointerException3() {
934 :     try {
935 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
936 :     new ArrayBlockingQueue<Runnable>(10),
937 :     (ThreadFactory) null);
938 : dl 1.1 shouldThrow();
939 : jsr166 1.5 } catch (NullPointerException success) {}
940 : dl 1.1 }
941 : jsr166 1.2
942 :     /**
943 :     * Constructor throws if corePoolSize argument is less than zero
944 : dl 1.1 */
945 :     public void testConstructor11() {
946 :     try {
947 : jsr166 1.36 new CustomTPE(-1, 1, 1L, SECONDS,
948 :     new ArrayBlockingQueue<Runnable>(10),
949 :     new NoOpREHandler());
950 : dl 1.1 shouldThrow();
951 : jsr166 1.5 } catch (IllegalArgumentException success) {}
952 : dl 1.1 }
953 :    
954 : jsr166 1.2 /**
955 :     * Constructor throws if maximumPoolSize is less than zero
956 : dl 1.1 */
957 :     public void testConstructor12() {
958 :     try {
959 : jsr166 1.36 new CustomTPE(1, -1, 1L, SECONDS,
960 :     new ArrayBlockingQueue<Runnable>(10),
961 :     new NoOpREHandler());
962 : dl 1.1 shouldThrow();
963 : jsr166 1.5 } catch (IllegalArgumentException success) {}
964 : dl 1.1 }
965 :    
966 : jsr166 1.2 /**
967 :     * Constructor throws if maximumPoolSize is equal to zero
968 : dl 1.1 */
969 :     public void testConstructor13() {
970 :     try {
971 : jsr166 1.36 new CustomTPE(1, 0, 1L, SECONDS,
972 :     new ArrayBlockingQueue<Runnable>(10),
973 :     new NoOpREHandler());
974 : dl 1.1 shouldThrow();
975 : jsr166 1.5 } catch (IllegalArgumentException success) {}
976 : dl 1.1 }
977 :    
978 : jsr166 1.2 /**
979 :     * Constructor throws if keepAliveTime is less than zero
980 : dl 1.1 */
981 :     public void testConstructor14() {
982 :     try {
983 : jsr166 1.36 new CustomTPE(1, 2, -1L, SECONDS,
984 :     new ArrayBlockingQueue<Runnable>(10),
985 :     new NoOpREHandler());
986 : dl 1.1 shouldThrow();
987 : jsr166 1.5 } catch (IllegalArgumentException success) {}
988 : dl 1.1 }
989 :    
990 : jsr166 1.2 /**
991 :     * Constructor throws if corePoolSize is greater than the maximumPoolSize
992 : dl 1.1 */
993 :     public void testConstructor15() {
994 :     try {
995 : jsr166 1.36 new CustomTPE(2, 1, 1L, SECONDS,
996 :     new ArrayBlockingQueue<Runnable>(10),
997 :     new NoOpREHandler());
998 : dl 1.1 shouldThrow();
999 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1000 : dl 1.1 }
1001 :    
1002 : jsr166 1.2 /**
1003 :     * Constructor throws if workQueue is set to null
1004 : dl 1.1 */
1005 :     public void testConstructorNullPointerException4() {
1006 :     try {
1007 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
1008 :     null,
1009 :     new NoOpREHandler());
1010 : dl 1.1 shouldThrow();
1011 : jsr166 1.5 } catch (NullPointerException success) {}
1012 : dl 1.1 }
1013 :    
1014 : jsr166 1.2 /**
1015 :     * Constructor throws if handler is set to null
1016 : dl 1.1 */
1017 :     public void testConstructorNullPointerException5() {
1018 :     try {
1019 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
1020 :     new ArrayBlockingQueue<Runnable>(10),
1021 :     (RejectedExecutionHandler) null);
1022 : dl 1.1 shouldThrow();
1023 : jsr166 1.5 } catch (NullPointerException success) {}
1024 : dl 1.1 }
1025 :    
1026 : jsr166 1.2 /**
1027 :     * Constructor throws if corePoolSize argument is less than zero
1028 : dl 1.1 */
1029 :     public void testConstructor16() {
1030 :     try {
1031 : jsr166 1.36 new CustomTPE(-1, 1, 1L, SECONDS,
1032 :     new ArrayBlockingQueue<Runnable>(10),
1033 :     new SimpleThreadFactory(),
1034 :     new NoOpREHandler());
1035 : dl 1.1 shouldThrow();
1036 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1037 : dl 1.1 }
1038 :    
1039 : jsr166 1.2 /**
1040 :     * Constructor throws if maximumPoolSize is less than zero
1041 : dl 1.1 */
1042 :     public void testConstructor17() {
1043 :     try {
1044 : jsr166 1.36 new CustomTPE(1, -1, 1L, SECONDS,
1045 :     new ArrayBlockingQueue<Runnable>(10),
1046 :     new SimpleThreadFactory(),
1047 :     new NoOpREHandler());
1048 : dl 1.1 shouldThrow();
1049 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1050 : dl 1.1 }
1051 :    
1052 : jsr166 1.2 /**
1053 :     * Constructor throws if maximumPoolSize is equal to zero
1054 : dl 1.1 */
1055 :     public void testConstructor18() {
1056 :     try {
1057 : jsr166 1.36 new CustomTPE(1, 0, 1L, SECONDS,
1058 :     new ArrayBlockingQueue<Runnable>(10),
1059 :     new SimpleThreadFactory(),
1060 :     new NoOpREHandler());
1061 : dl 1.1 shouldThrow();
1062 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1063 : dl 1.1 }
1064 :    
1065 : jsr166 1.2 /**
1066 :     * Constructor throws if keepAliveTime is less than zero
1067 : dl 1.1 */
1068 :     public void testConstructor19() {
1069 :     try {
1070 : jsr166 1.36 new CustomTPE(1, 2, -1L, SECONDS,
1071 :     new ArrayBlockingQueue<Runnable>(10),
1072 :     new SimpleThreadFactory(),
1073 :     new NoOpREHandler());
1074 : dl 1.1 shouldThrow();
1075 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1076 : dl 1.1 }
1077 :    
1078 : jsr166 1.2 /**
1079 :     * Constructor throws if corePoolSize is greater than the maximumPoolSize
1080 : dl 1.1 */
1081 :     public void testConstructor20() {
1082 :     try {
1083 : jsr166 1.36 new CustomTPE(2, 1, 1L, SECONDS,
1084 :     new ArrayBlockingQueue<Runnable>(10),
1085 :     new SimpleThreadFactory(),
1086 :     new NoOpREHandler());
1087 : dl 1.1 shouldThrow();
1088 : jsr166 1.5 } catch (IllegalArgumentException success) {}
1089 : dl 1.1 }
1090 :    
1091 : jsr166 1.2 /**
1092 : jsr166 1.20 * Constructor throws if workQueue is null
1093 : dl 1.1 */
1094 :     public void testConstructorNullPointerException6() {
1095 :     try {
1096 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
1097 :     null,
1098 :     new SimpleThreadFactory(),
1099 :     new NoOpREHandler());
1100 : dl 1.1 shouldThrow();
1101 : jsr166 1.5 } catch (NullPointerException success) {}
1102 : dl 1.1 }
1103 :    
1104 : jsr166 1.2 /**
1105 : jsr166 1.20 * Constructor throws if handler is null
1106 : dl 1.1 */
1107 :     public void testConstructorNullPointerException7() {
1108 :     try {
1109 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
1110 :     new ArrayBlockingQueue<Runnable>(10),
1111 :     new SimpleThreadFactory(),
1112 :     (RejectedExecutionHandler) null);
1113 : dl 1.1 shouldThrow();
1114 : jsr166 1.5 } catch (NullPointerException success) {}
1115 : dl 1.1 }
1116 :    
1117 : jsr166 1.2 /**
1118 : jsr166 1.20 * Constructor throws if ThreadFactory is null
1119 : dl 1.1 */
1120 :     public void testConstructorNullPointerException8() {
1121 :     try {
1122 : jsr166 1.36 new CustomTPE(1, 2, 1L, SECONDS,
1123 : jsr166 1.21 new ArrayBlockingQueue<Runnable>(10),
1124 :     (ThreadFactory) null,
1125 :     new NoOpREHandler());
1126 : dl 1.1 shouldThrow();
1127 : jsr166 1.5 } catch (NullPointerException success) {}
1128 : dl 1.1 }
1129 : jsr166 1.2
1130 : dl 1.1 /**
1131 : jsr166 1.19 * execute throws RejectedExecutionException if saturated.
1132 : dl 1.1 */
1133 :     public void testSaturatedExecute() {
1134 : jsr166 1.21 ThreadPoolExecutor p =
1135 :     new CustomTPE(1, 1,
1136 :     LONG_DELAY_MS, MILLISECONDS,
1137 :     new ArrayBlockingQueue<Runnable>(1));
1138 :     final CountDownLatch done = new CountDownLatch(1);
1139 :     try {
1140 :     Runnable task = new CheckedRunnable() {
1141 :     public void realRun() throws InterruptedException {
1142 :     done.await();
1143 :     }};
1144 :     for (int i = 0; i < 2; ++i)
1145 :     p.execute(task);
1146 :     for (int i = 0; i < 2; ++i) {
1147 :     try {
1148 :     p.execute(task);
1149 :     shouldThrow();
1150 :     } catch (RejectedExecutionException success) {}
1151 :     assertTrue(p.getTaskCount() <= 2);
1152 : dl 1.1 }
1153 : jsr166 1.21 } finally {
1154 :     done.countDown();
1155 :     joinPool(p);
1156 :     }
1157 : dl 1.1 }
1158 :    
1159 :     /**
1160 : jsr166 1.19 * executor using CallerRunsPolicy runs task if saturated.
1161 : dl 1.1 */
1162 :     public void testSaturatedExecute2() {
1163 :     RejectedExecutionHandler h = new CustomTPE.CallerRunsPolicy();
1164 : jsr166 1.21 ThreadPoolExecutor p = new CustomTPE(1, 1,
1165 :     LONG_DELAY_MS, MILLISECONDS,
1166 :     new ArrayBlockingQueue<Runnable>(1),
1167 :     h);
1168 : dl 1.1 try {
1169 :     TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1170 : jsr166 1.21 for (int i = 0; i < tasks.length; ++i)
1171 : dl 1.1 tasks[i] = new TrackedNoOpRunnable();
1172 :     TrackedLongRunnable mr = new TrackedLongRunnable();
1173 :     p.execute(mr);
1174 : jsr166 1.21 for (int i = 0; i < tasks.length; ++i)
1175 : dl 1.1 p.execute(tasks[i]);
1176 : jsr166 1.21 for (int i = 1; i < tasks.length; ++i)
1177 : dl 1.1 assertTrue(tasks[i].done);
1178 : jsr166 1.3 try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1179 : dl 1.1 } finally {
1180 :     joinPool(p);
1181 :     }
1182 :     }
1183 :    
1184 :     /**
1185 : jsr166 1.19 * executor using DiscardPolicy drops task if saturated.
1186 : dl 1.1 */
1187 :     public void testSaturatedExecute3() {
1188 :     RejectedExecutionHandler h = new CustomTPE.DiscardPolicy();
1189 : jsr166 1.21 ThreadPoolExecutor p =
1190 :     new CustomTPE(1, 1,
1191 :     LONG_DELAY_MS, MILLISECONDS,
1192 :     new ArrayBlockingQueue<Runnable>(1),
1193 :     h);
1194 : dl 1.1 try {
1195 :     TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1196 : jsr166 1.21 for (int i = 0; i < tasks.length; ++i)
1197 : dl 1.1 tasks[i] = new TrackedNoOpRunnable();
1198 :     p.execute(new TrackedLongRunnable());
1199 : jsr166 1.21 for (TrackedNoOpRunnable task : tasks)
1200 :     p.execute(task);
1201 :     for (TrackedNoOpRunnable task : tasks)
1202 :     assertFalse(task.done);
1203 : jsr166 1.3 try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1204 : dl 1.1 } finally {
1205 :     joinPool(p);
1206 :     }
1207 :     }
1208 :    
1209 :     /**
1210 : jsr166 1.19 * executor using DiscardOldestPolicy drops oldest task if saturated.
1211 : dl 1.1 */
1212 :     public void testSaturatedExecute4() {
1213 :     RejectedExecutionHandler h = new CustomTPE.DiscardOldestPolicy();
1214 : jsr166 1.9 ThreadPoolExecutor p = new CustomTPE(1,1, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(1), h);
1215 : dl 1.1 try {
1216 :     p.execute(new TrackedLongRunnable());
1217 :     TrackedLongRunnable r2 = new TrackedLongRunnable();
1218 :     p.execute(r2);
1219 :     assertTrue(p.getQueue().contains(r2));
1220 :     TrackedNoOpRunnable r3 = new TrackedNoOpRunnable();
1221 :     p.execute(r3);
1222 :     assertFalse(p.getQueue().contains(r2));
1223 :     assertTrue(p.getQueue().contains(r3));
1224 : jsr166 1.3 try { p.shutdownNow(); } catch (SecurityException ok) { return; }
1225 : dl 1.1 } finally {
1226 :     joinPool(p);
1227 :     }
1228 :     }
1229 :    
1230 :     /**
1231 : jsr166 1.19 * execute throws RejectedExecutionException if shutdown
1232 : dl 1.1 */
1233 :     public void testRejectedExecutionExceptionOnShutdown() {
1234 : jsr166 1.21 ThreadPoolExecutor p =
1235 : jsr166 1.9 new CustomTPE(1,1,LONG_DELAY_MS, MILLISECONDS,new ArrayBlockingQueue<Runnable>(1));
1236 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1237 : jsr166 1.8 try {
1238 : jsr166 1.21 p.execute(new NoOpRunnable());
1239 : jsr166 1.8 shouldThrow();
1240 :     } catch (RejectedExecutionException success) {}
1241 : jsr166 1.2
1242 : jsr166 1.21 joinPool(p);
1243 : dl 1.1 }
1244 :    
1245 :     /**
1246 : jsr166 1.19 * execute using CallerRunsPolicy drops task on shutdown
1247 : dl 1.1 */
1248 :     public void testCallerRunsOnShutdown() {
1249 :     RejectedExecutionHandler h = new CustomTPE.CallerRunsPolicy();
1250 : jsr166 1.9 ThreadPoolExecutor p = new CustomTPE(1,1, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(1), h);
1251 : dl 1.1
1252 : jsr166 1.3 try { p.shutdown(); } catch (SecurityException ok) { return; }
1253 : jsr166 1.8 try {
1254 : dl 1.1 TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1255 : jsr166 1.8 p.execute(r);
1256 : dl 1.1 assertFalse(r.done);
1257 :     } finally {
1258 :     joinPool(p);
1259 :     }
1260 :     }
1261 :    
1262 :     /**
1263 : jsr166 1.19 * execute using DiscardPolicy drops task on shutdown
1264 : dl 1.1 */
1265 :     public void testDiscardOnShutdown() {
1266 :     RejectedExecutionHandler h = new CustomTPE.DiscardPolicy();
1267 : jsr166 1.9 ThreadPoolExecutor p = new CustomTPE(1,1, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(1), h);
1268 : dl 1.1
1269 : jsr166 1.3 try { p.shutdown(); } catch (SecurityException ok) { return; }
1270 : jsr166 1.8 try {
1271 : dl 1.1 TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1272 : jsr166 1.8 p.execute(r);
1273 : dl 1.1 assertFalse(r.done);
1274 :     } finally {
1275 :     joinPool(p);
1276 :     }
1277 :     }
1278 :    
1279 :     /**
1280 : jsr166 1.19 * execute using DiscardOldestPolicy drops task on shutdown
1281 : dl 1.1 */
1282 :     public void testDiscardOldestOnShutdown() {
1283 :     RejectedExecutionHandler h = new CustomTPE.DiscardOldestPolicy();
1284 : jsr166 1.9 ThreadPoolExecutor p = new CustomTPE(1,1, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(1), h);
1285 : dl 1.1
1286 : jsr166 1.3 try { p.shutdown(); } catch (SecurityException ok) { return; }
1287 : jsr166 1.8 try {
1288 : dl 1.1 TrackedNoOpRunnable r = new TrackedNoOpRunnable();
1289 : jsr166 1.8 p.execute(r);
1290 : dl 1.1 assertFalse(r.done);
1291 :     } finally {
1292 :     joinPool(p);
1293 :     }
1294 :     }
1295 :    
1296 :     /**
1297 : jsr166 1.18 * execute(null) throws NPE
1298 : dl 1.1 */
1299 :     public void testExecuteNull() {
1300 : jsr166 1.37 ThreadPoolExecutor p =
1301 :     new CustomTPE(1, 2, 1L, SECONDS,
1302 :     new ArrayBlockingQueue<Runnable>(10));
1303 : dl 1.1 try {
1304 : jsr166 1.21 p.execute(null);
1305 : dl 1.1 shouldThrow();
1306 : jsr166 1.8 } catch (NullPointerException success) {}
1307 : jsr166 1.2
1308 : jsr166 1.21 joinPool(p);
1309 : dl 1.1 }
1310 : jsr166 1.2
1311 : dl 1.1 /**
1312 : jsr166 1.19 * setCorePoolSize of negative value throws IllegalArgumentException
1313 : dl 1.1 */
1314 :     public void testCorePoolSizeIllegalArgumentException() {
1315 : jsr166 1.21 ThreadPoolExecutor p =
1316 : jsr166 1.9 new CustomTPE(1,2,LONG_DELAY_MS, MILLISECONDS,new ArrayBlockingQueue<Runnable>(10));
1317 : jsr166 1.8 try {
1318 : jsr166 1.21 p.setCorePoolSize(-1);
1319 : jsr166 1.8 shouldThrow();
1320 :     } catch (IllegalArgumentException success) {
1321 : dl 1.1 } finally {
1322 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1323 : dl 1.1 }
1324 : jsr166 1.21 joinPool(p);
1325 : jsr166 1.2 }
1326 : dl 1.1
1327 :     /**
1328 : jsr166 1.19 * setMaximumPoolSize(int) throws IllegalArgumentException
1329 :     * if given a value less the core pool size
1330 : jsr166 1.2 */
1331 : dl 1.1 public void testMaximumPoolSizeIllegalArgumentException() {
1332 : jsr166 1.21 ThreadPoolExecutor p =
1333 : jsr166 1.10 new CustomTPE(2,3,LONG_DELAY_MS, MILLISECONDS,new ArrayBlockingQueue<Runnable>(10));
1334 : dl 1.1 try {
1335 : jsr166 1.21 p.setMaximumPoolSize(1);
1336 : dl 1.1 shouldThrow();
1337 : jsr166 1.4 } catch (IllegalArgumentException success) {
1338 : dl 1.1 } finally {
1339 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1340 : dl 1.1 }
1341 : jsr166 1.21 joinPool(p);
1342 : dl 1.1 }
1343 : jsr166 1.2
1344 : dl 1.1 /**
1345 : jsr166 1.19 * setMaximumPoolSize throws IllegalArgumentException
1346 :     * if given a negative value
1347 : dl 1.1 */
1348 :     public void testMaximumPoolSizeIllegalArgumentException2() {
1349 : jsr166 1.21 ThreadPoolExecutor p =
1350 : jsr166 1.10 new CustomTPE(2,3,LONG_DELAY_MS, MILLISECONDS,new ArrayBlockingQueue<Runnable>(10));
1351 : dl 1.1 try {
1352 : jsr166 1.21 p.setMaximumPoolSize(-1);
1353 : dl 1.1 shouldThrow();
1354 : jsr166 1.4 } catch (IllegalArgumentException success) {
1355 : dl 1.1 } finally {
1356 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1357 : dl 1.1 }
1358 : jsr166 1.21 joinPool(p);
1359 : dl 1.1 }
1360 : jsr166 1.2
1361 : dl 1.1 /**
1362 : jsr166 1.19 * setKeepAliveTime throws IllegalArgumentException
1363 :     * when given a negative value
1364 : dl 1.1 */
1365 :     public void testKeepAliveTimeIllegalArgumentException() {
1366 : jsr166 1.21 ThreadPoolExecutor p =
1367 : jsr166 1.10 new CustomTPE(2,3,LONG_DELAY_MS, MILLISECONDS,new ArrayBlockingQueue<Runnable>(10));
1368 : jsr166 1.2
1369 : jsr166 1.8 try {
1370 : jsr166 1.21 p.setKeepAliveTime(-1,MILLISECONDS);
1371 : dl 1.1 shouldThrow();
1372 : jsr166 1.4 } catch (IllegalArgumentException success) {
1373 : dl 1.1 } finally {
1374 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1375 : dl 1.1 }
1376 : jsr166 1.21 joinPool(p);
1377 : dl 1.1 }
1378 :    
1379 :     /**
1380 :     * terminated() is called on termination
1381 :     */
1382 :     public void testTerminated() {
1383 : jsr166 1.21 CustomTPE p = new CustomTPE();
1384 :     try { p.shutdown(); } catch (SecurityException ok) { return; }
1385 : jsr166 1.30 assertTrue(p.terminatedCalled());
1386 : jsr166 1.21 joinPool(p);
1387 : dl 1.1 }
1388 :    
1389 :     /**
1390 :     * beforeExecute and afterExecute are called when executing task
1391 :     */
1392 : jsr166 1.5 public void testBeforeAfter() throws InterruptedException {
1393 : jsr166 1.21 CustomTPE p = new CustomTPE();
1394 : dl 1.1 try {
1395 : jsr166 1.30 final CountDownLatch done = new CountDownLatch(1);
1396 : jsr166 1.32 p.execute(new CheckedRunnable() {
1397 : jsr166 1.30 public void realRun() {
1398 :     done.countDown();
1399 : jsr166 1.32 }});
1400 : jsr166 1.30 await(p.afterCalled);
1401 :     assertEquals(0, done.getCount());
1402 :     assertTrue(p.afterCalled());
1403 :     assertTrue(p.beforeCalled());
1404 : jsr166 1.21 try { p.shutdown(); } catch (SecurityException ok) { return; }
1405 : dl 1.1 } finally {
1406 : jsr166 1.21 joinPool(p);
1407 : dl 1.1 }
1408 :     }
1409 :    
1410 :     /**
1411 :     * completed submit of callable returns result
1412 :     */
1413 : jsr166 1.5 public void testSubmitCallable() throws Exception {
1414 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1415 : dl 1.1 try {
1416 :     Future<String> future = e.submit(new StringTask());
1417 :     String result = future.get();
1418 :     assertSame(TEST_STRING, result);
1419 :     } finally {
1420 :     joinPool(e);
1421 :     }
1422 :     }
1423 :    
1424 :     /**
1425 :     * completed submit of runnable returns successfully
1426 :     */
1427 : jsr166 1.5 public void testSubmitRunnable() throws Exception {
1428 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1429 : dl 1.1 try {
1430 :     Future<?> future = e.submit(new NoOpRunnable());
1431 :     future.get();
1432 :     assertTrue(future.isDone());
1433 :     } finally {
1434 :     joinPool(e);
1435 :     }
1436 :     }
1437 :    
1438 :     /**
1439 :     * completed submit of (runnable, result) returns result
1440 :     */
1441 : jsr166 1.5 public void testSubmitRunnable2() throws Exception {
1442 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1443 : dl 1.1 try {
1444 :     Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
1445 :     String result = future.get();
1446 :     assertSame(TEST_STRING, result);
1447 :     } finally {
1448 :     joinPool(e);
1449 :     }
1450 :     }
1451 :    
1452 :     /**
1453 :     * invokeAny(null) throws NPE
1454 :     */
1455 : jsr166 1.5 public void testInvokeAny1() throws Exception {
1456 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1457 : dl 1.1 try {
1458 :     e.invokeAny(null);
1459 : jsr166 1.5 shouldThrow();
1460 : dl 1.1 } catch (NullPointerException success) {
1461 :     } finally {
1462 :     joinPool(e);
1463 :     }
1464 :     }
1465 :    
1466 :     /**
1467 :     * invokeAny(empty collection) throws IAE
1468 :     */
1469 : jsr166 1.5 public void testInvokeAny2() throws Exception {
1470 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1471 : dl 1.1 try {
1472 :     e.invokeAny(new ArrayList<Callable<String>>());
1473 : jsr166 1.5 shouldThrow();
1474 : dl 1.1 } catch (IllegalArgumentException success) {
1475 :     } finally {
1476 :     joinPool(e);
1477 :     }
1478 :     }
1479 :    
1480 :     /**
1481 :     * invokeAny(c) throws NPE if c has null elements
1482 :     */
1483 : jsr166 1.5 public void testInvokeAny3() throws Exception {
1484 : jsr166 1.17 CountDownLatch latch = new CountDownLatch(1);
1485 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1486 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1487 :     l.add(latchAwaitingStringTask(latch));
1488 :     l.add(null);
1489 : dl 1.1 try {
1490 :     e.invokeAny(l);
1491 : jsr166 1.5 shouldThrow();
1492 : dl 1.1 } catch (NullPointerException success) {
1493 :     } finally {
1494 : jsr166 1.5 latch.countDown();
1495 : dl 1.1 joinPool(e);
1496 :     }
1497 :     }
1498 :    
1499 :     /**
1500 :     * invokeAny(c) throws ExecutionException if no task completes
1501 :     */
1502 : jsr166 1.5 public void testInvokeAny4() throws Exception {
1503 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1504 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1505 :     l.add(new NPETask());
1506 : dl 1.1 try {
1507 :     e.invokeAny(l);
1508 : jsr166 1.5 shouldThrow();
1509 : dl 1.1 } catch (ExecutionException success) {
1510 : jsr166 1.13 assertTrue(success.getCause() instanceof NullPointerException);
1511 : dl 1.1 } finally {
1512 :     joinPool(e);
1513 :     }
1514 :     }
1515 :    
1516 :     /**
1517 :     * invokeAny(c) returns result of some task
1518 :     */
1519 : jsr166 1.5 public void testInvokeAny5() throws Exception {
1520 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1521 : dl 1.1 try {
1522 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1523 : dl 1.1 l.add(new StringTask());
1524 :     l.add(new StringTask());
1525 :     String result = e.invokeAny(l);
1526 :     assertSame(TEST_STRING, result);
1527 :     } finally {
1528 :     joinPool(e);
1529 :     }
1530 :     }
1531 :    
1532 :     /**
1533 :     * invokeAll(null) throws NPE
1534 :     */
1535 : jsr166 1.5 public void testInvokeAll1() throws Exception {
1536 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1537 : dl 1.1 try {
1538 :     e.invokeAll(null);
1539 : jsr166 1.5 shouldThrow();
1540 : dl 1.1 } catch (NullPointerException success) {
1541 :     } finally {
1542 :     joinPool(e);
1543 :     }
1544 :     }
1545 :    
1546 :     /**
1547 :     * invokeAll(empty collection) returns empty collection
1548 :     */
1549 : jsr166 1.5 public void testInvokeAll2() throws Exception {
1550 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1551 : dl 1.1 try {
1552 :     List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
1553 :     assertTrue(r.isEmpty());
1554 :     } finally {
1555 :     joinPool(e);
1556 :     }
1557 :     }
1558 :    
1559 :     /**
1560 :     * invokeAll(c) throws NPE if c has null elements
1561 :     */
1562 : jsr166 1.5 public void testInvokeAll3() throws Exception {
1563 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1564 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1565 :     l.add(new StringTask());
1566 :     l.add(null);
1567 : dl 1.1 try {
1568 :     e.invokeAll(l);
1569 : jsr166 1.5 shouldThrow();
1570 : dl 1.1 } catch (NullPointerException success) {
1571 :     } finally {
1572 :     joinPool(e);
1573 :     }
1574 :     }
1575 :    
1576 :     /**
1577 :     * get of element of invokeAll(c) throws exception on failed task
1578 :     */
1579 : jsr166 1.5 public void testInvokeAll4() throws Exception {
1580 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1581 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1582 :     l.add(new NPETask());
1583 :     List<Future<String>> futures = e.invokeAll(l);
1584 :     assertEquals(1, futures.size());
1585 : dl 1.1 try {
1586 : jsr166 1.17 futures.get(0).get();
1587 : jsr166 1.5 shouldThrow();
1588 : jsr166 1.3 } catch (ExecutionException success) {
1589 : jsr166 1.15 assertTrue(success.getCause() instanceof NullPointerException);
1590 : dl 1.1 } finally {
1591 :     joinPool(e);
1592 :     }
1593 :     }
1594 :    
1595 :     /**
1596 :     * invokeAll(c) returns results of all completed tasks
1597 :     */
1598 : jsr166 1.5 public void testInvokeAll5() throws Exception {
1599 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1600 : dl 1.1 try {
1601 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1602 : dl 1.1 l.add(new StringTask());
1603 :     l.add(new StringTask());
1604 : jsr166 1.17 List<Future<String>> futures = e.invokeAll(l);
1605 :     assertEquals(2, futures.size());
1606 :     for (Future<String> future : futures)
1607 : jsr166 1.6 assertSame(TEST_STRING, future.get());
1608 : dl 1.1 } finally {
1609 :     joinPool(e);
1610 :     }
1611 :     }
1612 :    
1613 :     /**
1614 :     * timed invokeAny(null) throws NPE
1615 :     */
1616 : jsr166 1.5 public void testTimedInvokeAny1() throws Exception {
1617 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1618 : dl 1.1 try {
1619 : jsr166 1.9 e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1620 : jsr166 1.5 shouldThrow();
1621 : dl 1.1 } catch (NullPointerException success) {
1622 :     } finally {
1623 :     joinPool(e);
1624 :     }
1625 :     }
1626 :    
1627 :     /**
1628 :     * timed invokeAny(,,null) throws NPE
1629 :     */
1630 : jsr166 1.5 public void testTimedInvokeAnyNullTimeUnit() throws Exception {
1631 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1632 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1633 :     l.add(new StringTask());
1634 : dl 1.1 try {
1635 :     e.invokeAny(l, MEDIUM_DELAY_MS, null);
1636 : jsr166 1.5 shouldThrow();
1637 : dl 1.1 } catch (NullPointerException success) {
1638 :     } finally {
1639 :     joinPool(e);
1640 :     }
1641 :     }
1642 :    
1643 :     /**
1644 :     * timed invokeAny(empty collection) throws IAE
1645 :     */
1646 : jsr166 1.5 public void testTimedInvokeAny2() throws Exception {
1647 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1648 : dl 1.1 try {
1649 : jsr166 1.9 e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1650 : jsr166 1.5 shouldThrow();
1651 : dl 1.1 } catch (IllegalArgumentException success) {
1652 :     } finally {
1653 :     joinPool(e);
1654 :     }
1655 :     }
1656 :    
1657 :     /**
1658 :     * timed invokeAny(c) throws NPE if c has null elements
1659 :     */
1660 : jsr166 1.5 public void testTimedInvokeAny3() throws Exception {
1661 : jsr166 1.17 CountDownLatch latch = new CountDownLatch(1);
1662 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1663 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1664 :     l.add(latchAwaitingStringTask(latch));
1665 :     l.add(null);
1666 : dl 1.1 try {
1667 : jsr166 1.9 e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1668 : jsr166 1.5 shouldThrow();
1669 : dl 1.1 } catch (NullPointerException success) {
1670 :     } finally {
1671 : jsr166 1.12 latch.countDown();
1672 : dl 1.1 joinPool(e);
1673 :     }
1674 :     }
1675 :    
1676 :     /**
1677 :     * timed invokeAny(c) throws ExecutionException if no task completes
1678 :     */
1679 : jsr166 1.5 public void testTimedInvokeAny4() throws Exception {
1680 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1681 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1682 :     l.add(new NPETask());
1683 : dl 1.1 try {
1684 : jsr166 1.9 e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1685 : jsr166 1.5 shouldThrow();
1686 : jsr166 1.3 } catch (ExecutionException success) {
1687 : jsr166 1.13 assertTrue(success.getCause() instanceof NullPointerException);
1688 : dl 1.1 } finally {
1689 :     joinPool(e);
1690 :     }
1691 :     }
1692 :    
1693 :     /**
1694 :     * timed invokeAny(c) returns result of some task
1695 :     */
1696 : jsr166 1.5 public void testTimedInvokeAny5() throws Exception {
1697 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1698 : dl 1.1 try {
1699 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1700 : dl 1.1 l.add(new StringTask());
1701 :     l.add(new StringTask());
1702 : jsr166 1.9 String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1703 : dl 1.1 assertSame(TEST_STRING, result);
1704 :     } finally {
1705 :     joinPool(e);
1706 :     }
1707 :     }
1708 :    
1709 :     /**
1710 :     * timed invokeAll(null) throws NPE
1711 :     */
1712 : jsr166 1.5 public void testTimedInvokeAll1() throws Exception {
1713 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1714 : dl 1.1 try {
1715 : jsr166 1.9 e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1716 : jsr166 1.5 shouldThrow();
1717 : dl 1.1 } catch (NullPointerException success) {
1718 :     } finally {
1719 :     joinPool(e);
1720 :     }
1721 :     }
1722 :    
1723 :     /**
1724 :     * timed invokeAll(,,null) throws NPE
1725 :     */
1726 : jsr166 1.5 public void testTimedInvokeAllNullTimeUnit() throws Exception {
1727 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1728 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1729 :     l.add(new StringTask());
1730 : dl 1.1 try {
1731 :     e.invokeAll(l, MEDIUM_DELAY_MS, null);
1732 : jsr166 1.5 shouldThrow();
1733 : dl 1.1 } catch (NullPointerException success) {
1734 :     } finally {
1735 :     joinPool(e);
1736 :     }
1737 :     }
1738 :    
1739 :     /**
1740 :     * timed invokeAll(empty collection) returns empty collection
1741 :     */
1742 : jsr166 1.5 public void testTimedInvokeAll2() throws Exception {
1743 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1744 : dl 1.1 try {
1745 : jsr166 1.9 List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1746 : dl 1.1 assertTrue(r.isEmpty());
1747 :     } finally {
1748 :     joinPool(e);
1749 :     }
1750 :     }
1751 :    
1752 :     /**
1753 :     * timed invokeAll(c) throws NPE if c has null elements
1754 :     */
1755 : jsr166 1.5 public void testTimedInvokeAll3() throws Exception {
1756 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1757 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1758 :     l.add(new StringTask());
1759 :     l.add(null);
1760 : dl 1.1 try {
1761 : jsr166 1.9 e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1762 : jsr166 1.5 shouldThrow();
1763 : dl 1.1 } catch (NullPointerException success) {
1764 :     } finally {
1765 :     joinPool(e);
1766 :     }
1767 :     }
1768 :    
1769 :     /**
1770 :     * get of element of invokeAll(c) throws exception on failed task
1771 :     */
1772 : jsr166 1.5 public void testTimedInvokeAll4() throws Exception {
1773 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1774 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1775 :     l.add(new NPETask());
1776 :     List<Future<String>> futures =
1777 :     e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1778 :     assertEquals(1, futures.size());
1779 : dl 1.1 try {
1780 : jsr166 1.17 futures.get(0).get();
1781 : jsr166 1.5 shouldThrow();
1782 : jsr166 1.3 } catch (ExecutionException success) {
1783 : jsr166 1.6 assertTrue(success.getCause() instanceof NullPointerException);
1784 : dl 1.1 } finally {
1785 :     joinPool(e);
1786 :     }
1787 :     }
1788 :    
1789 :     /**
1790 :     * timed invokeAll(c) returns results of all completed tasks
1791 :     */
1792 : jsr166 1.5 public void testTimedInvokeAll5() throws Exception {
1793 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1794 : dl 1.1 try {
1795 : jsr166 1.17 List<Callable<String>> l = new ArrayList<Callable<String>>();
1796 : dl 1.1 l.add(new StringTask());
1797 :     l.add(new StringTask());
1798 : jsr166 1.17 List<Future<String>> futures =
1799 : jsr166 1.45 e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1800 : jsr166 1.17 assertEquals(2, futures.size());
1801 :     for (Future<String> future : futures)
1802 : jsr166 1.6 assertSame(TEST_STRING, future.get());
1803 : dl 1.1 } finally {
1804 :     joinPool(e);
1805 :     }
1806 :     }
1807 :    
1808 :     /**
1809 :     * timed invokeAll(c) cancels tasks not completed by timeout
1810 :     */
1811 : jsr166 1.5 public void testTimedInvokeAll6() throws Exception {
1812 : jsr166 1.9 ExecutorService e = new CustomTPE(2, 2, LONG_DELAY_MS, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1813 : dl 1.1 try {
1814 : jsr166 1.39 for (long timeout = timeoutMillis();;) {
1815 :     List<Callable<String>> tasks = new ArrayList<>();
1816 :     tasks.add(new StringTask("0"));
1817 :     tasks.add(Executors.callable(new LongPossiblyInterruptedRunnable(), TEST_STRING));
1818 :     tasks.add(new StringTask("2"));
1819 :     long startTime = System.nanoTime();
1820 :     List<Future<String>> futures =
1821 :     e.invokeAll(tasks, timeout, MILLISECONDS);
1822 :     assertEquals(tasks.size(), futures.size());
1823 :     assertTrue(millisElapsedSince(startTime) >= timeout);
1824 :     for (Future future : futures)
1825 :     assertTrue(future.isDone());
1826 :     assertTrue(futures.get(1).isCancelled());
1827 :     try {
1828 :     assertEquals("0", futures.get(0).get());
1829 :     assertEquals("2", futures.get(2).get());
1830 :     break;
1831 :     } catch (CancellationException retryWithLongerTimeout) {
1832 :     timeout *= 2;
1833 :     if (timeout >= LONG_DELAY_MS / 2)
1834 :     fail("expected exactly one task to be cancelled");
1835 :     }
1836 :     }
1837 : dl 1.1 } finally {
1838 :     joinPool(e);
1839 :     }
1840 :     }
1841 :    
1842 :     /**
1843 :     * Execution continues if there is at least one thread even if
1844 :     * thread factory fails to create more
1845 :     */
1846 : jsr166 1.5 public void testFailingThreadFactory() throws InterruptedException {
1847 : jsr166 1.21 final ExecutorService e =
1848 :     new CustomTPE(100, 100,
1849 :     LONG_DELAY_MS, MILLISECONDS,
1850 :     new LinkedBlockingQueue<Runnable>(),
1851 :     new FailingThreadFactory());
1852 :     try {
1853 :     final int TASKS = 100;
1854 :     final CountDownLatch done = new CountDownLatch(TASKS);
1855 :     for (int k = 0; k < TASKS; ++k)
1856 :     e.execute(new CheckedRunnable() {
1857 :     public void realRun() {
1858 :     done.countDown();
1859 :     }});
1860 :     assertTrue(done.await(LONG_DELAY_MS, MILLISECONDS));
1861 : dl 1.1 } finally {
1862 :     joinPool(e);
1863 :     }
1864 :     }
1865 :    
1866 :     /**
1867 :     * allowsCoreThreadTimeOut is by default false.
1868 :     */
1869 :     public void testAllowsCoreThreadTimeOut() {
1870 : jsr166 1.21 ThreadPoolExecutor p = new CustomTPE(2, 2, 1000, MILLISECONDS, new ArrayBlockingQueue<Runnable>(10));
1871 :     assertFalse(p.allowsCoreThreadTimeOut());
1872 :     joinPool(p);
1873 : dl 1.1 }
1874 :    
1875 :     /**
1876 :     * allowCoreThreadTimeOut(true) causes idle threads to time out
1877 :     */
1878 : jsr166 1.21 public void testAllowCoreThreadTimeOut_true() throws Exception {
1879 : jsr166 1.39 long keepAliveTime = timeoutMillis();
1880 : jsr166 1.21 final ThreadPoolExecutor p =
1881 :     new CustomTPE(2, 10,
1882 : jsr166 1.39 keepAliveTime, MILLISECONDS,
1883 : jsr166 1.21 new ArrayBlockingQueue<Runnable>(10));
1884 :     final CountDownLatch threadStarted = new CountDownLatch(1);
1885 :     try {
1886 :     p.allowCoreThreadTimeOut(true);
1887 :     p.execute(new CheckedRunnable() {
1888 : jsr166 1.39 public void realRun() {
1889 : jsr166 1.21 threadStarted.countDown();
1890 :     assertEquals(1, p.getPoolSize());
1891 :     }});
1892 : jsr166 1.29 await(threadStarted);
1893 : jsr166 1.39 delay(keepAliveTime);
1894 : jsr166 1.29 long startTime = System.nanoTime();
1895 :     while (p.getPoolSize() > 0
1896 :     && millisElapsedSince(startTime) < LONG_DELAY_MS)
1897 :     Thread.yield();
1898 :     assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1899 : jsr166 1.21 assertEquals(0, p.getPoolSize());
1900 : dl 1.1 } finally {
1901 : jsr166 1.21 joinPool(p);
1902 : dl 1.1 }
1903 :     }
1904 :    
1905 :     /**
1906 :     * allowCoreThreadTimeOut(false) causes idle threads not to time out
1907 :     */
1908 : jsr166 1.21 public void testAllowCoreThreadTimeOut_false() throws Exception {
1909 : jsr166 1.39 long keepAliveTime = timeoutMillis();
1910 : jsr166 1.21 final ThreadPoolExecutor p =
1911 :     new CustomTPE(2, 10,
1912 : jsr166 1.39 keepAliveTime, MILLISECONDS,
1913 : jsr166 1.21 new ArrayBlockingQueue<Runnable>(10));
1914 :     final CountDownLatch threadStarted = new CountDownLatch(1);
1915 :     try {
1916 :     p.allowCoreThreadTimeOut(false);
1917 :     p.execute(new CheckedRunnable() {
1918 :     public void realRun() throws InterruptedException {
1919 :     threadStarted.countDown();
1920 :     assertTrue(p.getPoolSize() >= 1);
1921 :     }});
1922 : jsr166 1.39 delay(2 * keepAliveTime);
1923 : jsr166 1.21 assertTrue(p.getPoolSize() >= 1);
1924 : dl 1.1 } finally {
1925 : jsr166 1.21 joinPool(p);
1926 : dl 1.1 }
1927 :     }
1928 :    
1929 : jsr166 1.44 /**
1930 :     * get(cancelled task) throws CancellationException
1931 :     * (in part, a test of CustomTPE itself)
1932 :     */
1933 :     public void testGet_cancelled() throws Exception {
1934 :     final ExecutorService e =
1935 :     new CustomTPE(1, 1,
1936 :     LONG_DELAY_MS, MILLISECONDS,
1937 :     new LinkedBlockingQueue<Runnable>());
1938 :     try {
1939 :     final CountDownLatch blockerStarted = new CountDownLatch(1);
1940 :     final CountDownLatch done = new CountDownLatch(1);
1941 :     final List<Future<?>> futures = new ArrayList<>();
1942 :     for (int i = 0; i < 2; i++) {
1943 :     Runnable r = new CheckedRunnable() { public void realRun()
1944 :     throws Throwable {
1945 :     blockerStarted.countDown();
1946 :     assertTrue(done.await(2 * LONG_DELAY_MS, MILLISECONDS));
1947 :     }};
1948 :     futures.add(e.submit(r));
1949 :     }
1950 :     assertTrue(blockerStarted.await(LONG_DELAY_MS, MILLISECONDS));
1951 :     for (Future<?> future : futures) future.cancel(false);
1952 :     for (Future<?> future : futures) {
1953 :     try {
1954 :     future.get();
1955 :     shouldThrow();
1956 :     } catch (CancellationException success) {}
1957 :     try {
1958 :     future.get(LONG_DELAY_MS, MILLISECONDS);
1959 :     shouldThrow();
1960 :     } catch (CancellationException success) {}
1961 :     assertTrue(future.isCancelled());
1962 :     assertTrue(future.isDone());
1963 :     }
1964 :     done.countDown();
1965 :     } finally {
1966 :     joinPool(e);
1967 :     }
1968 :     }
1969 :    
1970 : dl 1.1 }

Doug Lea
ViewVC Help
Powered by ViewVC 1.0.8