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 |
< |
* http://creativecommons.org/licenses/publicdomain |
4 |
> |
* http://creativecommons.org/publicdomain/zero/1.0/ |
5 |
|
*/ |
6 |
|
|
7 |
|
import junit.framework.*; |
10 |
|
import java.util.List; |
11 |
|
import java.util.concurrent.Executors; |
12 |
|
import java.util.concurrent.ExecutorService; |
13 |
– |
import java.util.concurrent.AbstractExecutorService; |
13 |
|
import java.util.concurrent.CountDownLatch; |
14 |
|
import java.util.concurrent.Callable; |
15 |
|
import java.util.concurrent.Future; |
16 |
|
import java.util.concurrent.ExecutionException; |
18 |
– |
import java.util.concurrent.CancellationException; |
17 |
|
import java.util.concurrent.RejectedExecutionException; |
18 |
|
import java.util.concurrent.ForkJoinPool; |
19 |
|
import java.util.concurrent.ForkJoinTask; |
20 |
|
import java.util.concurrent.ForkJoinWorkerThread; |
21 |
|
import java.util.concurrent.RecursiveTask; |
22 |
< |
import java.util.concurrent.TimeUnit; |
22 |
> |
import java.util.concurrent.atomic.AtomicBoolean; |
23 |
|
import java.util.concurrent.locks.ReentrantLock; |
24 |
|
import static java.util.concurrent.TimeUnit.MILLISECONDS; |
25 |
< |
import java.security.AccessControlException; |
28 |
< |
import java.security.Policy; |
25 |
> |
import static java.util.concurrent.TimeUnit.NANOSECONDS; |
26 |
|
import java.security.PrivilegedAction; |
27 |
|
import java.security.PrivilegedExceptionAction; |
28 |
|
|
35 |
|
return new TestSuite(ForkJoinPoolTest.class); |
36 |
|
} |
37 |
|
|
38 |
< |
/** |
38 |
> |
/* |
39 |
|
* Testing coverage notes: |
40 |
|
* |
41 |
|
* 1. shutdown and related methods are tested via super.joinPool. |
103 |
|
static final class FibTask extends RecursiveTask<Integer> { |
104 |
|
final int number; |
105 |
|
FibTask(int n) { number = n; } |
106 |
< |
public Integer compute() { |
106 |
> |
protected Integer compute() { |
107 |
|
int n = number; |
108 |
|
if (n <= 1) |
109 |
|
return n; |
131 |
|
this.locker = locker; |
132 |
|
this.lock = lock; |
133 |
|
} |
134 |
< |
public Integer compute() { |
134 |
> |
protected Integer compute() { |
135 |
|
int n; |
136 |
|
LockingFibTask f1 = null; |
137 |
|
LockingFibTask f2 = null; |
161 |
|
try { |
162 |
|
assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory, |
163 |
|
p.getFactory()); |
167 |
– |
assertTrue(p.isQuiescent()); |
164 |
|
assertFalse(p.getAsyncMode()); |
165 |
|
assertEquals(0, p.getActiveThreadCount()); |
166 |
|
assertEquals(0, p.getStealCount()); |
195 |
|
} catch (NullPointerException success) {} |
196 |
|
} |
197 |
|
|
202 |
– |
|
198 |
|
/** |
199 |
|
* getParallelism returns size set in constructor |
200 |
|
*/ |
222 |
|
} |
223 |
|
|
224 |
|
/** |
225 |
+ |
* awaitTermination on a non-shutdown pool times out |
226 |
+ |
*/ |
227 |
+ |
public void testAwaitTermination_timesOut() throws InterruptedException { |
228 |
+ |
ForkJoinPool p = new ForkJoinPool(1); |
229 |
+ |
assertFalse(p.isTerminated()); |
230 |
+ |
assertFalse(p.awaitTermination(Long.MIN_VALUE, NANOSECONDS)); |
231 |
+ |
assertFalse(p.awaitTermination(Long.MIN_VALUE, MILLISECONDS)); |
232 |
+ |
assertFalse(p.awaitTermination(-1L, NANOSECONDS)); |
233 |
+ |
assertFalse(p.awaitTermination(-1L, MILLISECONDS)); |
234 |
+ |
assertFalse(p.awaitTermination(0L, NANOSECONDS)); |
235 |
+ |
assertFalse(p.awaitTermination(0L, MILLISECONDS)); |
236 |
+ |
long timeoutNanos = 999999L; |
237 |
+ |
long startTime = System.nanoTime(); |
238 |
+ |
assertFalse(p.awaitTermination(timeoutNanos, NANOSECONDS)); |
239 |
+ |
assertTrue(System.nanoTime() - startTime >= timeoutNanos); |
240 |
+ |
assertFalse(p.isTerminated()); |
241 |
+ |
startTime = System.nanoTime(); |
242 |
+ |
long timeoutMillis = timeoutMillis(); |
243 |
+ |
assertFalse(p.awaitTermination(timeoutMillis, MILLISECONDS)); |
244 |
+ |
assertTrue(millisElapsedSince(startTime) >= timeoutMillis); |
245 |
+ |
assertFalse(p.isTerminated()); |
246 |
+ |
p.shutdown(); |
247 |
+ |
assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS)); |
248 |
+ |
assertTrue(p.isTerminated()); |
249 |
+ |
} |
250 |
+ |
|
251 |
+ |
/** |
252 |
|
* setUncaughtExceptionHandler changes handler for uncaught exceptions. |
253 |
|
* |
254 |
|
* Additionally tests: Overriding ForkJoinWorkerThread.onStart |
265 |
|
eh, false); |
266 |
|
try { |
267 |
|
assertSame(eh, p.getUncaughtExceptionHandler()); |
268 |
< |
p.execute(new FibTask(8)); |
269 |
< |
assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS)); |
268 |
> |
try { |
269 |
> |
p.execute(new FibTask(8)); |
270 |
> |
assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS)); |
271 |
> |
} catch (RejectedExecutionException ok) { |
272 |
> |
} |
273 |
|
} finally { |
274 |
|
p.shutdownNow(); // failure might have prevented processing task |
275 |
|
joinPool(p); |
277 |
|
} |
278 |
|
|
279 |
|
/** |
280 |
< |
* After invoking a single task, isQuiescent is true, |
281 |
< |
* queues are empty, threads are not active, and |
282 |
< |
* construction parameters continue to hold |
280 |
> |
* After invoking a single task, isQuiescent eventually becomes |
281 |
> |
* true, at which time queues are empty, threads are not active, |
282 |
> |
* the task has completed successfully, and construction |
283 |
> |
* parameters continue to hold |
284 |
|
*/ |
285 |
< |
public void testisQuiescent() throws InterruptedException { |
285 |
> |
public void testIsQuiescent() throws Exception { |
286 |
|
ForkJoinPool p = new ForkJoinPool(2); |
287 |
|
try { |
288 |
|
assertTrue(p.isQuiescent()); |
289 |
< |
p.invoke(new FibTask(20)); |
289 |
> |
long startTime = System.nanoTime(); |
290 |
> |
FibTask f = new FibTask(20); |
291 |
> |
p.invoke(f); |
292 |
|
assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory, |
293 |
|
p.getFactory()); |
294 |
< |
Thread.sleep(SMALL_DELAY_MS); |
294 |
> |
while (! p.isQuiescent()) { |
295 |
> |
if (millisElapsedSince(startTime) > LONG_DELAY_MS) |
296 |
> |
throw new AssertionFailedError("timed out"); |
297 |
> |
assertFalse(p.getAsyncMode()); |
298 |
> |
assertFalse(p.isShutdown()); |
299 |
> |
assertFalse(p.isTerminating()); |
300 |
> |
assertFalse(p.isTerminated()); |
301 |
> |
Thread.yield(); |
302 |
> |
} |
303 |
> |
|
304 |
|
assertTrue(p.isQuiescent()); |
305 |
|
assertFalse(p.getAsyncMode()); |
306 |
|
assertEquals(0, p.getActiveThreadCount()); |
310 |
|
assertFalse(p.isShutdown()); |
311 |
|
assertFalse(p.isTerminating()); |
312 |
|
assertFalse(p.isTerminated()); |
313 |
+ |
assertTrue(f.isDone()); |
314 |
+ |
assertEquals(6765, (int) f.get()); |
315 |
|
} finally { |
316 |
|
joinPool(p); |
317 |
|
} |
367 |
|
* pollSubmission returns unexecuted submitted task, if present |
368 |
|
*/ |
369 |
|
public void testPollSubmission() { |
370 |
+ |
final CountDownLatch done = new CountDownLatch(1); |
371 |
|
SubFJP p = new SubFJP(); |
372 |
|
try { |
373 |
< |
ForkJoinTask a = p.submit(new ShortRunnable()); |
374 |
< |
ForkJoinTask b = p.submit(new ShortRunnable()); |
375 |
< |
ForkJoinTask c = p.submit(new ShortRunnable()); |
373 |
> |
ForkJoinTask a = p.submit(awaiter(done)); |
374 |
> |
ForkJoinTask b = p.submit(awaiter(done)); |
375 |
> |
ForkJoinTask c = p.submit(awaiter(done)); |
376 |
|
ForkJoinTask r = p.pollSubmission(); |
377 |
|
assertTrue(r == a || r == b || r == c); |
378 |
|
assertFalse(r.isDone()); |
379 |
|
} finally { |
380 |
+ |
done.countDown(); |
381 |
|
joinPool(p); |
382 |
|
} |
383 |
|
} |
386 |
|
* drainTasksTo transfers unexecuted submitted tasks, if present |
387 |
|
*/ |
388 |
|
public void testDrainTasksTo() { |
389 |
+ |
final CountDownLatch done = new CountDownLatch(1); |
390 |
|
SubFJP p = new SubFJP(); |
391 |
|
try { |
392 |
< |
ForkJoinTask a = p.submit(new ShortRunnable()); |
393 |
< |
ForkJoinTask b = p.submit(new ShortRunnable()); |
394 |
< |
ForkJoinTask c = p.submit(new ShortRunnable()); |
392 |
> |
ForkJoinTask a = p.submit(awaiter(done)); |
393 |
> |
ForkJoinTask b = p.submit(awaiter(done)); |
394 |
> |
ForkJoinTask c = p.submit(awaiter(done)); |
395 |
|
ArrayList<ForkJoinTask> al = new ArrayList(); |
396 |
|
p.drainTasksTo(al); |
397 |
|
assertTrue(al.size() > 0); |
400 |
|
assertFalse(r.isDone()); |
401 |
|
} |
402 |
|
} finally { |
403 |
+ |
done.countDown(); |
404 |
|
joinPool(p); |
405 |
|
} |
406 |
|
} |
407 |
|
|
365 |
– |
|
408 |
|
// FJ Versions of AbstractExecutorService tests |
409 |
|
|
410 |
|
/** |
413 |
|
public void testExecuteRunnable() throws Throwable { |
414 |
|
ExecutorService e = new ForkJoinPool(1); |
415 |
|
try { |
416 |
< |
TrackedRunnable task = trackedRunnable(SHORT_DELAY_MS); |
417 |
< |
assertFalse(task.isDone()); |
418 |
< |
Future<?> future = e.submit(task); |
419 |
< |
future.get(); |
420 |
< |
assertTrue(task.isDone()); |
416 |
> |
final AtomicBoolean done = new AtomicBoolean(false); |
417 |
> |
Future<?> future = e.submit(new CheckedRunnable() { |
418 |
> |
public void realRun() { |
419 |
> |
done.set(true); |
420 |
> |
}}); |
421 |
> |
assertNull(future.get()); |
422 |
> |
assertNull(future.get(0, MILLISECONDS)); |
423 |
> |
assertTrue(done.get()); |
424 |
> |
assertTrue(future.isDone()); |
425 |
> |
assertFalse(future.isCancelled()); |
426 |
|
} finally { |
427 |
|
joinPool(e); |
428 |
|
} |
429 |
|
} |
430 |
|
|
384 |
– |
|
431 |
|
/** |
432 |
|
* Completed submit(callable) returns result |
433 |
|
*/ |
435 |
|
ExecutorService e = new ForkJoinPool(1); |
436 |
|
try { |
437 |
|
Future<String> future = e.submit(new StringTask()); |
438 |
< |
String result = future.get(); |
439 |
< |
assertSame(TEST_STRING, result); |
438 |
> |
assertSame(TEST_STRING, future.get()); |
439 |
> |
assertTrue(future.isDone()); |
440 |
> |
assertFalse(future.isCancelled()); |
441 |
|
} finally { |
442 |
|
joinPool(e); |
443 |
|
} |
450 |
|
ExecutorService e = new ForkJoinPool(1); |
451 |
|
try { |
452 |
|
Future<?> future = e.submit(new NoOpRunnable()); |
453 |
< |
future.get(); |
453 |
> |
assertNull(future.get()); |
454 |
|
assertTrue(future.isDone()); |
455 |
+ |
assertFalse(future.isCancelled()); |
456 |
|
} finally { |
457 |
|
joinPool(e); |
458 |
|
} |
465 |
|
ExecutorService e = new ForkJoinPool(1); |
466 |
|
try { |
467 |
|
Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING); |
468 |
< |
String result = future.get(); |
469 |
< |
assertSame(TEST_STRING, result); |
468 |
> |
assertSame(TEST_STRING, future.get()); |
469 |
> |
assertTrue(future.isDone()); |
470 |
> |
assertFalse(future.isCancelled()); |
471 |
|
} finally { |
472 |
|
joinPool(e); |
473 |
|
} |
474 |
|
} |
475 |
|
|
427 |
– |
|
476 |
|
/** |
477 |
< |
* A submitted privileged action to completion |
477 |
> |
* A submitted privileged action runs to completion |
478 |
|
*/ |
479 |
< |
public void testSubmitPrivilegedAction() throws Throwable { |
480 |
< |
Policy savedPolicy = null; |
481 |
< |
try { |
482 |
< |
savedPolicy = Policy.getPolicy(); |
483 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
436 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
437 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
438 |
< |
Policy.setPolicy(policy); |
439 |
< |
} catch (AccessControlException ok) { |
440 |
< |
return; |
441 |
< |
} |
442 |
< |
|
443 |
< |
try { |
479 |
> |
public void testSubmitPrivilegedAction() throws Exception { |
480 |
> |
final Callable callable = Executors.callable(new PrivilegedAction() { |
481 |
> |
public Object run() { return TEST_STRING; }}); |
482 |
> |
Runnable r = new CheckedRunnable() { |
483 |
> |
public void realRun() throws Exception { |
484 |
|
ExecutorService e = new ForkJoinPool(1); |
485 |
|
try { |
486 |
< |
Future future = e.submit(Executors.callable(new PrivilegedAction() { |
487 |
< |
public Object run() { |
448 |
< |
return TEST_STRING; |
449 |
< |
}})); |
450 |
< |
|
451 |
< |
Object result = future.get(); |
452 |
< |
assertSame(TEST_STRING, result); |
486 |
> |
Future future = e.submit(callable); |
487 |
> |
assertSame(TEST_STRING, future.get()); |
488 |
|
} finally { |
489 |
|
joinPool(e); |
490 |
|
} |
491 |
< |
} finally { |
492 |
< |
Policy.setPolicy(savedPolicy); |
493 |
< |
} |
491 |
> |
}}; |
492 |
> |
|
493 |
> |
runWithPermissions(r, new RuntimePermission("modifyThread")); |
494 |
|
} |
495 |
|
|
496 |
|
/** |
497 |
< |
* A submitted a privileged exception action runs to completion |
497 |
> |
* A submitted privileged exception action runs to completion |
498 |
|
*/ |
499 |
< |
public void testSubmitPrivilegedExceptionAction() throws Throwable { |
500 |
< |
Policy savedPolicy = null; |
501 |
< |
try { |
502 |
< |
savedPolicy = Policy.getPolicy(); |
503 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
504 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
470 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
471 |
< |
Policy.setPolicy(policy); |
472 |
< |
} catch (AccessControlException ok) { |
473 |
< |
return; |
474 |
< |
} |
475 |
< |
|
476 |
< |
try { |
499 |
> |
public void testSubmitPrivilegedExceptionAction() throws Exception { |
500 |
> |
final Callable callable = |
501 |
> |
Executors.callable(new PrivilegedExceptionAction() { |
502 |
> |
public Object run() { return TEST_STRING; }}); |
503 |
> |
Runnable r = new CheckedRunnable() { |
504 |
> |
public void realRun() throws Exception { |
505 |
|
ExecutorService e = new ForkJoinPool(1); |
506 |
|
try { |
507 |
< |
Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() { |
508 |
< |
public Object run() { |
481 |
< |
return TEST_STRING; |
482 |
< |
}})); |
483 |
< |
|
484 |
< |
Object result = future.get(); |
485 |
< |
assertSame(TEST_STRING, result); |
507 |
> |
Future future = e.submit(callable); |
508 |
> |
assertSame(TEST_STRING, future.get()); |
509 |
|
} finally { |
510 |
|
joinPool(e); |
511 |
|
} |
512 |
< |
} finally { |
513 |
< |
Policy.setPolicy(savedPolicy); |
514 |
< |
} |
512 |
> |
}}; |
513 |
> |
|
514 |
> |
runWithPermissions(r, new RuntimePermission("modifyThread")); |
515 |
|
} |
516 |
|
|
517 |
|
/** |
518 |
|
* A submitted failed privileged exception action reports exception |
519 |
|
*/ |
520 |
< |
public void testSubmitFailedPrivilegedExceptionAction() throws Throwable { |
521 |
< |
Policy savedPolicy = null; |
522 |
< |
try { |
523 |
< |
savedPolicy = Policy.getPolicy(); |
524 |
< |
AdjustablePolicy policy = new AdjustablePolicy(); |
525 |
< |
policy.addPermission(new RuntimePermission("getContextClassLoader")); |
503 |
< |
policy.addPermission(new RuntimePermission("setContextClassLoader")); |
504 |
< |
Policy.setPolicy(policy); |
505 |
< |
} catch (AccessControlException ok) { |
506 |
< |
return; |
507 |
< |
} |
508 |
< |
|
509 |
< |
try { |
520 |
> |
public void testSubmitFailedPrivilegedExceptionAction() throws Exception { |
521 |
> |
final Callable callable = |
522 |
> |
Executors.callable(new PrivilegedExceptionAction() { |
523 |
> |
public Object run() { throw new IndexOutOfBoundsException(); }}); |
524 |
> |
Runnable r = new CheckedRunnable() { |
525 |
> |
public void realRun() throws Exception { |
526 |
|
ExecutorService e = new ForkJoinPool(1); |
527 |
|
try { |
528 |
< |
Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() { |
529 |
< |
public Object run() throws Exception { |
530 |
< |
throw new IndexOutOfBoundsException(); |
531 |
< |
}})); |
532 |
< |
|
533 |
< |
Object result = future.get(); |
534 |
< |
shouldThrow(); |
519 |
< |
} catch (ExecutionException success) { |
520 |
< |
assertTrue(success.getCause() instanceof IndexOutOfBoundsException); |
528 |
> |
Future future = e.submit(callable); |
529 |
> |
try { |
530 |
> |
future.get(); |
531 |
> |
shouldThrow(); |
532 |
> |
} catch (ExecutionException success) { |
533 |
> |
assertTrue(success.getCause() instanceof IndexOutOfBoundsException); |
534 |
> |
} |
535 |
|
} finally { |
536 |
|
joinPool(e); |
537 |
|
} |
538 |
< |
} finally { |
539 |
< |
Policy.setPolicy(savedPolicy); |
540 |
< |
} |
538 |
> |
}}; |
539 |
> |
|
540 |
> |
runWithPermissions(r, new RuntimePermission("modifyThread")); |
541 |
|
} |
542 |
|
|
543 |
|
/** |
554 |
|
} |
555 |
|
} |
556 |
|
|
543 |
– |
|
557 |
|
/** |
558 |
|
* submit(null callable) throws NullPointerException |
559 |
|
*/ |
568 |
|
} |
569 |
|
} |
570 |
|
|
558 |
– |
|
571 |
|
/** |
572 |
|
* submit(callable).get() throws InterruptedException if interrupted |
573 |
|
*/ |
605 |
|
ForkJoinPool p = new ForkJoinPool(1); |
606 |
|
try { |
607 |
|
p.submit(new Callable() { |
608 |
< |
public Object call() { |
609 |
< |
int i = 5/0; |
598 |
< |
return Boolean.TRUE; |
599 |
< |
}}).get(); |
608 |
> |
public Object call() { throw new ArithmeticException(); }}) |
609 |
> |
.get(); |
610 |
|
shouldThrow(); |
611 |
|
} catch (ExecutionException success) { |
612 |
|
assertTrue(success.getCause() instanceof ArithmeticException); |
794 |
|
} |
795 |
|
} |
796 |
|
|
787 |
– |
|
797 |
|
/** |
798 |
|
* timed invokeAny(null) throws NullPointerException |
799 |
|
*/ |