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.math.BigInteger; |
11 |
< |
import java.security.*; |
10 |
> |
|
11 |
> |
import java.security.PrivilegedAction; |
12 |
> |
import java.security.PrivilegedExceptionAction; |
13 |
> |
import java.util.ArrayList; |
14 |
> |
import java.util.Collections; |
15 |
> |
import java.util.List; |
16 |
> |
import java.util.concurrent.AbstractExecutorService; |
17 |
> |
import java.util.concurrent.ArrayBlockingQueue; |
18 |
> |
import java.util.concurrent.Callable; |
19 |
> |
import java.util.concurrent.CountDownLatch; |
20 |
> |
import java.util.concurrent.ExecutionException; |
21 |
> |
import java.util.concurrent.Executors; |
22 |
> |
import java.util.concurrent.ExecutorService; |
23 |
> |
import java.util.concurrent.Future; |
24 |
> |
import java.util.concurrent.ThreadPoolExecutor; |
25 |
> |
import java.util.concurrent.TimeUnit; |
26 |
> |
import java.util.concurrent.atomic.AtomicBoolean; |
27 |
> |
|
28 |
> |
import junit.framework.Test; |
29 |
> |
import junit.framework.TestSuite; |
30 |
|
|
31 |
|
public class AbstractExecutorServiceTest extends JSR166TestCase { |
32 |
|
public static void main(String[] args) { |
43 |
|
static class DirectExecutorService extends AbstractExecutorService { |
44 |
|
public void execute(Runnable r) { r.run(); } |
45 |
|
public void shutdown() { shutdown = true; } |
46 |
< |
public List<Runnable> shutdownNow() { shutdown = true; return Collections.EMPTY_LIST; } |
46 |
> |
public List<Runnable> shutdownNow() { |
47 |
> |
shutdown = true; |
48 |
> |
return Collections.EMPTY_LIST; |
49 |
> |
} |
50 |
|
public boolean isShutdown() { return shutdown; } |
51 |
|
public boolean isTerminated() { return isShutdown(); } |
52 |
< |
public boolean awaitTermination(long timeout, TimeUnit unit) { return isShutdown(); } |
52 |
> |
public boolean awaitTermination(long timeout, TimeUnit unit) { |
53 |
> |
return isShutdown(); |
54 |
> |
} |
55 |
|
private volatile boolean shutdown = false; |
56 |
|
} |
57 |
|
|
60 |
|
*/ |
61 |
|
public void testExecuteRunnable() throws Exception { |
62 |
|
ExecutorService e = new DirectExecutorService(); |
63 |
< |
TrackedShortRunnable task = new TrackedShortRunnable(); |
64 |
< |
assertFalse(task.done); |
65 |
< |
Future<?> future = e.submit(task); |
66 |
< |
future.get(); |
67 |
< |
assertTrue(task.done); |
63 |
> |
final AtomicBoolean done = new AtomicBoolean(false); |
64 |
> |
Future<?> future = e.submit(new CheckedRunnable() { |
65 |
> |
public void realRun() { |
66 |
> |
done.set(true); |
67 |
> |
}}); |
68 |
> |
assertNull(future.get()); |
69 |
> |
assertNull(future.get(0, MILLISECONDS)); |
70 |
> |
assertTrue(done.get()); |
71 |
> |
assertTrue(future.isDone()); |
72 |
> |
assertFalse(future.isCancelled()); |
73 |
|
} |
74 |
|
|
51 |
– |
|
75 |
|
/** |
76 |
|
* Completed submit(callable) returns result |
77 |
|
*/ |
102 |
|
assertSame(TEST_STRING, result); |
103 |
|
} |
104 |
|
|
82 |
– |
|
105 |
|
/** |
106 |
|
* A submitted privileged action runs to completion |
107 |
|
*/ |
167 |
|
* execute(null runnable) throws NPE |
168 |
|
*/ |
169 |
|
public void testExecuteNullRunnable() { |
170 |
+ |
ExecutorService e = new DirectExecutorService(); |
171 |
|
try { |
149 |
– |
ExecutorService e = new DirectExecutorService(); |
172 |
|
e.submit((Runnable) null); |
173 |
|
shouldThrow(); |
174 |
|
} catch (NullPointerException success) {} |
175 |
|
} |
176 |
|
|
155 |
– |
|
177 |
|
/** |
178 |
|
* submit(null callable) throws NPE |
179 |
|
*/ |
180 |
|
public void testSubmitNullCallable() { |
181 |
+ |
ExecutorService e = new DirectExecutorService(); |
182 |
|
try { |
161 |
– |
ExecutorService e = new DirectExecutorService(); |
183 |
|
e.submit((Callable) null); |
184 |
|
shouldThrow(); |
185 |
|
} catch (NullPointerException success) {} |
186 |
|
} |
187 |
|
|
188 |
|
/** |
168 |
– |
* submit(runnable) throws RejectedExecutionException if |
169 |
– |
* executor is saturated. |
170 |
– |
*/ |
171 |
– |
public void testExecute1() { |
172 |
– |
ThreadPoolExecutor p = |
173 |
– |
new ThreadPoolExecutor(1, 1, |
174 |
– |
60, TimeUnit.SECONDS, |
175 |
– |
new ArrayBlockingQueue<Runnable>(1)); |
176 |
– |
try { |
177 |
– |
for (int i = 0; i < 2; ++i) |
178 |
– |
p.submit(new MediumRunnable()); |
179 |
– |
for (int i = 0; i < 2; ++i) { |
180 |
– |
try { |
181 |
– |
p.submit(new MediumRunnable()); |
182 |
– |
shouldThrow(); |
183 |
– |
} catch (RejectedExecutionException success) {} |
184 |
– |
} |
185 |
– |
} finally { |
186 |
– |
joinPool(p); |
187 |
– |
} |
188 |
– |
} |
189 |
– |
|
190 |
– |
/** |
191 |
– |
* submit(callable) throws RejectedExecutionException |
192 |
– |
* if executor is saturated. |
193 |
– |
*/ |
194 |
– |
public void testExecute2() { |
195 |
– |
ThreadPoolExecutor p = |
196 |
– |
new ThreadPoolExecutor(1, 1, |
197 |
– |
60, TimeUnit.SECONDS, |
198 |
– |
new ArrayBlockingQueue<Runnable>(1)); |
199 |
– |
try { |
200 |
– |
for (int i = 0; i < 2; ++i) |
201 |
– |
p.submit(new MediumRunnable()); |
202 |
– |
for (int i = 0; i < 2; ++i) { |
203 |
– |
try { |
204 |
– |
p.submit(new SmallCallable()); |
205 |
– |
shouldThrow(); |
206 |
– |
} catch (RejectedExecutionException success) {} |
207 |
– |
} |
208 |
– |
} finally { |
209 |
– |
joinPool(p); |
210 |
– |
} |
211 |
– |
} |
212 |
– |
|
213 |
– |
|
214 |
– |
/** |
189 |
|
* submit(callable).get() throws InterruptedException if interrupted |
190 |
|
*/ |
191 |
|
public void testInterruptedSubmit() throws InterruptedException { |
217 |
|
} |
218 |
|
|
219 |
|
/** |
220 |
< |
* get of submitted callable throws InterruptedException if callable |
221 |
< |
* interrupted |
248 |
< |
*/ |
249 |
< |
public void testSubmitIE() throws InterruptedException { |
250 |
< |
final ThreadPoolExecutor p = |
251 |
< |
new ThreadPoolExecutor(1, 1, |
252 |
< |
60, TimeUnit.SECONDS, |
253 |
< |
new ArrayBlockingQueue<Runnable>(10)); |
254 |
< |
|
255 |
< |
Thread t = new Thread(new CheckedInterruptedRunnable() { |
256 |
< |
public void realRun() throws Exception { |
257 |
< |
p.submit(new SmallCallable()).get(); |
258 |
< |
}}); |
259 |
< |
|
260 |
< |
t.start(); |
261 |
< |
Thread.sleep(SHORT_DELAY_MS); |
262 |
< |
t.interrupt(); |
263 |
< |
t.join(); |
264 |
< |
joinPool(p); |
265 |
< |
} |
266 |
< |
|
267 |
< |
/** |
268 |
< |
* get of submit(callable) throws ExecutionException if callable |
269 |
< |
* throws exception |
220 |
> |
* get of submit(callable) throws ExecutionException if callable |
221 |
> |
* throws exception |
222 |
|
*/ |
223 |
|
public void testSubmitEE() throws InterruptedException { |
224 |
|
ThreadPoolExecutor p = |
227 |
|
new ArrayBlockingQueue<Runnable>(10)); |
228 |
|
|
229 |
|
Callable c = new Callable() { |
230 |
< |
public Object call() { return 5/0; }}; |
230 |
> |
public Object call() { throw new ArithmeticException(); }}; |
231 |
|
|
232 |
|
try { |
233 |
|
p.submit(c).get(); |
241 |
|
/** |
242 |
|
* invokeAny(null) throws NPE |
243 |
|
*/ |
244 |
< |
public void testInvokeAny1() |
293 |
< |
throws InterruptedException, ExecutionException { |
244 |
> |
public void testInvokeAny1() throws Exception { |
245 |
|
ExecutorService e = new DirectExecutorService(); |
246 |
|
try { |
247 |
|
e.invokeAny(null); |
255 |
|
/** |
256 |
|
* invokeAny(empty collection) throws IAE |
257 |
|
*/ |
258 |
< |
public void testInvokeAny2() |
308 |
< |
throws InterruptedException, ExecutionException { |
258 |
> |
public void testInvokeAny2() throws Exception { |
259 |
|
ExecutorService e = new DirectExecutorService(); |
260 |
|
try { |
261 |
|
e.invokeAny(new ArrayList<Callable<String>>()); |
271 |
|
*/ |
272 |
|
public void testInvokeAny3() throws Exception { |
273 |
|
ExecutorService e = new DirectExecutorService(); |
274 |
< |
List<Callable<Integer>> l = new ArrayList<Callable<Integer>>(); |
275 |
< |
l.add(new Callable<Integer>() { |
276 |
< |
public Integer call() { return 5/0; }}); |
274 |
> |
List<Callable<Long>> l = new ArrayList<Callable<Long>>(); |
275 |
> |
l.add(new Callable<Long>() { |
276 |
> |
public Long call() { throw new ArithmeticException(); }}); |
277 |
|
l.add(null); |
278 |
|
try { |
279 |
|
e.invokeAny(l); |
400 |
|
} |
401 |
|
} |
402 |
|
|
453 |
– |
|
403 |
|
/** |
404 |
|
* timed invokeAny(null) throws NPE |
405 |
|
*/ |
449 |
|
*/ |
450 |
|
public void testTimedInvokeAny3() throws Exception { |
451 |
|
ExecutorService e = new DirectExecutorService(); |
452 |
< |
List<Callable<Integer>> l = new ArrayList<Callable<Integer>>(); |
453 |
< |
l.add(new Callable<Integer>() { |
454 |
< |
public Integer call() { return 5/0; }}); |
452 |
> |
List<Callable<Long>> l = new ArrayList<Callable<Long>>(); |
453 |
> |
l.add(new Callable<Long>() { |
454 |
> |
public Long call() { throw new ArithmeticException(); }}); |
455 |
|
l.add(null); |
456 |
|
try { |
457 |
|
e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS); |
604 |
|
try { |
605 |
|
List<Callable<String>> l = new ArrayList<Callable<String>>(); |
606 |
|
l.add(new StringTask()); |
607 |
< |
l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING)); |
607 |
> |
l.add(Executors.callable(possiblyInterruptedRunnable(2 * SHORT_DELAY_MS), TEST_STRING)); |
608 |
|
l.add(new StringTask()); |
609 |
|
List<Future<String>> futures = |
610 |
< |
e.invokeAll(l, SMALL_DELAY_MS, MILLISECONDS); |
611 |
< |
assertEquals(3, futures.size()); |
612 |
< |
Iterator<Future<String>> it = futures.iterator(); |
613 |
< |
Future<String> f1 = it.next(); |
614 |
< |
Future<String> f2 = it.next(); |
615 |
< |
Future<String> f3 = it.next(); |
616 |
< |
assertTrue(f1.isDone()); |
668 |
< |
assertFalse(f1.isCancelled()); |
669 |
< |
assertTrue(f2.isDone()); |
670 |
< |
assertTrue(f3.isDone()); |
671 |
< |
assertTrue(f3.isCancelled()); |
610 |
> |
e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS); |
611 |
> |
assertEquals(l.size(), futures.size()); |
612 |
> |
for (Future future : futures) |
613 |
> |
assertTrue(future.isDone()); |
614 |
> |
assertFalse(futures.get(0).isCancelled()); |
615 |
> |
assertFalse(futures.get(1).isCancelled()); |
616 |
> |
assertTrue(futures.get(2).isCancelled()); |
617 |
|
} finally { |
618 |
|
joinPool(e); |
619 |
|
} |