• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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/publicdomain/zero/1.0/
5  * Other contributors include Andrew Wright, Jeffrey Hayes,
6  * Pat Fisher, Mike Judd.
7  */
8 
9 /*
10  * Source:
11  * http://gee.cs.oswego.edu/cgi-bin/viewcvs.cgi/jsr166/src/test/tck/JSR166TestCase.java?revision=1.90
12  * (We have made some trivial local modifications (commented out
13  * uncompilable code).)
14  */
15 
16 package com.google.common.util.concurrent;
17 
18 import static java.util.concurrent.TimeUnit.MILLISECONDS;
19 import static java.util.concurrent.TimeUnit.NANOSECONDS;
20 
21 import junit.framework.*;
22 
23 import java.io.ByteArrayInputStream;
24 import java.io.ByteArrayOutputStream;
25 import java.io.ObjectInputStream;
26 import java.io.ObjectOutputStream;
27 import java.security.CodeSource;
28 import java.security.Permission;
29 import java.security.PermissionCollection;
30 import java.security.Permissions;
31 import java.security.Policy;
32 import java.security.ProtectionDomain;
33 import java.security.SecurityPermission;
34 import java.util.Arrays;
35 import java.util.Date;
36 import java.util.NoSuchElementException;
37 import java.util.PropertyPermission;
38 import java.util.concurrent.*;
39 import java.util.concurrent.atomic.AtomicBoolean;
40 import java.util.concurrent.atomic.AtomicReference;
41 
42 /**
43  * Base class for JSR166 Junit TCK tests.  Defines some constants,
44  * utility methods and classes, as well as a simple framework for
45  * helping to make sure that assertions failing in generated threads
46  * cause the associated test that generated them to itself fail (which
47  * JUnit does not otherwise arrange).  The rules for creating such
48  * tests are:
49  *
50  * <ol>
51  *
52  * <li> All assertions in code running in generated threads must use
53  * the forms {@link #threadFail}, {@link #threadAssertTrue}, {@link
54  * #threadAssertEquals}, or {@link #threadAssertNull}, (not
55  * {@code fail}, {@code assertTrue}, etc.) It is OK (but not
56  * particularly recommended) for other code to use these forms too.
57  * Only the most typically used JUnit assertion methods are defined
58  * this way, but enough to live with.</li>
59  *
60  * <li> If you override {@link #setUp} or {@link #tearDown}, make sure
61  * to invoke {@code super.setUp} and {@code super.tearDown} within
62  * them. These methods are used to clear and check for thread
63  * assertion failures.</li>
64  *
65  * <li>All delays and timeouts must use one of the constants {@code
66  * SHORT_DELAY_MS}, {@code SMALL_DELAY_MS}, {@code MEDIUM_DELAY_MS},
67  * {@code LONG_DELAY_MS}. The idea here is that a SHORT is always
68  * discriminable from zero time, and always allows enough time for the
69  * small amounts of computation (creating a thread, calling a few
70  * methods, etc) needed to reach a timeout point. Similarly, a SMALL
71  * is always discriminable as larger than SHORT and smaller than
72  * MEDIUM.  And so on. These constants are set to conservative values,
73  * but even so, if there is ever any doubt, they can all be increased
74  * in one spot to rerun tests on slower platforms.</li>
75  *
76  * <li> All threads generated must be joined inside each test case
77  * method (or {@code fail} to do so) before returning from the
78  * method. The {@code joinPool} method can be used to do this when
79  * using Executors.</li>
80  *
81  * </ol>
82  *
83  * <p> <b>Other notes</b>
84  * <ul>
85  *
86  * <li> Usually, there is one testcase method per JSR166 method
87  * covering "normal" operation, and then as many exception-testing
88  * methods as there are exceptions the method can throw. Sometimes
89  * there are multiple tests per JSR166 method when the different
90  * "normal" behaviors differ significantly. And sometimes testcases
91  * cover multiple methods when they cannot be tested in
92  * isolation.</li>
93  *
94  * <li> The documentation style for testcases is to provide as javadoc
95  * a simple sentence or two describing the property that the testcase
96  * method purports to test. The javadocs do not say anything about how
97  * the property is tested. To find out, read the code.</li>
98  *
99  * <li> These tests are "conformance tests", and do not attempt to
100  * test throughput, latency, scalability or other performance factors
101  * (see the separate "jtreg" tests for a set intended to check these
102  * for the most central aspects of functionality.) So, most tests use
103  * the smallest sensible numbers of threads, collection sizes, etc
104  * needed to check basic conformance.</li>
105  *
106  * <li>The test classes currently do not declare inclusion in
107  * any particular package to simplify things for people integrating
108  * them in TCK test suites.</li>
109  *
110  * <li> As a convenience, the {@code main} of this class (JSR166TestCase)
111  * runs all JSR166 unit tests.</li>
112  *
113  * </ul>
114  */
115 abstract class JSR166TestCase extends TestCase {
116     private static final boolean useSecurityManager =
117         Boolean.getBoolean("jsr166.useSecurityManager");
118 
119     protected static final boolean expensiveTests =
120         Boolean.getBoolean("jsr166.expensiveTests");
121 
122     /**
123      * If true, report on stdout all "slow" tests, that is, ones that
124      * take more than profileThreshold milliseconds to execute.
125      */
126     private static final boolean profileTests =
127         Boolean.getBoolean("jsr166.profileTests");
128 
129     /**
130      * The number of milliseconds that tests are permitted for
131      * execution without being reported, when profileTests is set.
132      */
133     private static final long profileThreshold =
134         Long.getLong("jsr166.profileThreshold", 100);
135 
runTest()136     protected void runTest() throws Throwable {
137         if (profileTests)
138             runTestProfiled();
139         else
140             super.runTest();
141     }
142 
runTestProfiled()143     protected void runTestProfiled() throws Throwable {
144         long t0 = System.nanoTime();
145         try {
146             super.runTest();
147         } finally {
148             long elapsedMillis =
149                 (System.nanoTime() - t0) / (1000L * 1000L);
150             if (elapsedMillis >= profileThreshold)
151                 System.out.printf("%n%s: %d%n", toString(), elapsedMillis);
152         }
153     }
154 
155 //     /**
156 //      * Runs all JSR166 unit tests using junit.textui.TestRunner
157 //      */
158 //     public static void main(String[] args) {
159 //         if (useSecurityManager) {
160 //             System.err.println("Setting a permissive security manager");
161 //             Policy.setPolicy(permissivePolicy());
162 //             System.setSecurityManager(new SecurityManager());
163 //         }
164 //         int iters = (args.length == 0) ? 1 : Integer.parseInt(args[0]);
165 
166 //         Test s = suite();
167 //         for (int i = 0; i < iters; ++i) {
168 //             junit.textui.TestRunner.run(s);
169 //             System.gc();
170 //             System.runFinalization();
171 //         }
172 //         System.exit(0);
173 //     }
174 
175 //     public static TestSuite newTestSuite(Object... suiteOrClasses) {
176 //         TestSuite suite = new TestSuite();
177 //         for (Object suiteOrClass : suiteOrClasses) {
178 //             if (suiteOrClass instanceof TestSuite)
179 //                 suite.addTest((TestSuite) suiteOrClass);
180 //             else if (suiteOrClass instanceof Class)
181 //                 suite.addTest(new TestSuite((Class<?>) suiteOrClass));
182 //             else
183 //                 throw new ClassCastException("not a test suite or class");
184 //         }
185 //         return suite;
186 //     }
187 
188 //     /**
189 //      * Collects all JSR166 unit tests as one suite.
190 //      */
191 //     public static Test suite() {
192 //         return newTestSuite(
193 //             ForkJoinPoolTest.suite(),
194 //             ForkJoinTaskTest.suite(),
195 //             RecursiveActionTest.suite(),
196 //             RecursiveTaskTest.suite(),
197 //             LinkedTransferQueueTest.suite(),
198 //             PhaserTest.suite(),
199 //             ThreadLocalRandomTest.suite(),
200 //             AbstractExecutorServiceTest.suite(),
201 //             AbstractQueueTest.suite(),
202 //             AbstractQueuedSynchronizerTest.suite(),
203 //             AbstractQueuedLongSynchronizerTest.suite(),
204 //             ArrayBlockingQueueTest.suite(),
205 //             ArrayDequeTest.suite(),
206 //             AtomicBooleanTest.suite(),
207 //             AtomicIntegerArrayTest.suite(),
208 //             AtomicIntegerFieldUpdaterTest.suite(),
209 //             AtomicIntegerTest.suite(),
210 //             AtomicLongArrayTest.suite(),
211 //             AtomicLongFieldUpdaterTest.suite(),
212 //             AtomicLongTest.suite(),
213 //             AtomicMarkableReferenceTest.suite(),
214 //             AtomicReferenceArrayTest.suite(),
215 //             AtomicReferenceFieldUpdaterTest.suite(),
216 //             AtomicReferenceTest.suite(),
217 //             AtomicStampedReferenceTest.suite(),
218 //             ConcurrentHashMapTest.suite(),
219 //             ConcurrentLinkedDequeTest.suite(),
220 //             ConcurrentLinkedQueueTest.suite(),
221 //             ConcurrentSkipListMapTest.suite(),
222 //             ConcurrentSkipListSubMapTest.suite(),
223 //             ConcurrentSkipListSetTest.suite(),
224 //             ConcurrentSkipListSubSetTest.suite(),
225 //             CopyOnWriteArrayListTest.suite(),
226 //             CopyOnWriteArraySetTest.suite(),
227 //             CountDownLatchTest.suite(),
228 //             CyclicBarrierTest.suite(),
229 //             DelayQueueTest.suite(),
230 //             EntryTest.suite(),
231 //             ExchangerTest.suite(),
232 //             ExecutorsTest.suite(),
233 //             ExecutorCompletionServiceTest.suite(),
234 //             FutureTaskTest.suite(),
235 //             LinkedBlockingDequeTest.suite(),
236 //             LinkedBlockingQueueTest.suite(),
237 //             LinkedListTest.suite(),
238 //             LockSupportTest.suite(),
239 //             PriorityBlockingQueueTest.suite(),
240 //             PriorityQueueTest.suite(),
241 //             ReentrantLockTest.suite(),
242 //             ReentrantReadWriteLockTest.suite(),
243 //             ScheduledExecutorTest.suite(),
244 //             ScheduledExecutorSubclassTest.suite(),
245 //             SemaphoreTest.suite(),
246 //             SynchronousQueueTest.suite(),
247 //             SystemTest.suite(),
248 //             ThreadLocalTest.suite(),
249 //             ThreadPoolExecutorTest.suite(),
250 //             ThreadPoolExecutorSubclassTest.suite(),
251 //             ThreadTest.suite(),
252 //             TimeUnitTest.suite(),
253 //             TreeMapTest.suite(),
254 //             TreeSetTest.suite(),
255 //             TreeSubMapTest.suite(),
256 //             TreeSubSetTest.suite());
257 //     }
258 
259     public static long SHORT_DELAY_MS;
260     public static long SMALL_DELAY_MS;
261     public static long MEDIUM_DELAY_MS;
262     public static long LONG_DELAY_MS;
263 
264     /**
265      * Returns the shortest timed delay. This could
266      * be reimplemented to use for example a Property.
267      */
getShortDelay()268     protected long getShortDelay() {
269         return 50;
270     }
271 
272     /**
273      * Sets delays as multiples of SHORT_DELAY.
274      */
setDelays()275     protected void setDelays() {
276         SHORT_DELAY_MS = getShortDelay();
277         SMALL_DELAY_MS  = SHORT_DELAY_MS * 5;
278         MEDIUM_DELAY_MS = SHORT_DELAY_MS * 10;
279         LONG_DELAY_MS   = SHORT_DELAY_MS * 200;
280     }
281 
282     /**
283      * Returns a timeout in milliseconds to be used in tests that
284      * verify that operations block or time out.
285      */
timeoutMillis()286     long timeoutMillis() {
287         return SHORT_DELAY_MS / 4;
288     }
289 
290     /**
291      * Returns a new Date instance representing a time delayMillis
292      * milliseconds in the future.
293      */
delayedDate(long delayMillis)294     Date delayedDate(long delayMillis) {
295         return new Date(System.currentTimeMillis() + delayMillis);
296     }
297 
298     /**
299      * The first exception encountered if any threadAssertXXX method fails.
300      */
301     private final AtomicReference<Throwable> threadFailure
302         = new AtomicReference<Throwable>(null);
303 
304     /**
305      * Records an exception so that it can be rethrown later in the test
306      * harness thread, triggering a test case failure.  Only the first
307      * failure is recorded; subsequent calls to this method from within
308      * the same test have no effect.
309      */
threadRecordFailure(Throwable t)310     public void threadRecordFailure(Throwable t) {
311         threadFailure.compareAndSet(null, t);
312     }
313 
setUp()314     public void setUp() {
315         setDelays();
316     }
317 
318     /**
319      * Extra checks that get done for all test cases.
320      *
321      * Triggers test case failure if any thread assertions have failed,
322      * by rethrowing, in the test harness thread, any exception recorded
323      * earlier by threadRecordFailure.
324      *
325      * Triggers test case failure if interrupt status is set in the main thread.
326      */
tearDown()327     public void tearDown() throws Exception {
328         Throwable t = threadFailure.getAndSet(null);
329         if (t != null) {
330             if (t instanceof Error)
331                 throw (Error) t;
332             else if (t instanceof RuntimeException)
333                 throw (RuntimeException) t;
334             else if (t instanceof Exception)
335                 throw (Exception) t;
336             else {
337                 AssertionFailedError afe =
338                     new AssertionFailedError(t.toString());
339                 afe.initCause(t);
340                 throw afe;
341             }
342         }
343 
344         if (Thread.interrupted())
345             throw new AssertionFailedError("interrupt status set in main thread");
346     }
347 
348     /**
349      * Just like fail(reason), but additionally recording (using
350      * threadRecordFailure) any AssertionFailedError thrown, so that
351      * the current testcase will fail.
352      */
threadFail(String reason)353     public void threadFail(String reason) {
354         try {
355             fail(reason);
356         } catch (AssertionFailedError t) {
357             threadRecordFailure(t);
358             fail(reason);
359         }
360     }
361 
362     /**
363      * Just like assertTrue(b), but additionally recording (using
364      * threadRecordFailure) any AssertionFailedError thrown, so that
365      * the current testcase will fail.
366      */
threadAssertTrue(boolean b)367     public void threadAssertTrue(boolean b) {
368         try {
369             assertTrue(b);
370         } catch (AssertionFailedError t) {
371             threadRecordFailure(t);
372             throw t;
373         }
374     }
375 
376     /**
377      * Just like assertFalse(b), but additionally recording (using
378      * threadRecordFailure) any AssertionFailedError thrown, so that
379      * the current testcase will fail.
380      */
threadAssertFalse(boolean b)381     public void threadAssertFalse(boolean b) {
382         try {
383             assertFalse(b);
384         } catch (AssertionFailedError t) {
385             threadRecordFailure(t);
386             throw t;
387         }
388     }
389 
390     /**
391      * Just like assertNull(x), but additionally recording (using
392      * threadRecordFailure) any AssertionFailedError thrown, so that
393      * the current testcase will fail.
394      */
threadAssertNull(Object x)395     public void threadAssertNull(Object x) {
396         try {
397             assertNull(x);
398         } catch (AssertionFailedError t) {
399             threadRecordFailure(t);
400             throw t;
401         }
402     }
403 
404     /**
405      * Just like assertEquals(x, y), but additionally recording (using
406      * threadRecordFailure) any AssertionFailedError thrown, so that
407      * the current testcase will fail.
408      */
threadAssertEquals(long x, long y)409     public void threadAssertEquals(long x, long y) {
410         try {
411             assertEquals(x, y);
412         } catch (AssertionFailedError t) {
413             threadRecordFailure(t);
414             throw t;
415         }
416     }
417 
418     /**
419      * Just like assertEquals(x, y), but additionally recording (using
420      * threadRecordFailure) any AssertionFailedError thrown, so that
421      * the current testcase will fail.
422      */
threadAssertEquals(Object x, Object y)423     public void threadAssertEquals(Object x, Object y) {
424         try {
425             assertEquals(x, y);
426         } catch (AssertionFailedError t) {
427             threadRecordFailure(t);
428             throw t;
429         } catch (Throwable t) {
430             threadUnexpectedException(t);
431         }
432     }
433 
434     /**
435      * Just like assertSame(x, y), but additionally recording (using
436      * threadRecordFailure) any AssertionFailedError thrown, so that
437      * the current testcase will fail.
438      */
threadAssertSame(Object x, Object y)439     public void threadAssertSame(Object x, Object y) {
440         try {
441             assertSame(x, y);
442         } catch (AssertionFailedError t) {
443             threadRecordFailure(t);
444             throw t;
445         }
446     }
447 
448     /**
449      * Calls threadFail with message "should throw exception".
450      */
threadShouldThrow()451     public void threadShouldThrow() {
452         threadFail("should throw exception");
453     }
454 
455     /**
456      * Calls threadFail with message "should throw" + exceptionName.
457      */
threadShouldThrow(String exceptionName)458     public void threadShouldThrow(String exceptionName) {
459         threadFail("should throw " + exceptionName);
460     }
461 
462     /**
463      * Records the given exception using {@link #threadRecordFailure},
464      * then rethrows the exception, wrapping it in an
465      * AssertionFailedError if necessary.
466      */
threadUnexpectedException(Throwable t)467     public void threadUnexpectedException(Throwable t) {
468         threadRecordFailure(t);
469         t.printStackTrace();
470         if (t instanceof RuntimeException)
471             throw (RuntimeException) t;
472         else if (t instanceof Error)
473             throw (Error) t;
474         else {
475             AssertionFailedError afe =
476                 new AssertionFailedError("unexpected exception: " + t);
477             afe.initCause(t);
478             throw afe;
479         }
480     }
481 
482     /**
483      * Delays, via Thread.sleep, for the given millisecond delay, but
484      * if the sleep is shorter than specified, may re-sleep or yield
485      * until time elapses.
486      */
delay(long millis)487     static void delay(long millis) throws InterruptedException {
488         long startTime = System.nanoTime();
489         long ns = millis * 1000 * 1000;
490         for (;;) {
491             if (millis > 0L)
492                 Thread.sleep(millis);
493             else // too short to sleep
494                 Thread.yield();
495             long d = ns - (System.nanoTime() - startTime);
496             if (d > 0L)
497                 millis = d / (1000 * 1000);
498             else
499                 break;
500         }
501     }
502 
503     /**
504      * Waits out termination of a thread pool or fails doing so.
505      */
joinPool(ExecutorService exec)506     void joinPool(ExecutorService exec) {
507         try {
508             exec.shutdown();
509             assertTrue("ExecutorService did not terminate in a timely manner",
510                        exec.awaitTermination(2 * LONG_DELAY_MS, MILLISECONDS));
511         } catch (SecurityException ok) {
512             // Allowed in case test doesn't have privs
513         } catch (InterruptedException ie) {
514             fail("Unexpected InterruptedException");
515         }
516     }
517 
518     /**
519      * Checks that thread does not terminate within the default
520      * millisecond delay of {@code timeoutMillis()}.
521      */
assertThreadStaysAlive(Thread thread)522     void assertThreadStaysAlive(Thread thread) {
523         assertThreadStaysAlive(thread, timeoutMillis());
524     }
525 
526     /**
527      * Checks that thread does not terminate within the given millisecond delay.
528      */
assertThreadStaysAlive(Thread thread, long millis)529     void assertThreadStaysAlive(Thread thread, long millis) {
530         try {
531             // No need to optimize the failing case via Thread.join.
532             delay(millis);
533             assertTrue(thread.isAlive());
534         } catch (InterruptedException ie) {
535             fail("Unexpected InterruptedException");
536         }
537     }
538 
539     /**
540      * Checks that the threads do not terminate within the default
541      * millisecond delay of {@code timeoutMillis()}.
542      */
assertThreadsStayAlive(Thread... threads)543     void assertThreadsStayAlive(Thread... threads) {
544         assertThreadsStayAlive(timeoutMillis(), threads);
545     }
546 
547     /**
548      * Checks that the threads do not terminate within the given millisecond delay.
549      */
assertThreadsStayAlive(long millis, Thread... threads)550     void assertThreadsStayAlive(long millis, Thread... threads) {
551         try {
552             // No need to optimize the failing case via Thread.join.
553             delay(millis);
554             for (Thread thread : threads)
555                 assertTrue(thread.isAlive());
556         } catch (InterruptedException ie) {
557             fail("Unexpected InterruptedException");
558         }
559     }
560 
561     /**
562      * Checks that future.get times out, with the default timeout of
563      * {@code timeoutMillis()}.
564      */
assertFutureTimesOut(Future future)565     void assertFutureTimesOut(Future future) {
566         assertFutureTimesOut(future, timeoutMillis());
567     }
568 
569     /**
570      * Checks that future.get times out, with the given millisecond timeout.
571      */
assertFutureTimesOut(Future future, long timeoutMillis)572     void assertFutureTimesOut(Future future, long timeoutMillis) {
573         long startTime = System.nanoTime();
574         try {
575             future.get(timeoutMillis, MILLISECONDS);
576             shouldThrow();
577         } catch (TimeoutException success) {
578         } catch (Exception e) {
579             threadUnexpectedException(e);
580         } finally { future.cancel(true); }
581         assertTrue(millisElapsedSince(startTime) >= timeoutMillis);
582     }
583 
584     /**
585      * Fails with message "should throw exception".
586      */
shouldThrow()587     public void shouldThrow() {
588         fail("Should throw exception");
589     }
590 
591     /**
592      * Fails with message "should throw " + exceptionName.
593      */
shouldThrow(String exceptionName)594     public void shouldThrow(String exceptionName) {
595         fail("Should throw " + exceptionName);
596     }
597 
598     /**
599      * The number of elements to place in collections, arrays, etc.
600      */
601     public static final int SIZE = 20;
602 
603     // Some convenient Integer constants
604 
605     public static final Integer zero  = new Integer(0);
606     public static final Integer one   = new Integer(1);
607     public static final Integer two   = new Integer(2);
608     public static final Integer three = new Integer(3);
609     public static final Integer four  = new Integer(4);
610     public static final Integer five  = new Integer(5);
611     public static final Integer six   = new Integer(6);
612     public static final Integer seven = new Integer(7);
613     public static final Integer eight = new Integer(8);
614     public static final Integer nine  = new Integer(9);
615     public static final Integer m1  = new Integer(-1);
616     public static final Integer m2  = new Integer(-2);
617     public static final Integer m3  = new Integer(-3);
618     public static final Integer m4  = new Integer(-4);
619     public static final Integer m5  = new Integer(-5);
620     public static final Integer m6  = new Integer(-6);
621     public static final Integer m10 = new Integer(-10);
622 
623     /**
624      * Runs Runnable r with a security policy that permits precisely
625      * the specified permissions.  If there is no current security
626      * manager, the runnable is run twice, both with and without a
627      * security manager.  We require that any security manager permit
628      * getPolicy/setPolicy.
629      */
runWithPermissions(Runnable r, Permission... permissions)630     public void runWithPermissions(Runnable r, Permission... permissions) {
631         SecurityManager sm = System.getSecurityManager();
632         if (sm == null) {
633             r.run();
634             Policy savedPolicy = Policy.getPolicy();
635             try {
636                 Policy.setPolicy(permissivePolicy());
637                 System.setSecurityManager(new SecurityManager());
638                 runWithPermissions(r, permissions);
639             } finally {
640                 System.setSecurityManager(null);
641                 Policy.setPolicy(savedPolicy);
642             }
643         } else {
644             Policy savedPolicy = Policy.getPolicy();
645             AdjustablePolicy policy = new AdjustablePolicy(permissions);
646             Policy.setPolicy(policy);
647 
648             try {
649                 r.run();
650             } finally {
651                 policy.addPermission(new SecurityPermission("setPolicy"));
652                 Policy.setPolicy(savedPolicy);
653             }
654         }
655     }
656 
657     /**
658      * Runs a runnable without any permissions.
659      */
runWithoutPermissions(Runnable r)660     public void runWithoutPermissions(Runnable r) {
661         runWithPermissions(r);
662     }
663 
664     /**
665      * A security policy where new permissions can be dynamically added
666      * or all cleared.
667      */
668     public static class AdjustablePolicy extends java.security.Policy {
669         Permissions perms = new Permissions();
AdjustablePolicy(Permission... permissions)670         AdjustablePolicy(Permission... permissions) {
671             for (Permission permission : permissions)
672                 perms.add(permission);
673         }
addPermission(Permission perm)674         void addPermission(Permission perm) { perms.add(perm); }
clearPermissions()675         void clearPermissions() { perms = new Permissions(); }
getPermissions(CodeSource cs)676         public PermissionCollection getPermissions(CodeSource cs) {
677             return perms;
678         }
getPermissions(ProtectionDomain pd)679         public PermissionCollection getPermissions(ProtectionDomain pd) {
680             return perms;
681         }
implies(ProtectionDomain pd, Permission p)682         public boolean implies(ProtectionDomain pd, Permission p) {
683             return perms.implies(p);
684         }
refresh()685         public void refresh() {}
686     }
687 
688     /**
689      * Returns a policy containing all the permissions we ever need.
690      */
permissivePolicy()691     public static Policy permissivePolicy() {
692         return new AdjustablePolicy
693             // Permissions j.u.c. needs directly
694             (new RuntimePermission("modifyThread"),
695              new RuntimePermission("getClassLoader"),
696              new RuntimePermission("setContextClassLoader"),
697              // Permissions needed to change permissions!
698              new SecurityPermission("getPolicy"),
699              new SecurityPermission("setPolicy"),
700              new RuntimePermission("setSecurityManager"),
701              // Permissions needed by the junit test harness
702              new RuntimePermission("accessDeclaredMembers"),
703              new PropertyPermission("*", "read"),
704              new java.io.FilePermission("<<ALL FILES>>", "read"));
705     }
706 
707     /**
708      * Sleeps until the given time has elapsed.
709      * Throws AssertionFailedError if interrupted.
710      */
sleep(long millis)711     void sleep(long millis) {
712         try {
713             delay(millis);
714         } catch (InterruptedException ie) {
715             AssertionFailedError afe =
716                 new AssertionFailedError("Unexpected InterruptedException");
717             afe.initCause(ie);
718             throw afe;
719         }
720     }
721 
722     /**
723      * Spin-waits up to the specified number of milliseconds for the given
724      * thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING.
725      */
waitForThreadToEnterWaitState(Thread thread, long timeoutMillis)726     void waitForThreadToEnterWaitState(Thread thread, long timeoutMillis) {
727         long startTime = System.nanoTime();
728         for (;;) {
729             Thread.State s = thread.getState();
730             if (s == Thread.State.BLOCKED ||
731                 s == Thread.State.WAITING ||
732                 s == Thread.State.TIMED_WAITING)
733                 return;
734             else if (s == Thread.State.TERMINATED)
735                 fail("Unexpected thread termination");
736             else if (millisElapsedSince(startTime) > timeoutMillis) {
737                 threadAssertTrue(thread.isAlive());
738                 return;
739             }
740             Thread.yield();
741         }
742     }
743 
744     /**
745      * Waits up to LONG_DELAY_MS for the given thread to enter a wait
746      * state: BLOCKED, WAITING, or TIMED_WAITING.
747      */
waitForThreadToEnterWaitState(Thread thread)748     void waitForThreadToEnterWaitState(Thread thread) {
749         waitForThreadToEnterWaitState(thread, LONG_DELAY_MS);
750     }
751 
752     /**
753      * Returns the number of milliseconds since time given by
754      * startNanoTime, which must have been previously returned from a
755      * call to {@link System.nanoTime()}.
756      */
millisElapsedSince(long startNanoTime)757     long millisElapsedSince(long startNanoTime) {
758         return NANOSECONDS.toMillis(System.nanoTime() - startNanoTime);
759     }
760 
761     /**
762      * Returns a new started daemon Thread running the given runnable.
763      */
newStartedThread(Runnable runnable)764     Thread newStartedThread(Runnable runnable) {
765         Thread t = new Thread(runnable);
766         t.setDaemon(true);
767         t.start();
768         return t;
769     }
770 
771     /**
772      * Waits for the specified time (in milliseconds) for the thread
773      * to terminate (using {@link Thread#join(long)}), else interrupts
774      * the thread (in the hope that it may terminate later) and fails.
775      */
awaitTermination(Thread t, long timeoutMillis)776     void awaitTermination(Thread t, long timeoutMillis) {
777         try {
778             t.join(timeoutMillis);
779         } catch (InterruptedException ie) {
780             threadUnexpectedException(ie);
781         } finally {
782             if (t.getState() != Thread.State.TERMINATED) {
783                 t.interrupt();
784                 fail("Test timed out");
785             }
786         }
787     }
788 
789     /**
790      * Waits for LONG_DELAY_MS milliseconds for the thread to
791      * terminate (using {@link Thread#join(long)}), else interrupts
792      * the thread (in the hope that it may terminate later) and fails.
793      */
awaitTermination(Thread t)794     void awaitTermination(Thread t) {
795         awaitTermination(t, LONG_DELAY_MS);
796     }
797 
798     // Some convenient Runnable classes
799 
800     public abstract class CheckedRunnable implements Runnable {
realRun()801         protected abstract void realRun() throws Throwable;
802 
run()803         public final void run() {
804             try {
805                 realRun();
806             } catch (Throwable t) {
807                 threadUnexpectedException(t);
808             }
809         }
810     }
811 
812     public abstract class RunnableShouldThrow implements Runnable {
realRun()813         protected abstract void realRun() throws Throwable;
814 
815         final Class<?> exceptionClass;
816 
RunnableShouldThrow(Class<T> exceptionClass)817         <T extends Throwable> RunnableShouldThrow(Class<T> exceptionClass) {
818             this.exceptionClass = exceptionClass;
819         }
820 
run()821         public final void run() {
822             try {
823                 realRun();
824                 threadShouldThrow(exceptionClass.getSimpleName());
825             } catch (Throwable t) {
826                 if (! exceptionClass.isInstance(t))
827                     threadUnexpectedException(t);
828             }
829         }
830     }
831 
832     public abstract class ThreadShouldThrow extends Thread {
realRun()833         protected abstract void realRun() throws Throwable;
834 
835         final Class<?> exceptionClass;
836 
ThreadShouldThrow(Class<T> exceptionClass)837         <T extends Throwable> ThreadShouldThrow(Class<T> exceptionClass) {
838             this.exceptionClass = exceptionClass;
839         }
840 
run()841         public final void run() {
842             try {
843                 realRun();
844                 threadShouldThrow(exceptionClass.getSimpleName());
845             } catch (Throwable t) {
846                 if (! exceptionClass.isInstance(t))
847                     threadUnexpectedException(t);
848             }
849         }
850     }
851 
852     public abstract class CheckedInterruptedRunnable implements Runnable {
realRun()853         protected abstract void realRun() throws Throwable;
854 
run()855         public final void run() {
856             try {
857                 realRun();
858                 threadShouldThrow("InterruptedException");
859             } catch (InterruptedException success) {
860                 threadAssertFalse(Thread.interrupted());
861             } catch (Throwable t) {
862                 threadUnexpectedException(t);
863             }
864         }
865     }
866 
867     public abstract class CheckedCallable<T> implements Callable<T> {
realCall()868         protected abstract T realCall() throws Throwable;
869 
call()870         public final T call() {
871             try {
872                 return realCall();
873             } catch (Throwable t) {
874                 threadUnexpectedException(t);
875                 return null;
876             }
877         }
878     }
879 
880     public abstract class CheckedInterruptedCallable<T>
881         implements Callable<T> {
realCall()882         protected abstract T realCall() throws Throwable;
883 
call()884         public final T call() {
885             try {
886                 T result = realCall();
887                 threadShouldThrow("InterruptedException");
888                 return result;
889             } catch (InterruptedException success) {
890                 threadAssertFalse(Thread.interrupted());
891             } catch (Throwable t) {
892                 threadUnexpectedException(t);
893             }
894             return null;
895         }
896     }
897 
898     public static class NoOpRunnable implements Runnable {
run()899         public void run() {}
900     }
901 
902     public static class NoOpCallable implements Callable {
call()903         public Object call() { return Boolean.TRUE; }
904     }
905 
906     public static final String TEST_STRING = "a test string";
907 
908     public static class StringTask implements Callable<String> {
call()909         public String call() { return TEST_STRING; }
910     }
911 
latchAwaitingStringTask(final CountDownLatch latch)912     public Callable<String> latchAwaitingStringTask(final CountDownLatch latch) {
913         return new CheckedCallable<String>() {
914             protected String realCall() {
915                 try {
916                     latch.await();
917                 } catch (InterruptedException quittingTime) {}
918                 return TEST_STRING;
919             }};
920     }
921 
922     public Runnable awaiter(final CountDownLatch latch) {
923         return new CheckedRunnable() {
924             public void realRun() throws InterruptedException {
925                 await(latch);
926             }};
927     }
928 
929     public void await(CountDownLatch latch) {
930         try {
931             assertTrue(latch.await(LONG_DELAY_MS, MILLISECONDS));
932         } catch (Throwable t) {
933             threadUnexpectedException(t);
934         }
935     }
936 
937     public void await(Semaphore semaphore) {
938         try {
939             assertTrue(semaphore.tryAcquire(LONG_DELAY_MS, MILLISECONDS));
940         } catch (Throwable t) {
941             threadUnexpectedException(t);
942         }
943     }
944 
945 //     /**
946 //      * Spin-waits up to LONG_DELAY_MS until flag becomes true.
947 //      */
948 //     public void await(AtomicBoolean flag) {
949 //         await(flag, LONG_DELAY_MS);
950 //     }
951 
952 //     /**
953 //      * Spin-waits up to the specified timeout until flag becomes true.
954 //      */
955 //     public void await(AtomicBoolean flag, long timeoutMillis) {
956 //         long startTime = System.nanoTime();
957 //         while (!flag.get()) {
958 //             if (millisElapsedSince(startTime) > timeoutMillis)
959 //                 throw new AssertionFailedError("timed out");
960 //             Thread.yield();
961 //         }
962 //     }
963 
964     public static class NPETask implements Callable<String> {
965         public String call() { throw new NullPointerException(); }
966     }
967 
968     public static class CallableOne implements Callable<Integer> {
969         public Integer call() { return one; }
970     }
971 
972     public class ShortRunnable extends CheckedRunnable {
973         protected void realRun() throws Throwable {
974             delay(SHORT_DELAY_MS);
975         }
976     }
977 
978     public class ShortInterruptedRunnable extends CheckedInterruptedRunnable {
979         protected void realRun() throws InterruptedException {
980             delay(SHORT_DELAY_MS);
981         }
982     }
983 
984     public class SmallRunnable extends CheckedRunnable {
985         protected void realRun() throws Throwable {
986             delay(SMALL_DELAY_MS);
987         }
988     }
989 
990     public class SmallPossiblyInterruptedRunnable extends CheckedRunnable {
991         protected void realRun() {
992             try {
993                 delay(SMALL_DELAY_MS);
994             } catch (InterruptedException ok) {}
995         }
996     }
997 
998     public class SmallCallable extends CheckedCallable {
999         protected Object realCall() throws InterruptedException {
1000             delay(SMALL_DELAY_MS);
1001             return Boolean.TRUE;
1002         }
1003     }
1004 
1005     public class MediumRunnable extends CheckedRunnable {
1006         protected void realRun() throws Throwable {
1007             delay(MEDIUM_DELAY_MS);
1008         }
1009     }
1010 
1011     public class MediumInterruptedRunnable extends CheckedInterruptedRunnable {
1012         protected void realRun() throws InterruptedException {
1013             delay(MEDIUM_DELAY_MS);
1014         }
1015     }
1016 
1017     public Runnable possiblyInterruptedRunnable(final long timeoutMillis) {
1018         return new CheckedRunnable() {
1019             protected void realRun() {
1020                 try {
1021                     delay(timeoutMillis);
1022                 } catch (InterruptedException ok) {}
1023             }};
1024     }
1025 
1026     public class MediumPossiblyInterruptedRunnable extends CheckedRunnable {
1027         protected void realRun() {
1028             try {
1029                 delay(MEDIUM_DELAY_MS);
1030             } catch (InterruptedException ok) {}
1031         }
1032     }
1033 
1034     public class LongPossiblyInterruptedRunnable extends CheckedRunnable {
1035         protected void realRun() {
1036             try {
1037                 delay(LONG_DELAY_MS);
1038             } catch (InterruptedException ok) {}
1039         }
1040     }
1041 
1042     /**
1043      * For use as ThreadFactory in constructors
1044      */
1045     public static class SimpleThreadFactory implements ThreadFactory {
1046         public Thread newThread(Runnable r) {
1047             return new Thread(r);
1048         }
1049     }
1050 
1051     public interface TrackedRunnable extends Runnable {
1052         boolean isDone();
1053     }
1054 
1055     public static TrackedRunnable trackedRunnable(final long timeoutMillis) {
1056         return new TrackedRunnable() {
1057                 private volatile boolean done = false;
1058                 public boolean isDone() { return done; }
1059                 public void run() {
1060                     try {
1061                         delay(timeoutMillis);
1062                         done = true;
1063                     } catch (InterruptedException ok) {}
1064                 }
1065             };
1066     }
1067 
1068     public static class TrackedShortRunnable implements Runnable {
1069         public volatile boolean done = false;
1070         public void run() {
1071             try {
1072                 delay(SHORT_DELAY_MS);
1073                 done = true;
1074             } catch (InterruptedException ok) {}
1075         }
1076     }
1077 
1078     public static class TrackedSmallRunnable implements Runnable {
1079         public volatile boolean done = false;
1080         public void run() {
1081             try {
1082                 delay(SMALL_DELAY_MS);
1083                 done = true;
1084             } catch (InterruptedException ok) {}
1085         }
1086     }
1087 
1088     public static class TrackedMediumRunnable implements Runnable {
1089         public volatile boolean done = false;
1090         public void run() {
1091             try {
1092                 delay(MEDIUM_DELAY_MS);
1093                 done = true;
1094             } catch (InterruptedException ok) {}
1095         }
1096     }
1097 
1098     public static class TrackedLongRunnable implements Runnable {
1099         public volatile boolean done = false;
1100         public void run() {
1101             try {
1102                 delay(LONG_DELAY_MS);
1103                 done = true;
1104             } catch (InterruptedException ok) {}
1105         }
1106     }
1107 
1108     public static class TrackedNoOpRunnable implements Runnable {
1109         public volatile boolean done = false;
1110         public void run() {
1111             done = true;
1112         }
1113     }
1114 
1115     public static class TrackedCallable implements Callable {
1116         public volatile boolean done = false;
1117         public Object call() {
1118             try {
1119                 delay(SMALL_DELAY_MS);
1120                 done = true;
1121             } catch (InterruptedException ok) {}
1122             return Boolean.TRUE;
1123         }
1124     }
1125 
1126 //     /**
1127 //      * Analog of CheckedRunnable for RecursiveAction
1128 //      */
1129 //     public abstract class CheckedRecursiveAction extends RecursiveAction {
1130 //         protected abstract void realCompute() throws Throwable;
1131 
1132 //         public final void compute() {
1133 //             try {
1134 //                 realCompute();
1135 //             } catch (Throwable t) {
1136 //                 threadUnexpectedException(t);
1137 //             }
1138 //         }
1139 //     }
1140 
1141 //     /**
1142 //      * Analog of CheckedCallable for RecursiveTask
1143 //      */
1144 //     public abstract class CheckedRecursiveTask<T> extends RecursiveTask<T> {
1145 //         protected abstract T realCompute() throws Throwable;
1146 
1147 //         public final T compute() {
1148 //             try {
1149 //                 return realCompute();
1150 //             } catch (Throwable t) {
1151 //                 threadUnexpectedException(t);
1152 //                 return null;
1153 //             }
1154 //         }
1155 //     }
1156 
1157     /**
1158      * For use as RejectedExecutionHandler in constructors
1159      */
1160     public static class NoOpREHandler implements RejectedExecutionHandler {
1161         public void rejectedExecution(Runnable r,
1162                                       ThreadPoolExecutor executor) {}
1163     }
1164 
1165     /**
1166      * A CyclicBarrier that uses timed await and fails with
1167      * AssertionFailedErrors instead of throwing checked exceptions.
1168      */
1169     public class CheckedBarrier extends CyclicBarrier {
1170         public CheckedBarrier(int parties) { super(parties); }
1171 
1172         public int await() {
1173             try {
1174                 return super.await(2 * LONG_DELAY_MS, MILLISECONDS);
1175             } catch (TimeoutException e) {
1176                 throw new AssertionFailedError("timed out");
1177             } catch (Exception e) {
1178                 AssertionFailedError afe =
1179                     new AssertionFailedError("Unexpected exception: " + e);
1180                 afe.initCause(e);
1181                 throw afe;
1182             }
1183         }
1184     }
1185 
1186     void checkEmpty(BlockingQueue q) {
1187         try {
1188             assertTrue(q.isEmpty());
1189             assertEquals(0, q.size());
1190             assertNull(q.peek());
1191             assertNull(q.poll());
1192             assertNull(q.poll(0, MILLISECONDS));
1193             assertEquals(q.toString(), "[]");
1194             assertTrue(Arrays.equals(q.toArray(), new Object[0]));
1195             assertFalse(q.iterator().hasNext());
1196             try {
1197                 q.element();
1198                 shouldThrow();
1199             } catch (NoSuchElementException success) {}
1200             try {
1201                 q.iterator().next();
1202                 shouldThrow();
1203             } catch (NoSuchElementException success) {}
1204             try {
1205                 q.remove();
1206                 shouldThrow();
1207             } catch (NoSuchElementException success) {}
1208         } catch (InterruptedException ie) {
1209             threadUnexpectedException(ie);
1210         }
1211     }
1212 
1213     @SuppressWarnings("unchecked")
1214     <T> T serialClone(T o) {
1215         try {
1216             ByteArrayOutputStream bos = new ByteArrayOutputStream();
1217             ObjectOutputStream oos = new ObjectOutputStream(bos);
1218             oos.writeObject(o);
1219             oos.flush();
1220             oos.close();
1221             ObjectInputStream ois = new ObjectInputStream
1222                 (new ByteArrayInputStream(bos.toByteArray()));
1223             T clone = (T) ois.readObject();
1224             assertSame(o.getClass(), clone.getClass());
1225             return clone;
1226         } catch (Throwable t) {
1227             threadUnexpectedException(t);
1228             return null;
1229         }
1230     }
1231 }
1232