ViewVC Help

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

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

Comparing jsr166/src/test/tck/ExecutorCompletionServiceTest.java (file contents):
Revision 1.9 by jsr166, Sat Nov 21 02:33:20 2009 UTC vs.
Revision 1.23 by jsr166, Tue May 3 23:06:12 2016 UTC

#	Line 1 | Line 1
1		/*
2		* Written by Doug Lea with assistance from members of JCP JSR-166
3		* Expert Group and released to the public domain, as explained at
4	<	* http://creativecommons.org/licenses/publicdomain
4	>	* http://creativecommons.org/publicdomain/zero/1.0/
5		* Other contributors include Andrew Wright, Jeffrey Hayes,
6		* Pat Fisher, Mike Judd.
7		*/
8
9	–
10	–	import junit.framework.*;
11	–	import java.util.*;
12	–	import java.util.concurrent.*;
9		import static java.util.concurrent.TimeUnit.MILLISECONDS;
10	<	import java.util.concurrent.atomic.*;
11	<	import java.math.BigInteger;
12	<	import java.security.*;
10	>
11	>	import java.util.concurrent.ArrayBlockingQueue;
12	>	import java.util.concurrent.Callable;
13	>	import java.util.concurrent.CompletionService;
14	>	import java.util.concurrent.ExecutorCompletionService;
15	>	import java.util.concurrent.Executors;
16	>	import java.util.concurrent.ExecutorService;
17	>	import java.util.concurrent.Future;
18	>	import java.util.concurrent.FutureTask;
19	>	import java.util.concurrent.RunnableFuture;
20	>	import java.util.concurrent.ThreadPoolExecutor;
21	>	import java.util.concurrent.TimeUnit;
22	>	import java.util.concurrent.atomic.AtomicBoolean;
23	>
24	>	import junit.framework.Test;
25	>	import junit.framework.TestSuite;
26
27		public class ExecutorCompletionServiceTest extends JSR166TestCase {
28		public static void main(String[] args) {
29	<	junit.textui.TestRunner.run (suite());
29	>	main(suite(), args);
30		}
31		public static Test suite() {
32		return new TestSuite(ExecutorCompletionServiceTest.class);
33		}
34
26	–
35		/**
36		* Creating a new ECS with null Executor throw NPE
37		*/
38		public void testConstructorNPE() {
39		try {
40	<	ExecutorCompletionService ecs = new ExecutorCompletionService(null);
40	>	new ExecutorCompletionService(null);
41		shouldThrow();
42	<	} catch (NullPointerException success) {
35	<	}
42	>	} catch (NullPointerException success) {}
43		}
44
45		/**
#	Line 41 | Line 48 | public class ExecutorCompletionServiceTe
48		public void testConstructorNPE2() {
49		try {
50		ExecutorService e = Executors.newCachedThreadPool();
51	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e, null);
51	>	new ExecutorCompletionService(e, null);
52		shouldThrow();
53	<	} catch (NullPointerException success) {
47	<	}
53	>	} catch (NullPointerException success) {}
54		}
55
56		/**
57		* Submitting a null callable throws NPE
58		*/
59		public void testSubmitNPE() {
60	<	ExecutorService e = Executors.newCachedThreadPool();
61	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
62	<	try {
60	>	final ExecutorService e = Executors.newCachedThreadPool();
61	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
62	>	try (PoolCleaner cleaner = cleaner(e)) {
63		Callable c = null;
64	<	ecs.submit(c);
65	<	shouldThrow();
66	<	} catch (NullPointerException success) {
67	<	} finally {
62	<	joinPool(e);
64	>	try {
65	>	ecs.submit(c);
66	>	shouldThrow();
67	>	} catch (NullPointerException success) {}
68		}
69		}
70
#	Line 67 | Line 72 | public class ExecutorCompletionServiceTe
72		* Submitting a null runnable throws NPE
73		*/
74		public void testSubmitNPE2() {
75	<	ExecutorService e = Executors.newCachedThreadPool();
76	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
77	<	try {
75	>	final ExecutorService e = Executors.newCachedThreadPool();
76	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
77	>	try (PoolCleaner cleaner = cleaner(e)) {
78		Runnable r = null;
79	<	ecs.submit(r, Boolean.TRUE);
80	<	shouldThrow();
81	<	} catch (NullPointerException success) {
82	<	} finally {
78	<	joinPool(e);
79	>	try {
80	>	ecs.submit(r, Boolean.TRUE);
81	>	shouldThrow();
82	>	} catch (NullPointerException success) {}
83		}
84		}
85
86		/**
87		* A taken submitted task is completed
88		*/
89	<	public void testTake() {
90	<	ExecutorService e = Executors.newCachedThreadPool();
91	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
92	<	try {
89	>	public void testTake() throws InterruptedException {
90	>	final ExecutorService e = Executors.newCachedThreadPool();
91	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
92	>	try (PoolCleaner cleaner = cleaner(e)) {
93		Callable c = new StringTask();
94		ecs.submit(c);
95		Future f = ecs.take();
96		assertTrue(f.isDone());
93	–	} catch (Exception ex) {
94	–	unexpectedException();
95	–	} finally {
96	–	joinPool(e);
97		}
98		}
99
100		/**
101		* Take returns the same future object returned by submit
102		*/
103	<	public void testTake2() {
104	<	ExecutorService e = Executors.newCachedThreadPool();
105	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
106	<	try {
103	>	public void testTake2() throws InterruptedException {
104	>	final ExecutorService e = Executors.newCachedThreadPool();
105	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
106	>	try (PoolCleaner cleaner = cleaner(e)) {
107		Callable c = new StringTask();
108		Future f1 = ecs.submit(c);
109		Future f2 = ecs.take();
110		assertSame(f1, f2);
111	–	} catch (Exception ex) {
112	–	unexpectedException();
113	–	} finally {
114	–	joinPool(e);
111		}
112		}
113
114		/**
115		* If poll returns non-null, the returned task is completed
116		*/
117	<	public void testPoll1() {
118	<	ExecutorService e = Executors.newCachedThreadPool();
119	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
120	<	try {
117	>	public void testPoll1() throws Exception {
118	>	final ExecutorService e = Executors.newCachedThreadPool();
119	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
120	>	try (PoolCleaner cleaner = cleaner(e)) {
121		assertNull(ecs.poll());
122		Callable c = new StringTask();
123		ecs.submit(c);
124	<	Thread.sleep(SHORT_DELAY_MS);
125	<	for (;;) {
126	<	Future f = ecs.poll();
127	<	if (f != null) {
128	<	assertTrue(f.isDone());
129	<	break;
130	<	}
124	>
125	>	long startTime = System.nanoTime();
126	>	Future f;
127	>	while ((f = ecs.poll()) == null) {
128	>	if (millisElapsedSince(startTime) > LONG_DELAY_MS)
129	>	fail("timed out");
130	>	Thread.yield();
131		}
132	<	} catch (Exception ex) {
133	<	unexpectedException();
138	<	} finally {
139	<	joinPool(e);
132	>	assertTrue(f.isDone());
133	>	assertSame(TEST_STRING, f.get());
134		}
135		}
136
137		/**
138		* If timed poll returns non-null, the returned task is completed
139		*/
140	<	public void testPoll2() {
141	<	ExecutorService e = Executors.newCachedThreadPool();
142	<	ExecutorCompletionService ecs = new ExecutorCompletionService(e);
143	<	try {
140	>	public void testPoll2() throws InterruptedException {
141	>	final ExecutorService e = Executors.newCachedThreadPool();
142	>	final ExecutorCompletionService ecs = new ExecutorCompletionService(e);
143	>	try (PoolCleaner cleaner = cleaner(e)) {
144		assertNull(ecs.poll());
145		Callable c = new StringTask();
146		ecs.submit(c);
147		Future f = ecs.poll(SHORT_DELAY_MS, MILLISECONDS);
148		if (f != null)
149		assertTrue(f.isDone());
150	<	} catch (Exception ex) {
151	<	unexpectedException();
158	<	} finally {
159	<	joinPool(e);
160	<	}
161	<	}
162	<	/**
163	<	* Submitting to underlying AES that overrides newTaskFor(Callable)
164	<	* returns and eventually runs Future returned by newTaskFor.
165	<	*/
166	<	public void testNewTaskForCallable() {
167	<	final AtomicBoolean done = new AtomicBoolean(false);
168	<	class MyCallableFuture<V> extends FutureTask<V> {
169	<	MyCallableFuture(Callable<V> c) { super(c); }
170	<	protected void done() { done.set(true); }
171	<	}
172	<	ExecutorService e = new ThreadPoolExecutor(
173	<	1, 1, 30L, TimeUnit.SECONDS,
174	<	new ArrayBlockingQueue<Runnable>(1)) {
175	<	protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
176	<	return new MyCallableFuture<T>(c);
177	<	}
178	<	};
179	<	ExecutorCompletionService<String> ecs =
180	<	new ExecutorCompletionService<String>(e);
181	<	try {
182	<	assertNull(ecs.poll());
183	<	Callable<String> c = new StringTask();
184	<	Future f1 = ecs.submit(c);
185	<	assertTrue("submit must return MyCallableFuture",
186	<	f1 instanceof MyCallableFuture);
187	<	Future f2 = ecs.take();
188	<	assertSame("submit and take must return same objects", f1, f2);
189	<	assertTrue("completed task must have set done", done.get());
190	<	} catch (Exception ex) {
191	<	unexpectedException();
192	<	} finally {
193	<	joinPool(e);
194	<	}
195	<	}
196	<
197	<	/**
198	<	* Submitting to underlying AES that overrides newTaskFor(Runnable,T)
199	<	* returns and eventually runs Future returned by newTaskFor.
200	<	*/
201	<	public void testNewTaskForRunnable() {
202	<	final AtomicBoolean done = new AtomicBoolean(false);
203	<	class MyRunnableFuture<V> extends FutureTask<V> {
204	<	MyRunnableFuture(Runnable t, V r) { super(t, r); }
205	<	protected void done() { done.set(true); }
206	<	}
207	<	ExecutorService e = new ThreadPoolExecutor(
208	<	1, 1, 30L, TimeUnit.SECONDS,
209	<	new ArrayBlockingQueue<Runnable>(1)) {
210	<	protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
211	<	return new MyRunnableFuture<T>(t, r);
212	<	}
213	<	};
214	<	ExecutorCompletionService<String> ecs =
215	<	new ExecutorCompletionService<String>(e);
216	<	try {
217	<	assertNull(ecs.poll());
218	<	Runnable r = new NoOpRunnable();
219	<	Future f1 = ecs.submit(r, null);
220	<	assertTrue("submit must return MyRunnableFuture",
221	<	f1 instanceof MyRunnableFuture);
222	<	Future f2 = ecs.take();
223	<	assertSame("submit and take must return same objects", f1, f2);
224	<	assertTrue("completed task must have set done", done.get());
225	<	} catch (Exception ex) {
226	<	unexpectedException();
227	<	} finally {
228	<	joinPool(e);
229	<	}
230	<	}
150	>	}
151	>	}
152
153	+	/**
154	+	* Submitting to underlying AES that overrides newTaskFor(Callable)
155	+	* returns and eventually runs Future returned by newTaskFor.
156	+	*/
157	+	public void testNewTaskForCallable() throws InterruptedException {
158	+	final AtomicBoolean done = new AtomicBoolean(false);
159	+	class MyCallableFuture<V> extends FutureTask<V> {
160	+	MyCallableFuture(Callable<V> c) { super(c); }
161	+	protected void done() { done.set(true); }
162	+	}
163	+	final ExecutorService e =
164	+	new ThreadPoolExecutor(1, 1,
165	+	30L, TimeUnit.SECONDS,
166	+	new ArrayBlockingQueue<Runnable>(1)) {
167	+	protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
168	+	return new MyCallableFuture<T>(c);
169	+	}};
170	+	CompletionService<String> ecs =
171	+	new ExecutorCompletionService<>(e);
172	+	try (PoolCleaner cleaner = cleaner(e)) {
173	+	assertNull(ecs.poll());
174	+	Callable<String> c = new StringTask();
175	+	Future f1 = ecs.submit(c);
176	+	assertTrue("submit must return MyCallableFuture",
177	+	f1 instanceof MyCallableFuture);
178	+	Future f2 = ecs.take();
179	+	assertSame("submit and take must return same objects", f1, f2);
180	+	assertTrue("completed task must have set done", done.get());
181	+	}
182	+	}
183
184	+	/**
185	+	* Submitting to underlying AES that overrides newTaskFor(Runnable,T)
186	+	* returns and eventually runs Future returned by newTaskFor.
187	+	*/
188	+	public void testNewTaskForRunnable() throws InterruptedException {
189	+	final AtomicBoolean done = new AtomicBoolean(false);
190	+	class MyRunnableFuture<V> extends FutureTask<V> {
191	+	MyRunnableFuture(Runnable t, V r) { super(t, r); }
192	+	protected void done() { done.set(true); }
193	+	}
194	+	final ExecutorService e =
195	+	new ThreadPoolExecutor(1, 1,
196	+	30L, TimeUnit.SECONDS,
197	+	new ArrayBlockingQueue<Runnable>(1)) {
198	+	protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
199	+	return new MyRunnableFuture<T>(t, r);
200	+	}};
201	+	final CompletionService<String> ecs =
202	+	new ExecutorCompletionService<>(e);
203	+	try (PoolCleaner cleaner = cleaner(e)) {
204	+	assertNull(ecs.poll());
205	+	Runnable r = new NoOpRunnable();
206	+	Future f1 = ecs.submit(r, null);
207	+	assertTrue("submit must return MyRunnableFuture",
208	+	f1 instanceof MyRunnableFuture);
209	+	Future f2 = ecs.take();
210	+	assertSame("submit and take must return same objects", f1, f2);
211	+	assertTrue("completed task must have set done", done.get());
212	+	}
213	+	}
214
215		}

Diff Legend

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