ViewVC Help

View File | Revision Log | Show Annotations | Download File | Root Listing

root/jsr166/jsr166/src/test/tck/ThreadPoolExecutorTest.java

Comparing jsr166/src/test/tck/ThreadPoolExecutorTest.java (file contents):
Revision 1.122 by jsr166, Sat Jul 15 18:42:01 2017 UTC vs.
Revision 1.123 by jsr166, Sat Jul 15 23:15:21 2017 UTC

#	Line 28 | Line 28 | import java.util.concurrent.RejectedExec
28		import java.util.concurrent.RejectedExecutionHandler;
29		import java.util.concurrent.SynchronousQueue;
30		import java.util.concurrent.ThreadFactory;
31	+	import java.util.concurrent.ThreadLocalRandom;
32		import java.util.concurrent.ThreadPoolExecutor;
33		import java.util.concurrent.ThreadPoolExecutor.AbortPolicy;
34		import java.util.concurrent.ThreadPoolExecutor.CallerRunsPolicy;
#	Line 1065 | Line 1066 | public class ThreadPoolExecutorTest exte
1066		}
1067
1068		/**
1069	<	* execute throws RejectedExecutionException if saturated.
1069	>	* Submitted tasks are rejected when saturated.
1070		*/
1071	<	public void testSaturatedExecute() {
1071	>	@SuppressWarnings("FutureReturnValueIgnored")
1072	>	public void testSubmittedTasksRejectedWhenSaturated() {
1073	>	final ThreadLocalRandom rnd = ThreadLocalRandom.current();
1074		final CountDownLatch done = new CountDownLatch(1);
1075	<	final ThreadPoolExecutor p =
1076	<	new ThreadPoolExecutor(1, 1,
1077	<	LONG_DELAY_MS, MILLISECONDS,
1078	<	new ArrayBlockingQueue<Runnable>(1));
1075	>	final Runnable r = awaiter(done);
1076	>	final Callable<Boolean> c = new CheckedCallable() {
1077	>	public Boolean realCall() throws InterruptedException {
1078	>	await(done);
1079	>	return Boolean.TRUE;
1080	>	}};
1081	>	final ThreadPoolExecutor p = new ThreadPoolExecutor(
1082	>	1, 1, 1, SECONDS, new ArrayBlockingQueue<Runnable>(1));
1083	>
1084		try (PoolCleaner cleaner = cleaner(p, done)) {
1085	<	Runnable task = new CheckedRunnable() {
1086	<	public void realRun() throws InterruptedException {
1087	<	await(done);
1088	<	}};
1089	<	for (int i = 0; i < 2; ++i)
1090	<	p.execute(task);
1091	<	for (int i = 0; i < 2; ++i) {
1085	>	// saturate
1086	>	for (int i = saturatedSize(p); i--> 0; ) {
1087	>	switch (rnd.nextInt(3)) {
1088	>	case 0: p.execute(r); break;
1089	>	case 1: assertFalse(p.submit(r).isDone()); break;
1090	>	case 2: assertFalse(p.submit(c).isDone()); break;
1091	>	}
1092	>	}
1093	>
1094	>	// check default handler
1095	>	assertTrue(p.getRejectedExecutionHandler() instanceof AbortPolicy);
1096	>	for (int i = 2; i--> 0; ) {
1097		try {
1098	<	p.execute(task);
1098	>	p.execute(r);
1099		shouldThrow();
1100		} catch (RejectedExecutionException success) {}
1088	–	assertTrue(p.getTaskCount() <= 2);
1089	–	}
1090	–	}
1091	–	}
1092	–
1093	–	/**
1094	–	* submit(runnable) throws RejectedExecutionException if saturated.
1095	–	*/
1096	–	public void testSaturatedSubmitRunnable() {
1097	–	final CountDownLatch done = new CountDownLatch(1);
1098	–	final ThreadPoolExecutor p =
1099	–	new ThreadPoolExecutor(1, 1,
1100	–	LONG_DELAY_MS, MILLISECONDS,
1101	–	new ArrayBlockingQueue<Runnable>(1));
1102	–	try (PoolCleaner cleaner = cleaner(p, done)) {
1103	–	Runnable task = new CheckedRunnable() {
1104	–	public void realRun() throws InterruptedException {
1105	–	await(done);
1106	–	}};
1107	–	for (int i = 0; i < 2; ++i)
1108	–	p.submit(task);
1109	–	for (int i = 0; i < 2; ++i) {
1101		try {
1102	<	p.execute(task);
1102	>	p.submit(r);
1103		shouldThrow();
1104		} catch (RejectedExecutionException success) {}
1114	–	assertTrue(p.getTaskCount() <= 2);
1115	–	}
1116	–	}
1117	–	}
1118	–
1119	–	/**
1120	–	* submit(callable) throws RejectedExecutionException if saturated.
1121	–	*/
1122	–	public void testSaturatedSubmitCallable() {
1123	–	final CountDownLatch done = new CountDownLatch(1);
1124	–	final ThreadPoolExecutor p =
1125	–	new ThreadPoolExecutor(1, 1,
1126	–	LONG_DELAY_MS, MILLISECONDS,
1127	–	new ArrayBlockingQueue<Runnable>(1));
1128	–	try (PoolCleaner cleaner = cleaner(p, done)) {
1129	–	Runnable task = new CheckedRunnable() {
1130	–	public void realRun() throws InterruptedException {
1131	–	await(done);
1132	–	}};
1133	–	for (int i = 0; i < 2; ++i)
1134	–	p.execute(task);
1135	–	for (int i = 0; i < 2; ++i) {
1105		try {
1106	<	p.execute(task);
1106	>	p.submit(c);
1107		shouldThrow();
1108		} catch (RejectedExecutionException success) {}
1140	–	assertTrue(p.getTaskCount() <= 2);
1109		}
1110	+
1111	+	// check CallerRunsPolicy runs task in caller thread
1112	+	{
1113	+	RejectedExecutionHandler handler = new CallerRunsPolicy();
1114	+	p.setRejectedExecutionHandler(handler);
1115	+	assertSame(handler, p.getRejectedExecutionHandler());
1116	+	final AtomicReference<Thread> thread = new AtomicReference<>();
1117	+	p.execute(new Runnable() { public void run() {
1118	+	thread.set(Thread.currentThread()); }});
1119	+	assertSame(Thread.currentThread(), thread.get());
1120	+	}
1121	+
1122	+	// check DiscardPolicy does nothing
1123	+	{
1124	+	RejectedExecutionHandler handler = new DiscardPolicy();
1125	+	p.setRejectedExecutionHandler(handler);
1126	+	assertSame(handler, p.getRejectedExecutionHandler());
1127	+	final AtomicReference<Thread> thread = new AtomicReference<>();
1128	+	p.execute(new Runnable() { public void run() {
1129	+	thread.set(Thread.currentThread()); }});
1130	+	assertNull(thread.get());
1131	+	}
1132	+
1133	+	class Recorder implements RejectedExecutionHandler {
1134	+	public volatile Runnable r = null;
1135	+	public volatile ThreadPoolExecutor p = null;
1136	+	public void reset() { r = null; p = null; }
1137	+	public void rejectedExecution(Runnable r, ThreadPoolExecutor p) {
1138	+	assertNull(this.r);
1139	+	assertNull(this.p);
1140	+	this.r = r;
1141	+	this.p = p;
1142	+	}
1143	+	}
1144	+
1145	+	// check custom handler is invoked exactly once per task
1146	+	Recorder recorder = new Recorder();
1147	+	p.setRejectedExecutionHandler(recorder);
1148	+	assertSame(recorder, p.getRejectedExecutionHandler());
1149	+	for (int i = 2; i--> 0; ) {
1150	+	recorder.reset();
1151	+	p.execute(r);
1152	+	assertSame(r, recorder.r);
1153	+	assertSame(p, recorder.p);
1154	+
1155	+	recorder.reset();
1156	+	assertFalse(p.submit(r).isDone());
1157	+	assertTrue(recorder.r instanceof FutureTask);
1158	+	assertSame(p, recorder.p);
1159	+
1160	+	recorder.reset();
1161	+	assertFalse(p.submit(c).isDone());
1162	+	assertTrue(recorder.r instanceof FutureTask);
1163	+	assertSame(p, recorder.p);
1164	+	}
1165	+
1166	+	// check that pool was not perturbed by handlers
1167	+	assertEquals(2, p.getTaskCount());
1168	+	assertEquals(0, p.getCompletedTaskCount());
1169	+	assertEquals(0, p.getQueue().remainingCapacity());
1170		}
1171	+	assertEquals(saturatedSize(p), p.getCompletedTaskCount());
1172		}
1173
1174		/**
1175		* executor using CallerRunsPolicy runs task if saturated.
1176		*/
1177		public void testSaturatedExecute2() {
1178	<	final ThreadPoolExecutor p =
1179	<	new ThreadPoolExecutor(1, 1,
1180	<	LONG_DELAY_MS,
1181	<	MILLISECONDS,
1182	<	new ArrayBlockingQueue<Runnable>(1),
1183	<	new CallerRunsPolicy());
1184	<	try (PoolCleaner cleaner = cleaner(p)) {
1185	<	final CountDownLatch done = new CountDownLatch(1);
1186	<	Runnable blocker = new CheckedRunnable() {
1187	<	public void realRun() throws InterruptedException {
1159	<	await(done);
1160	<	}};
1161	<	p.execute(blocker);
1162	<	TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1163	<	for (int i = 0; i < tasks.length; i++)
1164	<	tasks[i] = new TrackedNoOpRunnable();
1178	>	final RejectedExecutionHandler handler = new CallerRunsPolicy();
1179	>	final ThreadPoolExecutor p = new ThreadPoolExecutor(
1180	>	1, 1, LONG_DELAY_MS, SECONDS, new ArrayBlockingQueue<Runnable>(1),
1181	>	handler);
1182	>	assertSame(handler, p.getRejectedExecutionHandler());
1183	>	final TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1184	>	final CountDownLatch done = new CountDownLatch(1);
1185	>	try (PoolCleaner cleaner = cleaner(p, done)) {
1186	>	p.execute(awaiter(done));
1187	>
1188		for (int i = 0; i < tasks.length; i++)
1189	<	p.execute(tasks[i]);
1189	>	p.execute(tasks[i] = new TrackedNoOpRunnable());
1190	>
1191		for (int i = 1; i < tasks.length; i++)
1192		assertTrue(tasks[i].done);
1193		assertFalse(tasks[0].done); // waiting in queue
1170	–	done.countDown();
1194		}
1195	+	for (TrackedNoOpRunnable task : tasks)
1196	+	assertTrue(task.done);
1197		}
1198
1199		/**
1200		* executor using DiscardPolicy drops task if saturated.
1201		*/
1202		public void testSaturatedExecute3() {
1203	<	final CountDownLatch done = new CountDownLatch(1);
1203	>	final RejectedExecutionHandler handler = new DiscardPolicy();
1204	>	final ThreadPoolExecutor p = new ThreadPoolExecutor(
1205	>	1, 1, LONG_DELAY_MS, SECONDS, new ArrayBlockingQueue<Runnable>(1),
1206	>	handler);
1207	>	assertSame(handler, p.getRejectedExecutionHandler());
1208		final TrackedNoOpRunnable[] tasks = new TrackedNoOpRunnable[5];
1209	<	for (int i = 0; i < tasks.length; ++i)
1181	<	tasks[i] = new TrackedNoOpRunnable();
1182	<	final ThreadPoolExecutor p =
1183	<	new ThreadPoolExecutor(1, 1,
1184	<	LONG_DELAY_MS, MILLISECONDS,
1185	<	new ArrayBlockingQueue<Runnable>(1),
1186	<	new DiscardPolicy());
1209	>	final CountDownLatch done = new CountDownLatch(1);
1210		try (PoolCleaner cleaner = cleaner(p, done)) {
1211		p.execute(awaiter(done));
1212
1213	<	for (TrackedNoOpRunnable task : tasks)
1214	<	p.execute(task);
1213	>	for (int i = 0; i < tasks.length; i++)
1214	>	p.execute(tasks[i] = new TrackedNoOpRunnable());
1215	>
1216		for (int i = 1; i < tasks.length; i++)
1217		assertFalse(tasks[i].done);
1218		}

Diff Legend

–	Removed lines
+	Added lines
<	Changed lines
>	Changed lines