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.20 by dl, Wed Aug 11 19:50:02 2010 UTC vs.
Revision 1.28 by dl, Fri Sep 17 14:19:52 2010 UTC

# Line 4 | Line 4
4   * http://creativecommons.org/licenses/publicdomain
5   */
6  
7
7   import junit.framework.*;
8 < import java.util.*;
9 < import java.util.concurrent.Executor;
8 > import java.util.ArrayList;
9 > 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;
# Line 20 | Line 20 | import java.util.concurrent.RejectedExec
20   import java.util.concurrent.ForkJoinPool;
21   import java.util.concurrent.ForkJoinTask;
22   import java.util.concurrent.ForkJoinWorkerThread;
23 import java.util.concurrent.RecursiveAction;
23   import java.util.concurrent.RecursiveTask;
24   import java.util.concurrent.TimeUnit;
25 < import java.util.concurrent.locks.*;
26 < import java.security.*;
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;
30 > import java.security.PrivilegedExceptionAction;
31  
32   public class ForkJoinPoolTest extends JSR166TestCase {
33      public static void main(String[] args) {
34 <        junit.textui.TestRunner.run (suite());
34 >        junit.textui.TestRunner.run(suite());
35      }
36 +
37      public static Test suite() {
38          return new TestSuite(ForkJoinPoolTest.class);
39      }
# Line 156 | Line 160 | public class ForkJoinPoolTest extends JS
160       * tasks, and quiescent running state.
161       */
162      public void testDefaultInitialState() {
163 <        ForkJoinPool p = null;
163 >        ForkJoinPool p = new ForkJoinPool(1);
164          try {
165 <            p = new ForkJoinPool(1);
166 <            assertTrue(p.getFactory() ==
163 <                       ForkJoinPool.defaultForkJoinWorkerThreadFactory);
165 >            assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
166 >                       p.getFactory());
167              assertTrue(p.isQuiescent());
168              assertFalse(p.getAsyncMode());
169 <            assertTrue(p.getActiveThreadCount() == 0);
170 <            assertTrue(p.getStealCount() == 0);
171 <            assertTrue(p.getQueuedTaskCount() == 0);
172 <            assertTrue(p.getQueuedSubmissionCount() == 0);
169 >            assertEquals(0, p.getActiveThreadCount());
170 >            assertEquals(0, p.getStealCount());
171 >            assertEquals(0, p.getQueuedTaskCount());
172 >            assertEquals(0, p.getQueuedSubmissionCount());
173              assertFalse(p.hasQueuedSubmissions());
174              assertFalse(p.isShutdown());
175              assertFalse(p.isTerminating());
# Line 201 | Line 204 | public class ForkJoinPoolTest extends JS
204       * getParallelism returns size set in constructor
205       */
206      public void testGetParallelism() {
207 <        ForkJoinPool p = null;
207 >        ForkJoinPool p = new ForkJoinPool(1);
208          try {
209 <            p = new ForkJoinPool(1);
207 <            assertTrue(p.getParallelism() == 1);
209 >            assertEquals(1, p.getParallelism());
210          } finally {
211              joinPool(p);
212          }
# Line 214 | Line 216 | public class ForkJoinPoolTest extends JS
216       * getPoolSize returns number of started workers.
217       */
218      public void testGetPoolSize() {
219 <        ForkJoinPool p = null;
219 >        ForkJoinPool p = new ForkJoinPool(1);
220          try {
221 <            p = new ForkJoinPool(1);
220 <            assertTrue(p.getActiveThreadCount() == 0);
221 >            assertEquals(0, p.getActiveThreadCount());
222              Future<String> future = p.submit(new StringTask());
223 <            assertTrue(p.getPoolSize() == 1);
223 <
223 >            assertEquals(1, p.getPoolSize());
224          } finally {
225              joinPool(p);
226          }
# Line 233 | Line 233 | public class ForkJoinPoolTest extends JS
233       * performs its defined action
234       */
235      public void testSetUncaughtExceptionHandler() throws InterruptedException {
236 <        ForkJoinPool p = null;
236 >        final CountDownLatch uncaughtExceptionHappened = new CountDownLatch(1);
237 >        final Thread.UncaughtExceptionHandler eh =
238 >            new Thread.UncaughtExceptionHandler() {
239 >                public void uncaughtException(Thread t, Throwable e) {
240 >                    uncaughtExceptionHappened.countDown();
241 >                }};
242 >        ForkJoinPool p = new ForkJoinPool(1, new FailingThreadFactory(),
243 >                                          eh, false);
244          try {
245 <            MyHandler eh = new MyHandler();
239 <            p = new ForkJoinPool(1, new FailingThreadFactory(), eh, false);
240 <            assert(eh == p.getUncaughtExceptionHandler());
245 >            assertSame(eh, p.getUncaughtExceptionHandler());
246              p.execute(new FailingTask());
247 <            Thread.sleep(MEDIUM_DELAY_MS);
243 <            assertTrue(eh.catches > 0);
247 >            uncaughtExceptionHappened.await();
248          } finally {
249 <            p.shutdownNow();
249 >            p.shutdownNow(); // failure might have prevented processing task
250              joinPool(p);
251          }
252      }
# Line 253 | Line 257 | public class ForkJoinPoolTest extends JS
257       * construction parameters continue to hold
258       */
259      public void testisQuiescent() throws InterruptedException {
260 <        ForkJoinPool p = null;
260 >        ForkJoinPool p = new ForkJoinPool(2);
261          try {
258            p = new ForkJoinPool(2);
262              p.invoke(new FibTask(20));
263 <            assertTrue(p.getFactory() ==
264 <                       ForkJoinPool.defaultForkJoinWorkerThreadFactory);
263 >            assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
264 >                       p.getFactory());
265              Thread.sleep(MEDIUM_DELAY_MS);
266              assertTrue(p.isQuiescent());
267              assertFalse(p.getAsyncMode());
268 <            assertTrue(p.getActiveThreadCount() == 0);
269 <            assertTrue(p.getQueuedTaskCount() == 0);
270 <            assertTrue(p.getQueuedSubmissionCount() == 0);
268 >            assertEquals(0, p.getActiveThreadCount());
269 >            assertEquals(0, p.getQueuedTaskCount());
270 >            assertEquals(0, p.getQueuedSubmissionCount());
271              assertFalse(p.hasQueuedSubmissions());
272              assertFalse(p.isShutdown());
273              assertFalse(p.isTerminating());
# Line 278 | Line 281 | public class ForkJoinPoolTest extends JS
281       * Completed submit(ForkJoinTask) returns result
282       */
283      public void testSubmitForkJoinTask() throws Throwable {
284 <        ForkJoinPool p = null;
284 >        ForkJoinPool p = new ForkJoinPool(1);
285          try {
283            p = new ForkJoinPool(1);
286              ForkJoinTask<Integer> f = p.submit(new FibTask(8));
287 <            int r = f.get();
286 <            assertTrue(r == 21);
287 >            assertEquals(21, (int) f.get());
288          } finally {
289              joinPool(p);
290          }
# Line 293 | Line 294 | public class ForkJoinPoolTest extends JS
294       * A task submitted after shutdown is rejected
295       */
296      public void testSubmitAfterShutdown() {
297 <        ForkJoinPool p = null;
297 >        ForkJoinPool p = new ForkJoinPool(1);
298          try {
298            p = new ForkJoinPool(1);
299              p.shutdown();
300              assertTrue(p.isShutdown());
301 <            ForkJoinTask<Integer> f = p.submit(new FibTask(8));
302 <            shouldThrow();
303 <        } catch (RejectedExecutionException success) {
301 >            try {
302 >                ForkJoinTask<Integer> f = p.submit(new FibTask(8));
303 >                shouldThrow();
304 >            } catch (RejectedExecutionException success) {}
305          } finally {
306              joinPool(p);
307          }
# Line 310 | Line 311 | public class ForkJoinPoolTest extends JS
311       * Pool maintains parallelism when using ManagedBlocker
312       */
313      public void testBlockingForkJoinTask() throws Throwable {
314 <        ForkJoinPool p = null;
314 >        ForkJoinPool p = new ForkJoinPool(4);
315          try {
315            p = new ForkJoinPool(4);
316              ReentrantLock lock = new ReentrantLock();
317              ManagedLocker locker = new ManagedLocker(lock);
318              ForkJoinTask<Integer> f = new LockingFibTask(30, locker, lock);
319              p.execute(f);
320 <            int r = f.get();
321 <            assertTrue(r == 832040);
320 >            assertEquals(832040, (int) f.get());
321          } finally {
322              p.shutdownNow(); // don't wait out shutdown
323          }
# Line 328 | Line 327 | public class ForkJoinPoolTest extends JS
327       * pollSubmission returns unexecuted submitted task, if present
328       */
329      public void testPollSubmission() {
330 <        SubFJP p = null;
330 >        SubFJP p = new SubFJP();
331          try {
333            p = new SubFJP();
332              ForkJoinTask a = p.submit(new MediumRunnable());
333              ForkJoinTask b = p.submit(new MediumRunnable());
334              ForkJoinTask c = p.submit(new MediumRunnable());
# Line 346 | Line 344 | public class ForkJoinPoolTest extends JS
344       * drainTasksTo transfers unexecuted submitted tasks, if present
345       */
346      public void testDrainTasksTo() {
347 <        SubFJP p = null;
347 >        SubFJP p = new SubFJP();
348          try {
351            p = new SubFJP();
349              ForkJoinTask a = p.submit(new MediumRunnable());
350              ForkJoinTask b = p.submit(new MediumRunnable());
351              ForkJoinTask c = p.submit(new MediumRunnable());
# Line 372 | Line 369 | public class ForkJoinPoolTest extends JS
369       */
370      public void testExecuteRunnable() throws Throwable {
371          ExecutorService e = new ForkJoinPool(1);
372 <        TrackedShortRunnable task = new TrackedShortRunnable();
373 <        assertFalse(task.done);
374 <        Future<?> future = e.submit(task);
375 <        future.get();
376 <        assertTrue(task.done);
372 >        try {
373 >            TrackedShortRunnable task = new TrackedShortRunnable();
374 >            assertFalse(task.done);
375 >            Future<?> future = e.submit(task);
376 >            future.get();
377 >            assertTrue(task.done);
378 >        } finally {
379 >            joinPool(e);
380 >        }
381      }
382  
383  
# Line 385 | Line 386 | public class ForkJoinPoolTest extends JS
386       */
387      public void testSubmitCallable() throws Throwable {
388          ExecutorService e = new ForkJoinPool(1);
389 <        Future<String> future = e.submit(new StringTask());
390 <        String result = future.get();
391 <        assertSame(TEST_STRING, result);
389 >        try {
390 >            Future<String> future = e.submit(new StringTask());
391 >            String result = future.get();
392 >            assertSame(TEST_STRING, result);
393 >        } finally {
394 >            joinPool(e);
395 >        }
396      }
397  
398      /**
# Line 395 | Line 400 | public class ForkJoinPoolTest extends JS
400       */
401      public void testSubmitRunnable() throws Throwable {
402          ExecutorService e = new ForkJoinPool(1);
403 <        Future<?> future = e.submit(new NoOpRunnable());
404 <        future.get();
405 <        assertTrue(future.isDone());
403 >        try {
404 >            Future<?> future = e.submit(new NoOpRunnable());
405 >            future.get();
406 >            assertTrue(future.isDone());
407 >        } finally {
408 >            joinPool(e);
409 >        }
410      }
411  
412      /**
# Line 405 | Line 414 | public class ForkJoinPoolTest extends JS
414       */
415      public void testSubmitRunnable2() throws Throwable {
416          ExecutorService e = new ForkJoinPool(1);
417 <        Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
418 <        String result = future.get();
419 <        assertSame(TEST_STRING, result);
417 >        try {
418 >            Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
419 >            String result = future.get();
420 >            assertSame(TEST_STRING, result);
421 >        } finally {
422 >            joinPool(e);
423 >        }
424      }
425  
426  
# Line 425 | Line 438 | public class ForkJoinPoolTest extends JS
438          } catch (AccessControlException ok) {
439              return;
440          }
441 +
442          try {
443              ExecutorService e = new ForkJoinPool(1);
444 <            Future future = e.submit(Executors.callable(new PrivilegedAction() {
444 >            try {
445 >                Future future = e.submit(Executors.callable(new PrivilegedAction() {
446                      public Object run() {
447                          return TEST_STRING;
448                      }}));
449  
450 <            Object result = future.get();
451 <            assertSame(TEST_STRING, result);
452 <        }
453 <        finally {
450 >                Object result = future.get();
451 >                assertSame(TEST_STRING, result);
452 >            } finally {
453 >                joinPool(e);
454 >            }
455 >        } finally {
456              Policy.setPolicy(savedPolicy);
457          }
458      }
# Line 457 | Line 474 | public class ForkJoinPoolTest extends JS
474  
475          try {
476              ExecutorService e = new ForkJoinPool(1);
477 <            Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
477 >            try {
478 >                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
479                      public Object run() {
480                          return TEST_STRING;
481                      }}));
482  
483 <            Object result = future.get();
484 <            assertSame(TEST_STRING, result);
485 <        }
486 <        finally {
483 >                Object result = future.get();
484 >                assertSame(TEST_STRING, result);
485 >            } finally {
486 >                joinPool(e);
487 >            }
488 >        } finally {
489              Policy.setPolicy(savedPolicy);
490          }
491      }
# Line 485 | Line 505 | public class ForkJoinPoolTest extends JS
505              return;
506          }
507  
488
508          try {
509              ExecutorService e = new ForkJoinPool(1);
510 <            Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
510 >            try {
511 >                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
512                      public Object run() throws Exception {
513                          throw new IndexOutOfBoundsException();
514                      }}));
515  
516 <            Object result = future.get();
517 <            shouldThrow();
518 <        } catch (ExecutionException success) {
519 <            assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
516 >                Object result = future.get();
517 >                shouldThrow();
518 >            } catch (ExecutionException success) {
519 >                assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
520 >            } finally {
521 >                joinPool(e);
522 >            }
523          } finally {
524              Policy.setPolicy(savedPolicy);
525          }
# Line 506 | Line 529 | public class ForkJoinPoolTest extends JS
529       * execute(null runnable) throws NullPointerException
530       */
531      public void testExecuteNullRunnable() {
532 +        ExecutorService e = new ForkJoinPool(1);
533 +        TrackedShortRunnable task = null;
534          try {
510            ExecutorService e = new ForkJoinPool(1);
511            TrackedShortRunnable task = null;
535              Future<?> future = e.submit(task);
536              shouldThrow();
537 <        } catch (NullPointerException success) {}
537 >        } catch (NullPointerException success) {
538 >        } finally {
539 >            joinPool(e);
540 >        }
541      }
542  
543  
# Line 519 | Line 545 | public class ForkJoinPoolTest extends JS
545       * submit(null callable) throws NullPointerException
546       */
547      public void testSubmitNullCallable() {
548 +        ExecutorService e = new ForkJoinPool(1);
549 +        StringTask t = null;
550          try {
523            ExecutorService e = new ForkJoinPool(1);
524            StringTask t = null;
551              Future<String> future = e.submit(t);
552              shouldThrow();
553 <        } catch (NullPointerException success) {}
553 >        } catch (NullPointerException success) {
554 >        } finally {
555 >            joinPool(e);
556 >        }
557      }
558  
559  
560      /**
561 <     * Blocking on submit(callable) throws InterruptedException if
533 <     * caller interrupted.
561 >     * submit(callable).get() throws InterruptedException if interrupted
562       */
563      public void testInterruptedSubmit() throws InterruptedException {
564 <        final ForkJoinPool p = new ForkJoinPool(1);
565 <
566 <        Thread t = new Thread(new CheckedInterruptedRunnable() {
567 <            public void realRun() throws Throwable {
568 <                p.submit(new CheckedCallable<Object>() {
569 <                    public Object realCall() throws Throwable {
570 <                        try {
571 <                            Thread.sleep(MEDIUM_DELAY_MS);
572 <                        } catch (InterruptedException ok) {
573 <                        }
574 <                        return null;
575 <                    }}).get();
576 <            }});
577 <
578 <        t.start();
579 <        Thread.sleep(SHORT_DELAY_MS);
580 <        t.interrupt();
581 <        t.join();
582 <        p.shutdownNow();
583 <        joinPool(p);
564 >        final CountDownLatch submitted    = new CountDownLatch(1);
565 >        final CountDownLatch quittingTime = new CountDownLatch(1);
566 >        final ExecutorService p = new ForkJoinPool(1);
567 >        final Callable<Void> awaiter = new CheckedCallable<Void>() {
568 >            public Void realCall() throws InterruptedException {
569 >                assertTrue(quittingTime.await(MEDIUM_DELAY_MS, MILLISECONDS));
570 >                return null;
571 >            }};
572 >        try {
573 >            Thread t = new Thread(new CheckedInterruptedRunnable() {
574 >                public void realRun() throws Exception {
575 >                    Future<Void> future = p.submit(awaiter);
576 >                    submitted.countDown();
577 >                    future.get();
578 >                }});
579 >            t.start();
580 >            assertTrue(submitted.await(MEDIUM_DELAY_MS, MILLISECONDS));
581 >            t.interrupt();
582 >            t.join();
583 >        } finally {
584 >            quittingTime.countDown();
585 >            joinPool(p);
586 >        }
587      }
588  
589      /**
# Line 570 | Line 601 | public class ForkJoinPoolTest extends JS
601              shouldThrow();
602          } catch (ExecutionException success) {
603              assertTrue(success.getCause() instanceof ArithmeticException);
604 +        } finally {
605 +            joinPool(p);
606          }
574        
575        joinPool(p);
607      }
608  
609      /**
# Line 761 | Line 792 | public class ForkJoinPoolTest extends JS
792      public void testTimedInvokeAny1() throws Throwable {
793          ExecutorService e = new ForkJoinPool(1);
794          try {
795 <            e.invokeAny(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
795 >            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
796              shouldThrow();
797          } catch (NullPointerException success) {
798          } finally {
# Line 792 | Line 823 | public class ForkJoinPoolTest extends JS
823          ExecutorService e = new ForkJoinPool(1);
824          try {
825              e.invokeAny(new ArrayList<Callable<String>>(),
826 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
826 >                        MEDIUM_DELAY_MS, MILLISECONDS);
827              shouldThrow();
828          } catch (IllegalArgumentException success) {
829          } finally {
# Line 810 | Line 841 | public class ForkJoinPoolTest extends JS
841          l.add(latchAwaitingStringTask(latch));
842          l.add(null);
843          try {
844 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
844 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
845              shouldThrow();
846          } catch (NullPointerException success) {
847          } finally {
# Line 827 | Line 858 | public class ForkJoinPoolTest extends JS
858          List<Callable<String>> l = new ArrayList<Callable<String>>();
859          l.add(new NPETask());
860          try {
861 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
861 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
862              shouldThrow();
863          } catch (ExecutionException success) {
864              assertTrue(success.getCause() instanceof NullPointerException);
# Line 845 | Line 876 | public class ForkJoinPoolTest extends JS
876              List<Callable<String>> l = new ArrayList<Callable<String>>();
877              l.add(new StringTask());
878              l.add(new StringTask());
879 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
879 >            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
880              assertSame(TEST_STRING, result);
881          } finally {
882              joinPool(e);
# Line 858 | Line 889 | public class ForkJoinPoolTest extends JS
889      public void testTimedInvokeAll1() throws Throwable {
890          ExecutorService e = new ForkJoinPool(1);
891          try {
892 <            e.invokeAll(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
892 >            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
893              shouldThrow();
894          } catch (NullPointerException success) {
895          } finally {
# Line 890 | Line 921 | public class ForkJoinPoolTest extends JS
921          try {
922              List<Future<String>> r
923                  = e.invokeAll(new ArrayList<Callable<String>>(),
924 <                              MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
924 >                              MEDIUM_DELAY_MS, MILLISECONDS);
925              assertTrue(r.isEmpty());
926          } finally {
927              joinPool(e);
# Line 906 | Line 937 | public class ForkJoinPoolTest extends JS
937          l.add(new StringTask());
938          l.add(null);
939          try {
940 <            e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
940 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
941              shouldThrow();
942          } catch (NullPointerException success) {
943          } finally {
# Line 922 | Line 953 | public class ForkJoinPoolTest extends JS
953          List<Callable<String>> l = new ArrayList<Callable<String>>();
954          l.add(new NPETask());
955          List<Future<String>> futures
956 <            = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
956 >            = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
957          assertEquals(1, futures.size());
958          try {
959              futures.get(0).get();
# Line 944 | Line 975 | public class ForkJoinPoolTest extends JS
975              l.add(new StringTask());
976              l.add(new StringTask());
977              List<Future<String>> futures
978 <                = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
978 >                = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
979              assertEquals(2, futures.size());
980              for (Future<String> future : futures)
981                  assertSame(TEST_STRING, future.get());

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines