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

Comparing jsr166/src/test/tck/ForkJoinPoolTest.java (file contents):
Revision 1.31 by jsr166, Sat Oct 9 19:46:42 2010 UTC vs.
Revision 1.50 by jsr166, Mon May 20 16:46:23 2013 UTC

# Line 1 | Line 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 < * http://creativecommons.org/licenses/publicdomain
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5   */
6  
7   import junit.framework.*;
# Line 22 | Line 22 | import java.util.concurrent.ForkJoinTask
22   import java.util.concurrent.ForkJoinWorkerThread;
23   import java.util.concurrent.RecursiveTask;
24   import java.util.concurrent.TimeUnit;
25 + import java.util.concurrent.atomic.AtomicBoolean;
26   import java.util.concurrent.locks.ReentrantLock;
27   import static java.util.concurrent.TimeUnit.MILLISECONDS;
28 + import static java.util.concurrent.TimeUnit.NANOSECONDS;
29   import java.security.AccessControlException;
30   import java.security.Policy;
31   import java.security.PrivilegedAction;
# Line 38 | Line 40 | public class ForkJoinPoolTest extends JS
40          return new TestSuite(ForkJoinPoolTest.class);
41      }
42  
43 <    /**
43 >    /*
44       * Testing coverage notes:
45       *
46       * 1. shutdown and related methods are tested via super.joinPool.
# Line 164 | Line 166 | public class ForkJoinPoolTest extends JS
166          try {
167              assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
168                         p.getFactory());
167 –            assertTrue(p.isQuiescent());
169              assertFalse(p.getAsyncMode());
170              assertEquals(0, p.getActiveThreadCount());
171              assertEquals(0, p.getStealCount());
# Line 199 | Line 200 | public class ForkJoinPoolTest extends JS
200          } catch (NullPointerException success) {}
201      }
202  
202 –
203      /**
204       * getParallelism returns size set in constructor
205       */
# Line 227 | Line 227 | public class ForkJoinPoolTest extends JS
227      }
228  
229      /**
230 +     * awaitTermination on a non-shutdown pool times out
231 +     */
232 +    public void testAwaitTermination_timesOut() throws InterruptedException {
233 +        ForkJoinPool p = new ForkJoinPool(1);
234 +        assertFalse(p.isTerminated());
235 +        assertFalse(p.awaitTermination(Long.MIN_VALUE, NANOSECONDS));
236 +        assertFalse(p.awaitTermination(Long.MIN_VALUE, MILLISECONDS));
237 +        assertFalse(p.awaitTermination(-1L, NANOSECONDS));
238 +        assertFalse(p.awaitTermination(-1L, MILLISECONDS));
239 +        assertFalse(p.awaitTermination(0L, NANOSECONDS));
240 +        assertFalse(p.awaitTermination(0L, MILLISECONDS));
241 +        long timeoutNanos = 999999L;
242 +        long startTime = System.nanoTime();
243 +        assertFalse(p.awaitTermination(timeoutNanos, NANOSECONDS));
244 +        assertTrue(System.nanoTime() - startTime >= timeoutNanos);
245 +        assertFalse(p.isTerminated());
246 +        startTime = System.nanoTime();
247 +        long timeoutMillis = timeoutMillis();
248 +        assertFalse(p.awaitTermination(timeoutMillis, MILLISECONDS));
249 +        assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
250 +        assertFalse(p.isTerminated());
251 +        p.shutdown();
252 +        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
253 +        assertTrue(p.isTerminated());
254 +    }
255 +
256 +    /**
257       * setUncaughtExceptionHandler changes handler for uncaught exceptions.
258       *
259       * Additionally tests: Overriding ForkJoinWorkerThread.onStart
# Line 243 | Line 270 | public class ForkJoinPoolTest extends JS
270                                            eh, false);
271          try {
272              assertSame(eh, p.getUncaughtExceptionHandler());
273 <            p.execute(new FibTask(8));
274 <            assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS));
273 >            try {
274 >                p.execute(new FibTask(8));
275 >                assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS));
276 >            } catch (RejectedExecutionException ok) {
277 >            }
278          } finally {
279              p.shutdownNow(); // failure might have prevented processing task
280              joinPool(p);
# Line 252 | Line 282 | public class ForkJoinPoolTest extends JS
282      }
283  
284      /**
285 <     * After invoking a single task, isQuiescent is true,
286 <     * queues are empty, threads are not active, and
287 <     * construction parameters continue to hold
285 >     * After invoking a single task, isQuiescent eventually becomes
286 >     * true, at which time queues are empty, threads are not active,
287 >     * the task has completed successfully, and construction
288 >     * parameters continue to hold
289       */
290 <    public void testisQuiescent() throws InterruptedException {
290 >    public void testIsQuiescent() throws Exception {
291          ForkJoinPool p = new ForkJoinPool(2);
292          try {
293              assertTrue(p.isQuiescent());
294 <            p.invoke(new FibTask(20));
294 >            long startTime = System.nanoTime();
295 >            FibTask f = new FibTask(20);
296 >            p.invoke(f);
297              assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
298                         p.getFactory());
299 <            Thread.sleep(SMALL_DELAY_MS);
299 >            while (! p.isQuiescent()) {
300 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
301 >                    throw new AssertionFailedError("timed out");
302 >                assertFalse(p.getAsyncMode());
303 >                assertFalse(p.isShutdown());
304 >                assertFalse(p.isTerminating());
305 >                assertFalse(p.isTerminated());
306 >                Thread.yield();
307 >            }
308 >
309              assertTrue(p.isQuiescent());
310              assertFalse(p.getAsyncMode());
311              assertEquals(0, p.getActiveThreadCount());
# Line 273 | Line 315 | public class ForkJoinPoolTest extends JS
315              assertFalse(p.isShutdown());
316              assertFalse(p.isTerminating());
317              assertFalse(p.isTerminated());
318 +            assertTrue(f.isDone());
319 +            assertEquals(6765, (int) f.get());
320          } finally {
321              joinPool(p);
322          }
# Line 316 | Line 360 | public class ForkJoinPoolTest extends JS
360          try {
361              ReentrantLock lock = new ReentrantLock();
362              ManagedLocker locker = new ManagedLocker(lock);
363 <            ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock);
363 >            ForkJoinTask<Integer> f = new LockingFibTask(20, locker, lock);
364              p.execute(f);
365 <            assertEquals(832040, (int) f.get());
365 >            assertEquals(6765, (int) f.get());
366          } finally {
367              p.shutdownNow(); // don't wait out shutdown
368          }
# Line 328 | Line 372 | public class ForkJoinPoolTest extends JS
372       * pollSubmission returns unexecuted submitted task, if present
373       */
374      public void testPollSubmission() {
375 +        final CountDownLatch done = new CountDownLatch(1);
376          SubFJP p = new SubFJP();
377          try {
378 <            ForkJoinTask a = p.submit(new MediumRunnable());
379 <            ForkJoinTask b = p.submit(new MediumRunnable());
380 <            ForkJoinTask c = p.submit(new MediumRunnable());
378 >            ForkJoinTask a = p.submit(awaiter(done));
379 >            ForkJoinTask b = p.submit(awaiter(done));
380 >            ForkJoinTask c = p.submit(awaiter(done));
381              ForkJoinTask r = p.pollSubmission();
382              assertTrue(r == a || r == b || r == c);
383              assertFalse(r.isDone());
384          } finally {
385 +            done.countDown();
386              joinPool(p);
387          }
388      }
# Line 345 | Line 391 | public class ForkJoinPoolTest extends JS
391       * drainTasksTo transfers unexecuted submitted tasks, if present
392       */
393      public void testDrainTasksTo() {
394 +        final CountDownLatch done = new CountDownLatch(1);
395          SubFJP p = new SubFJP();
396          try {
397 <            ForkJoinTask a = p.submit(new MediumRunnable());
398 <            ForkJoinTask b = p.submit(new MediumRunnable());
399 <            ForkJoinTask c = p.submit(new MediumRunnable());
397 >            ForkJoinTask a = p.submit(awaiter(done));
398 >            ForkJoinTask b = p.submit(awaiter(done));
399 >            ForkJoinTask c = p.submit(awaiter(done));
400              ArrayList<ForkJoinTask> al = new ArrayList();
401              p.drainTasksTo(al);
402              assertTrue(al.size() > 0);
# Line 358 | Line 405 | public class ForkJoinPoolTest extends JS
405                  assertFalse(r.isDone());
406              }
407          } finally {
408 +            done.countDown();
409              joinPool(p);
410          }
411      }
412  
365 –
413      // FJ Versions of AbstractExecutorService tests
414  
415      /**
# Line 371 | Line 418 | public class ForkJoinPoolTest extends JS
418      public void testExecuteRunnable() throws Throwable {
419          ExecutorService e = new ForkJoinPool(1);
420          try {
421 <            TrackedShortRunnable task = new TrackedShortRunnable();
422 <            assertFalse(task.done);
421 >            final AtomicBoolean done = new AtomicBoolean(false);
422 >            CheckedRunnable task = new CheckedRunnable() {
423 >                public void realRun() {
424 >                    done.set(true);
425 >                }};
426              Future<?> future = e.submit(task);
427 <            future.get();
428 <            assertTrue(task.done);
427 >            assertNull(future.get());
428 >            assertNull(future.get(0, MILLISECONDS));
429 >            assertTrue(done.get());
430 >            assertTrue(future.isDone());
431 >            assertFalse(future.isCancelled());
432          } finally {
433              joinPool(e);
434          }
435      }
436  
384 –
437      /**
438       * Completed submit(callable) returns result
439       */
# Line 389 | Line 441 | public class ForkJoinPoolTest extends JS
441          ExecutorService e = new ForkJoinPool(1);
442          try {
443              Future<String> future = e.submit(new StringTask());
444 <            String result = future.get();
445 <            assertSame(TEST_STRING, result);
444 >            assertSame(TEST_STRING, future.get());
445 >            assertTrue(future.isDone());
446 >            assertFalse(future.isCancelled());
447          } finally {
448              joinPool(e);
449          }
# Line 403 | Line 456 | public class ForkJoinPoolTest extends JS
456          ExecutorService e = new ForkJoinPool(1);
457          try {
458              Future<?> future = e.submit(new NoOpRunnable());
459 <            future.get();
459 >            assertNull(future.get());
460              assertTrue(future.isDone());
461 +            assertFalse(future.isCancelled());
462          } finally {
463              joinPool(e);
464          }
# Line 417 | Line 471 | public class ForkJoinPoolTest extends JS
471          ExecutorService e = new ForkJoinPool(1);
472          try {
473              Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
474 <            String result = future.get();
475 <            assertSame(TEST_STRING, result);
474 >            assertSame(TEST_STRING, future.get());
475 >            assertTrue(future.isDone());
476 >            assertFalse(future.isCancelled());
477          } finally {
478              joinPool(e);
479          }
480      }
481  
427 –
482      /**
483 <     * A submitted privileged action to completion
483 >     * A submitted privileged action runs to completion
484       */
485 <    public void testSubmitPrivilegedAction() throws Throwable {
486 <        Policy savedPolicy = null;
487 <        try {
488 <            savedPolicy = Policy.getPolicy();
489 <            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 {
485 >    public void testSubmitPrivilegedAction() throws Exception {
486 >        final Callable callable = Executors.callable(new PrivilegedAction() {
487 >                public Object run() { return TEST_STRING; }});
488 >        Runnable r = new CheckedRunnable() {
489 >        public void realRun() throws Exception {
490              ExecutorService e = new ForkJoinPool(1);
491              try {
492 <                Future future = e.submit(Executors.callable(new PrivilegedAction() {
493 <                    public Object run() {
448 <                        return TEST_STRING;
449 <                    }}));
450 <
451 <                Object result = future.get();
452 <                assertSame(TEST_STRING, result);
492 >                Future future = e.submit(callable);
493 >                assertSame(TEST_STRING, future.get());
494              } finally {
495                  joinPool(e);
496              }
497 <        } finally {
498 <            Policy.setPolicy(savedPolicy);
499 <        }
497 >        }};
498 >
499 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
500      }
501  
502      /**
503 <     * A submitted a privileged exception action runs to completion
503 >     * A submitted privileged exception action runs to completion
504       */
505 <    public void testSubmitPrivilegedExceptionAction() throws Throwable {
506 <        Policy savedPolicy = null;
507 <        try {
508 <            savedPolicy = Policy.getPolicy();
509 <            AdjustablePolicy policy = new AdjustablePolicy();
510 <            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 {
505 >    public void testSubmitPrivilegedExceptionAction() throws Exception {
506 >        final Callable callable =
507 >            Executors.callable(new PrivilegedExceptionAction() {
508 >                public Object run() { return TEST_STRING; }});
509 >        Runnable r = new CheckedRunnable() {
510 >        public void realRun() throws Exception {
511              ExecutorService e = new ForkJoinPool(1);
512              try {
513 <                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
514 <                    public Object run() {
481 <                        return TEST_STRING;
482 <                    }}));
483 <
484 <                Object result = future.get();
485 <                assertSame(TEST_STRING, result);
513 >                Future future = e.submit(callable);
514 >                assertSame(TEST_STRING, future.get());
515              } finally {
516                  joinPool(e);
517              }
518 <        } finally {
519 <            Policy.setPolicy(savedPolicy);
520 <        }
518 >        }};
519 >
520 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
521      }
522  
523      /**
524       * A submitted failed privileged exception action reports exception
525       */
526 <    public void testSubmitFailedPrivilegedExceptionAction() throws Throwable {
527 <        Policy savedPolicy = null;
528 <        try {
529 <            savedPolicy = Policy.getPolicy();
530 <            AdjustablePolicy policy = new AdjustablePolicy();
531 <            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 {
526 >    public void testSubmitFailedPrivilegedExceptionAction() throws Exception {
527 >        final Callable callable =
528 >            Executors.callable(new PrivilegedExceptionAction() {
529 >                public Object run() { throw new IndexOutOfBoundsException(); }});
530 >        Runnable r = new CheckedRunnable() {
531 >        public void realRun() throws Exception {
532              ExecutorService e = new ForkJoinPool(1);
533              try {
534 <                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
535 <                    public Object run() throws Exception {
536 <                        throw new IndexOutOfBoundsException();
537 <                    }}));
538 <
539 <                Object result = future.get();
540 <                shouldThrow();
519 <            } catch (ExecutionException success) {
520 <                assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
534 >                Future future = e.submit(callable);
535 >                try {
536 >                    future.get();
537 >                    shouldThrow();
538 >                } catch (ExecutionException success) {
539 >                    assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
540 >                }
541              } finally {
542                  joinPool(e);
543              }
544 <        } finally {
545 <            Policy.setPolicy(savedPolicy);
546 <        }
544 >        }};
545 >
546 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
547      }
548  
549      /**
# Line 540 | Line 560 | public class ForkJoinPoolTest extends JS
560          }
561      }
562  
543 –
563      /**
564       * submit(null callable) throws NullPointerException
565       */
# Line 555 | Line 574 | public class ForkJoinPoolTest extends JS
574          }
575      }
576  
558 –
577      /**
578       * submit(callable).get() throws InterruptedException if interrupted
579       */
# Line 593 | Line 611 | public class ForkJoinPoolTest extends JS
611          ForkJoinPool p = new ForkJoinPool(1);
612          try {
613              p.submit(new Callable() {
614 <                public Object call() {
615 <                    int i = 5/0;
598 <                    return Boolean.TRUE;
599 <                }}).get();
614 >                public Object call() { throw new ArithmeticException(); }})
615 >                .get();
616              shouldThrow();
617          } catch (ExecutionException success) {
618              assertTrue(success.getCause() instanceof ArithmeticException);
# Line 784 | Line 800 | public class ForkJoinPoolTest extends JS
800          }
801      }
802  
787 –
803      /**
804       * timed invokeAny(null) throws NullPointerException
805       */

Diff Legend

– Removed lines
+ Added lines
< Changed lines
> Changed lines