ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/FutureTaskTest.java
(Generate patch)

Comparing jsr166/src/test/tck/FutureTaskTest.java (file contents):
Revision 1.33 by jsr166, Thu Dec 20 04:58:53 2012 UTC vs.
Revision 1.55 by jsr166, Thu Sep 5 21:37:25 2019 UTC

# Line 6 | Line 6
6   * Pat Fisher, Mike Judd.
7   */
8  
9 < import junit.framework.*;
10 < import java.security.Permission;
9 > import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 > import static java.util.concurrent.TimeUnit.NANOSECONDS;
11 >
12 > import java.util.ArrayList;
13 > import java.util.List;
14 > import java.util.NoSuchElementException;
15   import java.util.concurrent.Callable;
16   import java.util.concurrent.CancellationException;
17   import java.util.concurrent.CountDownLatch;
18   import java.util.concurrent.ExecutionException;
19 + import java.util.concurrent.Executors;
20 + import java.util.concurrent.ExecutorService;
21   import java.util.concurrent.Future;
22   import java.util.concurrent.FutureTask;
23   import java.util.concurrent.TimeoutException;
24   import java.util.concurrent.atomic.AtomicInteger;
25 < import static java.util.concurrent.TimeUnit.MILLISECONDS;
26 < import static java.util.concurrent.TimeUnit.SECONDS;
27 < import java.util.*;
25 >
26 > import junit.framework.Test;
27 > import junit.framework.TestSuite;
28  
29   public class FutureTaskTest extends JSR166TestCase {
30  
31      public static void main(String[] args) {
32 <        junit.textui.TestRunner.run(suite());
32 >        main(suite(), args);
33      }
34      public static Test suite() {
35          return new TestSuite(FutureTaskTest.class);
# Line 38 | Line 44 | public class FutureTaskTest extends JSR1
44              assertEquals(1, pf.doneCount());
45              assertFalse(pf.runAndReset());
46              assertEquals(1, pf.doneCount());
47 +            Object r = null; Object exInfo = null;
48 +            try {
49 +                r = f.get();
50 +            } catch (CancellationException t) {
51 +                exInfo = CancellationException.class;
52 +            } catch (ExecutionException t) {
53 +                exInfo = t.getCause();
54 +            } catch (Throwable t) {
55 +                threadUnexpectedException(t);
56 +            }
57  
58              // Check that run and runAndReset have no effect.
59              int savedRunCount = pf.runCount();
44            int savedSetCount = pf.setCount();
45            int savedSetExceptionCount = pf.setExceptionCount();
60              pf.run();
61              pf.runAndReset();
62              assertEquals(savedRunCount, pf.runCount());
63 <            assertEquals(savedSetCount, pf.setCount());
64 <            assertEquals(savedSetExceptionCount, pf.setExceptionCount());
63 >            Object r2 = null;
64 >            try {
65 >                r2 = f.get();
66 >            } catch (CancellationException t) {
67 >                assertSame(exInfo, CancellationException.class);
68 >            } catch (ExecutionException t) {
69 >                assertSame(exInfo, t.getCause());
70 >            } catch (Throwable t) {
71 >                threadUnexpectedException(t);
72 >            }
73 >            if (exInfo == null)
74 >                assertSame(r, r2);
75              assertTrue(f.isDone());
76          }
77      }
# Line 69 | Line 93 | public class FutureTaskTest extends JSR1
93              FutureTask ft = (FutureTask<?>) f;
94              // Check that run methods do nothing
95              ft.run();
96 <            if (f instanceof PublicFutureTask)
97 <                assertFalse(((PublicFutureTask) f).runAndReset());
96 >            if (f instanceof PublicFutureTask) {
97 >                PublicFutureTask pf = (PublicFutureTask) f;
98 >                int savedRunCount = pf.runCount();
99 >                pf.run();
100 >                assertFalse(pf.runAndReset());
101 >                assertEquals(savedRunCount, pf.runCount());
102 >            }
103              checkNotDone(f);
104          }
105      }
106  
107 <    <T> void checkCompletedNormally(Future<T> f, T expected) {
107 >    <T> void checkCompletedNormally(Future<T> f, T expectedValue) {
108          checkIsDone(f);
109          assertFalse(f.isCancelled());
110  
111 +        T v1 = null, v2 = null;
112          try {
113 <            assertSame(expected, f.get());
114 <        } catch (Throwable fail) { threadUnexpectedException(fail); }
85 <        try {
86 <            assertSame(expected, f.get(5L, SECONDS));
113 >            v1 = f.get();
114 >            v2 = f.get(randomTimeout(), randomTimeUnit());
115          } catch (Throwable fail) { threadUnexpectedException(fail); }
116 +        assertSame(expectedValue, v1);
117 +        assertSame(expectedValue, v2);
118      }
119  
120      void checkCancelled(Future<?> f) {
# Line 98 | Line 128 | public class FutureTaskTest extends JSR1
128          } catch (Throwable fail) { threadUnexpectedException(fail); }
129  
130          try {
131 <            f.get(5L, SECONDS);
131 >            f.get(randomTimeout(), randomTimeUnit());
132              shouldThrow();
133          } catch (CancellationException success) {
134          } catch (Throwable fail) { threadUnexpectedException(fail); }
# Line 108 | Line 138 | public class FutureTaskTest extends JSR1
138          pf.set(new Object());
139          pf.setException(new Error());
140          for (boolean mayInterruptIfRunning : new boolean[] { true, false }) {
141 <            pf.cancel(true);
141 >            pf.cancel(mayInterruptIfRunning);
142          }
143      }
144  
# Line 124 | Line 154 | public class FutureTaskTest extends JSR1
154          } catch (Throwable fail) { threadUnexpectedException(fail); }
155  
156          try {
157 <            f.get(5L, SECONDS);
157 >            f.get(randomTimeout(), randomTimeUnit());
158              shouldThrow();
159          } catch (ExecutionException success) {
160              assertSame(t, success.getCause());
# Line 240 | Line 270 | public class FutureTaskTest extends JSR1
270          for (int i = 0; i < 3; i++) {
271              assertTrue(task.runAndReset());
272              checkNotDone(task);
273 <            assertEquals(i+1, task.runCount());
274 <            assertEquals(i+1, task.runAndResetCount());
273 >            assertEquals(i + 1, task.runCount());
274 >            assertEquals(i + 1, task.runAndResetCount());
275              assertEquals(0, task.setCount());
276              assertEquals(0, task.setExceptionCount());
277          }
# Line 257 | Line 287 | public class FutureTaskTest extends JSR1
287              for (int i = 0; i < 3; i++) {
288                  assertFalse(task.runAndReset());
289                  assertEquals(0, task.runCount());
290 <                assertEquals(i+1, task.runAndResetCount());
290 >                assertEquals(i + 1, task.runAndResetCount());
291                  assertEquals(0, task.setCount());
292                  assertEquals(0, task.setExceptionCount());
293              }
# Line 390 | Line 420 | public class FutureTaskTest extends JSR1
420                          delay(LONG_DELAY_MS);
421                          shouldThrow();
422                      } catch (InterruptedException success) {}
423 +                    assertFalse(Thread.interrupted());
424                  }});
425  
426          Thread t = newStartedThread(task);
# Line 433 | Line 464 | public class FutureTaskTest extends JSR1
464          try {
465              task.cancel(true);
466              shouldThrow();
467 <        } catch (SecurityException expected) {}
467 >        } catch (SecurityException success) {}
468  
469          // We failed to deliver the interrupt, but the world retains
470          // its sanity, as if we had done task.cancel(false)
# Line 459 | Line 490 | public class FutureTaskTest extends JSR1
490          final PublicFutureTask task =
491              new PublicFutureTask(new Runnable() {
492                  public void run() {
493 +                    pleaseCancel.countDown();
494                      try {
463                        pleaseCancel.countDown();
495                          delay(LONG_DELAY_MS);
496 <                    } finally { throw new RuntimeException(); }
496 >                        threadShouldThrow();
497 >                    } catch (InterruptedException success) {
498 >                    } catch (Throwable t) { threadUnexpectedException(t); }
499 >                    throw new RuntimeException();
500                  }});
501  
502          Thread t = newStartedThread(task);
# Line 587 | Line 621 | public class FutureTaskTest extends JSR1
621       * CancellationException
622       */
623      public void testTimedGet_Cancellation() {
624 <        for (final boolean mayInterruptIfRunning :
625 <                 new boolean[] { true, false }) {
626 <            final CountDownLatch pleaseCancel = new CountDownLatch(3);
627 <            final CountDownLatch cancelled = new CountDownLatch(1);
628 <            final PublicFutureTask task =
629 <                new PublicFutureTask(new CheckedCallable<Object>() {
630 <                    public Object realCall() throws InterruptedException {
631 <                        pleaseCancel.countDown();
632 <                        if (mayInterruptIfRunning) {
633 <                            try {
634 <                                delay(2*LONG_DELAY_MS);
635 <                            } catch (InterruptedException success) {}
636 <                        } else {
637 <                            await(cancelled);
638 <                        }
639 <                        return two;
640 <                    }});
624 >        testTimedGet_Cancellation(false);
625 >    }
626 >    public void testTimedGet_Cancellation_interrupt() {
627 >        testTimedGet_Cancellation(true);
628 >    }
629 >    public void testTimedGet_Cancellation(final boolean mayInterruptIfRunning) {
630 >        final CountDownLatch pleaseCancel = new CountDownLatch(3);
631 >        final CountDownLatch cancelled = new CountDownLatch(1);
632 >        final Callable<Object> callable =
633 >            new CheckedCallable<Object>() {
634 >            public Object realCall() throws InterruptedException {
635 >                pleaseCancel.countDown();
636 >                if (mayInterruptIfRunning) {
637 >                    try {
638 >                        delay(2*LONG_DELAY_MS);
639 >                    } catch (InterruptedException success) {}
640 >                } else {
641 >                    await(cancelled);
642 >                }
643 >                return two;
644 >            }};
645 >        final PublicFutureTask task = new PublicFutureTask(callable);
646  
647 <            Thread t1 = new ThreadShouldThrow(CancellationException.class) {
647 >        Thread t1 = new ThreadShouldThrow(CancellationException.class) {
648                  public void realRun() throws Exception {
649                      pleaseCancel.countDown();
650                      task.get();
651                  }};
652 <            Thread t2 = new ThreadShouldThrow(CancellationException.class) {
652 >        Thread t2 = new ThreadShouldThrow(CancellationException.class) {
653                  public void realRun() throws Exception {
654                      pleaseCancel.countDown();
655                      task.get(2*LONG_DELAY_MS, MILLISECONDS);
656                  }};
657 <            t1.start();
658 <            t2.start();
659 <            Thread t3 = newStartedThread(task);
660 <            await(pleaseCancel);
661 <            checkIsRunning(task);
662 <            task.cancel(mayInterruptIfRunning);
663 <            checkCancelled(task);
664 <            awaitTermination(t1);
665 <            awaitTermination(t2);
666 <            cancelled.countDown();
667 <            awaitTermination(t3);
668 <            assertEquals(1, task.runCount());
669 <            assertEquals(1, task.setCount());
670 <            assertEquals(0, task.setExceptionCount());
671 <            tryToConfuseDoneTask(task);
672 <            checkCancelled(task);
634 <        }
657 >        t1.start();
658 >        t2.start();
659 >        Thread t3 = newStartedThread(task);
660 >        await(pleaseCancel);
661 >        checkIsRunning(task);
662 >        task.cancel(mayInterruptIfRunning);
663 >        checkCancelled(task);
664 >        awaitTermination(t1);
665 >        awaitTermination(t2);
666 >        cancelled.countDown();
667 >        awaitTermination(t3);
668 >        assertEquals(1, task.runCount());
669 >        assertEquals(1, task.setCount());
670 >        assertEquals(0, task.setExceptionCount());
671 >        tryToConfuseDoneTask(task);
672 >        checkCancelled(task);
673      }
674  
675      /**
# Line 718 | Line 756 | public class FutureTaskTest extends JSR1
756              public void realRun() throws Exception {
757                  Thread.currentThread().interrupt();
758                  try {
759 <                    task.get(2*LONG_DELAY_MS, MILLISECONDS);
759 >                    task.get(randomTimeout(), randomTimeUnit());
760                      shouldThrow();
761                  } catch (InterruptedException success) {}
762                  assertFalse(Thread.interrupted());
763  
764                  pleaseInterrupt.countDown();
765                  try {
766 <                    task.get(2*LONG_DELAY_MS, MILLISECONDS);
766 >                    task.get(LONGER_DELAY_MS, MILLISECONDS);
767                      shouldThrow();
768                  } catch (InterruptedException success) {}
769                  assertFalse(Thread.interrupted());
770              }});
771  
772          await(pleaseInterrupt);
773 +        if (randomBoolean()) assertThreadBlocks(t, Thread.State.TIMED_WAITING);
774          t.interrupt();
775          awaitTermination(t);
776          checkNotDone(task);
# Line 775 | Line 814 | public class FutureTaskTest extends JSR1
814          }
815      }
816  
817 +    /**
818 +     * timed get with most negative timeout works correctly (i.e. no
819 +     * underflow bug)
820 +     */
821 +    public void testGet_NegativeInfinityTimeout() throws Exception {
822 +        final ExecutorService pool = Executors.newFixedThreadPool(10);
823 +        final Runnable nop = new Runnable() { public void run() {}};
824 +        final FutureTask<Void> task = new FutureTask<>(nop, null);
825 +        final List<Future<?>> futures = new ArrayList<>();
826 +        Runnable r = new Runnable() { public void run() {
827 +            for (long timeout : new long[] { 0L, -1L, Long.MIN_VALUE }) {
828 +                try {
829 +                    task.get(timeout, NANOSECONDS);
830 +                    shouldThrow();
831 +                } catch (TimeoutException success) {
832 +                } catch (Throwable fail) {threadUnexpectedException(fail);}}}};
833 +        for (int i = 0; i < 10; i++)
834 +            futures.add(pool.submit(r));
835 +        try {
836 +            joinPool(pool);
837 +            for (Future<?> future : futures)
838 +                checkCompletedNormally(future, null);
839 +        } finally {
840 +            task.run();         // last resort to help terminate
841 +        }
842 +    }
843 +
844 +    /**
845 +     * toString indicates current completion state
846 +     */
847 +    public void testToString_incomplete() {
848 +        FutureTask<String> f = new FutureTask<>(() -> "");
849 +        assertTrue(f.toString().matches(".*\\[.*Not completed.*\\]"));
850 +        if (testImplementationDetails)
851 +            assertTrue(f.toString().startsWith(
852 +                               identityString(f) + "[Not completed, task ="));
853 +    }
854 +
855 +    public void testToString_normal() {
856 +        FutureTask<String> f = new FutureTask<>(() -> "");
857 +        f.run();
858 +        assertTrue(f.toString().matches(".*\\[.*Completed normally.*\\]"));
859 +        if (testImplementationDetails)
860 +            assertEquals(identityString(f) + "[Completed normally]",
861 +                         f.toString());
862 +    }
863 +
864 +    public void testToString_exception() {
865 +        FutureTask<String> f = new FutureTask<>(
866 +                () -> { throw new ArithmeticException(); });
867 +        f.run();
868 +        assertTrue(f.toString().matches(".*\\[.*Completed exceptionally.*\\]"));
869 +        if (testImplementationDetails)
870 +            assertTrue(f.toString().startsWith(
871 +                               identityString(f) + "[Completed exceptionally: "));
872 +    }
873 +
874 +    public void testToString_cancelled() {
875 +        for (boolean mayInterruptIfRunning : new boolean[] { true, false }) {
876 +            FutureTask<String> f = new FutureTask<>(() -> "");
877 +            assertTrue(f.cancel(mayInterruptIfRunning));
878 +            assertTrue(f.toString().matches(".*\\[.*Cancelled.*\\]"));
879 +            if (testImplementationDetails)
880 +                assertEquals(identityString(f) + "[Cancelled]",
881 +                             f.toString());
882 +        }
883 +    }
884 +
885   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines