76 |
|
import java.util.concurrent.CountDownLatch; |
77 |
|
import java.util.concurrent.CyclicBarrier; |
78 |
|
import java.util.concurrent.ExecutionException; |
79 |
+ |
import java.util.concurrent.Executor; |
80 |
|
import java.util.concurrent.Executors; |
81 |
|
import java.util.concurrent.ExecutorService; |
82 |
|
import java.util.concurrent.ForkJoinPool; |
83 |
|
import java.util.concurrent.Future; |
84 |
+ |
import java.util.concurrent.FutureTask; |
85 |
|
import java.util.concurrent.RecursiveAction; |
86 |
|
import java.util.concurrent.RecursiveTask; |
87 |
+ |
import java.util.concurrent.RejectedExecutionException; |
88 |
|
import java.util.concurrent.RejectedExecutionHandler; |
89 |
|
import java.util.concurrent.Semaphore; |
90 |
+ |
import java.util.concurrent.ScheduledExecutorService; |
91 |
+ |
import java.util.concurrent.ScheduledFuture; |
92 |
|
import java.util.concurrent.SynchronousQueue; |
93 |
|
import java.util.concurrent.ThreadFactory; |
94 |
|
import java.util.concurrent.ThreadLocalRandom; |
99 |
|
import java.util.concurrent.atomic.AtomicReference; |
100 |
|
import java.util.regex.Pattern; |
101 |
|
|
97 |
– |
import junit.framework.AssertionFailedError; |
102 |
|
import junit.framework.Test; |
103 |
|
import junit.framework.TestCase; |
104 |
|
import junit.framework.TestResult; |
418 |
|
for (String testClassName : testClassNames) { |
419 |
|
try { |
420 |
|
Class<?> testClass = Class.forName(testClassName); |
421 |
< |
Method m = testClass.getDeclaredMethod("suite", |
418 |
< |
new Class<?>[0]); |
421 |
> |
Method m = testClass.getDeclaredMethod("suite"); |
422 |
|
suite.addTest(newTestSuite((Test)m.invoke(null))); |
423 |
< |
} catch (Exception e) { |
424 |
< |
throw new Error("Missing test class", e); |
423 |
> |
} catch (ReflectiveOperationException e) { |
424 |
> |
throw new AssertionError("Missing test class", e); |
425 |
|
} |
426 |
|
} |
427 |
|
} |
443 |
|
} |
444 |
|
} |
445 |
|
|
446 |
< |
public static boolean atLeastJava6() { return JAVA_CLASS_VERSION >= 50.0; } |
447 |
< |
public static boolean atLeastJava7() { return JAVA_CLASS_VERSION >= 51.0; } |
448 |
< |
public static boolean atLeastJava8() { return JAVA_CLASS_VERSION >= 52.0; } |
449 |
< |
public static boolean atLeastJava9() { |
450 |
< |
return JAVA_CLASS_VERSION >= 53.0 |
448 |
< |
// As of 2015-09, java9 still uses 52.0 class file version |
449 |
< |
|| JAVA_SPECIFICATION_VERSION.matches("^(1\\.)?(9|[0-9][0-9])$"); |
450 |
< |
} |
451 |
< |
public static boolean atLeastJava10() { |
452 |
< |
return JAVA_CLASS_VERSION >= 54.0 |
453 |
< |
|| JAVA_SPECIFICATION_VERSION.matches("^(1\\.)?[0-9][0-9]$"); |
454 |
< |
} |
446 |
> |
public static boolean atLeastJava6() { return JAVA_CLASS_VERSION >= 50.0; } |
447 |
> |
public static boolean atLeastJava7() { return JAVA_CLASS_VERSION >= 51.0; } |
448 |
> |
public static boolean atLeastJava8() { return JAVA_CLASS_VERSION >= 52.0; } |
449 |
> |
public static boolean atLeastJava9() { return JAVA_CLASS_VERSION >= 53.0; } |
450 |
> |
public static boolean atLeastJava10() { return JAVA_CLASS_VERSION >= 54.0; } |
451 |
|
|
452 |
|
/** |
453 |
|
* Collects all JSR166 unit tests as one suite. |
535 |
|
"DoubleAdderTest", |
536 |
|
"ForkJoinPool8Test", |
537 |
|
"ForkJoinTask8Test", |
538 |
+ |
"HashMapTest", |
539 |
|
"LinkedBlockingDeque8Test", |
540 |
|
"LinkedBlockingQueue8Test", |
541 |
|
"LongAccumulatorTest", |
598 |
|
for (String methodName : testMethodNames(testClass)) |
599 |
|
suite.addTest((Test) c.newInstance(data, methodName)); |
600 |
|
return suite; |
601 |
< |
} catch (Exception e) { |
602 |
< |
throw new Error(e); |
601 |
> |
} catch (ReflectiveOperationException e) { |
602 |
> |
throw new AssertionError(e); |
603 |
|
} |
604 |
|
} |
605 |
|
|
615 |
|
if (atLeastJava8()) { |
616 |
|
String name = testClass.getName(); |
617 |
|
String name8 = name.replaceAll("Test$", "8Test"); |
618 |
< |
if (name.equals(name8)) throw new Error(name); |
618 |
> |
if (name.equals(name8)) throw new AssertionError(name); |
619 |
|
try { |
620 |
|
return (Test) |
621 |
|
Class.forName(name8) |
622 |
< |
.getMethod("testSuite", new Class[] { dataClass }) |
622 |
> |
.getMethod("testSuite", dataClass) |
623 |
|
.invoke(null, data); |
624 |
< |
} catch (Exception e) { |
625 |
< |
throw new Error(e); |
624 |
> |
} catch (ReflectiveOperationException e) { |
625 |
> |
throw new AssertionError(e); |
626 |
|
} |
627 |
|
} else { |
628 |
|
return new TestSuite(); |
731 |
|
String msg = toString() + ": " + String.format(format, args); |
732 |
|
System.err.println(msg); |
733 |
|
dumpTestThreads(); |
734 |
< |
throw new AssertionFailedError(msg); |
734 |
> |
throw new AssertionError(msg); |
735 |
|
} |
736 |
|
|
737 |
|
/** |
752 |
|
throw (RuntimeException) t; |
753 |
|
else if (t instanceof Exception) |
754 |
|
throw (Exception) t; |
755 |
< |
else { |
756 |
< |
AssertionFailedError afe = |
760 |
< |
new AssertionFailedError(t.toString()); |
761 |
< |
afe.initCause(t); |
762 |
< |
throw afe; |
763 |
< |
} |
755 |
> |
else |
756 |
> |
throw new AssertionError(t.toString(), t); |
757 |
|
} |
758 |
|
|
759 |
|
if (Thread.interrupted()) |
787 |
|
|
788 |
|
/** |
789 |
|
* Just like fail(reason), but additionally recording (using |
790 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
791 |
< |
* the current testcase will fail. |
790 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
791 |
> |
* current testcase will fail. |
792 |
|
*/ |
793 |
|
public void threadFail(String reason) { |
794 |
|
try { |
795 |
|
fail(reason); |
796 |
< |
} catch (AssertionFailedError t) { |
797 |
< |
threadRecordFailure(t); |
798 |
< |
throw t; |
796 |
> |
} catch (AssertionError fail) { |
797 |
> |
threadRecordFailure(fail); |
798 |
> |
throw fail; |
799 |
|
} |
800 |
|
} |
801 |
|
|
802 |
|
/** |
803 |
|
* Just like assertTrue(b), but additionally recording (using |
804 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
805 |
< |
* the current testcase will fail. |
804 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
805 |
> |
* current testcase will fail. |
806 |
|
*/ |
807 |
|
public void threadAssertTrue(boolean b) { |
808 |
|
try { |
809 |
|
assertTrue(b); |
810 |
< |
} catch (AssertionFailedError t) { |
811 |
< |
threadRecordFailure(t); |
812 |
< |
throw t; |
810 |
> |
} catch (AssertionError fail) { |
811 |
> |
threadRecordFailure(fail); |
812 |
> |
throw fail; |
813 |
|
} |
814 |
|
} |
815 |
|
|
816 |
|
/** |
817 |
|
* Just like assertFalse(b), but additionally recording (using |
818 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
819 |
< |
* the current testcase will fail. |
818 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
819 |
> |
* current testcase will fail. |
820 |
|
*/ |
821 |
|
public void threadAssertFalse(boolean b) { |
822 |
|
try { |
823 |
|
assertFalse(b); |
824 |
< |
} catch (AssertionFailedError t) { |
825 |
< |
threadRecordFailure(t); |
826 |
< |
throw t; |
824 |
> |
} catch (AssertionError fail) { |
825 |
> |
threadRecordFailure(fail); |
826 |
> |
throw fail; |
827 |
|
} |
828 |
|
} |
829 |
|
|
830 |
|
/** |
831 |
|
* Just like assertNull(x), but additionally recording (using |
832 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
833 |
< |
* the current testcase will fail. |
832 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
833 |
> |
* current testcase will fail. |
834 |
|
*/ |
835 |
|
public void threadAssertNull(Object x) { |
836 |
|
try { |
837 |
|
assertNull(x); |
838 |
< |
} catch (AssertionFailedError t) { |
839 |
< |
threadRecordFailure(t); |
840 |
< |
throw t; |
838 |
> |
} catch (AssertionError fail) { |
839 |
> |
threadRecordFailure(fail); |
840 |
> |
throw fail; |
841 |
|
} |
842 |
|
} |
843 |
|
|
844 |
|
/** |
845 |
|
* Just like assertEquals(x, y), but additionally recording (using |
846 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
847 |
< |
* the current testcase will fail. |
846 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
847 |
> |
* current testcase will fail. |
848 |
|
*/ |
849 |
|
public void threadAssertEquals(long x, long y) { |
850 |
|
try { |
851 |
|
assertEquals(x, y); |
852 |
< |
} catch (AssertionFailedError t) { |
853 |
< |
threadRecordFailure(t); |
854 |
< |
throw t; |
852 |
> |
} catch (AssertionError fail) { |
853 |
> |
threadRecordFailure(fail); |
854 |
> |
throw fail; |
855 |
|
} |
856 |
|
} |
857 |
|
|
858 |
|
/** |
859 |
|
* Just like assertEquals(x, y), but additionally recording (using |
860 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
861 |
< |
* the current testcase will fail. |
860 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
861 |
> |
* current testcase will fail. |
862 |
|
*/ |
863 |
|
public void threadAssertEquals(Object x, Object y) { |
864 |
|
try { |
865 |
|
assertEquals(x, y); |
866 |
< |
} catch (AssertionFailedError fail) { |
866 |
> |
} catch (AssertionError fail) { |
867 |
|
threadRecordFailure(fail); |
868 |
|
throw fail; |
869 |
|
} catch (Throwable fail) { |
873 |
|
|
874 |
|
/** |
875 |
|
* Just like assertSame(x, y), but additionally recording (using |
876 |
< |
* threadRecordFailure) any AssertionFailedError thrown, so that |
877 |
< |
* the current testcase will fail. |
876 |
> |
* threadRecordFailure) any AssertionError thrown, so that the |
877 |
> |
* current testcase will fail. |
878 |
|
*/ |
879 |
|
public void threadAssertSame(Object x, Object y) { |
880 |
|
try { |
881 |
|
assertSame(x, y); |
882 |
< |
} catch (AssertionFailedError fail) { |
882 |
> |
} catch (AssertionError fail) { |
883 |
|
threadRecordFailure(fail); |
884 |
|
throw fail; |
885 |
|
} |
901 |
|
|
902 |
|
/** |
903 |
|
* Records the given exception using {@link #threadRecordFailure}, |
904 |
< |
* then rethrows the exception, wrapping it in an |
905 |
< |
* AssertionFailedError if necessary. |
904 |
> |
* then rethrows the exception, wrapping it in an AssertionError |
905 |
> |
* if necessary. |
906 |
|
*/ |
907 |
|
public void threadUnexpectedException(Throwable t) { |
908 |
|
threadRecordFailure(t); |
911 |
|
throw (RuntimeException) t; |
912 |
|
else if (t instanceof Error) |
913 |
|
throw (Error) t; |
914 |
< |
else { |
915 |
< |
AssertionFailedError afe = |
923 |
< |
new AssertionFailedError("unexpected exception: " + t); |
924 |
< |
afe.initCause(t); |
925 |
< |
throw afe; |
926 |
< |
} |
914 |
> |
else |
915 |
> |
throw new AssertionError("unexpected exception: " + t, t); |
916 |
|
} |
917 |
|
|
918 |
|
/** |
1088 |
|
for (long retries = LONG_DELAY_MS * 3 / 4; retries-->0; ) { |
1089 |
|
try { delay(1); } |
1090 |
|
catch (InterruptedException fail) { |
1091 |
< |
fail("Unexpected InterruptedException"); |
1091 |
> |
throw new AssertionError("Unexpected InterruptedException", fail); |
1092 |
|
} |
1093 |
|
Thread.State s = thread.getState(); |
1094 |
|
if (s == expected) |
1100 |
|
} |
1101 |
|
|
1102 |
|
/** |
1114 |
– |
* Checks that thread does not terminate within the default |
1115 |
– |
* millisecond delay of {@code timeoutMillis()}. |
1116 |
– |
* TODO: REMOVEME |
1117 |
– |
*/ |
1118 |
– |
void assertThreadStaysAlive(Thread thread) { |
1119 |
– |
assertThreadStaysAlive(thread, timeoutMillis()); |
1120 |
– |
} |
1121 |
– |
|
1122 |
– |
/** |
1123 |
– |
* Checks that thread does not terminate within the given millisecond delay. |
1124 |
– |
* TODO: REMOVEME |
1125 |
– |
*/ |
1126 |
– |
void assertThreadStaysAlive(Thread thread, long millis) { |
1127 |
– |
try { |
1128 |
– |
// No need to optimize the failing case via Thread.join. |
1129 |
– |
delay(millis); |
1130 |
– |
assertTrue(thread.isAlive()); |
1131 |
– |
} catch (InterruptedException fail) { |
1132 |
– |
threadFail("Unexpected InterruptedException"); |
1133 |
– |
} |
1134 |
– |
} |
1135 |
– |
|
1136 |
– |
/** |
1137 |
– |
* Checks that the threads do not terminate within the default |
1138 |
– |
* millisecond delay of {@code timeoutMillis()}. |
1139 |
– |
* TODO: REMOVEME |
1140 |
– |
*/ |
1141 |
– |
void assertThreadsStayAlive(Thread... threads) { |
1142 |
– |
assertThreadsStayAlive(timeoutMillis(), threads); |
1143 |
– |
} |
1144 |
– |
|
1145 |
– |
/** |
1146 |
– |
* Checks that the threads do not terminate within the given millisecond delay. |
1147 |
– |
* TODO: REMOVEME |
1148 |
– |
*/ |
1149 |
– |
void assertThreadsStayAlive(long millis, Thread... threads) { |
1150 |
– |
try { |
1151 |
– |
// No need to optimize the failing case via Thread.join. |
1152 |
– |
delay(millis); |
1153 |
– |
for (Thread thread : threads) |
1154 |
– |
assertTrue(thread.isAlive()); |
1155 |
– |
} catch (InterruptedException fail) { |
1156 |
– |
threadFail("Unexpected InterruptedException"); |
1157 |
– |
} |
1158 |
– |
} |
1159 |
– |
|
1160 |
– |
/** |
1103 |
|
* Checks that future.get times out, with the default timeout of |
1104 |
|
* {@code timeoutMillis()}. |
1105 |
|
*/ |
1274 |
|
|
1275 |
|
/** |
1276 |
|
* Sleeps until the given time has elapsed. |
1277 |
< |
* Throws AssertionFailedError if interrupted. |
1277 |
> |
* Throws AssertionError if interrupted. |
1278 |
|
*/ |
1279 |
|
static void sleep(long millis) { |
1280 |
|
try { |
1281 |
|
delay(millis); |
1282 |
|
} catch (InterruptedException fail) { |
1283 |
< |
AssertionFailedError afe = |
1342 |
< |
new AssertionFailedError("Unexpected InterruptedException"); |
1343 |
< |
afe.initCause(fail); |
1344 |
< |
throw afe; |
1283 |
> |
throw new AssertionError("Unexpected InterruptedException", fail); |
1284 |
|
} |
1285 |
|
} |
1286 |
|
|
1371 |
|
// r.run(); |
1372 |
|
// } catch (Throwable fail) { threadUnexpectedException(fail); } |
1373 |
|
// if (millisElapsedSince(startTime) > timeoutMillis/2) |
1374 |
< |
// throw new AssertionFailedError("did not return promptly"); |
1374 |
> |
// throw new AssertionError("did not return promptly"); |
1375 |
|
// } |
1376 |
|
|
1377 |
|
// void assertTerminatesPromptly(Runnable r) { |
1388 |
|
assertEquals(expectedValue, f.get(timeoutMillis, MILLISECONDS)); |
1389 |
|
} catch (Throwable fail) { threadUnexpectedException(fail); } |
1390 |
|
if (millisElapsedSince(startTime) > timeoutMillis/2) |
1391 |
< |
throw new AssertionFailedError("timed get did not return promptly"); |
1391 |
> |
throw new AssertionError("timed get did not return promptly"); |
1392 |
|
} |
1393 |
|
|
1394 |
|
<T> void checkTimedGet(Future<T> f, T expectedValue) { |
1446 |
|
} |
1447 |
|
} |
1448 |
|
|
1510 |
– |
public abstract class RunnableShouldThrow implements Runnable { |
1511 |
– |
protected abstract void realRun() throws Throwable; |
1512 |
– |
|
1513 |
– |
final Class<?> exceptionClass; |
1514 |
– |
|
1515 |
– |
<T extends Throwable> RunnableShouldThrow(Class<T> exceptionClass) { |
1516 |
– |
this.exceptionClass = exceptionClass; |
1517 |
– |
} |
1518 |
– |
|
1519 |
– |
public final void run() { |
1520 |
– |
try { |
1521 |
– |
realRun(); |
1522 |
– |
threadShouldThrow(exceptionClass.getSimpleName()); |
1523 |
– |
} catch (Throwable t) { |
1524 |
– |
if (! exceptionClass.isInstance(t)) |
1525 |
– |
threadUnexpectedException(t); |
1526 |
– |
} |
1527 |
– |
} |
1528 |
– |
} |
1529 |
– |
|
1449 |
|
public abstract class ThreadShouldThrow extends Thread { |
1450 |
|
protected abstract void realRun() throws Throwable; |
1451 |
|
|
1610 |
|
// long startTime = System.nanoTime(); |
1611 |
|
// while (!flag.get()) { |
1612 |
|
// if (millisElapsedSince(startTime) > timeoutMillis) |
1613 |
< |
// throw new AssertionFailedError("timed out"); |
1613 |
> |
// throw new AssertionError("timed out"); |
1614 |
|
// Thread.yield(); |
1615 |
|
// } |
1616 |
|
// } |
1697 |
|
|
1698 |
|
/** |
1699 |
|
* A CyclicBarrier that uses timed await and fails with |
1700 |
< |
* AssertionFailedErrors instead of throwing checked exceptions. |
1700 |
> |
* AssertionErrors instead of throwing checked exceptions. |
1701 |
|
*/ |
1702 |
|
public static class CheckedBarrier extends CyclicBarrier { |
1703 |
|
public CheckedBarrier(int parties) { super(parties); } |
1706 |
|
try { |
1707 |
|
return super.await(2 * LONG_DELAY_MS, MILLISECONDS); |
1708 |
|
} catch (TimeoutException timedOut) { |
1709 |
< |
throw new AssertionFailedError("timed out"); |
1709 |
> |
throw new AssertionError("timed out"); |
1710 |
|
} catch (Exception fail) { |
1711 |
< |
AssertionFailedError afe = |
1793 |
< |
new AssertionFailedError("Unexpected exception: " + fail); |
1794 |
< |
afe.initCause(fail); |
1795 |
< |
throw afe; |
1711 |
> |
throw new AssertionError("Unexpected exception: " + fail, fail); |
1712 |
|
} |
1713 |
|
} |
1714 |
|
} |
1831 |
|
try { throwingAction.run(); } |
1832 |
|
catch (Throwable t) { |
1833 |
|
threw = true; |
1834 |
< |
if (!expectedExceptionClass.isInstance(t)) { |
1835 |
< |
AssertionFailedError afe = |
1836 |
< |
new AssertionFailedError |
1837 |
< |
("Expected " + expectedExceptionClass.getName() + |
1838 |
< |
", got " + t.getClass().getName()); |
1923 |
< |
afe.initCause(t); |
1924 |
< |
threadUnexpectedException(afe); |
1925 |
< |
} |
1834 |
> |
if (!expectedExceptionClass.isInstance(t)) |
1835 |
> |
throw new AssertionError( |
1836 |
> |
"Expected " + expectedExceptionClass.getName() + |
1837 |
> |
", got " + t.getClass().getName(), |
1838 |
> |
t); |
1839 |
|
} |
1840 |
|
if (!threw) |
1841 |
|
shouldThrow(expectedExceptionClass.getName()); |
1864 |
|
1000L, MILLISECONDS, |
1865 |
|
new SynchronousQueue<Runnable>()); |
1866 |
|
|
1867 |
+ |
static <T> void shuffle(T[] array) { |
1868 |
+ |
Collections.shuffle(Arrays.asList(array), ThreadLocalRandom.current()); |
1869 |
+ |
} |
1870 |
+ |
|
1871 |
+ |
/** |
1872 |
+ |
* Returns the same String as would be returned by {@link |
1873 |
+ |
* Object#toString}, whether or not the given object's class |
1874 |
+ |
* overrides toString(). |
1875 |
+ |
* |
1876 |
+ |
* @see System#identityHashCode |
1877 |
+ |
*/ |
1878 |
+ |
static String identityString(Object x) { |
1879 |
+ |
return x.getClass().getName() |
1880 |
+ |
+ "@" + Integer.toHexString(System.identityHashCode(x)); |
1881 |
+ |
} |
1882 |
+ |
|
1883 |
+ |
// --- Shared assertions for Executor tests --- |
1884 |
+ |
|
1885 |
|
/** |
1886 |
|
* Returns maximum number of tasks that can be submitted to given |
1887 |
|
* pool (with bounded queue) before saturation (when submission |
1892 |
|
return pool.getMaximumPoolSize() + q.size() + q.remainingCapacity(); |
1893 |
|
} |
1894 |
|
|
1895 |
< |
static <T> void shuffle(T[] array) { |
1896 |
< |
Collections.shuffle(Arrays.asList(array), ThreadLocalRandom.current()); |
1895 |
> |
@SuppressWarnings("FutureReturnValueIgnored") |
1896 |
> |
void assertNullTaskSubmissionThrowsNullPointerException(Executor e) { |
1897 |
> |
try { |
1898 |
> |
e.execute((Runnable) null); |
1899 |
> |
shouldThrow(); |
1900 |
> |
} catch (NullPointerException success) {} |
1901 |
> |
|
1902 |
> |
if (! (e instanceof ExecutorService)) return; |
1903 |
> |
ExecutorService es = (ExecutorService) e; |
1904 |
> |
try { |
1905 |
> |
es.submit((Runnable) null); |
1906 |
> |
shouldThrow(); |
1907 |
> |
} catch (NullPointerException success) {} |
1908 |
> |
try { |
1909 |
> |
es.submit((Runnable) null, Boolean.TRUE); |
1910 |
> |
shouldThrow(); |
1911 |
> |
} catch (NullPointerException success) {} |
1912 |
> |
try { |
1913 |
> |
es.submit((Callable) null); |
1914 |
> |
shouldThrow(); |
1915 |
> |
} catch (NullPointerException success) {} |
1916 |
> |
|
1917 |
> |
if (! (e instanceof ScheduledExecutorService)) return; |
1918 |
> |
ScheduledExecutorService ses = (ScheduledExecutorService) e; |
1919 |
> |
try { |
1920 |
> |
ses.schedule((Runnable) null, |
1921 |
> |
randomTimeout(), randomTimeUnit()); |
1922 |
> |
shouldThrow(); |
1923 |
> |
} catch (NullPointerException success) {} |
1924 |
> |
try { |
1925 |
> |
ses.schedule((Callable) null, |
1926 |
> |
randomTimeout(), randomTimeUnit()); |
1927 |
> |
shouldThrow(); |
1928 |
> |
} catch (NullPointerException success) {} |
1929 |
> |
try { |
1930 |
> |
ses.scheduleAtFixedRate((Runnable) null, |
1931 |
> |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1932 |
> |
shouldThrow(); |
1933 |
> |
} catch (NullPointerException success) {} |
1934 |
> |
try { |
1935 |
> |
ses.scheduleWithFixedDelay((Runnable) null, |
1936 |
> |
randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
1937 |
> |
shouldThrow(); |
1938 |
> |
} catch (NullPointerException success) {} |
1939 |
> |
} |
1940 |
> |
|
1941 |
> |
void setRejectedExecutionHandler( |
1942 |
> |
ThreadPoolExecutor p, RejectedExecutionHandler handler) { |
1943 |
> |
p.setRejectedExecutionHandler(handler); |
1944 |
> |
assertSame(handler, p.getRejectedExecutionHandler()); |
1945 |
> |
} |
1946 |
> |
|
1947 |
> |
void assertTaskSubmissionsAreRejected(ThreadPoolExecutor p) { |
1948 |
> |
final RejectedExecutionHandler savedHandler = p.getRejectedExecutionHandler(); |
1949 |
> |
final long savedTaskCount = p.getTaskCount(); |
1950 |
> |
final long savedCompletedTaskCount = p.getCompletedTaskCount(); |
1951 |
> |
final int savedQueueSize = p.getQueue().size(); |
1952 |
> |
final boolean stock = (p.getClass().getClassLoader() == null); |
1953 |
> |
|
1954 |
> |
Runnable r = () -> {}; |
1955 |
> |
Callable<Boolean> c = () -> Boolean.TRUE; |
1956 |
> |
|
1957 |
> |
class Recorder implements RejectedExecutionHandler { |
1958 |
> |
public volatile Runnable r = null; |
1959 |
> |
public volatile ThreadPoolExecutor p = null; |
1960 |
> |
public void reset() { r = null; p = null; } |
1961 |
> |
public void rejectedExecution(Runnable r, ThreadPoolExecutor p) { |
1962 |
> |
assertNull(this.r); |
1963 |
> |
assertNull(this.p); |
1964 |
> |
this.r = r; |
1965 |
> |
this.p = p; |
1966 |
> |
} |
1967 |
> |
} |
1968 |
> |
|
1969 |
> |
// check custom handler is invoked exactly once per task |
1970 |
> |
Recorder recorder = new Recorder(); |
1971 |
> |
setRejectedExecutionHandler(p, recorder); |
1972 |
> |
for (int i = 2; i--> 0; ) { |
1973 |
> |
recorder.reset(); |
1974 |
> |
p.execute(r); |
1975 |
> |
if (stock && p.getClass() == ThreadPoolExecutor.class) |
1976 |
> |
assertSame(r, recorder.r); |
1977 |
> |
assertSame(p, recorder.p); |
1978 |
> |
|
1979 |
> |
recorder.reset(); |
1980 |
> |
assertFalse(p.submit(r).isDone()); |
1981 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1982 |
> |
assertSame(p, recorder.p); |
1983 |
> |
|
1984 |
> |
recorder.reset(); |
1985 |
> |
assertFalse(p.submit(r, Boolean.TRUE).isDone()); |
1986 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1987 |
> |
assertSame(p, recorder.p); |
1988 |
> |
|
1989 |
> |
recorder.reset(); |
1990 |
> |
assertFalse(p.submit(c).isDone()); |
1991 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
1992 |
> |
assertSame(p, recorder.p); |
1993 |
> |
|
1994 |
> |
if (p instanceof ScheduledExecutorService) { |
1995 |
> |
ScheduledExecutorService s = (ScheduledExecutorService) p; |
1996 |
> |
ScheduledFuture<?> future; |
1997 |
> |
|
1998 |
> |
recorder.reset(); |
1999 |
> |
future = s.schedule(r, randomTimeout(), randomTimeUnit()); |
2000 |
> |
assertFalse(future.isDone()); |
2001 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2002 |
> |
assertSame(p, recorder.p); |
2003 |
> |
|
2004 |
> |
recorder.reset(); |
2005 |
> |
future = s.schedule(c, randomTimeout(), randomTimeUnit()); |
2006 |
> |
assertFalse(future.isDone()); |
2007 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2008 |
> |
assertSame(p, recorder.p); |
2009 |
> |
|
2010 |
> |
recorder.reset(); |
2011 |
> |
future = s.scheduleAtFixedRate(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
2012 |
> |
assertFalse(future.isDone()); |
2013 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2014 |
> |
assertSame(p, recorder.p); |
2015 |
> |
|
2016 |
> |
recorder.reset(); |
2017 |
> |
future = s.scheduleWithFixedDelay(r, randomTimeout(), LONG_DELAY_MS, MILLISECONDS); |
2018 |
> |
assertFalse(future.isDone()); |
2019 |
> |
if (stock) assertTrue(!((FutureTask) recorder.r).isDone()); |
2020 |
> |
assertSame(p, recorder.p); |
2021 |
> |
} |
2022 |
> |
} |
2023 |
> |
|
2024 |
> |
// Checking our custom handler above should be sufficient, but |
2025 |
> |
// we add some integration tests of standard handlers. |
2026 |
> |
final AtomicReference<Thread> thread = new AtomicReference<>(); |
2027 |
> |
final Runnable setThread = () -> thread.set(Thread.currentThread()); |
2028 |
> |
|
2029 |
> |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.AbortPolicy()); |
2030 |
> |
try { |
2031 |
> |
p.execute(setThread); |
2032 |
> |
shouldThrow(); |
2033 |
> |
} catch (RejectedExecutionException success) {} |
2034 |
> |
assertNull(thread.get()); |
2035 |
> |
|
2036 |
> |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.DiscardPolicy()); |
2037 |
> |
p.execute(setThread); |
2038 |
> |
assertNull(thread.get()); |
2039 |
> |
|
2040 |
> |
setRejectedExecutionHandler(p, new ThreadPoolExecutor.CallerRunsPolicy()); |
2041 |
> |
p.execute(setThread); |
2042 |
> |
if (p.isShutdown()) |
2043 |
> |
assertNull(thread.get()); |
2044 |
> |
else |
2045 |
> |
assertSame(Thread.currentThread(), thread.get()); |
2046 |
> |
|
2047 |
> |
setRejectedExecutionHandler(p, savedHandler); |
2048 |
> |
|
2049 |
> |
// check that pool was not perturbed by handlers |
2050 |
> |
assertEquals(savedTaskCount, p.getTaskCount()); |
2051 |
> |
assertEquals(savedCompletedTaskCount, p.getCompletedTaskCount()); |
2052 |
> |
assertEquals(savedQueueSize, p.getQueue().size()); |
2053 |
|
} |
2054 |
|
} |