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.26 by jsr166, Thu Sep 16 00:52:49 2010 UTC vs.
Revision 1.46 by jsr166, Tue Jan 10 16:31:30 2012 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 java.security.AccessControlException;
29   import java.security.Policy;
30   import java.security.PrivilegedAction;
# Line 163 | Line 165 | public class ForkJoinPoolTest extends JS
165          try {
166              assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
167                         p.getFactory());
166            assertTrue(p.isQuiescent());
168              assertFalse(p.getAsyncMode());
169              assertEquals(0, p.getActiveThreadCount());
170              assertEquals(0, p.getStealCount());
# Line 198 | Line 199 | public class ForkJoinPoolTest extends JS
199          } catch (NullPointerException success) {}
200      }
201  
201
202      /**
203       * getParallelism returns size set in constructor
204       */
# Line 232 | Line 232 | public class ForkJoinPoolTest extends JS
232       * performs its defined action
233       */
234      public void testSetUncaughtExceptionHandler() throws InterruptedException {
235 <        MyHandler eh = new MyHandler();
235 >        final CountDownLatch uehInvoked = new CountDownLatch(1);
236 >        final Thread.UncaughtExceptionHandler eh =
237 >            new Thread.UncaughtExceptionHandler() {
238 >                public void uncaughtException(Thread t, Throwable e) {
239 >                    uehInvoked.countDown();
240 >                }};
241          ForkJoinPool p = new ForkJoinPool(1, new FailingThreadFactory(),
242                                            eh, false);
243          try {
244              assertSame(eh, p.getUncaughtExceptionHandler());
245 <            p.execute(new FailingTask());
246 <            Thread.sleep(MEDIUM_DELAY_MS);
247 <            assertTrue(eh.catches > 0);
245 >            try {
246 >                p.execute(new FibTask(8));
247 >                assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS));
248 >            } catch (RejectedExecutionException ok) {
249 >            }
250          } finally {
251 <            p.shutdownNow();
251 >            p.shutdownNow(); // failure might have prevented processing task
252              joinPool(p);
253          }
254      }
255  
256      /**
257 <     * After invoking a single task, isQuiescent is true,
258 <     * queues are empty, threads are not active, and
259 <     * construction parameters continue to hold
257 >     * After invoking a single task, isQuiescent eventually becomes
258 >     * true, at which time queues are empty, threads are not active,
259 >     * the task has completed successfully, and construction
260 >     * parameters continue to hold
261       */
262 <    public void testisQuiescent() throws InterruptedException {
262 >    public void testIsQuiescent() throws Exception {
263          ForkJoinPool p = new ForkJoinPool(2);
264          try {
265 <            p.invoke(new FibTask(20));
265 >            assertTrue(p.isQuiescent());
266 >            long startTime = System.nanoTime();
267 >            FibTask f = new FibTask(20);
268 >            p.invoke(f);
269              assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
270                         p.getFactory());
271 <            Thread.sleep(MEDIUM_DELAY_MS);
271 >            while (! p.isQuiescent()) {
272 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
273 >                    throw new AssertionFailedError("timed out");
274 >                assertFalse(p.getAsyncMode());
275 >                assertFalse(p.isShutdown());
276 >                assertFalse(p.isTerminating());
277 >                assertFalse(p.isTerminated());
278 >                Thread.yield();
279 >            }
280 >
281              assertTrue(p.isQuiescent());
282              assertFalse(p.getAsyncMode());
283              assertEquals(0, p.getActiveThreadCount());
# Line 267 | Line 287 | public class ForkJoinPoolTest extends JS
287              assertFalse(p.isShutdown());
288              assertFalse(p.isTerminating());
289              assertFalse(p.isTerminated());
290 +            assertTrue(f.isDone());
291 +            assertEquals(6765, (int) f.get());
292          } finally {
293              joinPool(p);
294          }
# Line 310 | Line 332 | public class ForkJoinPoolTest extends JS
332          try {
333              ReentrantLock lock = new ReentrantLock();
334              ManagedLocker locker = new ManagedLocker(lock);
335 <            ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock);
335 >            ForkJoinTask<Integer> f = new LockingFibTask(20, locker, lock);
336              p.execute(f);
337 <            assertEquals(832040, (int) f.get());
337 >            assertEquals(6765, (int) f.get());
338          } finally {
339              p.shutdownNow(); // don't wait out shutdown
340          }
# Line 322 | Line 344 | public class ForkJoinPoolTest extends JS
344       * pollSubmission returns unexecuted submitted task, if present
345       */
346      public void testPollSubmission() {
347 +        final CountDownLatch done = new CountDownLatch(1);
348          SubFJP p = new SubFJP();
349          try {
350 <            ForkJoinTask a = p.submit(new MediumRunnable());
351 <            ForkJoinTask b = p.submit(new MediumRunnable());
352 <            ForkJoinTask c = p.submit(new MediumRunnable());
350 >            ForkJoinTask a = p.submit(awaiter(done));
351 >            ForkJoinTask b = p.submit(awaiter(done));
352 >            ForkJoinTask c = p.submit(awaiter(done));
353              ForkJoinTask r = p.pollSubmission();
354              assertTrue(r == a || r == b || r == c);
355              assertFalse(r.isDone());
356          } finally {
357 +            done.countDown();
358              joinPool(p);
359          }
360      }
# Line 339 | Line 363 | public class ForkJoinPoolTest extends JS
363       * drainTasksTo transfers unexecuted submitted tasks, if present
364       */
365      public void testDrainTasksTo() {
366 +        final CountDownLatch done = new CountDownLatch(1);
367          SubFJP p = new SubFJP();
368          try {
369 <            ForkJoinTask a = p.submit(new MediumRunnable());
370 <            ForkJoinTask b = p.submit(new MediumRunnable());
371 <            ForkJoinTask c = p.submit(new MediumRunnable());
369 >            ForkJoinTask a = p.submit(awaiter(done));
370 >            ForkJoinTask b = p.submit(awaiter(done));
371 >            ForkJoinTask c = p.submit(awaiter(done));
372              ArrayList<ForkJoinTask> al = new ArrayList();
373              p.drainTasksTo(al);
374              assertTrue(al.size() > 0);
# Line 352 | Line 377 | public class ForkJoinPoolTest extends JS
377                  assertFalse(r.isDone());
378              }
379          } finally {
380 +            done.countDown();
381              joinPool(p);
382          }
383      }
384  
359
385      // FJ Versions of AbstractExecutorService tests
386  
387      /**
# Line 365 | Line 390 | public class ForkJoinPoolTest extends JS
390      public void testExecuteRunnable() throws Throwable {
391          ExecutorService e = new ForkJoinPool(1);
392          try {
393 <            TrackedShortRunnable task = new TrackedShortRunnable();
394 <            assertFalse(task.done);
393 >            final AtomicBoolean done = new AtomicBoolean(false);
394 >            CheckedRunnable task = new CheckedRunnable() {
395 >                public void realRun() {
396 >                    done.set(true);
397 >                }};
398              Future<?> future = e.submit(task);
399 <            future.get();
400 <            assertTrue(task.done);
399 >            assertNull(future.get());
400 >            assertNull(future.get(0, MILLISECONDS));
401 >            assertTrue(done.get());
402 >            assertTrue(future.isDone());
403 >            assertFalse(future.isCancelled());
404          } finally {
405              joinPool(e);
406          }
407      }
408  
378
409      /**
410       * Completed submit(callable) returns result
411       */
# Line 383 | Line 413 | public class ForkJoinPoolTest extends JS
413          ExecutorService e = new ForkJoinPool(1);
414          try {
415              Future<String> future = e.submit(new StringTask());
416 <            String result = future.get();
417 <            assertSame(TEST_STRING, result);
416 >            assertSame(TEST_STRING, future.get());
417 >            assertTrue(future.isDone());
418 >            assertFalse(future.isCancelled());
419          } finally {
420              joinPool(e);
421          }
# Line 397 | Line 428 | public class ForkJoinPoolTest extends JS
428          ExecutorService e = new ForkJoinPool(1);
429          try {
430              Future<?> future = e.submit(new NoOpRunnable());
431 <            future.get();
431 >            assertNull(future.get());
432              assertTrue(future.isDone());
433 +            assertFalse(future.isCancelled());
434          } finally {
435              joinPool(e);
436          }
# Line 411 | Line 443 | public class ForkJoinPoolTest extends JS
443          ExecutorService e = new ForkJoinPool(1);
444          try {
445              Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
446 <            String result = future.get();
447 <            assertSame(TEST_STRING, result);
446 >            assertSame(TEST_STRING, future.get());
447 >            assertTrue(future.isDone());
448 >            assertFalse(future.isCancelled());
449          } finally {
450              joinPool(e);
451          }
452      }
453  
421
454      /**
455 <     * A submitted privileged action to completion
455 >     * A submitted privileged action runs to completion
456       */
457 <    public void testSubmitPrivilegedAction() throws Throwable {
458 <        Policy savedPolicy = null;
459 <        try {
460 <            savedPolicy = Policy.getPolicy();
429 <            AdjustablePolicy policy = new AdjustablePolicy();
430 <            policy.addPermission(new RuntimePermission("getContextClassLoader"));
431 <            policy.addPermission(new RuntimePermission("setContextClassLoader"));
432 <            Policy.setPolicy(policy);
433 <        } catch (AccessControlException ok) {
434 <            return;
435 <        }
436 <
437 <        try {
438 <            ExecutorService e = new ForkJoinPool(1);
439 <            try {
457 >    public void testSubmitPrivilegedAction() throws Exception {
458 >        Runnable r = new CheckedRunnable() {
459 >            public void realRun() throws Exception {
460 >                ExecutorService e = new ForkJoinPool(1);
461                  Future future = e.submit(Executors.callable(new PrivilegedAction() {
462                      public Object run() {
463                          return TEST_STRING;
464                      }}));
465  
466 <                Object result = future.get();
467 <                assertSame(TEST_STRING, result);
468 <            } finally {
469 <                joinPool(e);
470 <            }
450 <        } finally {
451 <            Policy.setPolicy(savedPolicy);
452 <        }
466 >                assertSame(TEST_STRING, future.get());
467 >            }};
468 >
469 >        runWithPermissions(r,
470 >                           new RuntimePermission("modifyThread"));
471      }
472  
473      /**
474 <     * A submitted a privileged exception action runs to completion
474 >     * A submitted privileged exception action runs to completion
475       */
476 <    public void testSubmitPrivilegedExceptionAction() throws Throwable {
477 <        Policy savedPolicy = null;
478 <        try {
479 <            savedPolicy = Policy.getPolicy();
462 <            AdjustablePolicy policy = new AdjustablePolicy();
463 <            policy.addPermission(new RuntimePermission("getContextClassLoader"));
464 <            policy.addPermission(new RuntimePermission("setContextClassLoader"));
465 <            Policy.setPolicy(policy);
466 <        } catch (AccessControlException ok) {
467 <            return;
468 <        }
469 <
470 <        try {
471 <            ExecutorService e = new ForkJoinPool(1);
472 <            try {
476 >    public void testSubmitPrivilegedExceptionAction() throws Exception {
477 >        Runnable r = new CheckedRunnable() {
478 >            public void realRun() throws Exception {
479 >                ExecutorService e = new ForkJoinPool(1);
480                  Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
481                      public Object run() {
482                          return TEST_STRING;
483                      }}));
484  
485 <                Object result = future.get();
486 <                assertSame(TEST_STRING, result);
487 <            } finally {
488 <                joinPool(e);
482 <            }
483 <        } finally {
484 <            Policy.setPolicy(savedPolicy);
485 <        }
485 >                assertSame(TEST_STRING, future.get());
486 >            }};
487 >
488 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
489      }
490  
491      /**
492       * A submitted failed privileged exception action reports exception
493       */
494 <    public void testSubmitFailedPrivilegedExceptionAction() throws Throwable {
495 <        Policy savedPolicy = null;
496 <        try {
497 <            savedPolicy = Policy.getPolicy();
495 <            AdjustablePolicy policy = new AdjustablePolicy();
496 <            policy.addPermission(new RuntimePermission("getContextClassLoader"));
497 <            policy.addPermission(new RuntimePermission("setContextClassLoader"));
498 <            Policy.setPolicy(policy);
499 <        } catch (AccessControlException ok) {
500 <            return;
501 <        }
502 <
503 <        try {
504 <            ExecutorService e = new ForkJoinPool(1);
505 <            try {
494 >    public void testSubmitFailedPrivilegedExceptionAction() throws Exception {
495 >        Runnable r = new CheckedRunnable() {
496 >            public void realRun() throws Exception {
497 >                ExecutorService e = new ForkJoinPool(1);
498                  Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
499                      public Object run() throws Exception {
500                          throw new IndexOutOfBoundsException();
501                      }}));
502  
503 <                Object result = future.get();
504 <                shouldThrow();
505 <            } catch (ExecutionException success) {
506 <                assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
507 <            } finally {
508 <                joinPool(e);
509 <            }
510 <        } finally {
519 <            Policy.setPolicy(savedPolicy);
520 <        }
503 >                try {
504 >                    future.get();
505 >                    shouldThrow();
506 >                } catch (ExecutionException success) {
507 >                    assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
508 >                }}};
509 >
510 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
511      }
512  
513      /**
# Line 525 | Line 515 | public class ForkJoinPoolTest extends JS
515       */
516      public void testExecuteNullRunnable() {
517          ExecutorService e = new ForkJoinPool(1);
528        TrackedShortRunnable task = null;
518          try {
519 <            Future<?> future = e.submit(task);
519 >            Future<?> future = e.submit((Runnable) null);
520              shouldThrow();
521          } catch (NullPointerException success) {
522          } finally {
# Line 535 | Line 524 | public class ForkJoinPoolTest extends JS
524          }
525      }
526  
538
527      /**
528       * submit(null callable) throws NullPointerException
529       */
530      public void testSubmitNullCallable() {
531          ExecutorService e = new ForkJoinPool(1);
544        StringTask t = null;
532          try {
533 <            Future<String> future = e.submit(t);
533 >            Future<String> future = e.submit((Callable) null);
534              shouldThrow();
535          } catch (NullPointerException success) {
536          } finally {
# Line 551 | Line 538 | public class ForkJoinPoolTest extends JS
538          }
539      }
540  
554
541      /**
542 <     * Blocking on submit(callable) throws InterruptedException if
557 <     * caller interrupted.
542 >     * submit(callable).get() throws InterruptedException if interrupted
543       */
544      public void testInterruptedSubmit() throws InterruptedException {
545 <        final ForkJoinPool p = new ForkJoinPool(1);
546 <
547 <        Thread t = new Thread(new CheckedInterruptedRunnable() {
548 <            public void realRun() throws Throwable {
549 <                p.submit(new CheckedCallable<Object>() {
550 <                    public Object realCall() throws Throwable {
551 <                        try {
552 <                            Thread.sleep(MEDIUM_DELAY_MS);
553 <                        } catch (InterruptedException ok) {
554 <                        }
555 <                        return null;
556 <                    }}).get();
557 <            }});
558 <
559 <        t.start();
560 <        Thread.sleep(SHORT_DELAY_MS);
561 <        t.interrupt();
562 <        t.join();
563 <        p.shutdownNow();
564 <        joinPool(p);
545 >        final CountDownLatch submitted    = new CountDownLatch(1);
546 >        final CountDownLatch quittingTime = new CountDownLatch(1);
547 >        final ExecutorService p = new ForkJoinPool(1);
548 >        final Callable<Void> awaiter = new CheckedCallable<Void>() {
549 >            public Void realCall() throws InterruptedException {
550 >                assertTrue(quittingTime.await(MEDIUM_DELAY_MS, MILLISECONDS));
551 >                return null;
552 >            }};
553 >        try {
554 >            Thread t = new Thread(new CheckedInterruptedRunnable() {
555 >                public void realRun() throws Exception {
556 >                    Future<Void> future = p.submit(awaiter);
557 >                    submitted.countDown();
558 >                    future.get();
559 >                }});
560 >            t.start();
561 >            assertTrue(submitted.await(MEDIUM_DELAY_MS, MILLISECONDS));
562 >            t.interrupt();
563 >            t.join();
564 >        } finally {
565 >            quittingTime.countDown();
566 >            joinPool(p);
567 >        }
568      }
569  
570      /**
# Line 778 | Line 766 | public class ForkJoinPoolTest extends JS
766          }
767      }
768  
781
769      /**
770       * timed invokeAny(null) throws NullPointerException
771       */
772      public void testTimedInvokeAny1() throws Throwable {
773          ExecutorService e = new ForkJoinPool(1);
774          try {
775 <            e.invokeAny(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
775 >            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
776              shouldThrow();
777          } catch (NullPointerException success) {
778          } finally {
# Line 816 | Line 803 | public class ForkJoinPoolTest extends JS
803          ExecutorService e = new ForkJoinPool(1);
804          try {
805              e.invokeAny(new ArrayList<Callable<String>>(),
806 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
806 >                        MEDIUM_DELAY_MS, MILLISECONDS);
807              shouldThrow();
808          } catch (IllegalArgumentException success) {
809          } finally {
# Line 834 | Line 821 | public class ForkJoinPoolTest extends JS
821          l.add(latchAwaitingStringTask(latch));
822          l.add(null);
823          try {
824 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
824 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
825              shouldThrow();
826          } catch (NullPointerException success) {
827          } finally {
# Line 851 | Line 838 | public class ForkJoinPoolTest extends JS
838          List<Callable<String>> l = new ArrayList<Callable<String>>();
839          l.add(new NPETask());
840          try {
841 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
841 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
842              shouldThrow();
843          } catch (ExecutionException success) {
844              assertTrue(success.getCause() instanceof NullPointerException);
# Line 869 | Line 856 | public class ForkJoinPoolTest extends JS
856              List<Callable<String>> l = new ArrayList<Callable<String>>();
857              l.add(new StringTask());
858              l.add(new StringTask());
859 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
859 >            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
860              assertSame(TEST_STRING, result);
861          } finally {
862              joinPool(e);
# Line 882 | Line 869 | public class ForkJoinPoolTest extends JS
869      public void testTimedInvokeAll1() throws Throwable {
870          ExecutorService e = new ForkJoinPool(1);
871          try {
872 <            e.invokeAll(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
872 >            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
873              shouldThrow();
874          } catch (NullPointerException success) {
875          } finally {
# Line 914 | Line 901 | public class ForkJoinPoolTest extends JS
901          try {
902              List<Future<String>> r
903                  = e.invokeAll(new ArrayList<Callable<String>>(),
904 <                              MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
904 >                              MEDIUM_DELAY_MS, MILLISECONDS);
905              assertTrue(r.isEmpty());
906          } finally {
907              joinPool(e);
# Line 930 | Line 917 | public class ForkJoinPoolTest extends JS
917          l.add(new StringTask());
918          l.add(null);
919          try {
920 <            e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
920 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
921              shouldThrow();
922          } catch (NullPointerException success) {
923          } finally {
# Line 946 | Line 933 | public class ForkJoinPoolTest extends JS
933          List<Callable<String>> l = new ArrayList<Callable<String>>();
934          l.add(new NPETask());
935          List<Future<String>> futures
936 <            = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
936 >            = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
937          assertEquals(1, futures.size());
938          try {
939              futures.get(0).get();
# Line 968 | Line 955 | public class ForkJoinPoolTest extends JS
955              l.add(new StringTask());
956              l.add(new StringTask());
957              List<Future<String>> futures
958 <                = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
958 >                = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
959              assertEquals(2, futures.size());
960              for (Future<String> future : futures)
961                  assertSame(TEST_STRING, future.get());

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines