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.23 by jsr166, Sat Sep 11 19:04:12 2010 UTC vs.
Revision 1.32 by jsr166, Mon Oct 11 04:39:12 2010 UTC

# Line 5 | Line 5
5   */
6  
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 19 | 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;
22 – 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) {
# Line 158 | Line 162 | public class ForkJoinPoolTest extends JS
162      public void testDefaultInitialState() {
163          ForkJoinPool p = new ForkJoinPool(1);
164          try {
165 <            assertTrue(p.getFactory() ==
166 <                       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 202 | Line 206 | public class ForkJoinPoolTest extends JS
206      public void testGetParallelism() {
207          ForkJoinPool p = new ForkJoinPool(1);
208          try {
209 <            assertTrue(p.getParallelism() == 1);
209 >            assertEquals(1, p.getParallelism());
210          } finally {
211              joinPool(p);
212          }
# Line 214 | Line 218 | public class ForkJoinPoolTest extends JS
218      public void testGetPoolSize() {
219          ForkJoinPool p = new ForkJoinPool(1);
220          try {
221 <            assertTrue(p.getActiveThreadCount() == 0);
221 >            assertEquals(0, p.getActiveThreadCount());
222              Future<String> future = p.submit(new StringTask());
223 <            assertTrue(p.getPoolSize() == 1);
223 >            assertEquals(1, p.getPoolSize());
224          } finally {
225              joinPool(p);
226          }
# Line 229 | Line 233 | public class ForkJoinPoolTest extends JS
233       * performs its defined action
234       */
235      public void testSetUncaughtExceptionHandler() throws InterruptedException {
236 <        MyHandler eh = new MyHandler();
237 <        ForkJoinPool p = new ForkJoinPool(1, new FailingThreadFactory(), eh, false);
238 <        try {
239 <            assert(eh == p.getUncaughtExceptionHandler());
240 <            p.execute(new FailingTask());
241 <            Thread.sleep(MEDIUM_DELAY_MS);
242 <            assertTrue(eh.catches > 0);
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 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 250 | 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 <            assertTrue(p.getFactory() ==
265 <                       ForkJoinPool.defaultForkJoinWorkerThreadFactory);
266 <            Thread.sleep(MEDIUM_DELAY_MS);
264 >            assertSame(ForkJoinPool.defaultForkJoinWorkerThreadFactory,
265 >                       p.getFactory());
266 >            Thread.sleep(SMALL_DELAY_MS);
267              assertTrue(p.isQuiescent());
268              assertFalse(p.getAsyncMode());
269 <            assertTrue(p.getActiveThreadCount() == 0);
270 <            assertTrue(p.getQueuedTaskCount() == 0);
271 <            assertTrue(p.getQueuedSubmissionCount() == 0);
269 >            assertEquals(0, p.getActiveThreadCount());
270 >            assertEquals(0, p.getQueuedTaskCount());
271 >            assertEquals(0, p.getQueuedSubmissionCount());
272              assertFalse(p.hasQueuedSubmissions());
273              assertFalse(p.isShutdown());
274              assertFalse(p.isTerminating());
# Line 275 | Line 285 | public class ForkJoinPoolTest extends JS
285          ForkJoinPool p = new ForkJoinPool(1);
286          try {
287              ForkJoinTask<Integer> f = p.submit(new FibTask(8));
288 <            int r = f.get();
279 <            assertTrue(r == 21);
288 >            assertEquals(21, (int) f.get());
289          } finally {
290              joinPool(p);
291          }
# Line 290 | Line 299 | public class ForkJoinPoolTest extends JS
299          try {
300              p.shutdown();
301              assertTrue(p.isShutdown());
302 <            ForkJoinTask<Integer> f = p.submit(new FibTask(8));
303 <            shouldThrow();
304 <        } catch (RejectedExecutionException success) {
302 >            try {
303 >                ForkJoinTask<Integer> f = p.submit(new FibTask(8));
304 >                shouldThrow();
305 >            } catch (RejectedExecutionException success) {}
306          } finally {
307              joinPool(p);
308          }
# Line 306 | 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 <            int r = f.get();
312 <            assertTrue(r == 832040);
321 >            assertEquals(6765, (int) f.get());
322          } finally {
323              p.shutdownNow(); // don't wait out shutdown
324          }
# Line 321 | Line 330 | public class ForkJoinPoolTest extends JS
330      public void testPollSubmission() {
331          SubFJP p = new SubFJP();
332          try {
333 <            ForkJoinTask a = p.submit(new MediumRunnable());
334 <            ForkJoinTask b = p.submit(new MediumRunnable());
335 <            ForkJoinTask c = p.submit(new MediumRunnable());
333 >            ForkJoinTask a = p.submit(new ShortRunnable());
334 >            ForkJoinTask b = p.submit(new ShortRunnable());
335 >            ForkJoinTask c = p.submit(new ShortRunnable());
336              ForkJoinTask r = p.pollSubmission();
337              assertTrue(r == a || r == b || r == c);
338              assertFalse(r.isDone());
# Line 338 | Line 347 | public class ForkJoinPoolTest extends JS
347      public void testDrainTasksTo() {
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(new ShortRunnable());
351 >            ForkJoinTask b = p.submit(new ShortRunnable());
352 >            ForkJoinTask c = p.submit(new ShortRunnable());
353              ArrayList<ForkJoinTask> al = new ArrayList();
354              p.drainTasksTo(al);
355              assertTrue(al.size() > 0);
# Line 362 | Line 371 | public class ForkJoinPoolTest extends JS
371      public void testExecuteRunnable() throws Throwable {
372          ExecutorService e = new ForkJoinPool(1);
373          try {
374 <            TrackedShortRunnable task = new TrackedShortRunnable();
375 <            assertFalse(task.done);
374 >            TrackedRunnable task = trackedRunnable(SHORT_DELAY_MS);
375 >            assertFalse(task.isDone());
376              Future<?> future = e.submit(task);
377              future.get();
378 <            assertTrue(task.done);
378 >            assertTrue(task.isDone());
379          } finally {
380              joinPool(e);
381          }
# Line 523 | Line 532 | public class ForkJoinPoolTest extends JS
532      public void testExecuteNullRunnable() {
533          ExecutorService e = new ForkJoinPool(1);
534          try {
535 <            TrackedShortRunnable task = null;
527 <            Future<?> future = e.submit(task);
535 >            Future<?> future = e.submit((Runnable) null);
536              shouldThrow();
537          } catch (NullPointerException success) {
538          } finally {
# Line 539 | Line 547 | public class ForkJoinPoolTest extends JS
547      public void testSubmitNullCallable() {
548          ExecutorService e = new ForkJoinPool(1);
549          try {
550 <            StringTask t = null;
543 <            Future<String> future = e.submit(t);
550 >            Future<String> future = e.submit((Callable) null);
551              shouldThrow();
552          } catch (NullPointerException success) {
553          } finally {
# Line 550 | Line 557 | public class ForkJoinPoolTest extends JS
557  
558  
559      /**
560 <     * Blocking on submit(callable) throws InterruptedException if
554 <     * caller interrupted.
560 >     * submit(callable).get() throws InterruptedException if interrupted
561       */
562      public void testInterruptedSubmit() throws InterruptedException {
563 <        final ForkJoinPool p = new ForkJoinPool(1);
564 <
565 <        Thread t = new Thread(new CheckedInterruptedRunnable() {
566 <            public void realRun() throws Throwable {
567 <                p.submit(new CheckedCallable<Object>() {
568 <                    public Object realCall() throws Throwable {
569 <                        try {
570 <                            Thread.sleep(MEDIUM_DELAY_MS);
571 <                        } catch (InterruptedException ok) {
572 <                        }
573 <                        return null;
574 <                    }}).get();
575 <            }});
576 <
577 <        t.start();
578 <        Thread.sleep(SHORT_DELAY_MS);
579 <        t.interrupt();
580 <        t.join();
581 <        p.shutdownNow();
582 <        joinPool(p);
563 >        final CountDownLatch submitted    = new CountDownLatch(1);
564 >        final CountDownLatch quittingTime = new CountDownLatch(1);
565 >        final ExecutorService p = new ForkJoinPool(1);
566 >        final Callable<Void> awaiter = new CheckedCallable<Void>() {
567 >            public Void realCall() throws InterruptedException {
568 >                assertTrue(quittingTime.await(MEDIUM_DELAY_MS, MILLISECONDS));
569 >                return null;
570 >            }};
571 >        try {
572 >            Thread t = new Thread(new CheckedInterruptedRunnable() {
573 >                public void realRun() throws Exception {
574 >                    Future<Void> future = p.submit(awaiter);
575 >                    submitted.countDown();
576 >                    future.get();
577 >                }});
578 >            t.start();
579 >            assertTrue(submitted.await(MEDIUM_DELAY_MS, MILLISECONDS));
580 >            t.interrupt();
581 >            t.join();
582 >        } finally {
583 >            quittingTime.countDown();
584 >            joinPool(p);
585 >        }
586      }
587  
588      /**
# Line 782 | Line 791 | public class ForkJoinPoolTest extends JS
791      public void testTimedInvokeAny1() throws Throwable {
792          ExecutorService e = new ForkJoinPool(1);
793          try {
794 <            e.invokeAny(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
794 >            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
795              shouldThrow();
796          } catch (NullPointerException success) {
797          } finally {
# Line 813 | Line 822 | public class ForkJoinPoolTest extends JS
822          ExecutorService e = new ForkJoinPool(1);
823          try {
824              e.invokeAny(new ArrayList<Callable<String>>(),
825 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
825 >                        MEDIUM_DELAY_MS, MILLISECONDS);
826              shouldThrow();
827          } catch (IllegalArgumentException success) {
828          } finally {
# Line 831 | Line 840 | public class ForkJoinPoolTest extends JS
840          l.add(latchAwaitingStringTask(latch));
841          l.add(null);
842          try {
843 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
843 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
844              shouldThrow();
845          } catch (NullPointerException success) {
846          } finally {
# Line 848 | Line 857 | public class ForkJoinPoolTest extends JS
857          List<Callable<String>> l = new ArrayList<Callable<String>>();
858          l.add(new NPETask());
859          try {
860 <            e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
860 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
861              shouldThrow();
862          } catch (ExecutionException success) {
863              assertTrue(success.getCause() instanceof NullPointerException);
# Line 866 | Line 875 | public class ForkJoinPoolTest extends JS
875              List<Callable<String>> l = new ArrayList<Callable<String>>();
876              l.add(new StringTask());
877              l.add(new StringTask());
878 <            String result = e.invokeAny(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
878 >            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
879              assertSame(TEST_STRING, result);
880          } finally {
881              joinPool(e);
# Line 879 | Line 888 | public class ForkJoinPoolTest extends JS
888      public void testTimedInvokeAll1() throws Throwable {
889          ExecutorService e = new ForkJoinPool(1);
890          try {
891 <            e.invokeAll(null, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
891 >            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
892              shouldThrow();
893          } catch (NullPointerException success) {
894          } finally {
# Line 911 | Line 920 | public class ForkJoinPoolTest extends JS
920          try {
921              List<Future<String>> r
922                  = e.invokeAll(new ArrayList<Callable<String>>(),
923 <                              MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
923 >                              MEDIUM_DELAY_MS, MILLISECONDS);
924              assertTrue(r.isEmpty());
925          } finally {
926              joinPool(e);
# Line 927 | Line 936 | public class ForkJoinPoolTest extends JS
936          l.add(new StringTask());
937          l.add(null);
938          try {
939 <            e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
939 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
940              shouldThrow();
941          } catch (NullPointerException success) {
942          } finally {
# Line 943 | Line 952 | public class ForkJoinPoolTest extends JS
952          List<Callable<String>> l = new ArrayList<Callable<String>>();
953          l.add(new NPETask());
954          List<Future<String>> futures
955 <            = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
955 >            = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
956          assertEquals(1, futures.size());
957          try {
958              futures.get(0).get();
# Line 965 | Line 974 | public class ForkJoinPoolTest extends JS
974              l.add(new StringTask());
975              l.add(new StringTask());
976              List<Future<String>> futures
977 <                = e.invokeAll(l, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
977 >                = e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
978              assertEquals(2, futures.size());
979              for (Future<String> future : futures)
980                  assertSame(TEST_STRING, future.get());

Diff Legend

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