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

Diff Legend

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