398 |
|
* getTaskCount increases, but doesn't overestimate, when tasks submitted |
399 |
|
*/ |
400 |
|
public void testGetTaskCount() throws InterruptedException { |
401 |
+ |
final int TASKS = 3; |
402 |
+ |
final CountDownLatch done = new CountDownLatch(1); |
403 |
|
final ThreadPoolExecutor p = |
404 |
|
new ThreadPoolExecutor(1, 1, |
405 |
|
LONG_DELAY_MS, MILLISECONDS, |
406 |
|
new ArrayBlockingQueue<Runnable>(10)); |
407 |
< |
try (PoolCleaner cleaner = cleaner(p)) { |
407 |
> |
try (PoolCleaner cleaner = cleaner(p, done)) { |
408 |
|
final CountDownLatch threadStarted = new CountDownLatch(1); |
407 |
– |
final CountDownLatch done = new CountDownLatch(1); |
409 |
|
assertEquals(0, p.getTaskCount()); |
410 |
+ |
assertEquals(0, p.getCompletedTaskCount()); |
411 |
|
p.execute(new CheckedRunnable() { |
412 |
|
public void realRun() throws InterruptedException { |
413 |
|
threadStarted.countDown(); |
412 |
– |
assertEquals(1, p.getTaskCount()); |
414 |
|
done.await(); |
415 |
|
}}); |
416 |
< |
assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS)); |
416 |
> |
assertTrue(threadStarted.await(LONG_DELAY_MS, MILLISECONDS)); |
417 |
|
assertEquals(1, p.getTaskCount()); |
418 |
< |
done.countDown(); |
418 |
> |
assertEquals(0, p.getCompletedTaskCount()); |
419 |
> |
for (int i = 0; i < TASKS; i++) { |
420 |
> |
assertEquals(1 + i, p.getTaskCount()); |
421 |
> |
p.execute(new CheckedRunnable() { |
422 |
> |
public void realRun() throws InterruptedException { |
423 |
> |
threadStarted.countDown(); |
424 |
> |
assertEquals(1 + TASKS, p.getTaskCount()); |
425 |
> |
done.await(); |
426 |
> |
}}); |
427 |
> |
} |
428 |
> |
assertEquals(1 + TASKS, p.getTaskCount()); |
429 |
> |
assertEquals(0, p.getCompletedTaskCount()); |
430 |
|
} |
431 |
+ |
assertEquals(1 + TASKS, p.getTaskCount()); |
432 |
+ |
assertEquals(1 + TASKS, p.getCompletedTaskCount()); |
433 |
|
} |
434 |
|
|
435 |
|
/** |
614 |
|
new ThreadPoolExecutor(1, 1, |
615 |
|
LONG_DELAY_MS, MILLISECONDS, |
616 |
|
q); |
617 |
< |
try (PoolCleaner cleaner = cleaner(p)) { |
617 |
> |
try (PoolCleaner cleaner = cleaner(p, done)) { |
618 |
|
FutureTask[] tasks = new FutureTask[5]; |
619 |
|
for (int i = 0; i < tasks.length; i++) { |
620 |
|
Callable task = new CheckedCallable<Boolean>() { |
626 |
|
tasks[i] = new FutureTask(task); |
627 |
|
p.execute(tasks[i]); |
628 |
|
} |
629 |
< |
assertTrue(threadStarted.await(MEDIUM_DELAY_MS, MILLISECONDS)); |
629 |
> |
assertTrue(threadStarted.await(LONG_DELAY_MS, MILLISECONDS)); |
630 |
|
assertEquals(tasks.length, p.getTaskCount()); |
631 |
|
assertEquals(tasks.length - 1, q.size()); |
632 |
|
assertEquals(1L, p.getActiveCount()); |
639 |
|
p.purge(); // Nothing to do |
640 |
|
assertEquals(tasks.length - 3, q.size()); |
641 |
|
assertEquals(tasks.length - 2, p.getTaskCount()); |
628 |
– |
done.countDown(); |
642 |
|
} |
643 |
|
} |
644 |
|
|