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.63 by jsr166, Mon Oct 11 08:30:01 2010 UTC vs.
Revision 1.134 by jsr166, Sun Jun 14 20:58:14 2015 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
4 < * http://creativecommons.org/licenses/publicdomain
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5   * Other contributors include Andrew Wright, Jeffrey Hayes,
6   * Pat Fisher, Mike Judd.
7   */
8  
9 import junit.framework.*;
10 import java.util.PropertyPermission;
11 import java.util.concurrent.*;
12 import java.util.concurrent.atomic.AtomicReference;
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 + import static java.util.concurrent.TimeUnit.NANOSECONDS;
11 +
12 + import java.io.ByteArrayInputStream;
13 + import java.io.ByteArrayOutputStream;
14 + import java.io.ObjectInputStream;
15 + import java.io.ObjectOutputStream;
16 + import java.lang.management.ManagementFactory;
17 + import java.lang.management.ThreadInfo;
18 + import java.lang.reflect.Constructor;
19 + import java.lang.reflect.Method;
20 + import java.lang.reflect.Modifier;
21   import java.security.CodeSource;
22   import java.security.Permission;
23   import java.security.PermissionCollection;
# Line 18 | Line 25 | import java.security.Permissions;
25   import java.security.Policy;
26   import java.security.ProtectionDomain;
27   import java.security.SecurityPermission;
28 + import java.util.ArrayList;
29 + import java.util.Arrays;
30 + import java.util.Date;
31 + import java.util.Enumeration;
32 + import java.util.Iterator;
33 + import java.util.List;
34 + import java.util.NoSuchElementException;
35 + import java.util.PropertyPermission;
36 + import java.util.concurrent.BlockingQueue;
37 + import java.util.concurrent.Callable;
38 + import java.util.concurrent.CountDownLatch;
39 + import java.util.concurrent.CyclicBarrier;
40 + import java.util.concurrent.ExecutorService;
41 + import java.util.concurrent.Future;
42 + import java.util.concurrent.RecursiveAction;
43 + import java.util.concurrent.RecursiveTask;
44 + import java.util.concurrent.RejectedExecutionHandler;
45 + import java.util.concurrent.Semaphore;
46 + import java.util.concurrent.ThreadFactory;
47 + import java.util.concurrent.ThreadPoolExecutor;
48 + import java.util.concurrent.TimeoutException;
49 + import java.util.concurrent.atomic.AtomicReference;
50 + import java.util.regex.Pattern;
51 +
52 + import junit.framework.AssertionFailedError;
53 + import junit.framework.Test;
54 + import junit.framework.TestCase;
55 + import junit.framework.TestResult;
56 + import junit.framework.TestSuite;
57  
58   /**
59   * Base class for JSR166 Junit TCK tests.  Defines some constants,
# Line 60 | Line 96 | import java.security.SecurityPermission;
96   *
97   * </ol>
98   *
99 < * <p> <b>Other notes</b>
99 > * <p><b>Other notes</b>
100   * <ul>
101   *
102   * <li> Usually, there is one testcase method per JSR166 method
# Line 100 | Line 136 | public class JSR166TestCase extends Test
136          Boolean.getBoolean("jsr166.expensiveTests");
137  
138      /**
139 +     * If true, also run tests that are not part of the official tck
140 +     * because they test unspecified implementation details.
141 +     */
142 +    protected static final boolean testImplementationDetails =
143 +        Boolean.getBoolean("jsr166.testImplementationDetails");
144 +
145 +    /**
146       * If true, report on stdout all "slow" tests, that is, ones that
147       * take more than profileThreshold milliseconds to execute.
148       */
# Line 113 | Line 156 | public class JSR166TestCase extends Test
156      private static final long profileThreshold =
157          Long.getLong("jsr166.profileThreshold", 100);
158  
159 +    /**
160 +     * The number of repetitions per test (for tickling rare bugs).
161 +     */
162 +    private static final int runsPerTest =
163 +        Integer.getInteger("jsr166.runsPerTest", 1);
164 +
165 +    /**
166 +     * The number of repetitions of the test suite (for finding leaks?).
167 +     */
168 +    private static final int suiteRuns =
169 +        Integer.getInteger("jsr166.suiteRuns", 1);
170 +
171 +    public JSR166TestCase() { super(); }
172 +    public JSR166TestCase(String name) { super(name); }
173 +
174 +    /**
175 +     * A filter for tests to run, matching strings of the form
176 +     * methodName(className), e.g. "testInvokeAll5(ForkJoinPoolTest)"
177 +     * Usefully combined with jsr166.runsPerTest.
178 +     */
179 +    private static final Pattern methodFilter = methodFilter();
180 +
181 +    private static Pattern methodFilter() {
182 +        String regex = System.getProperty("jsr166.methodFilter");
183 +        return (regex == null) ? null : Pattern.compile(regex);
184 +    }
185 +
186      protected void runTest() throws Throwable {
187 <        if (profileTests)
188 <            runTestProfiled();
189 <        else
190 <            super.runTest();
187 >        if (methodFilter == null
188 >            || methodFilter.matcher(toString()).find()) {
189 >            for (int i = 0; i < runsPerTest; i++) {
190 >                if (profileTests)
191 >                    runTestProfiled();
192 >                else
193 >                    super.runTest();
194 >            }
195 >        }
196      }
197  
198      protected void runTestProfiled() throws Throwable {
199 +        // Warmup run, notably to trigger all needed classloading.
200 +        super.runTest();
201          long t0 = System.nanoTime();
202          try {
203              super.runTest();
204          } finally {
205 <            long elapsedMillis =
129 <                (System.nanoTime() - t0) / (1000L * 1000L);
205 >            long elapsedMillis = millisElapsedSince(t0);
206              if (elapsedMillis >= profileThreshold)
207                  System.out.printf("%n%s: %d%n", toString(), elapsedMillis);
208          }
209      }
210  
211      /**
212 <     * Runs all JSR166 unit tests using junit.textui.TestRunner
212 >     * Runs all JSR166 unit tests using junit.textui.TestRunner.
213       */
214      public static void main(String[] args) {
215 +        main(suite(), args);
216 +    }
217 +
218 +    /**
219 +     * Runs all unit tests in the given test suite.
220 +     * Actual behavior influenced by jsr166.* system properties.
221 +     */
222 +    static void main(Test suite, String[] args) {
223          if (useSecurityManager) {
224              System.err.println("Setting a permissive security manager");
225              Policy.setPolicy(permissivePolicy());
226              System.setSecurityManager(new SecurityManager());
227          }
228 <        int iters = (args.length == 0) ? 1 : Integer.parseInt(args[0]);
229 <
230 <        Test s = suite();
231 <        for (int i = 0; i < iters; ++i) {
148 <            junit.textui.TestRunner.run(s);
228 >        for (int i = 0; i < suiteRuns; i++) {
229 >            TestResult result = junit.textui.TestRunner.run(suite);
230 >            if (!result.wasSuccessful())
231 >                System.exit(1);
232              System.gc();
233              System.runFinalization();
234          }
152        System.exit(0);
235      }
236  
237      public static TestSuite newTestSuite(Object... suiteOrClasses) {
# Line 165 | Line 247 | public class JSR166TestCase extends Test
247          return suite;
248      }
249  
250 +    public static void addNamedTestClasses(TestSuite suite,
251 +                                           String... testClassNames) {
252 +        for (String testClassName : testClassNames) {
253 +            try {
254 +                Class<?> testClass = Class.forName(testClassName);
255 +                Method m = testClass.getDeclaredMethod("suite",
256 +                                                       new Class<?>[0]);
257 +                suite.addTest(newTestSuite((Test)m.invoke(null)));
258 +            } catch (Exception e) {
259 +                throw new Error("Missing test class", e);
260 +            }
261 +        }
262 +    }
263 +
264 +    public static final double JAVA_CLASS_VERSION;
265 +    public static final String JAVA_SPECIFICATION_VERSION;
266 +    static {
267 +        try {
268 +            JAVA_CLASS_VERSION = java.security.AccessController.doPrivileged(
269 +                new java.security.PrivilegedAction<Double>() {
270 +                public Double run() {
271 +                    return Double.valueOf(System.getProperty("java.class.version"));}});
272 +            JAVA_SPECIFICATION_VERSION = java.security.AccessController.doPrivileged(
273 +                new java.security.PrivilegedAction<String>() {
274 +                public String run() {
275 +                    return System.getProperty("java.specification.version");}});
276 +        } catch (Throwable t) {
277 +            throw new Error(t);
278 +        }
279 +    }
280 +
281 +    public static boolean atLeastJava6() { return JAVA_CLASS_VERSION >= 50.0; }
282 +    public static boolean atLeastJava7() { return JAVA_CLASS_VERSION >= 51.0; }
283 +    public static boolean atLeastJava8() { return JAVA_CLASS_VERSION >= 52.0; }
284 +    public static boolean atLeastJava9() { return JAVA_CLASS_VERSION >= 53.0; }
285 +
286      /**
287       * Collects all JSR166 unit tests as one suite.
288       */
289      public static Test suite() {
290 <        return newTestSuite(
290 >        // Java7+ test classes
291 >        TestSuite suite = newTestSuite(
292              ForkJoinPoolTest.suite(),
293              ForkJoinTaskTest.suite(),
294              RecursiveActionTest.suite(),
# Line 234 | Line 353 | public class JSR166TestCase extends Test
353              TreeSetTest.suite(),
354              TreeSubMapTest.suite(),
355              TreeSubSetTest.suite());
356 +
357 +        // Java8+ test classes
358 +        if (atLeastJava8()) {
359 +            String[] java8TestClassNames = {
360 +                "Atomic8Test",
361 +                "CompletableFutureTest",
362 +                "ConcurrentHashMap8Test",
363 +                "CountedCompleterTest",
364 +                "DoubleAccumulatorTest",
365 +                "DoubleAdderTest",
366 +                "ForkJoinPool8Test",
367 +                "ForkJoinTask8Test",
368 +                "LongAccumulatorTest",
369 +                "LongAdderTest",
370 +                "SplittableRandomTest",
371 +                "StampedLockTest",
372 +                "ThreadLocalRandom8Test",
373 +            };
374 +            addNamedTestClasses(suite, java8TestClassNames);
375 +        }
376 +
377 +        // Java9+ test classes
378 +        if (atLeastJava9()) {
379 +            String[] java9TestClassNames = {
380 +                "ThreadPoolExecutor9Test",
381 +            };
382 +            addNamedTestClasses(suite, java9TestClassNames);
383 +        }
384 +
385 +        return suite;
386      }
387  
388 +    /** Returns list of junit-style test method names in given class. */
389 +    public static ArrayList<String> testMethodNames(Class<?> testClass) {
390 +        Method[] methods = testClass.getDeclaredMethods();
391 +        ArrayList<String> names = new ArrayList<String>(methods.length);
392 +        for (Method method : methods) {
393 +            if (method.getName().startsWith("test")
394 +                && Modifier.isPublic(method.getModifiers())
395 +                // method.getParameterCount() requires jdk8+
396 +                && method.getParameterTypes().length == 0) {
397 +                names.add(method.getName());
398 +            }
399 +        }
400 +        return names;
401 +    }
402 +
403 +    /**
404 +     * Returns junit-style testSuite for the given test class, but
405 +     * parameterized by passing extra data to each test.
406 +     */
407 +    public static <ExtraData> Test parameterizedTestSuite
408 +        (Class<? extends JSR166TestCase> testClass,
409 +         Class<ExtraData> dataClass,
410 +         ExtraData data) {
411 +        try {
412 +            TestSuite suite = new TestSuite();
413 +            Constructor c =
414 +                testClass.getDeclaredConstructor(dataClass, String.class);
415 +            for (String methodName : testMethodNames(testClass))
416 +                suite.addTest((Test) c.newInstance(data, methodName));
417 +            return suite;
418 +        } catch (Exception e) {
419 +            throw new Error(e);
420 +        }
421 +    }
422 +
423 +    /**
424 +     * Returns junit-style testSuite for the jdk8 extension of the
425 +     * given test class, but parameterized by passing extra data to
426 +     * each test.  Uses reflection to allow compilation in jdk7.
427 +     */
428 +    public static <ExtraData> Test jdk8ParameterizedTestSuite
429 +        (Class<? extends JSR166TestCase> testClass,
430 +         Class<ExtraData> dataClass,
431 +         ExtraData data) {
432 +        if (atLeastJava8()) {
433 +            String name = testClass.getName();
434 +            String name8 = name.replaceAll("Test$", "8Test");
435 +            if (name.equals(name8)) throw new Error(name);
436 +            try {
437 +                return (Test)
438 +                    Class.forName(name8)
439 +                    .getMethod("testSuite", new Class[] { dataClass })
440 +                    .invoke(null, data);
441 +            } catch (Exception e) {
442 +                throw new Error(e);
443 +            }
444 +        } else {
445 +            return new TestSuite();
446 +        }
447 +
448 +    }
449 +
450 +    // Delays for timing-dependent tests, in milliseconds.
451  
452      public static long SHORT_DELAY_MS;
453      public static long SMALL_DELAY_MS;
454      public static long MEDIUM_DELAY_MS;
455      public static long LONG_DELAY_MS;
456  
245
457      /**
458       * Returns the shortest timed delay. This could
459       * be reimplemented to use for example a Property.
# Line 251 | Line 462 | public class JSR166TestCase extends Test
462          return 50;
463      }
464  
254
465      /**
466       * Sets delays as multiples of SHORT_DELAY.
467       */
# Line 259 | Line 469 | public class JSR166TestCase extends Test
469          SHORT_DELAY_MS = getShortDelay();
470          SMALL_DELAY_MS  = SHORT_DELAY_MS * 5;
471          MEDIUM_DELAY_MS = SHORT_DELAY_MS * 10;
472 <        LONG_DELAY_MS   = SHORT_DELAY_MS * 50;
472 >        LONG_DELAY_MS   = SHORT_DELAY_MS * 200;
473 >    }
474 >
475 >    /**
476 >     * Returns a timeout in milliseconds to be used in tests that
477 >     * verify that operations block or time out.
478 >     */
479 >    long timeoutMillis() {
480 >        return SHORT_DELAY_MS / 4;
481 >    }
482 >
483 >    /**
484 >     * Returns a new Date instance representing a time delayMillis
485 >     * milliseconds in the future.
486 >     */
487 >    Date delayedDate(long delayMillis) {
488 >        return new Date(System.currentTimeMillis() + delayMillis);
489      }
490  
491      /**
# Line 283 | Line 509 | public class JSR166TestCase extends Test
509      }
510  
511      /**
512 +     * Extra checks that get done for all test cases.
513 +     *
514       * Triggers test case failure if any thread assertions have failed,
515       * by rethrowing, in the test harness thread, any exception recorded
516       * earlier by threadRecordFailure.
517 +     *
518 +     * Triggers test case failure if interrupt status is set in the main thread.
519       */
520      public void tearDown() throws Exception {
521 <        Throwable t = threadFailure.get();
521 >        Throwable t = threadFailure.getAndSet(null);
522          if (t != null) {
523              if (t instanceof Error)
524                  throw (Error) t;
# Line 303 | Line 533 | public class JSR166TestCase extends Test
533                  throw afe;
534              }
535          }
536 +
537 +        if (Thread.interrupted())
538 +            throw new AssertionFailedError("interrupt status set in main thread");
539 +
540 +        checkForkJoinPoolThreadLeaks();
541 +    }
542 +
543 +    /**
544 +     * Finds missing try { ... } finally { joinPool(e); }
545 +     */
546 +    void checkForkJoinPoolThreadLeaks() throws InterruptedException {
547 +        Thread[] survivors = new Thread[5];
548 +        int count = Thread.enumerate(survivors);
549 +        for (int i = 0; i < count; i++) {
550 +            Thread thread = survivors[i];
551 +            String name = thread.getName();
552 +            if (name.startsWith("ForkJoinPool-")) {
553 +                // give thread some time to terminate
554 +                thread.join(LONG_DELAY_MS);
555 +                if (!thread.isAlive()) continue;
556 +                throw new AssertionFailedError
557 +                    (String.format("Found leaked ForkJoinPool thread test=%s thread=%s%n",
558 +                                   toString(), name));
559 +            }
560 +        }
561      }
562  
563      /**
# Line 383 | Line 638 | public class JSR166TestCase extends Test
638      public void threadAssertEquals(Object x, Object y) {
639          try {
640              assertEquals(x, y);
641 <        } catch (AssertionFailedError t) {
642 <            threadRecordFailure(t);
643 <            throw t;
644 <        } catch (Throwable t) {
645 <            threadUnexpectedException(t);
641 >        } catch (AssertionFailedError fail) {
642 >            threadRecordFailure(fail);
643 >            throw fail;
644 >        } catch (Throwable fail) {
645 >            threadUnexpectedException(fail);
646          }
647      }
648  
# Line 399 | Line 654 | public class JSR166TestCase extends Test
654      public void threadAssertSame(Object x, Object y) {
655          try {
656              assertSame(x, y);
657 <        } catch (AssertionFailedError t) {
658 <            threadRecordFailure(t);
659 <            throw t;
657 >        } catch (AssertionFailedError fail) {
658 >            threadRecordFailure(fail);
659 >            throw fail;
660          }
661      }
662  
# Line 434 | Line 689 | public class JSR166TestCase extends Test
689          else {
690              AssertionFailedError afe =
691                  new AssertionFailedError("unexpected exception: " + t);
692 <            t.initCause(t);
692 >            afe.initCause(t);
693              throw afe;
694          }
695      }
696  
697      /**
698 +     * Delays, via Thread.sleep, for the given millisecond delay, but
699 +     * if the sleep is shorter than specified, may re-sleep or yield
700 +     * until time elapses.
701 +     */
702 +    static void delay(long millis) throws InterruptedException {
703 +        long startTime = System.nanoTime();
704 +        long ns = millis * 1000 * 1000;
705 +        for (;;) {
706 +            if (millis > 0L)
707 +                Thread.sleep(millis);
708 +            else // too short to sleep
709 +                Thread.yield();
710 +            long d = ns - (System.nanoTime() - startTime);
711 +            if (d > 0L)
712 +                millis = d / (1000 * 1000);
713 +            else
714 +                break;
715 +        }
716 +    }
717 +
718 +    /**
719       * Waits out termination of a thread pool or fails doing so.
720       */
721 <    public void joinPool(ExecutorService exec) {
721 >    void joinPool(ExecutorService exec) {
722          try {
723              exec.shutdown();
724 <            assertTrue("ExecutorService did not terminate in a timely manner",
725 <                       exec.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
724 >            if (!exec.awaitTermination(2 * LONG_DELAY_MS, MILLISECONDS))
725 >                fail("ExecutorService " + exec +
726 >                     " did not terminate in a timely manner");
727          } catch (SecurityException ok) {
728              // Allowed in case test doesn't have privs
729 <        } catch (InterruptedException ie) {
729 >        } catch (InterruptedException fail) {
730              fail("Unexpected InterruptedException");
731          }
732      }
733  
734 +    /**
735 +     * A debugging tool to print all stack traces, as jstack does.
736 +     */
737 +    static void printAllStackTraces() {
738 +        for (ThreadInfo info :
739 +                 ManagementFactory.getThreadMXBean()
740 +                 .dumpAllThreads(true, true))
741 +            System.err.print(info);
742 +    }
743 +
744 +    /**
745 +     * Checks that thread does not terminate within the default
746 +     * millisecond delay of {@code timeoutMillis()}.
747 +     */
748 +    void assertThreadStaysAlive(Thread thread) {
749 +        assertThreadStaysAlive(thread, timeoutMillis());
750 +    }
751 +
752 +    /**
753 +     * Checks that thread does not terminate within the given millisecond delay.
754 +     */
755 +    void assertThreadStaysAlive(Thread thread, long millis) {
756 +        try {
757 +            // No need to optimize the failing case via Thread.join.
758 +            delay(millis);
759 +            assertTrue(thread.isAlive());
760 +        } catch (InterruptedException fail) {
761 +            fail("Unexpected InterruptedException");
762 +        }
763 +    }
764 +
765 +    /**
766 +     * Checks that the threads do not terminate within the default
767 +     * millisecond delay of {@code timeoutMillis()}.
768 +     */
769 +    void assertThreadsStayAlive(Thread... threads) {
770 +        assertThreadsStayAlive(timeoutMillis(), threads);
771 +    }
772 +
773 +    /**
774 +     * Checks that the threads do not terminate within the given millisecond delay.
775 +     */
776 +    void assertThreadsStayAlive(long millis, Thread... threads) {
777 +        try {
778 +            // No need to optimize the failing case via Thread.join.
779 +            delay(millis);
780 +            for (Thread thread : threads)
781 +                assertTrue(thread.isAlive());
782 +        } catch (InterruptedException fail) {
783 +            fail("Unexpected InterruptedException");
784 +        }
785 +    }
786 +
787 +    /**
788 +     * Checks that future.get times out, with the default timeout of
789 +     * {@code timeoutMillis()}.
790 +     */
791 +    void assertFutureTimesOut(Future future) {
792 +        assertFutureTimesOut(future, timeoutMillis());
793 +    }
794 +
795 +    /**
796 +     * Checks that future.get times out, with the given millisecond timeout.
797 +     */
798 +    void assertFutureTimesOut(Future future, long timeoutMillis) {
799 +        long startTime = System.nanoTime();
800 +        try {
801 +            future.get(timeoutMillis, MILLISECONDS);
802 +            shouldThrow();
803 +        } catch (TimeoutException success) {
804 +        } catch (Exception fail) {
805 +            threadUnexpectedException(fail);
806 +        } finally { future.cancel(true); }
807 +        assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
808 +    }
809  
810      /**
811       * Fails with message "should throw exception".
# Line 494 | Line 846 | public class JSR166TestCase extends Test
846      public static final Integer m6  = new Integer(-6);
847      public static final Integer m10 = new Integer(-10);
848  
497
849      /**
850       * Runs Runnable r with a security policy that permits precisely
851       * the specified permissions.  If there is no current security
# Line 506 | Line 857 | public class JSR166TestCase extends Test
857          SecurityManager sm = System.getSecurityManager();
858          if (sm == null) {
859              r.run();
860 +        }
861 +        runWithSecurityManagerWithPermissions(r, permissions);
862 +    }
863 +
864 +    /**
865 +     * Runs Runnable r with a security policy that permits precisely
866 +     * the specified permissions.  If there is no current security
867 +     * manager, a temporary one is set for the duration of the
868 +     * Runnable.  We require that any security manager permit
869 +     * getPolicy/setPolicy.
870 +     */
871 +    public void runWithSecurityManagerWithPermissions(Runnable r,
872 +                                                      Permission... permissions) {
873 +        SecurityManager sm = System.getSecurityManager();
874 +        if (sm == null) {
875              Policy savedPolicy = Policy.getPolicy();
876              try {
877                  Policy.setPolicy(permissivePolicy());
878                  System.setSecurityManager(new SecurityManager());
879 <                runWithPermissions(r, permissions);
879 >                runWithSecurityManagerWithPermissions(r, permissions);
880              } finally {
881                  System.setSecurityManager(null);
882                  Policy.setPolicy(savedPolicy);
# Line 558 | Line 924 | public class JSR166TestCase extends Test
924              return perms.implies(p);
925          }
926          public void refresh() {}
927 +        public String toString() {
928 +            List<Permission> ps = new ArrayList<Permission>();
929 +            for (Enumeration<Permission> e = perms.elements(); e.hasMoreElements();)
930 +                ps.add(e.nextElement());
931 +            return "AdjustablePolicy with permissions " + ps;
932 +        }
933      }
934  
935      /**
# Line 585 | Line 957 | public class JSR166TestCase extends Test
957       */
958      void sleep(long millis) {
959          try {
960 <            Thread.sleep(millis);
961 <        } catch (InterruptedException ie) {
960 >            delay(millis);
961 >        } catch (InterruptedException fail) {
962              AssertionFailedError afe =
963                  new AssertionFailedError("Unexpected InterruptedException");
964 <            afe.initCause(ie);
964 >            afe.initCause(fail);
965              throw afe;
966          }
967      }
968  
969      /**
970 <     * Sleeps until the timeout has elapsed, or interrupted.
971 <     * Does <em>NOT</em> throw InterruptedException.
970 >     * Spin-waits up to the specified number of milliseconds for the given
971 >     * thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING.
972 >     */
973 >    void waitForThreadToEnterWaitState(Thread thread, long timeoutMillis) {
974 >        long startTime = System.nanoTime();
975 >        for (;;) {
976 >            Thread.State s = thread.getState();
977 >            if (s == Thread.State.BLOCKED ||
978 >                s == Thread.State.WAITING ||
979 >                s == Thread.State.TIMED_WAITING)
980 >                return;
981 >            else if (s == Thread.State.TERMINATED)
982 >                fail("Unexpected thread termination");
983 >            else if (millisElapsedSince(startTime) > timeoutMillis) {
984 >                threadAssertTrue(thread.isAlive());
985 >                return;
986 >            }
987 >            Thread.yield();
988 >        }
989 >    }
990 >
991 >    /**
992 >     * Waits up to LONG_DELAY_MS for the given thread to enter a wait
993 >     * state: BLOCKED, WAITING, or TIMED_WAITING.
994 >     */
995 >    void waitForThreadToEnterWaitState(Thread thread) {
996 >        waitForThreadToEnterWaitState(thread, LONG_DELAY_MS);
997 >    }
998 >
999 >    /**
1000 >     * Returns the number of milliseconds since time given by
1001 >     * startNanoTime, which must have been previously returned from a
1002 >     * call to {@link System#nanoTime()}.
1003 >     */
1004 >    static long millisElapsedSince(long startNanoTime) {
1005 >        return NANOSECONDS.toMillis(System.nanoTime() - startNanoTime);
1006 >    }
1007 >
1008 > //     void assertTerminatesPromptly(long timeoutMillis, Runnable r) {
1009 > //         long startTime = System.nanoTime();
1010 > //         try {
1011 > //             r.run();
1012 > //         } catch (Throwable fail) { threadUnexpectedException(fail); }
1013 > //         if (millisElapsedSince(startTime) > timeoutMillis/2)
1014 > //             throw new AssertionFailedError("did not return promptly");
1015 > //     }
1016 >
1017 > //     void assertTerminatesPromptly(Runnable r) {
1018 > //         assertTerminatesPromptly(LONG_DELAY_MS/2, r);
1019 > //     }
1020 >
1021 >    /**
1022 >     * Checks that timed f.get() returns the expected value, and does not
1023 >     * wait for the timeout to elapse before returning.
1024       */
1025 <    void sleepTillInterrupted(long timeoutMillis) {
1025 >    <T> void checkTimedGet(Future<T> f, T expectedValue, long timeoutMillis) {
1026 >        long startTime = System.nanoTime();
1027          try {
1028 <            Thread.sleep(timeoutMillis);
1029 <        } catch (InterruptedException wakeup) {}
1028 >            assertEquals(expectedValue, f.get(timeoutMillis, MILLISECONDS));
1029 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
1030 >        if (millisElapsedSince(startTime) > timeoutMillis/2)
1031 >            throw new AssertionFailedError("timed get did not return promptly");
1032 >    }
1033 >
1034 >    <T> void checkTimedGet(Future<T> f, T expectedValue) {
1035 >        checkTimedGet(f, expectedValue, LONG_DELAY_MS);
1036      }
1037  
1038      /**
# Line 622 | Line 1053 | public class JSR166TestCase extends Test
1053      void awaitTermination(Thread t, long timeoutMillis) {
1054          try {
1055              t.join(timeoutMillis);
1056 <        } catch (InterruptedException ie) {
1057 <            threadUnexpectedException(ie);
1056 >        } catch (InterruptedException fail) {
1057 >            threadUnexpectedException(fail);
1058          } finally {
1059 <            if (t.isAlive()) {
1059 >            if (t.getState() != Thread.State.TERMINATED) {
1060                  t.interrupt();
1061                  fail("Test timed out");
1062              }
1063          }
1064      }
1065  
1066 +    /**
1067 +     * Waits for LONG_DELAY_MS milliseconds for the thread to
1068 +     * terminate (using {@link Thread#join(long)}), else interrupts
1069 +     * the thread (in the hope that it may terminate later) and fails.
1070 +     */
1071 +    void awaitTermination(Thread t) {
1072 +        awaitTermination(t, LONG_DELAY_MS);
1073 +    }
1074 +
1075      // Some convenient Runnable classes
1076  
1077      public abstract class CheckedRunnable implements Runnable {
# Line 640 | Line 1080 | public class JSR166TestCase extends Test
1080          public final void run() {
1081              try {
1082                  realRun();
1083 <            } catch (Throwable t) {
1084 <                threadUnexpectedException(t);
1083 >            } catch (Throwable fail) {
1084 >                threadUnexpectedException(fail);
1085              }
1086          }
1087      }
# Line 694 | Line 1134 | public class JSR166TestCase extends Test
1134                  realRun();
1135                  threadShouldThrow("InterruptedException");
1136              } catch (InterruptedException success) {
1137 <            } catch (Throwable t) {
1138 <                threadUnexpectedException(t);
1137 >                threadAssertFalse(Thread.interrupted());
1138 >            } catch (Throwable fail) {
1139 >                threadUnexpectedException(fail);
1140              }
1141          }
1142      }
# Line 706 | Line 1147 | public class JSR166TestCase extends Test
1147          public final T call() {
1148              try {
1149                  return realCall();
1150 <            } catch (Throwable t) {
1151 <                threadUnexpectedException(t);
1150 >            } catch (Throwable fail) {
1151 >                threadUnexpectedException(fail);
1152                  return null;
1153              }
1154          }
# Line 723 | Line 1164 | public class JSR166TestCase extends Test
1164                  threadShouldThrow("InterruptedException");
1165                  return result;
1166              } catch (InterruptedException success) {
1167 <            } catch (Throwable t) {
1168 <                threadUnexpectedException(t);
1167 >                threadAssertFalse(Thread.interrupted());
1168 >            } catch (Throwable fail) {
1169 >                threadUnexpectedException(fail);
1170              }
1171              return null;
1172          }
# Line 746 | Line 1188 | public class JSR166TestCase extends Test
1188  
1189      public Callable<String> latchAwaitingStringTask(final CountDownLatch latch) {
1190          return new CheckedCallable<String>() {
1191 <            public String realCall() {
1191 >            protected String realCall() {
1192                  try {
1193                      latch.await();
1194                  } catch (InterruptedException quittingTime) {}
# Line 754 | Line 1196 | public class JSR166TestCase extends Test
1196              }};
1197      }
1198  
1199 +    public Runnable awaiter(final CountDownLatch latch) {
1200 +        return new CheckedRunnable() {
1201 +            public void realRun() throws InterruptedException {
1202 +                await(latch);
1203 +            }};
1204 +    }
1205 +
1206 +    public void await(CountDownLatch latch) {
1207 +        try {
1208 +            assertTrue(latch.await(LONG_DELAY_MS, MILLISECONDS));
1209 +        } catch (Throwable fail) {
1210 +            threadUnexpectedException(fail);
1211 +        }
1212 +    }
1213 +
1214 +    public void await(Semaphore semaphore) {
1215 +        try {
1216 +            assertTrue(semaphore.tryAcquire(LONG_DELAY_MS, MILLISECONDS));
1217 +        } catch (Throwable fail) {
1218 +            threadUnexpectedException(fail);
1219 +        }
1220 +    }
1221 +
1222 + //     /**
1223 + //      * Spin-waits up to LONG_DELAY_MS until flag becomes true.
1224 + //      */
1225 + //     public void await(AtomicBoolean flag) {
1226 + //         await(flag, LONG_DELAY_MS);
1227 + //     }
1228 +
1229 + //     /**
1230 + //      * Spin-waits up to the specified timeout until flag becomes true.
1231 + //      */
1232 + //     public void await(AtomicBoolean flag, long timeoutMillis) {
1233 + //         long startTime = System.nanoTime();
1234 + //         while (!flag.get()) {
1235 + //             if (millisElapsedSince(startTime) > timeoutMillis)
1236 + //                 throw new AssertionFailedError("timed out");
1237 + //             Thread.yield();
1238 + //         }
1239 + //     }
1240 +
1241      public static class NPETask implements Callable<String> {
1242          public String call() { throw new NullPointerException(); }
1243      }
# Line 764 | Line 1248 | public class JSR166TestCase extends Test
1248  
1249      public class ShortRunnable extends CheckedRunnable {
1250          protected void realRun() throws Throwable {
1251 <            Thread.sleep(SHORT_DELAY_MS);
1251 >            delay(SHORT_DELAY_MS);
1252          }
1253      }
1254  
1255      public class ShortInterruptedRunnable extends CheckedInterruptedRunnable {
1256          protected void realRun() throws InterruptedException {
1257 <            Thread.sleep(SHORT_DELAY_MS);
1257 >            delay(SHORT_DELAY_MS);
1258          }
1259      }
1260  
1261      public class SmallRunnable extends CheckedRunnable {
1262          protected void realRun() throws Throwable {
1263 <            Thread.sleep(SMALL_DELAY_MS);
1263 >            delay(SMALL_DELAY_MS);
1264          }
1265      }
1266  
1267      public class SmallPossiblyInterruptedRunnable extends CheckedRunnable {
1268          protected void realRun() {
1269              try {
1270 <                Thread.sleep(SMALL_DELAY_MS);
1270 >                delay(SMALL_DELAY_MS);
1271              } catch (InterruptedException ok) {}
1272          }
1273      }
1274  
1275      public class SmallCallable extends CheckedCallable {
1276          protected Object realCall() throws InterruptedException {
1277 <            Thread.sleep(SMALL_DELAY_MS);
1277 >            delay(SMALL_DELAY_MS);
1278              return Boolean.TRUE;
1279          }
1280      }
1281  
1282      public class MediumRunnable extends CheckedRunnable {
1283          protected void realRun() throws Throwable {
1284 <            Thread.sleep(MEDIUM_DELAY_MS);
1284 >            delay(MEDIUM_DELAY_MS);
1285          }
1286      }
1287  
1288      public class MediumInterruptedRunnable extends CheckedInterruptedRunnable {
1289          protected void realRun() throws InterruptedException {
1290 <            Thread.sleep(MEDIUM_DELAY_MS);
1290 >            delay(MEDIUM_DELAY_MS);
1291          }
1292      }
1293  
# Line 811 | Line 1295 | public class JSR166TestCase extends Test
1295          return new CheckedRunnable() {
1296              protected void realRun() {
1297                  try {
1298 <                    Thread.sleep(timeoutMillis);
1298 >                    delay(timeoutMillis);
1299                  } catch (InterruptedException ok) {}
1300              }};
1301      }
# Line 819 | Line 1303 | public class JSR166TestCase extends Test
1303      public class MediumPossiblyInterruptedRunnable extends CheckedRunnable {
1304          protected void realRun() {
1305              try {
1306 <                Thread.sleep(MEDIUM_DELAY_MS);
1306 >                delay(MEDIUM_DELAY_MS);
1307              } catch (InterruptedException ok) {}
1308          }
1309      }
# Line 827 | Line 1311 | public class JSR166TestCase extends Test
1311      public class LongPossiblyInterruptedRunnable extends CheckedRunnable {
1312          protected void realRun() {
1313              try {
1314 <                Thread.sleep(LONG_DELAY_MS);
1314 >                delay(LONG_DELAY_MS);
1315              } catch (InterruptedException ok) {}
1316          }
1317      }
# Line 851 | Line 1335 | public class JSR166TestCase extends Test
1335                  public boolean isDone() { return done; }
1336                  public void run() {
1337                      try {
1338 <                        Thread.sleep(timeoutMillis);
1338 >                        delay(timeoutMillis);
1339                          done = true;
1340                      } catch (InterruptedException ok) {}
1341                  }
# Line 862 | Line 1346 | public class JSR166TestCase extends Test
1346          public volatile boolean done = false;
1347          public void run() {
1348              try {
1349 <                Thread.sleep(SHORT_DELAY_MS);
1349 >                delay(SHORT_DELAY_MS);
1350                  done = true;
1351              } catch (InterruptedException ok) {}
1352          }
# Line 872 | Line 1356 | public class JSR166TestCase extends Test
1356          public volatile boolean done = false;
1357          public void run() {
1358              try {
1359 <                Thread.sleep(SMALL_DELAY_MS);
1359 >                delay(SMALL_DELAY_MS);
1360                  done = true;
1361              } catch (InterruptedException ok) {}
1362          }
# Line 882 | Line 1366 | public class JSR166TestCase extends Test
1366          public volatile boolean done = false;
1367          public void run() {
1368              try {
1369 <                Thread.sleep(MEDIUM_DELAY_MS);
1369 >                delay(MEDIUM_DELAY_MS);
1370                  done = true;
1371              } catch (InterruptedException ok) {}
1372          }
# Line 892 | Line 1376 | public class JSR166TestCase extends Test
1376          public volatile boolean done = false;
1377          public void run() {
1378              try {
1379 <                Thread.sleep(LONG_DELAY_MS);
1379 >                delay(LONG_DELAY_MS);
1380                  done = true;
1381              } catch (InterruptedException ok) {}
1382          }
# Line 909 | Line 1393 | public class JSR166TestCase extends Test
1393          public volatile boolean done = false;
1394          public Object call() {
1395              try {
1396 <                Thread.sleep(SMALL_DELAY_MS);
1396 >                delay(SMALL_DELAY_MS);
1397                  done = true;
1398              } catch (InterruptedException ok) {}
1399              return Boolean.TRUE;
# Line 922 | Line 1406 | public class JSR166TestCase extends Test
1406      public abstract class CheckedRecursiveAction extends RecursiveAction {
1407          protected abstract void realCompute() throws Throwable;
1408  
1409 <        public final void compute() {
1409 >        @Override protected final void compute() {
1410              try {
1411                  realCompute();
1412 <            } catch (Throwable t) {
1413 <                threadUnexpectedException(t);
1412 >            } catch (Throwable fail) {
1413 >                threadUnexpectedException(fail);
1414              }
1415          }
1416      }
# Line 937 | Line 1421 | public class JSR166TestCase extends Test
1421      public abstract class CheckedRecursiveTask<T> extends RecursiveTask<T> {
1422          protected abstract T realCompute() throws Throwable;
1423  
1424 <        public final T compute() {
1424 >        @Override protected final T compute() {
1425              try {
1426                  return realCompute();
1427 <            } catch (Throwable t) {
1428 <                threadUnexpectedException(t);
1427 >            } catch (Throwable fail) {
1428 >                threadUnexpectedException(fail);
1429                  return null;
1430              }
1431          }
# Line 956 | Line 1440 | public class JSR166TestCase extends Test
1440      }
1441  
1442      /**
1443 <     * A CyclicBarrier that fails with AssertionFailedErrors instead
1444 <     * of throwing checked exceptions.
1443 >     * A CyclicBarrier that uses timed await and fails with
1444 >     * AssertionFailedErrors instead of throwing checked exceptions.
1445       */
1446      public class CheckedBarrier extends CyclicBarrier {
1447          public CheckedBarrier(int parties) { super(parties); }
1448  
1449          public int await() {
1450              try {
1451 <                return super.await();
1452 <            } catch (Exception e) {
1451 >                return super.await(2 * LONG_DELAY_MS, MILLISECONDS);
1452 >            } catch (TimeoutException timedOut) {
1453 >                throw new AssertionFailedError("timed out");
1454 >            } catch (Exception fail) {
1455                  AssertionFailedError afe =
1456 <                    new AssertionFailedError("Unexpected exception: " + e);
1457 <                afe.initCause(e);
1456 >                    new AssertionFailedError("Unexpected exception: " + fail);
1457 >                afe.initCause(fail);
1458                  throw afe;
1459              }
1460          }
1461      }
1462  
1463 +    void checkEmpty(BlockingQueue q) {
1464 +        try {
1465 +            assertTrue(q.isEmpty());
1466 +            assertEquals(0, q.size());
1467 +            assertNull(q.peek());
1468 +            assertNull(q.poll());
1469 +            assertNull(q.poll(0, MILLISECONDS));
1470 +            assertEquals(q.toString(), "[]");
1471 +            assertTrue(Arrays.equals(q.toArray(), new Object[0]));
1472 +            assertFalse(q.iterator().hasNext());
1473 +            try {
1474 +                q.element();
1475 +                shouldThrow();
1476 +            } catch (NoSuchElementException success) {}
1477 +            try {
1478 +                q.iterator().next();
1479 +                shouldThrow();
1480 +            } catch (NoSuchElementException success) {}
1481 +            try {
1482 +                q.remove();
1483 +                shouldThrow();
1484 +            } catch (NoSuchElementException success) {}
1485 +        } catch (InterruptedException fail) { threadUnexpectedException(fail); }
1486 +    }
1487 +
1488 +    void assertSerialEquals(Object x, Object y) {
1489 +        assertTrue(Arrays.equals(serialBytes(x), serialBytes(y)));
1490 +    }
1491 +
1492 +    void assertNotSerialEquals(Object x, Object y) {
1493 +        assertFalse(Arrays.equals(serialBytes(x), serialBytes(y)));
1494 +    }
1495 +
1496 +    byte[] serialBytes(Object o) {
1497 +        try {
1498 +            ByteArrayOutputStream bos = new ByteArrayOutputStream();
1499 +            ObjectOutputStream oos = new ObjectOutputStream(bos);
1500 +            oos.writeObject(o);
1501 +            oos.flush();
1502 +            oos.close();
1503 +            return bos.toByteArray();
1504 +        } catch (Throwable fail) {
1505 +            threadUnexpectedException(fail);
1506 +            return new byte[0];
1507 +        }
1508 +    }
1509 +
1510 +    @SuppressWarnings("unchecked")
1511 +    <T> T serialClone(T o) {
1512 +        try {
1513 +            ObjectInputStream ois = new ObjectInputStream
1514 +                (new ByteArrayInputStream(serialBytes(o)));
1515 +            T clone = (T) ois.readObject();
1516 +            assertSame(o.getClass(), clone.getClass());
1517 +            return clone;
1518 +        } catch (Throwable fail) {
1519 +            threadUnexpectedException(fail);
1520 +            return null;
1521 +        }
1522 +    }
1523 +
1524 +    public void assertThrows(Class<? extends Throwable> expectedExceptionClass,
1525 +                             Runnable... throwingActions) {
1526 +        for (Runnable throwingAction : throwingActions) {
1527 +            boolean threw = false;
1528 +            try { throwingAction.run(); }
1529 +            catch (Throwable t) {
1530 +                threw = true;
1531 +                if (!expectedExceptionClass.isInstance(t)) {
1532 +                    AssertionFailedError afe =
1533 +                        new AssertionFailedError
1534 +                        ("Expected " + expectedExceptionClass.getName() +
1535 +                         ", got " + t.getClass().getName());
1536 +                    afe.initCause(t);
1537 +                    threadUnexpectedException(afe);
1538 +                }
1539 +            }
1540 +            if (!threw)
1541 +                shouldThrow(expectedExceptionClass.getName());
1542 +        }
1543 +    }
1544 +
1545 +    public void assertIteratorExhausted(Iterator<?> it) {
1546 +        try {
1547 +            it.next();
1548 +            shouldThrow();
1549 +        } catch (NoSuchElementException success) {}
1550 +        assertFalse(it.hasNext());
1551 +    }
1552   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines