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

Comparing jsr166/src/test/tck/JSR166TestCase.java (file contents):
Revision 1.212 by jsr166, Fri Dec 9 04:24:07 2016 UTC vs.
Revision 1.235 by jsr166, Sat Jul 22 18:23:59 2017 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
2 > * Written by Doug Lea and Martin Buchholz with assistance from
3 > * members of JCP JSR-166 Expert Group and released to the public
4 > * domain, as explained at
5   * http://creativecommons.org/publicdomain/zero/1.0/
6   * Other contributors include Andrew Wright, Jeffrey Hayes,
7   * Pat Fisher, Mike Judd.
# Line 8 | Line 9
9  
10   /*
11   * @test
12 < * @summary JSR-166 tck tests
13 < * @modules java.base/java.util.concurrent:open
14 < *          java.management
12 > * @summary JSR-166 tck tests, in a number of variations.
13 > *          The first is the conformance testing variant,
14 > *          while others also test implementation details.
15   * @build *
16 < * @run junit/othervm/timeout=1000 -Djsr166.testImplementationDetails=true JSR166TestCase
17 < * @run junit/othervm/timeout=1000 -Djava.util.concurrent.ForkJoinPool.common.parallelism=0 -Djsr166.testImplementationDetails=true JSR166TestCase
18 < * @run junit/othervm/timeout=1000 -Djava.util.concurrent.ForkJoinPool.common.parallelism=1 -Djava.util.secureRandomSeed=true JSR166TestCase
16 > * @modules java.management
17 > * @run junit/othervm/timeout=1000 JSR166TestCase
18 > * @run junit/othervm/timeout=1000
19 > *      --add-opens java.base/java.util.concurrent=ALL-UNNAMED
20 > *      --add-opens java.base/java.lang=ALL-UNNAMED
21 > *      -Djsr166.testImplementationDetails=true
22 > *      JSR166TestCase
23 > * @run junit/othervm/timeout=1000
24 > *      --add-opens java.base/java.util.concurrent=ALL-UNNAMED
25 > *      --add-opens java.base/java.lang=ALL-UNNAMED
26 > *      -Djsr166.testImplementationDetails=true
27 > *      -Djava.util.concurrent.ForkJoinPool.common.parallelism=0
28 > *      JSR166TestCase
29 > * @run junit/othervm/timeout=1000
30 > *      --add-opens java.base/java.util.concurrent=ALL-UNNAMED
31 > *      --add-opens java.base/java.lang=ALL-UNNAMED
32 > *      -Djsr166.testImplementationDetails=true
33 > *      -Djava.util.concurrent.ForkJoinPool.common.parallelism=1
34 > *      -Djava.util.secureRandomSeed=true
35 > *      JSR166TestCase
36 > * @run junit/othervm/timeout=1000/policy=tck.policy
37 > *      --add-opens java.base/java.util.concurrent=ALL-UNNAMED
38 > *      --add-opens java.base/java.lang=ALL-UNNAMED
39 > *      -Djsr166.testImplementationDetails=true
40 > *      JSR166TestCase
41   */
42  
43   import static java.util.concurrent.TimeUnit.MILLISECONDS;
# Line 31 | Line 54 | import java.lang.management.ThreadMXBean
54   import java.lang.reflect.Constructor;
55   import java.lang.reflect.Method;
56   import java.lang.reflect.Modifier;
34 import java.nio.file.Files;
35 import java.nio.file.Paths;
57   import java.security.CodeSource;
58   import java.security.Permission;
59   import java.security.PermissionCollection;
# Line 55 | Line 76 | import java.util.concurrent.Callable;
76   import java.util.concurrent.CountDownLatch;
77   import java.util.concurrent.CyclicBarrier;
78   import java.util.concurrent.ExecutionException;
79 + import java.util.concurrent.Executor;
80   import java.util.concurrent.Executors;
81   import java.util.concurrent.ExecutorService;
82   import java.util.concurrent.ForkJoinPool;
83   import java.util.concurrent.Future;
84 + import java.util.concurrent.FutureTask;
85   import java.util.concurrent.RecursiveAction;
86   import java.util.concurrent.RecursiveTask;
87 + import java.util.concurrent.RejectedExecutionException;
88   import java.util.concurrent.RejectedExecutionHandler;
89   import java.util.concurrent.Semaphore;
90 + import java.util.concurrent.ScheduledExecutorService;
91 + import java.util.concurrent.ScheduledFuture;
92   import java.util.concurrent.SynchronousQueue;
93   import java.util.concurrent.ThreadFactory;
94   import java.util.concurrent.ThreadLocalRandom;
95   import java.util.concurrent.ThreadPoolExecutor;
96 + import java.util.concurrent.TimeUnit;
97   import java.util.concurrent.TimeoutException;
98   import java.util.concurrent.atomic.AtomicBoolean;
99   import java.util.concurrent.atomic.AtomicReference;
73 import java.util.regex.Matcher;
100   import java.util.regex.Pattern;
101  
102   import junit.framework.AssertionFailedError;
# Line 280 | Line 306 | public class JSR166TestCase extends Test
306  
307   //     public static String cpuModel() {
308   //         try {
309 < //             Matcher matcher = Pattern.compile("model name\\s*: (.*)")
309 > //             java.util.regex.Matcher matcher
310 > //               = Pattern.compile("model name\\s*: (.*)")
311   //                 .matcher(new String(
312 < //                      Files.readAllBytes(Paths.get("/proc/cpuinfo")), "UTF-8"));
312 > //                     java.nio.file.Files.readAllBytes(
313 > //                         java.nio.file.Paths.get("/proc/cpuinfo")), "UTF-8"));
314   //             matcher.find();
315   //             return matcher.group(1);
316   //         } catch (Exception ex) { return null; }
# Line 516 | Line 544 | public class JSR166TestCase extends Test
544                  "DoubleAdderTest",
545                  "ForkJoinPool8Test",
546                  "ForkJoinTask8Test",
547 +                "LinkedBlockingDeque8Test",
548 +                "LinkedBlockingQueue8Test",
549                  "LongAccumulatorTest",
550                  "LongAdderTest",
551                  "SplittableRandomTest",
# Line 538 | Line 568 | public class JSR166TestCase extends Test
568                  "AtomicReference9Test",
569                  "AtomicReferenceArray9Test",
570                  "ExecutorCompletionService9Test",
571 +                "ForkJoinPool9Test",
572              };
573              addNamedTestClasses(suite, java9TestClassNames);
574          }
# Line 548 | Line 579 | public class JSR166TestCase extends Test
579      /** Returns list of junit-style test method names in given class. */
580      public static ArrayList<String> testMethodNames(Class<?> testClass) {
581          Method[] methods = testClass.getDeclaredMethods();
582 <        ArrayList<String> names = new ArrayList<String>(methods.length);
582 >        ArrayList<String> names = new ArrayList<>(methods.length);
583          for (Method method : methods) {
584              if (method.getName().startsWith("test")
585                  && Modifier.isPublic(method.getModifiers())
# Line 613 | Line 644 | public class JSR166TestCase extends Test
644      public static long MEDIUM_DELAY_MS;
645      public static long LONG_DELAY_MS;
646  
647 +    private static final long RANDOM_TIMEOUT;
648 +    private static final long RANDOM_EXPIRED_TIMEOUT;
649 +    private static final TimeUnit RANDOM_TIMEUNIT;
650 +    static {
651 +        ThreadLocalRandom rnd = ThreadLocalRandom.current();
652 +        long[] timeouts = { Long.MIN_VALUE, -1, 0, 1, Long.MAX_VALUE };
653 +        RANDOM_TIMEOUT = timeouts[rnd.nextInt(timeouts.length)];
654 +        RANDOM_EXPIRED_TIMEOUT = timeouts[rnd.nextInt(3)];
655 +        TimeUnit[] timeUnits = TimeUnit.values();
656 +        RANDOM_TIMEUNIT = timeUnits[rnd.nextInt(timeUnits.length)];
657 +    }
658 +
659 +    /**
660 +     * Returns a timeout for use when any value at all will do.
661 +     */
662 +    static long randomTimeout() { return RANDOM_TIMEOUT; }
663 +
664 +    /**
665 +     * Returns a timeout that means "no waiting", i.e. not positive.
666 +     */
667 +    static long randomExpiredTimeout() { return RANDOM_EXPIRED_TIMEOUT; }
668 +
669 +    /**
670 +     * Returns a random non-null TimeUnit.
671 +     */
672 +    static TimeUnit randomTimeUnit() { return RANDOM_TIMEUNIT; }
673 +
674      /**
675       * Returns the shortest timed delay. This can be scaled up for
676       * slow machines using the jsr166.delay.factor system property,
# Line 633 | Line 691 | public class JSR166TestCase extends Test
691          LONG_DELAY_MS   = SHORT_DELAY_MS * 200;
692      }
693  
694 +    private static final long TIMEOUT_DELAY_MS
695 +        = (long) (12.0 * Math.cbrt(delayFactor));
696 +
697      /**
698 <     * Returns a timeout in milliseconds to be used in tests that
699 <     * verify that operations block or time out.
698 >     * Returns a timeout in milliseconds to be used in tests that verify
699 >     * that operations block or time out.  We want this to be longer
700 >     * than the OS scheduling quantum, but not too long, so don't scale
701 >     * linearly with delayFactor; we use "crazy" cube root instead.
702       */
703 <    long timeoutMillis() {
704 <        return SHORT_DELAY_MS / 4;
703 >    static long timeoutMillis() {
704 >        return TIMEOUT_DELAY_MS;
705      }
706  
707      /**
# Line 654 | Line 717 | public class JSR166TestCase extends Test
717       * The first exception encountered if any threadAssertXXX method fails.
718       */
719      private final AtomicReference<Throwable> threadFailure
720 <        = new AtomicReference<Throwable>(null);
720 >        = new AtomicReference<>(null);
721  
722      /**
723       * Records an exception so that it can be rethrown later in the test
# Line 1033 | Line 1096 | public class JSR166TestCase extends Test
1096      }
1097  
1098      /**
1099 <     * Checks that thread does not terminate within the default
1037 <     * millisecond delay of {@code timeoutMillis()}.
1038 <     */
1039 <    void assertThreadStaysAlive(Thread thread) {
1040 <        assertThreadStaysAlive(thread, timeoutMillis());
1041 <    }
1042 <
1043 <    /**
1044 <     * Checks that thread does not terminate within the given millisecond delay.
1045 <     */
1046 <    void assertThreadStaysAlive(Thread thread, long millis) {
1047 <        try {
1048 <            // No need to optimize the failing case via Thread.join.
1049 <            delay(millis);
1050 <            assertTrue(thread.isAlive());
1051 <        } catch (InterruptedException fail) {
1052 <            threadFail("Unexpected InterruptedException");
1053 <        }
1054 <    }
1055 <
1056 <    /**
1057 <     * Checks that the threads do not terminate within the default
1058 <     * millisecond delay of {@code timeoutMillis()}.
1099 >     * Checks that thread eventually enters the expected blocked thread state.
1100       */
1101 <    void assertThreadsStayAlive(Thread... threads) {
1102 <        assertThreadsStayAlive(timeoutMillis(), threads);
1103 <    }
1104 <
1105 <    /**
1106 <     * Checks that the threads do not terminate within the given millisecond delay.
1107 <     */
1108 <    void assertThreadsStayAlive(long millis, Thread... threads) {
1109 <        try {
1110 <            // No need to optimize the failing case via Thread.join.
1111 <            delay(millis);
1112 <            for (Thread thread : threads)
1113 <                assertTrue(thread.isAlive());
1073 <        } catch (InterruptedException fail) {
1074 <            threadFail("Unexpected InterruptedException");
1101 >    void assertThreadBlocks(Thread thread, Thread.State expected) {
1102 >        // always sleep at least 1 ms, with high probability avoiding
1103 >        // transitory states
1104 >        for (long retries = LONG_DELAY_MS * 3 / 4; retries-->0; ) {
1105 >            try { delay(1); }
1106 >            catch (InterruptedException fail) {
1107 >                fail("Unexpected InterruptedException");
1108 >            }
1109 >            Thread.State s = thread.getState();
1110 >            if (s == expected)
1111 >                return;
1112 >            else if (s == Thread.State.TERMINATED)
1113 >                fail("Unexpected thread termination");
1114          }
1115 +        fail("timed out waiting for thread to enter thread state " + expected);
1116      }
1117  
1118      /**
# Line 1113 | Line 1153 | public class JSR166TestCase extends Test
1153      }
1154  
1155      /**
1156 +     * The maximum number of consecutive spurious wakeups we should
1157 +     * tolerate (from APIs like LockSupport.park) before failing a test.
1158 +     */
1159 +    static final int MAX_SPURIOUS_WAKEUPS = 10;
1160 +
1161 +    /**
1162       * The number of elements to place in collections, arrays, etc.
1163       */
1164      public static final int SIZE = 20;
# Line 1216 | Line 1262 | public class JSR166TestCase extends Test
1262          }
1263          public void refresh() {}
1264          public String toString() {
1265 <            List<Permission> ps = new ArrayList<Permission>();
1265 >            List<Permission> ps = new ArrayList<>();
1266              for (Enumeration<Permission> e = perms.elements(); e.hasMoreElements();)
1267                  ps.add(e.nextElement());
1268              return "AdjustablePolicy with permissions " + ps;
# Line 1275 | Line 1321 | public class JSR166TestCase extends Test
1321                  startTime = System.nanoTime();
1322              else if (millisElapsedSince(startTime) > timeoutMillis) {
1323                  threadAssertTrue(thread.isAlive());
1324 <                return;
1324 >                fail("timed out waiting for thread to enter wait state");
1325 >            }
1326 >            Thread.yield();
1327 >        }
1328 >    }
1329 >
1330 >    /**
1331 >     * Spin-waits up to the specified number of milliseconds for the given
1332 >     * thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING,
1333 >     * and additionally satisfy the given condition.
1334 >     */
1335 >    void waitForThreadToEnterWaitState(
1336 >        Thread thread, long timeoutMillis, Callable<Boolean> waitingForGodot) {
1337 >        long startTime = 0L;
1338 >        for (;;) {
1339 >            Thread.State s = thread.getState();
1340 >            if (s == Thread.State.BLOCKED ||
1341 >                s == Thread.State.WAITING ||
1342 >                s == Thread.State.TIMED_WAITING) {
1343 >                try {
1344 >                    if (waitingForGodot.call())
1345 >                        return;
1346 >                } catch (Throwable fail) { threadUnexpectedException(fail); }
1347 >            }
1348 >            else if (s == Thread.State.TERMINATED)
1349 >                fail("Unexpected thread termination");
1350 >            else if (startTime == 0L)
1351 >                startTime = System.nanoTime();
1352 >            else if (millisElapsedSince(startTime) > timeoutMillis) {
1353 >                threadAssertTrue(thread.isAlive());
1354 >                fail("timed out waiting for thread to enter wait state");
1355              }
1356              Thread.yield();
1357          }
1358      }
1359  
1360      /**
1361 <     * Waits up to LONG_DELAY_MS for the given thread to enter a wait
1362 <     * state: BLOCKED, WAITING, or TIMED_WAITING.
1361 >     * Spin-waits up to LONG_DELAY_MS milliseconds for the given thread to
1362 >     * enter a wait state: BLOCKED, WAITING, or TIMED_WAITING.
1363       */
1364      void waitForThreadToEnterWaitState(Thread thread) {
1365          waitForThreadToEnterWaitState(thread, LONG_DELAY_MS);
1366      }
1367  
1368      /**
1369 +     * Spin-waits up to LONG_DELAY_MS milliseconds for the given thread to
1370 +     * enter a wait state: BLOCKED, WAITING, or TIMED_WAITING,
1371 +     * and additionally satisfy the given condition.
1372 +     */
1373 +    void waitForThreadToEnterWaitState(
1374 +        Thread thread, Callable<Boolean> waitingForGodot) {
1375 +        waitForThreadToEnterWaitState(thread, LONG_DELAY_MS, waitingForGodot);
1376 +    }
1377 +
1378 +    /**
1379       * Returns the number of milliseconds since time given by
1380       * startNanoTime, which must have been previously returned from a
1381       * call to {@link System#nanoTime()}.
# Line 1541 | Line 1627 | public class JSR166TestCase extends Test
1627          }
1628      }
1629  
1630 +    public void await(CyclicBarrier barrier) {
1631 +        try {
1632 +            barrier.await(LONG_DELAY_MS, MILLISECONDS);
1633 +        } catch (Throwable fail) {
1634 +            threadUnexpectedException(fail);
1635 +        }
1636 +    }
1637 +
1638   //     /**
1639   //      * Spin-waits up to LONG_DELAY_MS until flag becomes true.
1640   //      */
# Line 1564 | Line 1658 | public class JSR166TestCase extends Test
1658          public String call() { throw new NullPointerException(); }
1659      }
1660  
1567    public static class CallableOne implements Callable<Integer> {
1568        public Integer call() { return one; }
1569    }
1570
1571    public class ShortRunnable extends CheckedRunnable {
1572        protected void realRun() throws Throwable {
1573            delay(SHORT_DELAY_MS);
1574        }
1575    }
1576
1577    public class ShortInterruptedRunnable extends CheckedInterruptedRunnable {
1578        protected void realRun() throws InterruptedException {
1579            delay(SHORT_DELAY_MS);
1580        }
1581    }
1582
1583    public class SmallRunnable extends CheckedRunnable {
1584        protected void realRun() throws Throwable {
1585            delay(SMALL_DELAY_MS);
1586        }
1587    }
1588
1661      public class SmallPossiblyInterruptedRunnable extends CheckedRunnable {
1662          protected void realRun() {
1663              try {
# Line 1594 | Line 1666 | public class JSR166TestCase extends Test
1666          }
1667      }
1668  
1597    public class SmallCallable extends CheckedCallable {
1598        protected Object realCall() throws InterruptedException {
1599            delay(SMALL_DELAY_MS);
1600            return Boolean.TRUE;
1601        }
1602    }
1603
1604    public class MediumRunnable extends CheckedRunnable {
1605        protected void realRun() throws Throwable {
1606            delay(MEDIUM_DELAY_MS);
1607        }
1608    }
1609
1610    public class MediumInterruptedRunnable extends CheckedInterruptedRunnable {
1611        protected void realRun() throws InterruptedException {
1612            delay(MEDIUM_DELAY_MS);
1613        }
1614    }
1615
1669      public Runnable possiblyInterruptedRunnable(final long timeoutMillis) {
1670          return new CheckedRunnable() {
1671              protected void realRun() {
# Line 1622 | Line 1675 | public class JSR166TestCase extends Test
1675              }};
1676      }
1677  
1625    public class MediumPossiblyInterruptedRunnable extends CheckedRunnable {
1626        protected void realRun() {
1627            try {
1628                delay(MEDIUM_DELAY_MS);
1629            } catch (InterruptedException ok) {}
1630        }
1631    }
1632
1633    public class LongPossiblyInterruptedRunnable extends CheckedRunnable {
1634        protected void realRun() {
1635            try {
1636                delay(LONG_DELAY_MS);
1637            } catch (InterruptedException ok) {}
1638        }
1639    }
1640
1678      /**
1679       * For use as ThreadFactory in constructors
1680       */
# Line 1651 | Line 1688 | public class JSR166TestCase extends Test
1688          boolean isDone();
1689      }
1690  
1654    public static TrackedRunnable trackedRunnable(final long timeoutMillis) {
1655        return new TrackedRunnable() {
1656                private volatile boolean done = false;
1657                public boolean isDone() { return done; }
1658                public void run() {
1659                    try {
1660                        delay(timeoutMillis);
1661                        done = true;
1662                    } catch (InterruptedException ok) {}
1663                }
1664            };
1665    }
1666
1667    public static class TrackedShortRunnable implements Runnable {
1668        public volatile boolean done = false;
1669        public void run() {
1670            try {
1671                delay(SHORT_DELAY_MS);
1672                done = true;
1673            } catch (InterruptedException ok) {}
1674        }
1675    }
1676
1677    public static class TrackedSmallRunnable implements Runnable {
1678        public volatile boolean done = false;
1679        public void run() {
1680            try {
1681                delay(SMALL_DELAY_MS);
1682                done = true;
1683            } catch (InterruptedException ok) {}
1684        }
1685    }
1686
1687    public static class TrackedMediumRunnable implements Runnable {
1688        public volatile boolean done = false;
1689        public void run() {
1690            try {
1691                delay(MEDIUM_DELAY_MS);
1692                done = true;
1693            } catch (InterruptedException ok) {}
1694        }
1695    }
1696
1697    public static class TrackedLongRunnable implements Runnable {
1698        public volatile boolean done = false;
1699        public void run() {
1700            try {
1701                delay(LONG_DELAY_MS);
1702                done = true;
1703            } catch (InterruptedException ok) {}
1704        }
1705    }
1706
1691      public static class TrackedNoOpRunnable implements Runnable {
1692          public volatile boolean done = false;
1693          public void run() {
# Line 1711 | Line 1695 | public class JSR166TestCase extends Test
1695          }
1696      }
1697  
1714    public static class TrackedCallable implements Callable {
1715        public volatile boolean done = false;
1716        public Object call() {
1717            try {
1718                delay(SMALL_DELAY_MS);
1719                done = true;
1720            } catch (InterruptedException ok) {}
1721            return Boolean.TRUE;
1722        }
1723    }
1724
1698      /**
1699       * Analog of CheckedRunnable for RecursiveAction
1700       */
# Line 1788 | Line 1761 | public class JSR166TestCase extends Test
1761              assertEquals(0, q.size());
1762              assertNull(q.peek());
1763              assertNull(q.poll());
1764 <            assertNull(q.poll(0, MILLISECONDS));
1764 >            assertNull(q.poll(randomExpiredTimeout(), randomTimeUnit()));
1765              assertEquals(q.toString(), "[]");
1766              assertTrue(Arrays.equals(q.toArray(), new Object[0]));
1767              assertFalse(q.iterator().hasNext());
# Line 1939 | Line 1912 | public class JSR166TestCase extends Test
1912      static <T> void shuffle(T[] array) {
1913          Collections.shuffle(Arrays.asList(array), ThreadLocalRandom.current());
1914      }
1915 +
1916 +    // --- Shared assertions for Executor tests ---
1917 +
1918 +    /**
1919 +     * Returns maximum number of tasks that can be submitted to given
1920 +     * pool (with bounded queue) before saturation (when submission
1921 +     * throws RejectedExecutionException).
1922 +     */
1923 +    static final int saturatedSize(ThreadPoolExecutor pool) {
1924 +        BlockingQueue<Runnable> q = pool.getQueue();
1925 +        return pool.getMaximumPoolSize() + q.size() + q.remainingCapacity();
1926 +    }
1927 +
1928 +    @SuppressWarnings("FutureReturnValueIgnored")
1929 +    void assertNullTaskSubmissionThrowsNullPointerException(Executor e) {
1930 +        try {
1931 +            e.execute((Runnable) null);
1932 +            shouldThrow();
1933 +        } catch (NullPointerException success) {}
1934 +
1935 +        if (! (e instanceof ExecutorService)) return;
1936 +        ExecutorService es = (ExecutorService) e;
1937 +        try {
1938 +            es.submit((Runnable) null);
1939 +            shouldThrow();
1940 +        } catch (NullPointerException success) {}
1941 +        try {
1942 +            es.submit((Runnable) null, Boolean.TRUE);
1943 +            shouldThrow();
1944 +        } catch (NullPointerException success) {}
1945 +        try {
1946 +            es.submit((Callable) null);
1947 +            shouldThrow();
1948 +        } catch (NullPointerException success) {}
1949 +
1950 +        if (! (e instanceof ScheduledExecutorService)) return;
1951 +        ScheduledExecutorService ses = (ScheduledExecutorService) e;
1952 +        try {
1953 +            ses.schedule((Runnable) null,
1954 +                         randomTimeout(), randomTimeUnit());
1955 +            shouldThrow();
1956 +        } catch (NullPointerException success) {}
1957 +        try {
1958 +            ses.schedule((Callable) null,
1959 +                         randomTimeout(), randomTimeUnit());
1960 +            shouldThrow();
1961 +        } catch (NullPointerException success) {}
1962 +        try {
1963 +            ses.scheduleAtFixedRate((Runnable) null,
1964 +                                    randomTimeout(), LONG_DELAY_MS, MILLISECONDS);
1965 +            shouldThrow();
1966 +        } catch (NullPointerException success) {}
1967 +        try {
1968 +            ses.scheduleWithFixedDelay((Runnable) null,
1969 +                                       randomTimeout(), LONG_DELAY_MS, MILLISECONDS);
1970 +            shouldThrow();
1971 +        } catch (NullPointerException success) {}
1972 +    }
1973 +
1974 +    void setRejectedExecutionHandler(
1975 +        ThreadPoolExecutor p, RejectedExecutionHandler handler) {
1976 +        p.setRejectedExecutionHandler(handler);
1977 +        assertSame(handler, p.getRejectedExecutionHandler());
1978 +    }
1979 +
1980 +    void assertTaskSubmissionsAreRejected(ThreadPoolExecutor p) {
1981 +        final RejectedExecutionHandler savedHandler = p.getRejectedExecutionHandler();
1982 +        final long savedTaskCount = p.getTaskCount();
1983 +        final long savedCompletedTaskCount = p.getCompletedTaskCount();
1984 +        final int savedQueueSize = p.getQueue().size();
1985 +        final boolean stock = (p.getClass().getClassLoader() == null);
1986 +
1987 +        Runnable r = () -> {};
1988 +        Callable<Boolean> c = () -> Boolean.TRUE;
1989 +
1990 +        class Recorder implements RejectedExecutionHandler {
1991 +            public volatile Runnable r = null;
1992 +            public volatile ThreadPoolExecutor p = null;
1993 +            public void reset() { r = null; p = null; }
1994 +            public void rejectedExecution(Runnable r, ThreadPoolExecutor p) {
1995 +                assertNull(this.r);
1996 +                assertNull(this.p);
1997 +                this.r = r;
1998 +                this.p = p;
1999 +            }
2000 +        }
2001 +
2002 +        // check custom handler is invoked exactly once per task
2003 +        Recorder recorder = new Recorder();
2004 +        setRejectedExecutionHandler(p, recorder);
2005 +        for (int i = 2; i--> 0; ) {
2006 +            recorder.reset();
2007 +            p.execute(r);
2008 +            if (stock && p.getClass() == ThreadPoolExecutor.class)
2009 +                assertSame(r, recorder.r);
2010 +            assertSame(p, recorder.p);
2011 +
2012 +            recorder.reset();
2013 +            assertFalse(p.submit(r).isDone());
2014 +            if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2015 +            assertSame(p, recorder.p);
2016 +
2017 +            recorder.reset();
2018 +            assertFalse(p.submit(r, Boolean.TRUE).isDone());
2019 +            if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2020 +            assertSame(p, recorder.p);
2021 +
2022 +            recorder.reset();
2023 +            assertFalse(p.submit(c).isDone());
2024 +            if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2025 +            assertSame(p, recorder.p);
2026 +
2027 +            if (p instanceof ScheduledExecutorService) {
2028 +                ScheduledExecutorService s = (ScheduledExecutorService) p;
2029 +                ScheduledFuture<?> future;
2030 +
2031 +                recorder.reset();
2032 +                future = s.schedule(r, randomTimeout(), randomTimeUnit());
2033 +                assertFalse(future.isDone());
2034 +                if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2035 +                assertSame(p, recorder.p);
2036 +
2037 +                recorder.reset();
2038 +                future = s.schedule(c, randomTimeout(), randomTimeUnit());
2039 +                assertFalse(future.isDone());
2040 +                if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2041 +                assertSame(p, recorder.p);
2042 +
2043 +                recorder.reset();
2044 +                future = s.scheduleAtFixedRate(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS);
2045 +                assertFalse(future.isDone());
2046 +                if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2047 +                assertSame(p, recorder.p);
2048 +
2049 +                recorder.reset();
2050 +                future = s.scheduleWithFixedDelay(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS);
2051 +                assertFalse(future.isDone());
2052 +                if (stock) assertTrue(!((FutureTask) recorder.r).isDone());
2053 +                assertSame(p, recorder.p);
2054 +            }
2055 +        }
2056 +
2057 +        // Checking our custom handler above should be sufficient, but
2058 +        // we add some integration tests of standard handlers.
2059 +        final AtomicReference<Thread> thread = new AtomicReference<>();
2060 +        final Runnable setThread = () -> thread.set(Thread.currentThread());
2061 +
2062 +        setRejectedExecutionHandler(p, new ThreadPoolExecutor.AbortPolicy());
2063 +        try {
2064 +            p.execute(setThread);
2065 +            shouldThrow();
2066 +        } catch (RejectedExecutionException success) {}
2067 +        assertNull(thread.get());
2068 +
2069 +        setRejectedExecutionHandler(p, new ThreadPoolExecutor.DiscardPolicy());
2070 +        p.execute(setThread);
2071 +        assertNull(thread.get());
2072 +
2073 +        setRejectedExecutionHandler(p, new ThreadPoolExecutor.CallerRunsPolicy());
2074 +        p.execute(setThread);
2075 +        if (p.isShutdown())
2076 +            assertNull(thread.get());
2077 +        else
2078 +            assertSame(Thread.currentThread(), thread.get());
2079 +
2080 +        setRejectedExecutionHandler(p, savedHandler);
2081 +
2082 +        // check that pool was not perturbed by handlers
2083 +        assertEquals(savedTaskCount, p.getTaskCount());
2084 +        assertEquals(savedCompletedTaskCount, p.getCompletedTaskCount());
2085 +        assertEquals(savedQueueSize, p.getQueue().size());
2086 +    }
2087   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines