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.36 by jsr166, Mon Nov 29 07:42:58 2010 UTC

# Line 23 | Line 23 | import java.util.concurrent.ForkJoinWork
23   import java.util.concurrent.RecursiveTask;
24   import java.util.concurrent.TimeUnit;
25   import java.util.concurrent.locks.ReentrantLock;
26 + import static java.util.concurrent.TimeUnit.MILLISECONDS;
27   import java.security.AccessControlException;
28   import java.security.Policy;
29   import java.security.PrivilegedAction;
# Line 232 | Line 233 | public class ForkJoinPoolTest extends JS
233       * performs its defined action
234       */
235      public void testSetUncaughtExceptionHandler() throws InterruptedException {
236 <        MyHandler eh = new MyHandler();
236 >        final CountDownLatch uehInvoked = new CountDownLatch(1);
237 >        final Thread.UncaughtExceptionHandler eh =
238 >            new Thread.UncaughtExceptionHandler() {
239 >                public void uncaughtException(Thread t, Throwable e) {
240 >                    uehInvoked.countDown();
241 >                }};
242          ForkJoinPool p = new ForkJoinPool(1, new FailingThreadFactory(),
243                                            eh, false);
244          try {
245              assertSame(eh, p.getUncaughtExceptionHandler());
246 <            p.execute(new FailingTask());
247 <            Thread.sleep(MEDIUM_DELAY_MS);
242 <            assertTrue(eh.catches > 0);
246 >            p.execute(new FibTask(8));
247 >            assertTrue(uehInvoked.await(MEDIUM_DELAY_MS, MILLISECONDS));
248          } finally {
249 <            p.shutdownNow();
249 >            p.shutdownNow(); // failure might have prevented processing task
250              joinPool(p);
251          }
252      }
# Line 254 | Line 259 | public class ForkJoinPoolTest extends JS
259      public void testisQuiescent() throws InterruptedException {
260          ForkJoinPool p = new ForkJoinPool(2);
261          try {
262 +            assertTrue(p.isQuiescent());
263              p.invoke(new FibTask(20));
264              assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
265                         p.getFactory());
266 <            Thread.sleep(MEDIUM_DELAY_MS);
266 >            Thread.sleep(SMALL_DELAY_MS);
267              assertTrue(p.isQuiescent());
268              assertFalse(p.getAsyncMode());
269              assertEquals(0, p.getActiveThreadCount());
# Line 310 | Line 316 | public class ForkJoinPoolTest extends JS
316          try {
317              ReentrantLock lock = new ReentrantLock();
318              ManagedLocker locker = new ManagedLocker(lock);
319 <            ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock);
319 >            ForkJoinTask<Integer> f = new LockingFibTask(20, locker, lock);
320              p.execute(f);
321 <            assertEquals(832040, (int) f.get());
321 >            assertEquals(6765, (int) f.get());
322          } finally {
323              p.shutdownNow(); // don't wait out shutdown
324          }
# Line 322 | Line 328 | public class ForkJoinPoolTest extends JS
328       * pollSubmission returns unexecuted submitted task, if present
329       */
330      public void testPollSubmission() {
331 +        final CountDownLatch done = new CountDownLatch(1);
332          SubFJP p = new SubFJP();
333          try {
334 <            ForkJoinTask a = p.submit(new MediumRunnable());
335 <            ForkJoinTask b = p.submit(new MediumRunnable());
336 <            ForkJoinTask c = p.submit(new MediumRunnable());
334 >            ForkJoinTask a = p.submit(awaiter(done));
335 >            ForkJoinTask b = p.submit(awaiter(done));
336 >            ForkJoinTask c = p.submit(awaiter(done));
337              ForkJoinTask r = p.pollSubmission();
338              assertTrue(r == a || r == b || r == c);
339              assertFalse(r.isDone());
340          } finally {
341 +            done.countDown();
342              joinPool(p);
343          }
344      }
# Line 339 | Line 347 | public class ForkJoinPoolTest extends JS
347       * drainTasksTo transfers unexecuted submitted tasks, if present
348       */
349      public void testDrainTasksTo() {
350 +        final CountDownLatch done = new CountDownLatch(1);
351          SubFJP p = new SubFJP();
352          try {
353 <            ForkJoinTask a = p.submit(new MediumRunnable());
354 <            ForkJoinTask b = p.submit(new MediumRunnable());
355 <            ForkJoinTask c = p.submit(new MediumRunnable());
353 >            ForkJoinTask a = p.submit(awaiter(done));
354 >            ForkJoinTask b = p.submit(awaiter(done));
355 >            ForkJoinTask c = p.submit(awaiter(done));
356              ArrayList<ForkJoinTask> al = new ArrayList();
357              p.drainTasksTo(al);
358              assertTrue(al.size() > 0);
# Line 352 | Line 361 | public class ForkJoinPoolTest extends JS
361                  assertFalse(r.isDone());
362              }
363          } finally {
364 +            done.countDown();
365              joinPool(p);
366          }
367      }
# Line 365 | Line 375 | public class ForkJoinPoolTest extends JS
375      public void testExecuteRunnable() throws Throwable {
376          ExecutorService e = new ForkJoinPool(1);
377          try {
378 <            TrackedShortRunnable task = new TrackedShortRunnable();
379 <            assertFalse(task.done);
378 >            TrackedRunnable task = trackedRunnable(SHORT_DELAY_MS);
379 >            assertFalse(task.isDone());
380              Future<?> future = e.submit(task);
381 <            future.get();
382 <            assertTrue(task.done);
381 >            assertNull(future.get());
382 >            assertNull(future.get(MEDIUM_DELAY_MS, MILLISECONDS));
383 >            assertTrue(task.isDone());
384 >            assertTrue(future.isDone());
385 >            assertFalse(future.isCancelled());
386          } finally {
387              joinPool(e);
388          }
# Line 383 | Line 396 | public class ForkJoinPoolTest extends JS
396          ExecutorService e = new ForkJoinPool(1);
397          try {
398              Future<String> future = e.submit(new StringTask());
399 <            String result = future.get();
400 <            assertSame(TEST_STRING, result);
399 >            assertSame(TEST_STRING, future.get());
400 >            assertTrue(future.isDone());
401 >            assertFalse(future.isCancelled());
402          } finally {
403              joinPool(e);
404          }
# Line 397 | Line 411 | public class ForkJoinPoolTest extends JS
411          ExecutorService e = new ForkJoinPool(1);
412          try {
413              Future<?> future = e.submit(new NoOpRunnable());
414 <            future.get();
414 >            assertNull(future.get());
415              assertTrue(future.isDone());
416 +            assertFalse(future.isCancelled());
417          } finally {
418              joinPool(e);
419          }
# Line 411 | Line 426 | public class ForkJoinPoolTest extends JS
426          ExecutorService e = new ForkJoinPool(1);
427          try {
428              Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
429 <            String result = future.get();
430 <            assertSame(TEST_STRING, result);
429 >            assertSame(TEST_STRING, future.get());
430 >            assertTrue(future.isDone());
431 >            assertFalse(future.isCancelled());
432          } finally {
433              joinPool(e);
434          }
435      }
436  
421
437      /**
438 <     * A submitted privileged action to completion
438 >     * A submitted privileged action runs to completion
439       */
440 <    public void testSubmitPrivilegedAction() throws Throwable {
441 <        Policy savedPolicy = null;
442 <        try {
443 <            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 {
440 >    public void testSubmitPrivilegedAction() throws Exception {
441 >        Runnable r = new CheckedRunnable() {
442 >            public void realRun() throws Exception {
443 >                ExecutorService e = new ForkJoinPool(1);
444                  Future future = e.submit(Executors.callable(new PrivilegedAction() {
445                      public Object run() {
446                          return TEST_STRING;
447                      }}));
448  
449 <                Object result = future.get();
450 <                assertSame(TEST_STRING, result);
451 <            } finally {
452 <                joinPool(e);
453 <            }
450 <        } finally {
451 <            Policy.setPolicy(savedPolicy);
452 <        }
449 >                assertSame(TEST_STRING, future.get());
450 >            }};
451 >
452 >        runWithPermissions(r,
453 >                           new RuntimePermission("modifyThread"));
454      }
455  
456      /**
457 <     * A submitted a privileged exception action runs to completion
457 >     * A submitted privileged exception action runs to completion
458       */
459 <    public void testSubmitPrivilegedExceptionAction() throws Throwable {
460 <        Policy savedPolicy = null;
461 <        try {
462 <            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 {
459 >    public void testSubmitPrivilegedExceptionAction() throws Exception {
460 >        Runnable r = new CheckedRunnable() {
461 >            public void realRun() throws Exception {
462 >                ExecutorService e = new ForkJoinPool(1);
463                  Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
464                      public Object run() {
465                          return TEST_STRING;
466                      }}));
467  
468 <                Object result = future.get();
469 <                assertSame(TEST_STRING, result);
470 <            } finally {
471 <                joinPool(e);
482 <            }
483 <        } finally {
484 <            Policy.setPolicy(savedPolicy);
485 <        }
468 >                assertSame(TEST_STRING, future.get());
469 >            }};
470 >
471 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
472      }
473  
474      /**
475       * A submitted failed privileged exception action reports exception
476       */
477 <    public void testSubmitFailedPrivilegedExceptionAction() throws Throwable {
478 <        Policy savedPolicy = null;
479 <        try {
480 <            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 {
477 >    public void testSubmitFailedPrivilegedExceptionAction() throws Exception {
478 >        Runnable r = new CheckedRunnable() {
479 >            public void realRun() throws Exception {
480 >                ExecutorService e = new ForkJoinPool(1);
481                  Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
482                      public Object run() throws Exception {
483                          throw new IndexOutOfBoundsException();
484                      }}));
485  
486 <                Object result = future.get();
487 <                shouldThrow();
488 <            } catch (ExecutionException success) {
489 <                assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
490 <            } finally {
491 <                joinPool(e);
492 <            }
493 <        } finally {
519 <            Policy.setPolicy(savedPolicy);
520 <        }
486 >                try {
487 >                    future.get();
488 >                    shouldThrow();
489 >                } catch (ExecutionException success) {
490 >                    assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
491 >                }}};
492 >
493 >        runWithPermissions(r, new RuntimePermission("modifyThread"));
494      }
495  
496      /**
# Line 525 | Line 498 | public class ForkJoinPoolTest extends JS
498       */
499      public void testExecuteNullRunnable() {
500          ExecutorService e = new ForkJoinPool(1);
528        TrackedShortRunnable task = null;
501          try {
502 <            Future<?> future = e.submit(task);
502 >            Future<?> future = e.submit((Runnable) null);
503              shouldThrow();
504          } catch (NullPointerException success) {
505          } finally {
# Line 541 | Line 513 | public class ForkJoinPoolTest extends JS
513       */
514      public void testSubmitNullCallable() {
515          ExecutorService e = new ForkJoinPool(1);
544        StringTask t = null;
516          try {
517 <            Future<String> future = e.submit(t);
517 >            Future<String> future = e.submit((Callable) null);
518              shouldThrow();
519          } catch (NullPointerException success) {
520          } finally {
# Line 553 | Line 524 | public class ForkJoinPoolTest extends JS
524  
525  
526      /**
527 <     * Blocking on submit(callable) throws InterruptedException if
557 <     * caller interrupted.
527 >     * submit(callable).get() throws InterruptedException if interrupted
528       */
529      public void testInterruptedSubmit() throws InterruptedException {
530 <        final ForkJoinPool p = new ForkJoinPool(1);
531 <
532 <        Thread t = new Thread(new CheckedInterruptedRunnable() {
533 <            public void realRun() throws Throwable {
534 <                p.submit(new CheckedCallable<Object>() {
535 <                    public Object realCall() throws Throwable {
536 <                        try {
537 <                            Thread.sleep(MEDIUM_DELAY_MS);
538 <                        } catch (InterruptedException ok) {
539 <                        }
540 <                        return null;
541 <                    }}).get();
542 <            }});
543 <
544 <        t.start();
545 <        Thread.sleep(SHORT_DELAY_MS);
546 <        t.interrupt();
547 <        t.join();
548 <        p.shutdownNow();
549 <        joinPool(p);
530 >        final CountDownLatch submitted    = new CountDownLatch(1);
531 >        final CountDownLatch quittingTime = new CountDownLatch(1);
532 >        final ExecutorService p = new ForkJoinPool(1);
533 >        final Callable<Void> awaiter = new CheckedCallable<Void>() {
534 >            public Void realCall() throws InterruptedException {
535 >                assertTrue(quittingTime.await(MEDIUM_DELAY_MS, MILLISECONDS));
536 >                return null;
537 >            }};
538 >        try {
539 >            Thread t = new Thread(new CheckedInterruptedRunnable() {
540 >                public void realRun() throws Exception {
541 >                    Future<Void> future = p.submit(awaiter);
542 >                    submitted.countDown();
543 >                    future.get();
544 >                }});
545 >            t.start();
546 >            assertTrue(submitted.await(MEDIUM_DELAY_MS, MILLISECONDS));
547 >            t.interrupt();
548 >            t.join();
549 >        } finally {
550 >            quittingTime.countDown();
551 >            joinPool(p);
552 >        }
553      }
554  
555      /**
# Line 785 | Line 758 | public class ForkJoinPoolTest extends JS
758      public void testTimedInvokeAny1() throws Throwable {
759          ExecutorService e = new ForkJoinPool(1);
760          try {
761 <            e.invokeAny(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
761 >            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
762              shouldThrow();
763          } catch (NullPointerException success) {
764          } finally {
# Line 816 | Line 789 | public class ForkJoinPoolTest extends JS
789          ExecutorService e = new ForkJoinPool(1);
790          try {
791              e.invokeAny(new ArrayList<Callable<String>>(),
792 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
792 >                        MEDIUM_DELAY_MS, MILLISECONDS);
793              shouldThrow();
794          } catch (IllegalArgumentException success) {
795          } finally {
# Line 834 | Line 807 | public class ForkJoinPoolTest extends JS
807          l.add(latchAwaitingStringTask(latch));
808          l.add(null);
809          try {
810 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
810 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
811              shouldThrow();
812          } catch (NullPointerException success) {
813          } finally {
# Line 851 | Line 824 | public class ForkJoinPoolTest extends JS
824          List<Callable<String>> l = new ArrayList<Callable<String>>();
825          l.add(new NPETask());
826          try {
827 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
827 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
828              shouldThrow();
829          } catch (ExecutionException success) {
830              assertTrue(success.getCause() instanceof NullPointerException);
# Line 869 | Line 842 | public class ForkJoinPoolTest extends JS
842              List<Callable<String>> l = new ArrayList<Callable<String>>();
843              l.add(new StringTask());
844              l.add(new StringTask());
845 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
845 >            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
846              assertSame(TEST_STRING, result);
847          } finally {
848              joinPool(e);
# Line 882 | Line 855 | public class ForkJoinPoolTest extends JS
855      public void testTimedInvokeAll1() throws Throwable {
856          ExecutorService e = new ForkJoinPool(1);
857          try {
858 <            e.invokeAll(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
858 >            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
859              shouldThrow();
860          } catch (NullPointerException success) {
861          } finally {
# Line 914 | Line 887 | public class ForkJoinPoolTest extends JS
887          try {
888              List<Future<String>> r
889                  = e.invokeAll(new ArrayList<Callable<String>>(),
890 <                              MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
890 >                              MEDIUM_DELAY_MS, MILLISECONDS);
891              assertTrue(r.isEmpty());
892          } finally {
893              joinPool(e);
# Line 930 | Line 903 | public class ForkJoinPoolTest extends JS
903          l.add(new StringTask());
904          l.add(null);
905          try {
906 <            e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
906 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
907              shouldThrow();
908          } catch (NullPointerException success) {
909          } finally {
# Line 946 | Line 919 | public class ForkJoinPoolTest extends JS
919          List<Callable<String>> l = new ArrayList<Callable<String>>();
920          l.add(new NPETask());
921          List<Future<String>> futures
922 <            = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
922 >            = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
923          assertEquals(1, futures.size());
924          try {
925              futures.get(0).get();
# Line 968 | Line 941 | public class ForkJoinPoolTest extends JS
941              l.add(new StringTask());
942              l.add(new StringTask());
943              List<Future<String>> futures
944 <                = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
944 >                = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
945              assertEquals(2, futures.size());
946              for (Future<String> future : futures)
947                  assertSame(TEST_STRING, future.get());

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines