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.53 by jsr166, Wed Dec 31 16:44:01 2014 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 10 | Line 10 | import java.util.Collection;
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  
# Line 38 | Line 35 | public class ForkJoinPoolTest extends JS
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.
# Line 106 | Line 103 | public class ForkJoinPoolTest extends JS
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;
# Line 134 | Line 131 | public class ForkJoinPoolTest extends JS
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;
# Line 164 | Line 161 | public class ForkJoinPoolTest extends JS
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());
# Line 199 | Line 195 | public class ForkJoinPoolTest extends JS
195          } catch (NullPointerException success) {}
196      }
197  
202 –
198      /**
199       * getParallelism returns size set in constructor
200       */
# Line 227 | Line 222 | public class ForkJoinPoolTest extends JS
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
# Line 243 | Line 265 | public class ForkJoinPoolTest extends JS
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);
# Line 252 | Line 277 | public class ForkJoinPoolTest extends JS
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());
# Line 273 | Line 310 | public class ForkJoinPoolTest extends JS
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          }
# Line 316 | Line 355 | public class ForkJoinPoolTest extends JS
355          try {
356              ReentrantLock lock = new ReentrantLock();
357              ManagedLocker locker = new ManagedLocker(lock);
358 <            ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock);
358 >            ForkJoinTask<Integer> f = new LockingFibTask(20, locker, lock);
359              p.execute(f);
360 <            assertEquals(832040, (int) f.get());
360 >            assertEquals(6765, (int) f.get());
361          } finally {
362              p.shutdownNow(); // don't wait out shutdown
363          }
# Line 328 | Line 367 | public class ForkJoinPoolTest extends JS
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 MediumRunnable());
374 <            ForkJoinTask b = p.submit(new MediumRunnable());
375 <            ForkJoinTask c = p.submit(new MediumRunnable());
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      }
# Line 345 | Line 386 | public class ForkJoinPoolTest extends JS
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 MediumRunnable());
393 <            ForkJoinTask b = p.submit(new MediumRunnable());
394 <            ForkJoinTask c = p.submit(new MediumRunnable());
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);
# Line 358 | Line 400 | public class ForkJoinPoolTest extends JS
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      /**
# Line 371 | Line 413 | public class ForkJoinPoolTest extends JS
413      public void testExecuteRunnable() throws Throwable {
414          ExecutorService e = new ForkJoinPool(1);
415          try {
416 <            TrackedShortRunnable task = new TrackedShortRunnable();
417 <            assertFalse(task.done);
418 <            Future<?> future = e.submit(task);
419 <            future.get();
420 <            assertTrue(task.done);
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       */
# Line 389 | Line 435 | public class ForkJoinPoolTest extends JS
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          }
# Line 403 | Line 450 | public class ForkJoinPoolTest extends JS
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          }
# Line 417 | Line 465 | public class ForkJoinPoolTest extends JS
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      /**
# Line 540 | Line 554 | public class ForkJoinPoolTest extends JS
554          }
555      }
556  
543 –
557      /**
558       * submit(null callable) throws NullPointerException
559       */
# Line 555 | Line 568 | public class ForkJoinPoolTest extends JS
568          }
569      }
570  
558 –
571      /**
572       * submit(callable).get() throws InterruptedException if interrupted
573       */
# Line 593 | Line 605 | public class ForkJoinPoolTest extends JS
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);
# Line 784 | Line 794 | public class ForkJoinPoolTest extends JS
794          }
795      }
796  
787 –
797      /**
798       * timed invokeAny(null) throws NullPointerException
799       */

Diff Legend

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