66 |
|
import java.util.Collection; |
67 |
|
import java.util.Collections; |
68 |
|
import java.util.Date; |
69 |
+ |
import java.util.Deque; |
70 |
|
import java.util.Enumeration; |
71 |
+ |
import java.util.HashSet; |
72 |
|
import java.util.Iterator; |
73 |
|
import java.util.List; |
74 |
|
import java.util.NoSuchElementException; |
75 |
|
import java.util.PropertyPermission; |
76 |
+ |
import java.util.Set; |
77 |
|
import java.util.concurrent.BlockingQueue; |
78 |
|
import java.util.concurrent.Callable; |
79 |
|
import java.util.concurrent.CountDownLatch; |
80 |
|
import java.util.concurrent.CyclicBarrier; |
81 |
|
import java.util.concurrent.ExecutionException; |
82 |
+ |
import java.util.concurrent.Executor; |
83 |
|
import java.util.concurrent.Executors; |
84 |
|
import java.util.concurrent.ExecutorService; |
85 |
|
import java.util.concurrent.ForkJoinPool; |
86 |
|
import java.util.concurrent.Future; |
87 |
+ |
import java.util.concurrent.FutureTask; |
88 |
|
import java.util.concurrent.RecursiveAction; |
89 |
|
import java.util.concurrent.RecursiveTask; |
90 |
+ |
import java.util.concurrent.RejectedExecutionException; |
91 |
|
import java.util.concurrent.RejectedExecutionHandler; |
92 |
|
import java.util.concurrent.Semaphore; |
93 |
+ |
import java.util.concurrent.ScheduledExecutorService; |
94 |
+ |
import java.util.concurrent.ScheduledFuture; |
95 |
|
import java.util.concurrent.SynchronousQueue; |
96 |
|
import java.util.concurrent.ThreadFactory; |
97 |
|
import java.util.concurrent.ThreadLocalRandom; |
98 |
|
import java.util.concurrent.ThreadPoolExecutor; |
99 |
+ |
import java.util.concurrent.TimeUnit; |
100 |
|
import java.util.concurrent.TimeoutException; |
101 |
|
import java.util.concurrent.atomic.AtomicBoolean; |
102 |
|
import java.util.concurrent.atomic.AtomicReference; |
103 |
|
import java.util.regex.Pattern; |
104 |
|
|
96 |
– |
import junit.framework.AssertionFailedError; |
105 |
|
import junit.framework.Test; |
106 |
|
import junit.framework.TestCase; |
107 |
|
import junit.framework.TestResult; |
117 |
|
* |
118 |
|
* <ol> |
119 |
|
* |
120 |
< |
* <li>All assertions in code running in generated threads must use |
121 |
< |
* the forms {@link #threadFail}, {@link #threadAssertTrue}, {@link |
122 |
< |
* #threadAssertEquals}, or {@link #threadAssertNull}, (not |
123 |
< |
* {@code fail}, {@code assertTrue}, etc.) It is OK (but not |
124 |
< |
* particularly recommended) for other code to use these forms too. |
125 |
< |
* Only the most typically used JUnit assertion methods are defined |
126 |
< |
* this way, but enough to live with. |
120 |
> |
* <li>All code not running in the main test thread (manually spawned threads |
121 |
> |
* or the common fork join pool) must be checked for failure (and completion!). |
122 |
> |
* Mechanisms that can be used to ensure this are: |
123 |
> |
* <ol> |
124 |
> |
* <li>Signalling via a synchronizer like AtomicInteger or CountDownLatch |
125 |
> |
* that the task completed normally, which is checked before returning from |
126 |
> |
* the test method in the main thread. |
127 |
> |
* <li>Using the forms {@link #threadFail}, {@link #threadAssertTrue}, |
128 |
> |
* or {@link #threadAssertNull}, (not {@code fail}, {@code assertTrue}, etc.) |
129 |
> |
* Only the most typically used JUnit assertion methods are defined |
130 |
> |
* this way, but enough to live with. |
131 |
> |
* <li>Recording failure explicitly using {@link #threadUnexpectedException} |
132 |
> |
* or {@link #threadRecordFailure}. |
133 |
> |
* <li>Using a wrapper like CheckedRunnable that uses one the mechanisms above. |
134 |
> |
* </ol> |
135 |
|
* |
136 |
|
* <li>If you override {@link #setUp} or {@link #tearDown}, make sure |
137 |
|
* to invoke {@code super.setUp} and {@code super.tearDown} within |
429 |
|
for (String testClassName : testClassNames) { |
430 |
|
try { |
431 |
|
Class<?> testClass = Class.forName(testClassName); |
432 |
< |
Method m = testClass.getDeclaredMethod("suite", |
417 |
< |
new Class<?>[0]); |
432 |
> |
Method m = testClass.getDeclaredMethod("suite"); |
433 |
|
suite.addTest(newTestSuite((Test)m.invoke(null))); |
434 |
< |
} catch (Exception e) { |
435 |
< |
throw new Error("Missing test class", e); |
434 |
> |
} catch (ReflectiveOperationException e) { |
435 |
> |
throw new AssertionError("Missing test class", e); |
436 |
|
} |
437 |
|
} |
438 |
|
} |
454 |
|
} |
455 |
|
} |
456 |
|
|
457 |
< |
public static boolean atLeastJava6() { return JAVA_CLASS_VERSION >= 50.0; } |
458 |
< |
public static boolean atLeastJava7() { return JAVA_CLASS_VERSION >= 51.0; } |
459 |
< |
public static boolean atLeastJava8() { return JAVA_CLASS_VERSION >= 52.0; } |
460 |
< |
public static boolean atLeastJava9() { |
461 |
< |
return JAVA_CLASS_VERSION >= 53.0 |
462 |
< |
// As of 2015-09, java9 still uses 52.0 class file version |
448 |
< |
|| JAVA_SPECIFICATION_VERSION.matches("^(1\\.)?(9|[0-9][0-9])$"); |
449 |
< |
} |
450 |
< |
public static boolean atLeastJava10() { |
451 |
< |
return JAVA_CLASS_VERSION >= 54.0 |
452 |
< |
|| JAVA_SPECIFICATION_VERSION.matches("^(1\\.)?[0-9][0-9]$"); |
453 |
< |
} |
457 |
> |
public static boolean atLeastJava6() { return JAVA_CLASS_VERSION >= 50.0; } |
458 |
> |
public static boolean atLeastJava7() { return JAVA_CLASS_VERSION >= 51.0; } |
459 |
> |
public static boolean atLeastJava8() { return JAVA_CLASS_VERSION >= 52.0; } |
460 |
> |
public static boolean atLeastJava9() { return JAVA_CLASS_VERSION >= 53.0; } |
461 |
> |
public static boolean atLeastJava10() { return JAVA_CLASS_VERSION >= 54.0; } |
462 |
> |
public static boolean atLeastJava11() { return JAVA_CLASS_VERSION >= 55.0; } |
463 |
|
|
464 |
|
/** |
465 |
|
* Collects all JSR166 unit tests as one suite. |
547 |
|
"DoubleAdderTest", |
548 |
|
"ForkJoinPool8Test", |
549 |
|
"ForkJoinTask8Test", |
550 |
+ |
"HashMapTest", |
551 |
|
"LinkedBlockingDeque8Test", |
552 |
|
"LinkedBlockingQueue8Test", |
553 |
|
"LongAccumulatorTest", |
610 |
|
for (String methodName : testMethodNames(testClass)) |
611 |
|
suite.addTest((Test) c.newInstance(data, methodName)); |
612 |
|
return suite; |
613 |
< |
} catch (Exception e) { |
614 |
< |
throw new Error(e); |
613 |
> |
} catch (ReflectiveOperationException e) { |
614 |
> |
throw new AssertionError(e); |
615 |
|
} |
616 |
|
} |
617 |
|
|
627 |
|
if (atLeastJava8()) { |
628 |
|
String name = testClass.getName(); |
629 |
|
String name8 = name.replaceAll("Test$", "8Test"); |
630 |
< |
if (name.equals(name8)) throw new Error(name); |
630 |
> |
if (name.equals(name8)) throw new AssertionError(name); |
631 |
|
try { |
632 |
|
return (Test) |
633 |
|
Class.forName(name8) |
634 |
< |
.getMethod("testSuite", new Class[] { dataClass }) |
634 |
> |
.getMethod("testSuite", dataClass) |
635 |
|
.invoke(null, data); |
636 |
< |
} catch (Exception e) { |
637 |
< |
throw new Error(e); |
636 |
> |
} catch (ReflectiveOperationException e) { |
637 |
> |
throw new AssertionError(e); |
638 |
|
} |
639 |
|
} else { |
640 |
|
return new TestSuite(); |
648 |
|
public static long MEDIUM_DELAY_MS; |
649 |
|
public static long LONG_DELAY_MS; |
650 |
|
|
651 |
+ |
private static final long RANDOM_TIMEOUT; |
652 |
+ |
private static final long RANDOM_EXPIRED_TIMEOUT; |
653 |
+ |
private static final TimeUnit RANDOM_TIMEUNIT; |
654 |
+ |
static { |
655 |
+ |
ThreadLocalRandom rnd = ThreadLocalRandom.current(); |
656 |
+ |
long[] timeouts = { Long.MIN_VALUE, -1, 0, 1, Long.MAX_VALUE }; |
657 |
+ |
RANDOM_TIMEOUT = timeouts[rnd.nextInt(timeouts.length)]; |
658 |
+ |
RANDOM_EXPIRED_TIMEOUT = timeouts[rnd.nextInt(3)]; |
659 |
+ |
TimeUnit[] timeUnits = TimeUnit.values(); |
660 |
+ |
RANDOM_TIMEUNIT = timeUnits[rnd.nextInt(timeUnits.length)]; |
661 |
+ |
} |
662 |
+ |
|
663 |
+ |
/** |
664 |
+ |
* Returns a timeout for use when any value at all will do. |
665 |
+ |
*/ |
666 |
+ |
static long randomTimeout() { return RANDOM_TIMEOUT; } |
667 |
+ |
|
668 |
+ |
/** |
669 |
+ |
* Returns a timeout that means "no waiting", i.e. not positive. |
670 |
+ |
*/ |
671 |
+ |
static long randomExpiredTimeout() { return RANDOM_EXPIRED_TIMEOUT; } |
672 |
+ |
|
673 |
+ |
/** |
674 |
+ |
* Returns a random non-null TimeUnit. |
675 |
+ |
*/ |
676 |
+ |
static TimeUnit randomTimeUnit() { return RANDOM_TIMEUNIT; } |
677 |
+ |
|
678 |
|
/** |
679 |
|
* Returns the shortest timed delay. This can be scaled up for |
680 |
|
* slow machines using the jsr166.delay.factor system property, |
743 |
|
String msg = toString() + ": " + String.format(format, args); |
744 |
|
System.err.println(msg); |
745 |
|
dumpTestThreads(); |
746 |
< |
throw new AssertionFailedError(msg); |
746 |
> |
throw new AssertionError(msg); |
747 |
|
} |
748 |
|
|
749 |
|
/** |
764 |
|
throw (RuntimeException) t; |
765 |
|
else if (t instanceof Exception) |
766 |
|
throw (Exception) t; |
767 |
< |
else { |
768 |
< |
AssertionFailedError afe = |
732 |
< |
new AssertionFailedError(t.toString()); |
733 |
< |
afe.initCause(t); |
734 |
< |
throw afe; |
735 |
< |
} |
767 |
> |
else |
768 |
> |
throw new AssertionError(t.toString(), t); |
769 |
|
} |
770 |
|
|
771 |
|
if (Thread.interrupted()) |
799 |
|
|
800 |
|
/** |
801 |
|
* Just like fail(reason), but additionally recording (using |
802 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
803 |
< |
* the current testcase will fail. |
802 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
803 |
> |
* current testcase will fail. |
804 |
|
*/ |
805 |
|
public void threadFail(String reason) { |
806 |
|
try { |
807 |
|
fail(reason); |
808 |
< |
} catch (AssertionFailedError t) { |
809 |
< |
threadRecordFailure(t); |
810 |
< |
throw t; |
808 |
> |
} catch (AssertionError fail) { |
809 |
> |
threadRecordFailure(fail); |
810 |
> |
throw fail; |
811 |
|
} |
812 |
|
} |
813 |
|
|
814 |
|
/** |
815 |
|
* Just like assertTrue(b), but additionally recording (using |
816 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
817 |
< |
* the current testcase will fail. |
816 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
817 |
> |
* current testcase will fail. |
818 |
|
*/ |
819 |
|
public void threadAssertTrue(boolean b) { |
820 |
|
try { |
821 |
|
assertTrue(b); |
822 |
< |
} catch (AssertionFailedError t) { |
823 |
< |
threadRecordFailure(t); |
824 |
< |
throw t; |
822 |
> |
} catch (AssertionError fail) { |
823 |
> |
threadRecordFailure(fail); |
824 |
> |
throw fail; |
825 |
|
} |
826 |
|
} |
827 |
|
|
828 |
|
/** |
829 |
|
* Just like assertFalse(b), but additionally recording (using |
830 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
831 |
< |
* the current testcase will fail. |
830 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
831 |
> |
* current testcase will fail. |
832 |
|
*/ |
833 |
|
public void threadAssertFalse(boolean b) { |
834 |
|
try { |
835 |
|
assertFalse(b); |
836 |
< |
} catch (AssertionFailedError t) { |
837 |
< |
threadRecordFailure(t); |
838 |
< |
throw t; |
836 |
> |
} catch (AssertionError fail) { |
837 |
> |
threadRecordFailure(fail); |
838 |
> |
throw fail; |
839 |
|
} |
840 |
|
} |
841 |
|
|
842 |
|
/** |
843 |
|
* Just like assertNull(x), but additionally recording (using |
844 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
845 |
< |
* the current testcase will fail. |
844 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
845 |
> |
* current testcase will fail. |
846 |
|
*/ |
847 |
|
public void threadAssertNull(Object x) { |
848 |
|
try { |
849 |
|
assertNull(x); |
850 |
< |
} catch (AssertionFailedError t) { |
851 |
< |
threadRecordFailure(t); |
852 |
< |
throw t; |
850 |
> |
} catch (AssertionError fail) { |
851 |
> |
threadRecordFailure(fail); |
852 |
> |
throw fail; |
853 |
|
} |
854 |
|
} |
855 |
|
|
856 |
|
/** |
857 |
|
* Just like assertEquals(x, y), but additionally recording (using |
858 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
859 |
< |
* the current testcase will fail. |
858 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
859 |
> |
* current testcase will fail. |
860 |
|
*/ |
861 |
|
public void threadAssertEquals(long x, long y) { |
862 |
|
try { |
863 |
|
assertEquals(x, y); |
864 |
< |
} catch (AssertionFailedError t) { |
865 |
< |
threadRecordFailure(t); |
866 |
< |
throw t; |
864 |
> |
} catch (AssertionError fail) { |
865 |
> |
threadRecordFailure(fail); |
866 |
> |
throw fail; |
867 |
|
} |
868 |
|
} |
869 |
|
|
870 |
|
/** |
871 |
|
* Just like assertEquals(x, y), but additionally recording (using |
872 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
873 |
< |
* the current testcase will fail. |
872 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
873 |
> |
* current testcase will fail. |
874 |
|
*/ |
875 |
|
public void threadAssertEquals(Object x, Object y) { |
876 |
|
try { |
877 |
|
assertEquals(x, y); |
878 |
< |
} catch (AssertionFailedError fail) { |
878 |
> |
} catch (AssertionError fail) { |
879 |
|
threadRecordFailure(fail); |
880 |
|
throw fail; |
881 |
|
} catch (Throwable fail) { |
885 |
|
|
886 |
|
/** |
887 |
|
* Just like assertSame(x, y), but additionally recording (using |
888 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
889 |
< |
* the current testcase will fail. |
888 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
889 |
> |
* current testcase will fail. |
890 |
|
*/ |
891 |
|
public void threadAssertSame(Object x, Object y) { |
892 |
|
try { |
893 |
|
assertSame(x, y); |
894 |
< |
} catch (AssertionFailedError fail) { |
894 |
> |
} catch (AssertionError fail) { |
895 |
|
threadRecordFailure(fail); |
896 |
|
throw fail; |
897 |
|
} |
913 |
|
|
914 |
|
/** |
915 |
|
* Records the given exception using {@link #threadRecordFailure}, |
916 |
< |
* then rethrows the exception, wrapping it in an |
917 |
< |
* AssertionFailedError if necessary. |
916 |
> |
* then rethrows the exception, wrapping it in an AssertionError |
917 |
> |
* if necessary. |
918 |
|
*/ |
919 |
|
public void threadUnexpectedException(Throwable t) { |
920 |
|
threadRecordFailure(t); |
923 |
|
throw (RuntimeException) t; |
924 |
|
else if (t instanceof Error) |
925 |
|
throw (Error) t; |
926 |
< |
else { |
927 |
< |
AssertionFailedError afe = |
895 |
< |
new AssertionFailedError("unexpected exception: " + t); |
896 |
< |
afe.initCause(t); |
897 |
< |
throw afe; |
898 |
< |
} |
926 |
> |
else |
927 |
> |
throw new AssertionError("unexpected exception: " + t, t); |
928 |
|
} |
929 |
|
|
930 |
|
/** |
1095 |
|
* Checks that thread eventually enters the expected blocked thread state. |
1096 |
|
*/ |
1097 |
|
void assertThreadBlocks(Thread thread, Thread.State expected) { |
1098 |
< |
// always sleep at least 1 ms, avoiding transitional states |
1099 |
< |
// with high probability |
1098 |
> |
// always sleep at least 1 ms, with high probability avoiding |
1099 |
> |
// transitory states |
1100 |
|
for (long retries = LONG_DELAY_MS * 3 / 4; retries-->0; ) { |
1101 |
|
try { delay(1); } |
1102 |
|
catch (InterruptedException fail) { |
1103 |
< |
fail("Unexpected InterruptedException"); |
1103 |
> |
throw new AssertionError("Unexpected InterruptedException", fail); |
1104 |
|
} |
1105 |
|
Thread.State s = thread.getState(); |
1106 |
|
if (s == expected) |
1112 |
|
} |
1113 |
|
|
1114 |
|
/** |
1086 |
– |
* Checks that thread does not terminate within the default |
1087 |
– |
* millisecond delay of {@code timeoutMillis()}. |
1088 |
– |
*/ |
1089 |
– |
void assertThreadStaysAlive(Thread thread) { |
1090 |
– |
assertThreadStaysAlive(thread, timeoutMillis()); |
1091 |
– |
} |
1092 |
– |
|
1093 |
– |
/** |
1094 |
– |
* Checks that thread does not terminate within the given millisecond delay. |
1095 |
– |
*/ |
1096 |
– |
void assertThreadStaysAlive(Thread thread, long millis) { |
1097 |
– |
try { |
1098 |
– |
// No need to optimize the failing case via Thread.join. |
1099 |
– |
delay(millis); |
1100 |
– |
assertTrue(thread.isAlive()); |
1101 |
– |
} catch (InterruptedException fail) { |
1102 |
– |
threadFail("Unexpected InterruptedException"); |
1103 |
– |
} |
1104 |
– |
} |
1105 |
– |
|
1106 |
– |
/** |
1115 |
|
* Checks that future.get times out, with the default timeout of |
1116 |
|
* {@code timeoutMillis()}. |
1117 |
|
*/ |
1286 |
|
|
1287 |
|
/** |
1288 |
|
* Sleeps until the given time has elapsed. |
1289 |
< |
* Throws AssertionFailedError if interrupted. |
1289 |
> |
* Throws AssertionError if interrupted. |
1290 |
|
*/ |
1291 |
|
static void sleep(long millis) { |
1292 |
|
try { |
1293 |
|
delay(millis); |
1294 |
|
} catch (InterruptedException fail) { |
1295 |
< |
AssertionFailedError afe = |
1288 |
< |
new AssertionFailedError("Unexpected InterruptedException"); |
1289 |
< |
afe.initCause(fail); |
1290 |
< |
throw afe; |
1295 |
> |
throw new AssertionError("Unexpected InterruptedException", fail); |
1296 |
|
} |
1297 |
|
} |
1298 |
|
|
1299 |
|
/** |
1300 |
|
* Spin-waits up to the specified number of milliseconds for the given |
1301 |
|
* thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING. |
1302 |
+ |
* @param waitingForGodot if non-null, an additional condition to satisfy |
1303 |
|
*/ |
1304 |
< |
void waitForThreadToEnterWaitState(Thread thread, long timeoutMillis) { |
1305 |
< |
long startTime = 0L; |
1306 |
< |
for (;;) { |
1307 |
< |
Thread.State s = thread.getState(); |
1308 |
< |
if (s == Thread.State.BLOCKED || |
1309 |
< |
s == Thread.State.WAITING || |
1310 |
< |
s == Thread.State.TIMED_WAITING) |
1311 |
< |
return; |
1312 |
< |
else if (s == Thread.State.TERMINATED) |
1304 |
> |
void waitForThreadToEnterWaitState(Thread thread, long timeoutMillis, |
1305 |
> |
Callable<Boolean> waitingForGodot) { |
1306 |
> |
for (long startTime = 0L;;) { |
1307 |
> |
switch (thread.getState()) { |
1308 |
> |
default: break; |
1309 |
> |
case BLOCKED: case WAITING: case TIMED_WAITING: |
1310 |
> |
try { |
1311 |
> |
if (waitingForGodot == null || waitingForGodot.call()) |
1312 |
> |
return; |
1313 |
> |
} catch (Throwable fail) { threadUnexpectedException(fail); } |
1314 |
> |
break; |
1315 |
> |
case TERMINATED: |
1316 |
|
fail("Unexpected thread termination"); |
1317 |
< |
else if (startTime == 0L) |
1317 |
> |
} |
1318 |
> |
|
1319 |
> |
if (startTime == 0L) |
1320 |
|
startTime = System.nanoTime(); |
1321 |
|
else if (millisElapsedSince(startTime) > timeoutMillis) { |
1322 |
< |
threadAssertTrue(thread.isAlive()); |
1323 |
< |
fail("timed out waiting for thread to enter wait state"); |
1322 |
> |
assertTrue(thread.isAlive()); |
1323 |
> |
if (waitingForGodot == null |
1324 |
> |
|| thread.getState() == Thread.State.RUNNABLE) |
1325 |
> |
fail("timed out waiting for thread to enter wait state"); |
1326 |
> |
else |
1327 |
> |
fail("timed out waiting for condition, thread state=" |
1328 |
> |
+ thread.getState()); |
1329 |
|
} |
1330 |
|
Thread.yield(); |
1331 |
|
} |
1333 |
|
|
1334 |
|
/** |
1335 |
|
* Spin-waits up to the specified number of milliseconds for the given |
1336 |
< |
* thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING, |
1321 |
< |
* and additionally satisfy the given condition. |
1336 |
> |
* thread to enter a wait state: BLOCKED, WAITING, or TIMED_WAITING. |
1337 |
|
*/ |
1338 |
< |
void waitForThreadToEnterWaitState( |
1339 |
< |
Thread thread, long timeoutMillis, Callable<Boolean> waitingForGodot) { |
1325 |
< |
long startTime = 0L; |
1326 |
< |
for (;;) { |
1327 |
< |
Thread.State s = thread.getState(); |
1328 |
< |
if (s == Thread.State.BLOCKED || |
1329 |
< |
s == Thread.State.WAITING || |
1330 |
< |
s == Thread.State.TIMED_WAITING) { |
1331 |
< |
try { |
1332 |
< |
if (waitingForGodot.call()) |
1333 |
< |
return; |
1334 |
< |
} catch (Throwable fail) { threadUnexpectedException(fail); } |
1335 |
< |
} |
1336 |
< |
else if (s == Thread.State.TERMINATED) |
1337 |
< |
fail("Unexpected thread termination"); |
1338 |
< |
else if (startTime == 0L) |
1339 |
< |
startTime = System.nanoTime(); |
1340 |
< |
else if (millisElapsedSince(startTime) > timeoutMillis) { |
1341 |
< |
threadAssertTrue(thread.isAlive()); |
1342 |
< |
fail("timed out waiting for thread to enter wait state"); |
1343 |
< |
} |
1344 |
< |
Thread.yield(); |
1345 |
< |
} |
1338 |
> |
void waitForThreadToEnterWaitState(Thread thread, long timeoutMillis) { |
1339 |
> |
waitForThreadToEnterWaitState(thread, timeoutMillis, null); |
1340 |
|
} |
1341 |
|
|
1342 |
|
/** |
1344 |
|
* enter a wait state: BLOCKED, WAITING, or TIMED_WAITING. |
1345 |
|
*/ |
1346 |
|
void waitForThreadToEnterWaitState(Thread thread) { |
1347 |
< |
waitForThreadToEnterWaitState(thread, LONG_DELAY_MS); |
1347 |
> |
waitForThreadToEnterWaitState(thread, LONG_DELAY_MS, null); |
1348 |
|
} |
1349 |
|
|
1350 |
|
/** |
1352 |
|
* enter a wait state: BLOCKED, WAITING, or TIMED_WAITING, |
1353 |
|
* and additionally satisfy the given condition. |
1354 |
|
*/ |
1355 |
< |
void waitForThreadToEnterWaitState( |
1356 |
< |
Thread thread, Callable<Boolean> waitingForGodot) { |
1355 |
> |
void waitForThreadToEnterWaitState(Thread thread, |
1356 |
> |
Callable<Boolean> waitingForGodot) { |
1357 |
|
waitForThreadToEnterWaitState(thread, LONG_DELAY_MS, waitingForGodot); |
1358 |
|
} |
1359 |
|
|
1372 |
|
// r.run(); |
1373 |
|
// } catch (Throwable fail) { threadUnexpectedException(fail); } |
1374 |
|
// if (millisElapsedSince(startTime) > timeoutMillis/2) |
1375 |
< |
// throw new AssertionFailedError("did not return promptly"); |
1375 |
> |
// throw new AssertionError("did not return promptly"); |
1376 |
|
// } |
1377 |
|
|
1378 |
|
// void assertTerminatesPromptly(Runnable r) { |
1385 |
|
*/ |
1386 |
|
<T> void checkTimedGet(Future<T> f, T expectedValue, long timeoutMillis) { |
1387 |
|
long startTime = System.nanoTime(); |
1388 |
+ |
T actual = null; |
1389 |
|
try { |
1390 |
< |
assertEquals(expectedValue, f.get(timeoutMillis, MILLISECONDS)); |
1390 |
> |
actual = f.get(timeoutMillis, MILLISECONDS); |
1391 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
1392 |
+ |
assertEquals(expectedValue, actual); |
1393 |
|
if (millisElapsedSince(startTime) > timeoutMillis/2) |
1394 |
< |
throw new AssertionFailedError("timed get did not return promptly"); |
1394 |
> |
throw new AssertionError("timed get did not return promptly"); |
1395 |
|
} |
1396 |
|
|
1397 |
|
<T> void checkTimedGet(Future<T> f, T expectedValue) { |
1449 |
|
} |
1450 |
|
} |
1451 |
|
|
1456 |
– |
public abstract class RunnableShouldThrow implements Runnable { |
1457 |
– |
protected abstract void realRun() throws Throwable; |
1458 |
– |
|
1459 |
– |
final Class<?> exceptionClass; |
1460 |
– |
|
1461 |
– |
<T extends Throwable> RunnableShouldThrow(Class<T> exceptionClass) { |
1462 |
– |
this.exceptionClass = exceptionClass; |
1463 |
– |
} |
1464 |
– |
|
1465 |
– |
public final void run() { |
1466 |
– |
try { |
1467 |
– |
realRun(); |
1468 |
– |
threadShouldThrow(exceptionClass.getSimpleName()); |
1469 |
– |
} catch (Throwable t) { |
1470 |
– |
if (! exceptionClass.isInstance(t)) |
1471 |
– |
threadUnexpectedException(t); |
1472 |
– |
} |
1473 |
– |
} |
1474 |
– |
} |
1475 |
– |
|
1452 |
|
public abstract class ThreadShouldThrow extends Thread { |
1453 |
|
protected abstract void realRun() throws Throwable; |
1454 |
|
|
1461 |
|
public final void run() { |
1462 |
|
try { |
1463 |
|
realRun(); |
1488 |
– |
threadShouldThrow(exceptionClass.getSimpleName()); |
1464 |
|
} catch (Throwable t) { |
1465 |
|
if (! exceptionClass.isInstance(t)) |
1466 |
|
threadUnexpectedException(t); |
1467 |
+ |
return; |
1468 |
|
} |
1469 |
+ |
threadShouldThrow(exceptionClass.getSimpleName()); |
1470 |
|
} |
1471 |
|
} |
1472 |
|
|
1476 |
|
public final void run() { |
1477 |
|
try { |
1478 |
|
realRun(); |
1502 |
– |
threadShouldThrow("InterruptedException"); |
1479 |
|
} catch (InterruptedException success) { |
1480 |
|
threadAssertFalse(Thread.interrupted()); |
1481 |
+ |
return; |
1482 |
|
} catch (Throwable fail) { |
1483 |
|
threadUnexpectedException(fail); |
1484 |
|
} |
1485 |
+ |
threadShouldThrow("InterruptedException"); |
1486 |
|
} |
1487 |
|
} |
1488 |
|
|
1494 |
|
return realCall(); |
1495 |
|
} catch (Throwable fail) { |
1496 |
|
threadUnexpectedException(fail); |
1519 |
– |
return null; |
1497 |
|
} |
1498 |
< |
} |
1522 |
< |
} |
1523 |
< |
|
1524 |
< |
public abstract class CheckedInterruptedCallable<T> |
1525 |
< |
implements Callable<T> { |
1526 |
< |
protected abstract T realCall() throws Throwable; |
1527 |
< |
|
1528 |
< |
public final T call() { |
1529 |
< |
try { |
1530 |
< |
T result = realCall(); |
1531 |
< |
threadShouldThrow("InterruptedException"); |
1532 |
< |
return result; |
1533 |
< |
} catch (InterruptedException success) { |
1534 |
< |
threadAssertFalse(Thread.interrupted()); |
1535 |
< |
} catch (Throwable fail) { |
1536 |
< |
threadUnexpectedException(fail); |
1537 |
< |
} |
1538 |
< |
return null; |
1498 |
> |
throw new AssertionError("unreached"); |
1499 |
|
} |
1500 |
|
} |
1501 |
|
|
1552 |
|
} |
1553 |
|
|
1554 |
|
public void await(CountDownLatch latch, long timeoutMillis) { |
1555 |
+ |
boolean timedOut = false; |
1556 |
|
try { |
1557 |
< |
if (!latch.await(timeoutMillis, MILLISECONDS)) |
1597 |
< |
fail("timed out waiting for CountDownLatch for " |
1598 |
< |
+ (timeoutMillis/1000) + " sec"); |
1557 |
> |
timedOut = !latch.await(timeoutMillis, MILLISECONDS); |
1558 |
|
} catch (Throwable fail) { |
1559 |
|
threadUnexpectedException(fail); |
1560 |
|
} |
1561 |
+ |
if (timedOut) |
1562 |
+ |
fail("timed out waiting for CountDownLatch for " |
1563 |
+ |
+ (timeoutMillis/1000) + " sec"); |
1564 |
|
} |
1565 |
|
|
1566 |
|
public void await(CountDownLatch latch) { |
1568 |
|
} |
1569 |
|
|
1570 |
|
public void await(Semaphore semaphore) { |
1571 |
+ |
boolean timedOut = false; |
1572 |
|
try { |
1573 |
< |
if (!semaphore.tryAcquire(LONG_DELAY_MS, MILLISECONDS)) |
1611 |
< |
fail("timed out waiting for Semaphore for " |
1612 |
< |
+ (LONG_DELAY_MS/1000) + " sec"); |
1573 |
> |
timedOut = !semaphore.tryAcquire(LONG_DELAY_MS, MILLISECONDS); |
1574 |
|
} catch (Throwable fail) { |
1575 |
|
threadUnexpectedException(fail); |
1576 |
|
} |
1577 |
+ |
if (timedOut) |
1578 |
+ |
fail("timed out waiting for Semaphore for " |
1579 |
+ |
+ (LONG_DELAY_MS/1000) + " sec"); |
1580 |
|
} |
1581 |
|
|
1582 |
|
public void await(CyclicBarrier barrier) { |
1601 |
|
// long startTime = System.nanoTime(); |
1602 |
|
// while (!flag.get()) { |
1603 |
|
// if (millisElapsedSince(startTime) > timeoutMillis) |
1604 |
< |
// throw new AssertionFailedError("timed out"); |
1604 |
> |
// throw new AssertionError("timed out"); |
1605 |
|
// Thread.yield(); |
1606 |
|
// } |
1607 |
|
// } |
1610 |
|
public String call() { throw new NullPointerException(); } |
1611 |
|
} |
1612 |
|
|
1649 |
– |
public static class CallableOne implements Callable<Integer> { |
1650 |
– |
public Integer call() { return one; } |
1651 |
– |
} |
1652 |
– |
|
1653 |
– |
public class ShortRunnable extends CheckedRunnable { |
1654 |
– |
protected void realRun() throws Throwable { |
1655 |
– |
delay(SHORT_DELAY_MS); |
1656 |
– |
} |
1657 |
– |
} |
1658 |
– |
|
1659 |
– |
public class ShortInterruptedRunnable extends CheckedInterruptedRunnable { |
1660 |
– |
protected void realRun() throws InterruptedException { |
1661 |
– |
delay(SHORT_DELAY_MS); |
1662 |
– |
} |
1663 |
– |
} |
1664 |
– |
|
1665 |
– |
public class SmallRunnable extends CheckedRunnable { |
1666 |
– |
protected void realRun() throws Throwable { |
1667 |
– |
delay(SMALL_DELAY_MS); |
1668 |
– |
} |
1669 |
– |
} |
1670 |
– |
|
1671 |
– |
public class SmallPossiblyInterruptedRunnable extends CheckedRunnable { |
1672 |
– |
protected void realRun() { |
1673 |
– |
try { |
1674 |
– |
delay(SMALL_DELAY_MS); |
1675 |
– |
} catch (InterruptedException ok) {} |
1676 |
– |
} |
1677 |
– |
} |
1678 |
– |
|
1679 |
– |
public class SmallCallable extends CheckedCallable { |
1680 |
– |
protected Object realCall() throws InterruptedException { |
1681 |
– |
delay(SMALL_DELAY_MS); |
1682 |
– |
return Boolean.TRUE; |
1683 |
– |
} |
1684 |
– |
} |
1685 |
– |
|
1686 |
– |
public class MediumRunnable extends CheckedRunnable { |
1687 |
– |
protected void realRun() throws Throwable { |
1688 |
– |
delay(MEDIUM_DELAY_MS); |
1689 |
– |
} |
1690 |
– |
} |
1691 |
– |
|
1692 |
– |
public class MediumInterruptedRunnable extends CheckedInterruptedRunnable { |
1693 |
– |
protected void realRun() throws InterruptedException { |
1694 |
– |
delay(MEDIUM_DELAY_MS); |
1695 |
– |
} |
1696 |
– |
} |
1697 |
– |
|
1613 |
|
public Runnable possiblyInterruptedRunnable(final long timeoutMillis) { |
1614 |
|
return new CheckedRunnable() { |
1615 |
|
protected void realRun() { |
1619 |
|
}}; |
1620 |
|
} |
1621 |
|
|
1707 |
– |
public class MediumPossiblyInterruptedRunnable extends CheckedRunnable { |
1708 |
– |
protected void realRun() { |
1709 |
– |
try { |
1710 |
– |
delay(MEDIUM_DELAY_MS); |
1711 |
– |
} catch (InterruptedException ok) {} |
1712 |
– |
} |
1713 |
– |
} |
1714 |
– |
|
1715 |
– |
public class LongPossiblyInterruptedRunnable extends CheckedRunnable { |
1716 |
– |
protected void realRun() { |
1717 |
– |
try { |
1718 |
– |
delay(LONG_DELAY_MS); |
1719 |
– |
} catch (InterruptedException ok) {} |
1720 |
– |
} |
1721 |
– |
} |
1722 |
– |
|
1622 |
|
/** |
1623 |
|
* For use as ThreadFactory in constructors |
1624 |
|
*/ |
1632 |
|
boolean isDone(); |
1633 |
|
} |
1634 |
|
|
1736 |
– |
public static TrackedRunnable trackedRunnable(final long timeoutMillis) { |
1737 |
– |
return new TrackedRunnable() { |
1738 |
– |
private volatile boolean done = false; |
1739 |
– |
public boolean isDone() { return done; } |
1740 |
– |
public void run() { |
1741 |
– |
try { |
1742 |
– |
delay(timeoutMillis); |
1743 |
– |
done = true; |
1744 |
– |
} catch (InterruptedException ok) {} |
1745 |
– |
} |
1746 |
– |
}; |
1747 |
– |
} |
1748 |
– |
|
1749 |
– |
public static class TrackedShortRunnable implements Runnable { |
1750 |
– |
public volatile boolean done = false; |
1751 |
– |
public void run() { |
1752 |
– |
try { |
1753 |
– |
delay(SHORT_DELAY_MS); |
1754 |
– |
done = true; |
1755 |
– |
} catch (InterruptedException ok) {} |
1756 |
– |
} |
1757 |
– |
} |
1758 |
– |
|
1759 |
– |
public static class TrackedSmallRunnable implements Runnable { |
1760 |
– |
public volatile boolean done = false; |
1761 |
– |
public void run() { |
1762 |
– |
try { |
1763 |
– |
delay(SMALL_DELAY_MS); |
1764 |
– |
done = true; |
1765 |
– |
} catch (InterruptedException ok) {} |
1766 |
– |
} |
1767 |
– |
} |
1768 |
– |
|
1769 |
– |
public static class TrackedMediumRunnable implements Runnable { |
1770 |
– |
public volatile boolean done = false; |
1771 |
– |
public void run() { |
1772 |
– |
try { |
1773 |
– |
delay(MEDIUM_DELAY_MS); |
1774 |
– |
done = true; |
1775 |
– |
} catch (InterruptedException ok) {} |
1776 |
– |
} |
1777 |
– |
} |
1778 |
– |
|
1779 |
– |
public static class TrackedLongRunnable implements Runnable { |
1780 |
– |
public volatile boolean done = false; |
1781 |
– |
public void run() { |
1782 |
– |
try { |
1783 |
– |
delay(LONG_DELAY_MS); |
1784 |
– |
done = true; |
1785 |
– |
} catch (InterruptedException ok) {} |
1786 |
– |
} |
1787 |
– |
} |
1788 |
– |
|
1635 |
|
public static class TrackedNoOpRunnable implements Runnable { |
1636 |
|
public volatile boolean done = false; |
1637 |
|
public void run() { |
1639 |
|
} |
1640 |
|
} |
1641 |
|
|
1796 |
– |
public static class TrackedCallable implements Callable { |
1797 |
– |
public volatile boolean done = false; |
1798 |
– |
public Object call() { |
1799 |
– |
try { |
1800 |
– |
delay(SMALL_DELAY_MS); |
1801 |
– |
done = true; |
1802 |
– |
} catch (InterruptedException ok) {} |
1803 |
– |
return Boolean.TRUE; |
1804 |
– |
} |
1805 |
– |
} |
1806 |
– |
|
1642 |
|
/** |
1643 |
|
* Analog of CheckedRunnable for RecursiveAction |
1644 |
|
*/ |
1665 |
|
return realCompute(); |
1666 |
|
} catch (Throwable fail) { |
1667 |
|
threadUnexpectedException(fail); |
1833 |
– |
return null; |
1668 |
|
} |
1669 |
+ |
throw new AssertionError("unreached"); |
1670 |
|
} |
1671 |
|
} |
1672 |
|
|
1680 |
|
|
1681 |
|
/** |
1682 |
|
* A CyclicBarrier that uses timed await and fails with |
1683 |
< |
* AssertionFailedErrors instead of throwing checked exceptions. |
1683 |
> |
* AssertionErrors instead of throwing checked exceptions. |
1684 |
|
*/ |
1685 |
|
public static class CheckedBarrier extends CyclicBarrier { |
1686 |
|
public CheckedBarrier(int parties) { super(parties); } |
1689 |
|
try { |
1690 |
|
return super.await(2 * LONG_DELAY_MS, MILLISECONDS); |
1691 |
|
} catch (TimeoutException timedOut) { |
1692 |
< |
throw new AssertionFailedError("timed out"); |
1692 |
> |
throw new AssertionError("timed out"); |
1693 |
|
} catch (Exception fail) { |
1694 |
< |
AssertionFailedError afe = |
1860 |
< |
new AssertionFailedError("Unexpected exception: " + fail); |
1861 |
< |
afe.initCause(fail); |
1862 |
< |
throw afe; |
1694 |
> |
throw new AssertionError("Unexpected exception: " + fail, fail); |
1695 |
|
} |
1696 |
|
} |
1697 |
|
} |
1702 |
|
assertEquals(0, q.size()); |
1703 |
|
assertNull(q.peek()); |
1704 |
|
assertNull(q.poll()); |
1705 |
< |
assertNull(q.poll(0, MILLISECONDS)); |
1705 |
> |
assertNull(q.poll(randomExpiredTimeout(), randomTimeUnit())); |
1706 |
|
assertEquals(q.toString(), "[]"); |
1707 |
|
assertTrue(Arrays.equals(q.toArray(), new Object[0])); |
1708 |
|
assertFalse(q.iterator().hasNext()); |
1754 |
|
|
1755 |
|
@SuppressWarnings("unchecked") |
1756 |
|
<T> T serialClone(T o) { |
1757 |
+ |
T clone = null; |
1758 |
|
try { |
1759 |
|
ObjectInputStream ois = new ObjectInputStream |
1760 |
|
(new ByteArrayInputStream(serialBytes(o))); |
1761 |
< |
T clone = (T) ois.readObject(); |
1929 |
< |
if (o == clone) assertImmutable(o); |
1930 |
< |
assertSame(o.getClass(), clone.getClass()); |
1931 |
< |
return clone; |
1761 |
> |
clone = (T) ois.readObject(); |
1762 |
|
} catch (Throwable fail) { |
1763 |
|
threadUnexpectedException(fail); |
1934 |
– |
return null; |
1764 |
|
} |
1765 |
+ |
if (o == clone) assertImmutable(o); |
1766 |
+ |
else assertSame(o.getClass(), clone.getClass()); |
1767 |
+ |
return clone; |
1768 |
|
} |
1769 |
|
|
1770 |
|
/** |
1783 |
|
(new ByteArrayInputStream(bos.toByteArray())); |
1784 |
|
T clone = (T) ois.readObject(); |
1785 |
|
if (o == clone) assertImmutable(o); |
1786 |
< |
assertSame(o.getClass(), clone.getClass()); |
1786 |
> |
else assertSame(o.getClass(), clone.getClass()); |
1787 |
|
return clone; |
1788 |
|
} |
1789 |
|
|
1814 |
|
try { throwingAction.run(); } |
1815 |
|
catch (Throwable t) { |
1816 |
|
threw = true; |
1817 |
< |
if (!expectedExceptionClass.isInstance(t)) { |
1818 |
< |
AssertionFailedError afe = |
1819 |
< |
new AssertionFailedError |
1820 |
< |
("Expected " + expectedExceptionClass.getName() + |
1821 |
< |
", got " + t.getClass().getName()); |
1990 |
< |
afe.initCause(t); |
1991 |
< |
threadUnexpectedException(afe); |
1992 |
< |
} |
1817 |
> |
if (!expectedExceptionClass.isInstance(t)) |
1818 |
> |
throw new AssertionError( |
1819 |
> |
"Expected " + expectedExceptionClass.getName() + |
1820 |
> |
", got " + t.getClass().getName(), |
1821 |
> |
t); |
1822 |
|
} |
1823 |
|
if (!threw) |
1824 |
|
shouldThrow(expectedExceptionClass.getName()); |
1850 |
|
static <T> void shuffle(T[] array) { |
1851 |
|
Collections.shuffle(Arrays.asList(array), ThreadLocalRandom.current()); |
1852 |
|
} |
1853 |
+ |
|
1854 |
+ |
/** |
1855 |
+ |
* Returns the same String as would be returned by {@link |
1856 |
+ |
* Object#toString}, whether or not the given object's class |
1857 |
+ |
* overrides toString(). |
1858 |
+ |
* |
1859 |
+ |
* @see System#identityHashCode |
1860 |
+ |
*/ |
1861 |
+ |
static String identityString(Object x) { |
1862 |
+ |
return x.getClass().getName() |
1863 |
+ |
+ "@" + Integer.toHexString(System.identityHashCode(x)); |
1864 |
+ |
} |
1865 |
+ |
|
1866 |
+ |
// --- Shared assertions for Executor tests --- |
1867 |
+ |
|
1868 |
+ |
/** |
1869 |
+ |
* Returns maximum number of tasks that can be submitted to given |
1870 |
+ |
* pool (with bounded queue) before saturation (when submission |
1871 |
+ |
* throws RejectedExecutionException). |
1872 |
+ |
*/ |
1873 |
+ |
static final int saturatedSize(ThreadPoolExecutor pool) { |
1874 |
+ |
BlockingQueue<Runnable> q = pool.getQueue(); |
1875 |
+ |
return pool.getMaximumPoolSize() + q.size() + q.remainingCapacity(); |
1876 |
+ |
} |
1877 |
+ |
|
1878 |
+ |
@SuppressWarnings("FutureReturnValueIgnored") |
1879 |
+ |
void assertNullTaskSubmissionThrowsNullPointerException(Executor e) { |
1880 |
+ |
try { |
1881 |
+ |
e.execute((Runnable) null); |
1882 |
+ |
shouldThrow(); |
1883 |
+ |
} catch (NullPointerException success) {} |
1884 |
+ |
|
1885 |
+ |
if (! (e instanceof ExecutorService)) return; |
1886 |
+ |
ExecutorService es = (ExecutorService) e; |
1887 |
+ |
try { |
1888 |
+ |
es.submit((Runnable) null); |
1889 |
+ |
shouldThrow(); |
1890 |
+ |
} catch (NullPointerException success) {} |
1891 |
+ |
try { |
1892 |
+ |
es.submit((Runnable) null, Boolean.TRUE); |
1893 |
+ |
shouldThrow(); |
1894 |
+ |
} catch (NullPointerException success) {} |
1895 |
+ |
try { |
1896 |
+ |
es.submit((Callable) null); |
1897 |
+ |
shouldThrow(); |
1898 |
+ |
} catch (NullPointerException success) {} |
1899 |
+ |
|
1900 |
+ |
if (! (e instanceof ScheduledExecutorService)) return; |
1901 |
+ |
ScheduledExecutorService ses = (ScheduledExecutorService) e; |
1902 |
+ |
try { |
1903 |
+ |
ses.schedule((Runnable) null, |
1904 |
+ |
randomTimeout(), randomTimeUnit()); |
1905 |
+ |
shouldThrow(); |
1906 |
+ |
} catch (NullPointerException success) {} |
1907 |
+ |
try { |
1908 |
+ |
ses.schedule((Callable) null, |
1909 |
+ |
randomTimeout(), randomTimeUnit()); |
1910 |
+ |
shouldThrow(); |
1911 |
+ |
} catch (NullPointerException success) {} |
1912 |
+ |
try { |
1913 |
+ |
ses.scheduleAtFixedRate((Runnable) null, |
1914 |
+ |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1915 |
+ |
shouldThrow(); |
1916 |
+ |
} catch (NullPointerException success) {} |
1917 |
+ |
try { |
1918 |
+ |
ses.scheduleWithFixedDelay((Runnable) null, |
1919 |
+ |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1920 |
+ |
shouldThrow(); |
1921 |
+ |
} catch (NullPointerException success) {} |
1922 |
+ |
} |
1923 |
+ |
|
1924 |
+ |
void setRejectedExecutionHandler( |
1925 |
+ |
ThreadPoolExecutor p, RejectedExecutionHandler handler) { |
1926 |
+ |
p.setRejectedExecutionHandler(handler); |
1927 |
+ |
assertSame(handler, p.getRejectedExecutionHandler()); |
1928 |
+ |
} |
1929 |
+ |
|
1930 |
+ |
void assertTaskSubmissionsAreRejected(ThreadPoolExecutor p) { |
1931 |
+ |
final RejectedExecutionHandler savedHandler = p.getRejectedExecutionHandler(); |
1932 |
+ |
final long savedTaskCount = p.getTaskCount(); |
1933 |
+ |
final long savedCompletedTaskCount = p.getCompletedTaskCount(); |
1934 |
+ |
final int savedQueueSize = p.getQueue().size(); |
1935 |
+ |
final boolean stock = (p.getClass().getClassLoader() == null); |
1936 |
+ |
|
1937 |
+ |
Runnable r = () -> {}; |
1938 |
+ |
Callable<Boolean> c = () -> Boolean.TRUE; |
1939 |
+ |
|
1940 |
+ |
class Recorder implements RejectedExecutionHandler { |
1941 |
+ |
public volatile Runnable r = null; |
1942 |
+ |
public volatile ThreadPoolExecutor p = null; |
1943 |
+ |
public void reset() { r = null; p = null; } |
1944 |
+ |
public void rejectedExecution(Runnable r, ThreadPoolExecutor p) { |
1945 |
+ |
assertNull(this.r); |
1946 |
+ |
assertNull(this.p); |
1947 |
+ |
this.r = r; |
1948 |
+ |
this.p = p; |
1949 |
+ |
} |
1950 |
+ |
} |
1951 |
+ |
|
1952 |
+ |
// check custom handler is invoked exactly once per task |
1953 |
+ |
Recorder recorder = new Recorder(); |
1954 |
+ |
setRejectedExecutionHandler(p, recorder); |
1955 |
+ |
for (int i = 2; i--> 0; ) { |
1956 |
+ |
recorder.reset(); |
1957 |
+ |
p.execute(r); |
1958 |
+ |
if (stock && p.getClass() == ThreadPoolExecutor.class) |
1959 |
+ |
assertSame(r, recorder.r); |
1960 |
+ |
assertSame(p, recorder.p); |
1961 |
+ |
|
1962 |
+ |
recorder.reset(); |
1963 |
+ |
assertFalse(p.submit(r).isDone()); |
1964 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1965 |
+ |
assertSame(p, recorder.p); |
1966 |
+ |
|
1967 |
+ |
recorder.reset(); |
1968 |
+ |
assertFalse(p.submit(r, Boolean.TRUE).isDone()); |
1969 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1970 |
+ |
assertSame(p, recorder.p); |
1971 |
+ |
|
1972 |
+ |
recorder.reset(); |
1973 |
+ |
assertFalse(p.submit(c).isDone()); |
1974 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1975 |
+ |
assertSame(p, recorder.p); |
1976 |
+ |
|
1977 |
+ |
if (p instanceof ScheduledExecutorService) { |
1978 |
+ |
ScheduledExecutorService s = (ScheduledExecutorService) p; |
1979 |
+ |
ScheduledFuture<?> future; |
1980 |
+ |
|
1981 |
+ |
recorder.reset(); |
1982 |
+ |
future = s.schedule(r, randomTimeout(), randomTimeUnit()); |
1983 |
+ |
assertFalse(future.isDone()); |
1984 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1985 |
+ |
assertSame(p, recorder.p); |
1986 |
+ |
|
1987 |
+ |
recorder.reset(); |
1988 |
+ |
future = s.schedule(c, randomTimeout(), randomTimeUnit()); |
1989 |
+ |
assertFalse(future.isDone()); |
1990 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1991 |
+ |
assertSame(p, recorder.p); |
1992 |
+ |
|
1993 |
+ |
recorder.reset(); |
1994 |
+ |
future = s.scheduleAtFixedRate(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1995 |
+ |
assertFalse(future.isDone()); |
1996 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1997 |
+ |
assertSame(p, recorder.p); |
1998 |
+ |
|
1999 |
+ |
recorder.reset(); |
2000 |
+ |
future = s.scheduleWithFixedDelay(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
2001 |
+ |
assertFalse(future.isDone()); |
2002 |
+ |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2003 |
+ |
assertSame(p, recorder.p); |
2004 |
+ |
} |
2005 |
+ |
} |
2006 |
+ |
|
2007 |
+ |
// Checking our custom handler above should be sufficient, but |
2008 |
+ |
// we add some integration tests of standard handlers. |
2009 |
+ |
final AtomicReference<Thread> thread = new AtomicReference<>(); |
2010 |
+ |
final Runnable setThread = () -> thread.set(Thread.currentThread()); |
2011 |
+ |
|
2012 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.AbortPolicy()); |
2013 |
+ |
try { |
2014 |
+ |
p.execute(setThread); |
2015 |
+ |
shouldThrow(); |
2016 |
+ |
} catch (RejectedExecutionException success) {} |
2017 |
+ |
assertNull(thread.get()); |
2018 |
+ |
|
2019 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.DiscardPolicy()); |
2020 |
+ |
p.execute(setThread); |
2021 |
+ |
assertNull(thread.get()); |
2022 |
+ |
|
2023 |
+ |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.CallerRunsPolicy()); |
2024 |
+ |
p.execute(setThread); |
2025 |
+ |
if (p.isShutdown()) |
2026 |
+ |
assertNull(thread.get()); |
2027 |
+ |
else |
2028 |
+ |
assertSame(Thread.currentThread(), thread.get()); |
2029 |
+ |
|
2030 |
+ |
setRejectedExecutionHandler(p, savedHandler); |
2031 |
+ |
|
2032 |
+ |
// check that pool was not perturbed by handlers |
2033 |
+ |
assertEquals(savedTaskCount, p.getTaskCount()); |
2034 |
+ |
assertEquals(savedCompletedTaskCount, p.getCompletedTaskCount()); |
2035 |
+ |
assertEquals(savedQueueSize, p.getQueue().size()); |
2036 |
+ |
} |
2037 |
+ |
|
2038 |
+ |
void assertCollectionsEquals(Collection<?> x, Collection<?> y) { |
2039 |
+ |
assertEquals(x, y); |
2040 |
+ |
assertEquals(y, x); |
2041 |
+ |
assertEquals(x.isEmpty(), y.isEmpty()); |
2042 |
+ |
assertEquals(x.size(), y.size()); |
2043 |
+ |
if (x instanceof List) { |
2044 |
+ |
assertEquals(x.toString(), y.toString()); |
2045 |
+ |
} |
2046 |
+ |
if (x instanceof List || x instanceof Set) { |
2047 |
+ |
assertEquals(x.hashCode(), y.hashCode()); |
2048 |
+ |
} |
2049 |
+ |
if (x instanceof List || x instanceof Deque) { |
2050 |
+ |
assertTrue(Arrays.equals(x.toArray(), y.toArray())); |
2051 |
+ |
assertTrue(Arrays.equals(x.toArray(new Object[0]), |
2052 |
+ |
y.toArray(new Object[0]))); |
2053 |
+ |
} |
2054 |
+ |
} |
2055 |
+ |
|
2056 |
+ |
/** |
2057 |
+ |
* A weaker form of assertCollectionsEquals which does not insist |
2058 |
+ |
* that the two collections satisfy Object#equals(Object), since |
2059 |
+ |
* they may use identity semantics as Deques do. |
2060 |
+ |
*/ |
2061 |
+ |
void assertCollectionsEquivalent(Collection<?> x, Collection<?> y) { |
2062 |
+ |
if (x instanceof List || x instanceof Set) |
2063 |
+ |
assertCollectionsEquals(x, y); |
2064 |
+ |
else { |
2065 |
+ |
assertEquals(x.isEmpty(), y.isEmpty()); |
2066 |
+ |
assertEquals(x.size(), y.size()); |
2067 |
+ |
assertEquals(new HashSet(x), new HashSet(y)); |
2068 |
+ |
if (x instanceof Deque) { |
2069 |
+ |
assertTrue(Arrays.equals(x.toArray(), y.toArray())); |
2070 |
+ |
assertTrue(Arrays.equals(x.toArray(new Object[0]), |
2071 |
+ |
y.toArray(new Object[0]))); |
2072 |
+ |
} |
2073 |
+ |
} |
2074 |
+ |
} |
2075 |
|
} |