ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/ScheduledExecutorTest.java
Revision: 1.90
Committed: Tue Mar 28 23:21:24 2017 UTC (7 years, 1 month ago) by jsr166
Branch: MAIN
Changes since 1.89: +35 -16 lines
Log Message:
testShutdown_cancellation: ensure periodic tasks continue executing if policy set

File Contents

# User Rev Content
1 dl 1.1 /*
2 dl 1.13 * 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 jsr166 1.37 * http://creativecommons.org/publicdomain/zero/1.0/
5 jsr166 1.22 * Other contributors include Andrew Wright, Jeffrey Hayes,
6     * Pat Fisher, Mike Judd.
7 dl 1.1 */
8    
9 jsr166 1.25 import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 jsr166 1.80 import static java.util.concurrent.TimeUnit.NANOSECONDS;
11 jsr166 1.54 import static java.util.concurrent.TimeUnit.SECONDS;
12 jsr166 1.50
13     import java.util.ArrayList;
14 jsr166 1.54 import java.util.HashSet;
15 jsr166 1.50 import java.util.List;
16     import java.util.concurrent.BlockingQueue;
17     import java.util.concurrent.Callable;
18 jsr166 1.52 import java.util.concurrent.CancellationException;
19 jsr166 1.50 import java.util.concurrent.CountDownLatch;
20     import java.util.concurrent.ExecutionException;
21     import java.util.concurrent.ExecutorService;
22     import java.util.concurrent.Future;
23     import java.util.concurrent.RejectedExecutionException;
24     import java.util.concurrent.ScheduledFuture;
25     import java.util.concurrent.ScheduledThreadPoolExecutor;
26     import java.util.concurrent.ThreadFactory;
27 jsr166 1.88 import java.util.concurrent.ThreadLocalRandom;
28 jsr166 1.50 import java.util.concurrent.ThreadPoolExecutor;
29 jsr166 1.80 import java.util.concurrent.atomic.AtomicBoolean;
30 jsr166 1.46 import java.util.concurrent.atomic.AtomicInteger;
31 jsr166 1.80 import java.util.concurrent.atomic.AtomicLong;
32 jsr166 1.89 import java.util.stream.Stream;
33 dl 1.1
34 jsr166 1.50 import junit.framework.Test;
35     import junit.framework.TestSuite;
36    
37 dl 1.4 public class ScheduledExecutorTest extends JSR166TestCase {
38 dl 1.1 public static void main(String[] args) {
39 jsr166 1.51 main(suite(), args);
40 dl 1.1 }
41     public static Test suite() {
42 jsr166 1.26 return new TestSuite(ScheduledExecutorTest.class);
43 dl 1.1 }
44    
45 dl 1.5 /**
46 dl 1.6 * execute successfully executes a runnable
47 dl 1.5 */
48 jsr166 1.25 public void testExecute() throws InterruptedException {
49 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
50 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
51     final CountDownLatch done = new CountDownLatch(1);
52     final Runnable task = new CheckedRunnable() {
53     public void realRun() { done.countDown(); }};
54 jsr166 1.33 p.execute(task);
55 jsr166 1.86 await(done);
56 jsr166 1.33 }
57 dl 1.1 }
58    
59 dl 1.5 /**
60 dl 1.6 * delayed schedule of callable successfully executes after delay
61 dl 1.5 */
62 jsr166 1.25 public void testSchedule1() throws Exception {
63 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
64 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
65     final long startTime = System.nanoTime();
66     final CountDownLatch done = new CountDownLatch(1);
67 jsr166 1.33 Callable task = new CheckedCallable<Boolean>() {
68     public Boolean realCall() {
69     done.countDown();
70 jsr166 1.44 assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
71 jsr166 1.33 return Boolean.TRUE;
72     }};
73 jsr166 1.44 Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
74     assertSame(Boolean.TRUE, f.get());
75     assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
76 jsr166 1.33 assertTrue(done.await(0L, MILLISECONDS));
77     }
78 dl 1.1 }
79    
80     /**
81 jsr166 1.32 * delayed schedule of runnable successfully executes after delay
82 dl 1.1 */
83 jsr166 1.33 public void testSchedule3() throws Exception {
84 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
85 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
86     final long startTime = System.nanoTime();
87     final CountDownLatch done = new CountDownLatch(1);
88 jsr166 1.33 Runnable task = new CheckedRunnable() {
89     public void realRun() {
90     done.countDown();
91 jsr166 1.44 assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
92 jsr166 1.33 }};
93 jsr166 1.44 Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
94     await(done);
95     assertNull(f.get(LONG_DELAY_MS, MILLISECONDS));
96     assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
97 jsr166 1.33 }
98 dl 1.1 }
99 jsr166 1.22
100 dl 1.1 /**
101 dl 1.6 * scheduleAtFixedRate executes runnable after given initial delay
102 dl 1.1 */
103 jsr166 1.33 public void testSchedule4() throws Exception {
104 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
105 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
106     final long startTime = System.nanoTime();
107     final CountDownLatch done = new CountDownLatch(1);
108 jsr166 1.33 Runnable task = new CheckedRunnable() {
109     public void realRun() {
110     done.countDown();
111 jsr166 1.44 assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
112 jsr166 1.33 }};
113     ScheduledFuture f =
114 jsr166 1.44 p.scheduleAtFixedRate(task, timeoutMillis(),
115     LONG_DELAY_MS, MILLISECONDS);
116     await(done);
117     assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
118 jsr166 1.33 f.cancel(true);
119     }
120 dl 1.18 }
121    
122 dl 1.6 /**
123     * scheduleWithFixedDelay executes runnable after given initial delay
124     */
125 jsr166 1.44 public void testSchedule5() throws Exception {
126 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
127 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
128     final long startTime = System.nanoTime();
129     final CountDownLatch done = new CountDownLatch(1);
130 jsr166 1.33 Runnable task = new CheckedRunnable() {
131     public void realRun() {
132     done.countDown();
133 jsr166 1.44 assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
134 jsr166 1.33 }};
135     ScheduledFuture f =
136 jsr166 1.44 p.scheduleWithFixedDelay(task, timeoutMillis(),
137     LONG_DELAY_MS, MILLISECONDS);
138     await(done);
139     assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
140 jsr166 1.33 f.cancel(true);
141     }
142     }
143    
144     static class RunnableCounter implements Runnable {
145     AtomicInteger count = new AtomicInteger(0);
146     public void run() { count.getAndIncrement(); }
147 dl 1.1 }
148 jsr166 1.22
149 dl 1.6 /**
150 jsr166 1.80 * scheduleAtFixedRate executes series of tasks at given rate.
151     * Eventually, it must hold that:
152     * cycles - 1 <= elapsedMillis/delay < cycles
153 dl 1.18 */
154 jsr166 1.25 public void testFixedRateSequence() throws InterruptedException {
155 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
156 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
157 jsr166 1.47 for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
158 jsr166 1.80 final long startTime = System.nanoTime();
159     final int cycles = 8;
160 jsr166 1.47 final CountDownLatch done = new CountDownLatch(cycles);
161 jsr166 1.80 final Runnable task = new CheckedRunnable() {
162 jsr166 1.47 public void realRun() { done.countDown(); }};
163 jsr166 1.80 final ScheduledFuture periodicTask =
164 jsr166 1.47 p.scheduleAtFixedRate(task, 0, delay, MILLISECONDS);
165 jsr166 1.80 final int totalDelayMillis = (cycles - 1) * delay;
166     await(done, totalDelayMillis + LONG_DELAY_MS);
167 jsr166 1.79 periodicTask.cancel(true);
168 jsr166 1.80 final long elapsedMillis = millisElapsedSince(startTime);
169     assertTrue(elapsedMillis >= totalDelayMillis);
170     if (elapsedMillis <= cycles * delay)
171 jsr166 1.47 return;
172 jsr166 1.80 // else retry with longer delay
173 jsr166 1.47 }
174 jsr166 1.78 fail("unexpected execution rate");
175 jsr166 1.47 }
176 dl 1.18 }
177    
178     /**
179 jsr166 1.80 * scheduleWithFixedDelay executes series of tasks with given period.
180     * Eventually, it must hold that each task starts at least delay and at
181     * most 2 * delay after the termination of the previous task.
182 dl 1.18 */
183 jsr166 1.25 public void testFixedDelaySequence() throws InterruptedException {
184 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
185 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
186 jsr166 1.47 for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
187 jsr166 1.80 final long startTime = System.nanoTime();
188     final AtomicLong previous = new AtomicLong(startTime);
189     final AtomicBoolean tryLongerDelay = new AtomicBoolean(false);
190     final int cycles = 8;
191 jsr166 1.47 final CountDownLatch done = new CountDownLatch(cycles);
192 jsr166 1.80 final int d = delay;
193     final Runnable task = new CheckedRunnable() {
194     public void realRun() {
195     long now = System.nanoTime();
196     long elapsedMillis
197     = NANOSECONDS.toMillis(now - previous.get());
198     if (done.getCount() == cycles) { // first execution
199     if (elapsedMillis >= d)
200     tryLongerDelay.set(true);
201     } else {
202     assertTrue(elapsedMillis >= d);
203     if (elapsedMillis >= 2 * d)
204     tryLongerDelay.set(true);
205     }
206     previous.set(now);
207     done.countDown();
208     }};
209     final ScheduledFuture periodicTask =
210 jsr166 1.47 p.scheduleWithFixedDelay(task, 0, delay, MILLISECONDS);
211 jsr166 1.80 final int totalDelayMillis = (cycles - 1) * delay;
212     await(done, totalDelayMillis + cycles * LONG_DELAY_MS);
213 jsr166 1.79 periodicTask.cancel(true);
214 jsr166 1.80 final long elapsedMillis = millisElapsedSince(startTime);
215     assertTrue(elapsedMillis >= totalDelayMillis);
216     if (!tryLongerDelay.get())
217 jsr166 1.47 return;
218 jsr166 1.80 // else retry with longer delay
219 jsr166 1.47 }
220 jsr166 1.78 fail("unexpected execution rate");
221 jsr166 1.47 }
222 dl 1.18 }
223    
224     /**
225 jsr166 1.32 * execute(null) throws NPE
226 dl 1.6 */
227 jsr166 1.25 public void testExecuteNull() throws InterruptedException {
228 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
229 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
230     try {
231     p.execute(null);
232     shouldThrow();
233     } catch (NullPointerException success) {}
234     }
235 dl 1.6 }
236    
237     /**
238 jsr166 1.30 * schedule(null) throws NPE
239 dl 1.6 */
240 jsr166 1.25 public void testScheduleNull() throws InterruptedException {
241 jsr166 1.62 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
242     try (PoolCleaner cleaner = cleaner(p)) {
243     try {
244     TrackedCallable callable = null;
245     Future f = p.schedule(callable, SHORT_DELAY_MS, MILLISECONDS);
246     shouldThrow();
247     } catch (NullPointerException success) {}
248     }
249 dl 1.6 }
250 jsr166 1.22
251 dl 1.1 /**
252 dl 1.6 * execute throws RejectedExecutionException if shutdown
253 dl 1.1 */
254 jsr166 1.25 public void testSchedule1_RejectedExecutionException() throws InterruptedException {
255 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
256 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
257     try {
258     p.shutdown();
259     p.schedule(new NoOpRunnable(),
260     MEDIUM_DELAY_MS, MILLISECONDS);
261     shouldThrow();
262     } catch (RejectedExecutionException success) {
263     } catch (SecurityException ok) {}
264 dl 1.4 }
265 dl 1.1 }
266    
267     /**
268 dl 1.6 * schedule throws RejectedExecutionException if shutdown
269 dl 1.1 */
270 jsr166 1.25 public void testSchedule2_RejectedExecutionException() throws InterruptedException {
271 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
272 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
273     try {
274     p.shutdown();
275     p.schedule(new NoOpCallable(),
276     MEDIUM_DELAY_MS, MILLISECONDS);
277     shouldThrow();
278     } catch (RejectedExecutionException success) {
279     } catch (SecurityException ok) {}
280 dl 1.4 }
281 dl 1.1 }
282    
283     /**
284 dl 1.6 * schedule callable throws RejectedExecutionException if shutdown
285 dl 1.1 */
286 jsr166 1.38 public void testSchedule3_RejectedExecutionException() throws InterruptedException {
287 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
288 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
289     try {
290     p.shutdown();
291     p.schedule(new NoOpCallable(),
292     MEDIUM_DELAY_MS, MILLISECONDS);
293     shouldThrow();
294     } catch (RejectedExecutionException success) {
295     } catch (SecurityException ok) {}
296 dl 1.17 }
297 dl 1.1 }
298    
299     /**
300 jsr166 1.32 * scheduleAtFixedRate throws RejectedExecutionException if shutdown
301 dl 1.1 */
302 jsr166 1.25 public void testScheduleAtFixedRate1_RejectedExecutionException() throws InterruptedException {
303 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
304 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
305     try {
306     p.shutdown();
307     p.scheduleAtFixedRate(new NoOpRunnable(),
308     MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
309     shouldThrow();
310     } catch (RejectedExecutionException success) {
311     } catch (SecurityException ok) {}
312 jsr166 1.22 }
313 dl 1.1 }
314 jsr166 1.22
315 dl 1.1 /**
316 dl 1.6 * scheduleWithFixedDelay throws RejectedExecutionException if shutdown
317 dl 1.1 */
318 jsr166 1.25 public void testScheduleWithFixedDelay1_RejectedExecutionException() throws InterruptedException {
319 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
320 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
321     try {
322     p.shutdown();
323     p.scheduleWithFixedDelay(new NoOpRunnable(),
324     MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
325     shouldThrow();
326     } catch (RejectedExecutionException success) {
327     } catch (SecurityException ok) {}
328 jsr166 1.22 }
329 dl 1.1 }
330    
331     /**
332 jsr166 1.32 * getActiveCount increases but doesn't overestimate, when a
333     * thread becomes active
334 dl 1.2 */
335 jsr166 1.25 public void testGetActiveCount() throws InterruptedException {
336 jsr166 1.73 final CountDownLatch done = new CountDownLatch(1);
337 jsr166 1.33 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
338 jsr166 1.73 try (PoolCleaner cleaner = cleaner(p, done)) {
339 jsr166 1.62 final CountDownLatch threadStarted = new CountDownLatch(1);
340 jsr166 1.33 assertEquals(0, p.getActiveCount());
341     p.execute(new CheckedRunnable() {
342     public void realRun() throws InterruptedException {
343     threadStarted.countDown();
344     assertEquals(1, p.getActiveCount());
345 jsr166 1.70 await(done);
346 jsr166 1.33 }});
347 jsr166 1.72 await(threadStarted);
348 jsr166 1.33 assertEquals(1, p.getActiveCount());
349     }
350 dl 1.2 }
351 jsr166 1.22
352 dl 1.2 /**
353 jsr166 1.32 * getCompletedTaskCount increases, but doesn't overestimate,
354     * when tasks complete
355 dl 1.2 */
356 jsr166 1.25 public void testGetCompletedTaskCount() throws InterruptedException {
357 jsr166 1.33 final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
358 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
359     final CountDownLatch threadStarted = new CountDownLatch(1);
360     final CountDownLatch threadProceed = new CountDownLatch(1);
361     final CountDownLatch threadDone = new CountDownLatch(1);
362 jsr166 1.33 assertEquals(0, p.getCompletedTaskCount());
363     p.execute(new CheckedRunnable() {
364     public void realRun() throws InterruptedException {
365     threadStarted.countDown();
366     assertEquals(0, p.getCompletedTaskCount());
367 jsr166 1.86 await(threadProceed);
368 jsr166 1.33 threadDone.countDown();
369     }});
370 jsr166 1.45 await(threadStarted);
371 jsr166 1.33 assertEquals(0, p.getCompletedTaskCount());
372     threadProceed.countDown();
373 jsr166 1.86 await(threadDone);
374 jsr166 1.44 long startTime = System.nanoTime();
375     while (p.getCompletedTaskCount() != 1) {
376     if (millisElapsedSince(startTime) > LONG_DELAY_MS)
377     fail("timed out");
378     Thread.yield();
379     }
380 jsr166 1.33 }
381 dl 1.2 }
382 jsr166 1.22
383 dl 1.2 /**
384 jsr166 1.32 * getCorePoolSize returns size given in constructor if not otherwise set
385 dl 1.2 */
386 jsr166 1.25 public void testGetCorePoolSize() throws InterruptedException {
387 jsr166 1.33 ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
388 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
389     assertEquals(1, p.getCorePoolSize());
390     }
391 dl 1.2 }
392 jsr166 1.22
393 dl 1.2 /**
394 jsr166 1.32 * getLargestPoolSize increases, but doesn't overestimate, when
395     * multiple threads active
396 dl 1.2 */
397 jsr166 1.25 public void testGetLargestPoolSize() throws InterruptedException {
398 jsr166 1.33 final int THREADS = 3;
399     final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(THREADS);
400     final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
401     final CountDownLatch done = new CountDownLatch(1);
402 jsr166 1.71 try (PoolCleaner cleaner = cleaner(p, done)) {
403 jsr166 1.33 assertEquals(0, p.getLargestPoolSize());
404     for (int i = 0; i < THREADS; i++)
405     p.execute(new CheckedRunnable() {
406     public void realRun() throws InterruptedException {
407     threadsStarted.countDown();
408 jsr166 1.70 await(done);
409 jsr166 1.33 assertEquals(THREADS, p.getLargestPoolSize());
410     }});
411 jsr166 1.71 await(threadsStarted);
412 jsr166 1.33 assertEquals(THREADS, p.getLargestPoolSize());
413     }
414 jsr166 1.62 assertEquals(THREADS, p.getLargestPoolSize());
415 dl 1.2 }
416 jsr166 1.22
417 dl 1.2 /**
418 jsr166 1.32 * getPoolSize increases, but doesn't overestimate, when threads
419     * become active
420 dl 1.2 */
421 jsr166 1.25 public void testGetPoolSize() throws InterruptedException {
422 jsr166 1.33 final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
423     final CountDownLatch threadStarted = new CountDownLatch(1);
424     final CountDownLatch done = new CountDownLatch(1);
425 jsr166 1.71 try (PoolCleaner cleaner = cleaner(p, done)) {
426 jsr166 1.33 assertEquals(0, p.getPoolSize());
427     p.execute(new CheckedRunnable() {
428     public void realRun() throws InterruptedException {
429     threadStarted.countDown();
430     assertEquals(1, p.getPoolSize());
431 jsr166 1.70 await(done);
432 jsr166 1.33 }});
433 jsr166 1.71 await(threadStarted);
434 jsr166 1.33 assertEquals(1, p.getPoolSize());
435     }
436 dl 1.2 }
437 jsr166 1.22
438 dl 1.2 /**
439 jsr166 1.32 * getTaskCount increases, but doesn't overestimate, when tasks
440     * submitted
441 dl 1.2 */
442 jsr166 1.25 public void testGetTaskCount() throws InterruptedException {
443 jsr166 1.65 final int TASKS = 3;
444     final CountDownLatch done = new CountDownLatch(1);
445 jsr166 1.33 final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
446 jsr166 1.65 try (PoolCleaner cleaner = cleaner(p, done)) {
447 jsr166 1.62 final CountDownLatch threadStarted = new CountDownLatch(1);
448 jsr166 1.33 assertEquals(0, p.getTaskCount());
449 jsr166 1.65 assertEquals(0, p.getCompletedTaskCount());
450     p.execute(new CheckedRunnable() {
451     public void realRun() throws InterruptedException {
452     threadStarted.countDown();
453 jsr166 1.70 await(done);
454 jsr166 1.65 }});
455 jsr166 1.72 await(threadStarted);
456 jsr166 1.65 assertEquals(1, p.getTaskCount());
457     assertEquals(0, p.getCompletedTaskCount());
458     for (int i = 0; i < TASKS; i++) {
459     assertEquals(1 + i, p.getTaskCount());
460 jsr166 1.33 p.execute(new CheckedRunnable() {
461     public void realRun() throws InterruptedException {
462     threadStarted.countDown();
463 jsr166 1.65 assertEquals(1 + TASKS, p.getTaskCount());
464 jsr166 1.70 await(done);
465 jsr166 1.33 }});
466 jsr166 1.65 }
467     assertEquals(1 + TASKS, p.getTaskCount());
468     assertEquals(0, p.getCompletedTaskCount());
469 jsr166 1.33 }
470 jsr166 1.65 assertEquals(1 + TASKS, p.getTaskCount());
471     assertEquals(1 + TASKS, p.getCompletedTaskCount());
472 dl 1.2 }
473 dl 1.8
474 jsr166 1.22 /**
475 dl 1.8 * getThreadFactory returns factory in constructor if not set
476     */
477 jsr166 1.25 public void testGetThreadFactory() throws InterruptedException {
478 jsr166 1.63 final ThreadFactory threadFactory = new SimpleThreadFactory();
479     final ScheduledThreadPoolExecutor p =
480     new ScheduledThreadPoolExecutor(1, threadFactory);
481     try (PoolCleaner cleaner = cleaner(p)) {
482     assertSame(threadFactory, p.getThreadFactory());
483     }
484 dl 1.8 }
485    
486 jsr166 1.22 /**
487 dl 1.8 * setThreadFactory sets the thread factory returned by getThreadFactory
488     */
489 jsr166 1.25 public void testSetThreadFactory() throws InterruptedException {
490 jsr166 1.62 ThreadFactory threadFactory = new SimpleThreadFactory();
491 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
492 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
493     p.setThreadFactory(threadFactory);
494     assertSame(threadFactory, p.getThreadFactory());
495     }
496 dl 1.8 }
497    
498 jsr166 1.22 /**
499 dl 1.8 * setThreadFactory(null) throws NPE
500     */
501 jsr166 1.25 public void testSetThreadFactoryNull() throws InterruptedException {
502 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
503 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
504     try {
505     p.setThreadFactory(null);
506     shouldThrow();
507     } catch (NullPointerException success) {}
508 dl 1.8 }
509     }
510 jsr166 1.22
511 dl 1.2 /**
512 jsr166 1.85 * The default rejected execution handler is AbortPolicy.
513     */
514     public void testDefaultRejectedExecutionHandler() {
515     final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
516     try (PoolCleaner cleaner = cleaner(p)) {
517     assertTrue(p.getRejectedExecutionHandler()
518     instanceof ThreadPoolExecutor.AbortPolicy);
519     }
520     }
521    
522     /**
523 jsr166 1.41 * isShutdown is false before shutdown, true after
524 dl 1.2 */
525 dl 1.5 public void testIsShutdown() {
526 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
527 jsr166 1.82 assertFalse(p.isShutdown());
528     try (PoolCleaner cleaner = cleaner(p)) {
529     try {
530     p.shutdown();
531     assertTrue(p.isShutdown());
532     } catch (SecurityException ok) {}
533 dl 1.2 }
534     }
535    
536     /**
537 jsr166 1.32 * isTerminated is false before termination, true after
538 dl 1.2 */
539 jsr166 1.25 public void testIsTerminated() throws InterruptedException {
540 jsr166 1.33 final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
541 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
542     final CountDownLatch threadStarted = new CountDownLatch(1);
543     final CountDownLatch done = new CountDownLatch(1);
544     assertFalse(p.isTerminated());
545 jsr166 1.33 p.execute(new CheckedRunnable() {
546     public void realRun() throws InterruptedException {
547 jsr166 1.36 assertFalse(p.isTerminated());
548 jsr166 1.33 threadStarted.countDown();
549 jsr166 1.70 await(done);
550 jsr166 1.33 }});
551 jsr166 1.72 await(threadStarted);
552 jsr166 1.36 assertFalse(p.isTerminating());
553 jsr166 1.33 done.countDown();
554     try { p.shutdown(); } catch (SecurityException ok) { return; }
555 jsr166 1.62 assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
556     assertTrue(p.isTerminated());
557 dl 1.2 }
558 dl 1.5 }
559    
560     /**
561 jsr166 1.32 * isTerminating is not true when running or when terminated
562 dl 1.5 */
563 jsr166 1.25 public void testIsTerminating() throws InterruptedException {
564 jsr166 1.33 final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
565     final CountDownLatch threadStarted = new CountDownLatch(1);
566     final CountDownLatch done = new CountDownLatch(1);
567 jsr166 1.62 try (PoolCleaner cleaner = cleaner(p)) {
568 jsr166 1.33 assertFalse(p.isTerminating());
569     p.execute(new CheckedRunnable() {
570     public void realRun() throws InterruptedException {
571 jsr166 1.35 assertFalse(p.isTerminating());
572 jsr166 1.33 threadStarted.countDown();
573 jsr166 1.70 await(done);
574 jsr166 1.33 }});
575 jsr166 1.72 await(threadStarted);
576 jsr166 1.33 assertFalse(p.isTerminating());
577     done.countDown();
578     try { p.shutdown(); } catch (SecurityException ok) { return; }
579 jsr166 1.62 assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
580     assertTrue(p.isTerminated());
581     assertFalse(p.isTerminating());
582 jsr166 1.33 }
583 dl 1.2 }
584    
585     /**
586 dl 1.8 * getQueue returns the work queue, which contains queued tasks
587     */
588 jsr166 1.25 public void testGetQueue() throws InterruptedException {
589 jsr166 1.73 final CountDownLatch done = new CountDownLatch(1);
590     final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
591     try (PoolCleaner cleaner = cleaner(p, done)) {
592 jsr166 1.62 final CountDownLatch threadStarted = new CountDownLatch(1);
593 jsr166 1.33 ScheduledFuture[] tasks = new ScheduledFuture[5];
594     for (int i = 0; i < tasks.length; i++) {
595     Runnable r = new CheckedRunnable() {
596     public void realRun() throws InterruptedException {
597     threadStarted.countDown();
598 jsr166 1.70 await(done);
599 jsr166 1.33 }};
600     tasks[i] = p.schedule(r, 1, MILLISECONDS);
601     }
602 jsr166 1.72 await(threadStarted);
603 jsr166 1.33 BlockingQueue<Runnable> q = p.getQueue();
604     assertTrue(q.contains(tasks[tasks.length - 1]));
605 dl 1.8 assertFalse(q.contains(tasks[0]));
606     }
607     }
608    
609     /**
610     * remove(task) removes queued task, and fails to remove active task
611     */
612 jsr166 1.25 public void testRemove() throws InterruptedException {
613 jsr166 1.73 final CountDownLatch done = new CountDownLatch(1);
614 jsr166 1.33 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
615 jsr166 1.73 try (PoolCleaner cleaner = cleaner(p, done)) {
616 jsr166 1.62 ScheduledFuture[] tasks = new ScheduledFuture[5];
617     final CountDownLatch threadStarted = new CountDownLatch(1);
618 jsr166 1.33 for (int i = 0; i < tasks.length; i++) {
619     Runnable r = new CheckedRunnable() {
620     public void realRun() throws InterruptedException {
621     threadStarted.countDown();
622 jsr166 1.70 await(done);
623 jsr166 1.33 }};
624     tasks[i] = p.schedule(r, 1, MILLISECONDS);
625     }
626 jsr166 1.72 await(threadStarted);
627 jsr166 1.33 BlockingQueue<Runnable> q = p.getQueue();
628     assertFalse(p.remove((Runnable)tasks[0]));
629 dl 1.8 assertTrue(q.contains((Runnable)tasks[4]));
630     assertTrue(q.contains((Runnable)tasks[3]));
631 jsr166 1.33 assertTrue(p.remove((Runnable)tasks[4]));
632     assertFalse(p.remove((Runnable)tasks[4]));
633 dl 1.8 assertFalse(q.contains((Runnable)tasks[4]));
634     assertTrue(q.contains((Runnable)tasks[3]));
635 jsr166 1.33 assertTrue(p.remove((Runnable)tasks[3]));
636 dl 1.8 assertFalse(q.contains((Runnable)tasks[3]));
637     }
638     }
639    
640     /**
641 jsr166 1.40 * purge eventually removes cancelled tasks from the queue
642 dl 1.2 */
643 jsr166 1.25 public void testPurge() throws InterruptedException {
644 jsr166 1.66 final ScheduledFuture[] tasks = new ScheduledFuture[5];
645     final Runnable releaser = new Runnable() { public void run() {
646     for (ScheduledFuture task : tasks)
647     if (task != null) task.cancel(true); }};
648     final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
649     try (PoolCleaner cleaner = cleaner(p, releaser)) {
650     for (int i = 0; i < tasks.length; i++)
651     tasks[i] = p.schedule(new SmallPossiblyInterruptedRunnable(),
652     LONG_DELAY_MS, MILLISECONDS);
653 jsr166 1.33 int max = tasks.length;
654 dl 1.19 if (tasks[4].cancel(true)) --max;
655     if (tasks[3].cancel(true)) --max;
656     // There must eventually be an interference-free point at
657     // which purge will not fail. (At worst, when queue is empty.)
658 jsr166 1.40 long startTime = System.nanoTime();
659     do {
660 jsr166 1.33 p.purge();
661     long count = p.getTaskCount();
662 jsr166 1.40 if (count == max)
663     return;
664 jsr166 1.66 } while (millisElapsedSince(startTime) < LONG_DELAY_MS);
665 jsr166 1.40 fail("Purge failed to remove cancelled tasks");
666 dl 1.2 }
667     }
668    
669     /**
670 jsr166 1.53 * shutdownNow returns a list containing tasks that were not run,
671     * and those tasks are drained from the queue
672 dl 1.2 */
673 jsr166 1.56 public void testShutdownNow() throws InterruptedException {
674     final int poolSize = 2;
675     final int count = 5;
676     final AtomicInteger ran = new AtomicInteger(0);
677     final ScheduledThreadPoolExecutor p =
678     new ScheduledThreadPoolExecutor(poolSize);
679 jsr166 1.64 final CountDownLatch threadsStarted = new CountDownLatch(poolSize);
680 jsr166 1.58 Runnable waiter = new CheckedRunnable() { public void realRun() {
681 jsr166 1.56 threadsStarted.countDown();
682     try {
683     MILLISECONDS.sleep(2 * LONG_DELAY_MS);
684     } catch (InterruptedException success) {}
685     ran.getAndIncrement();
686     }};
687     for (int i = 0; i < count; i++)
688     p.execute(waiter);
689 jsr166 1.71 await(threadsStarted);
690 jsr166 1.57 assertEquals(poolSize, p.getActiveCount());
691     assertEquals(0, p.getCompletedTaskCount());
692 jsr166 1.56 final List<Runnable> queuedTasks;
693     try {
694     queuedTasks = p.shutdownNow();
695     } catch (SecurityException ok) {
696     return; // Allowed in case test doesn't have privs
697     }
698     assertTrue(p.isShutdown());
699     assertTrue(p.getQueue().isEmpty());
700     assertEquals(count - poolSize, queuedTasks.size());
701     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
702     assertTrue(p.isTerminated());
703     assertEquals(poolSize, ran.get());
704 jsr166 1.57 assertEquals(poolSize, p.getCompletedTaskCount());
705 jsr166 1.56 }
706    
707     /**
708     * shutdownNow returns a list containing tasks that were not run,
709     * and those tasks are drained from the queue
710     */
711 jsr166 1.54 public void testShutdownNow_delayedTasks() throws InterruptedException {
712 jsr166 1.73 final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
713 jsr166 1.54 List<ScheduledFuture> tasks = new ArrayList<>();
714     for (int i = 0; i < 3; i++) {
715     Runnable r = new NoOpRunnable();
716     tasks.add(p.schedule(r, 9, SECONDS));
717     tasks.add(p.scheduleAtFixedRate(r, 9, 9, SECONDS));
718     tasks.add(p.scheduleWithFixedDelay(r, 9, 9, SECONDS));
719     }
720 jsr166 1.55 if (testImplementationDetails)
721     assertEquals(new HashSet(tasks), new HashSet(p.getQueue()));
722 jsr166 1.54 final List<Runnable> queuedTasks;
723 dl 1.17 try {
724 jsr166 1.54 queuedTasks = p.shutdownNow();
725 jsr166 1.22 } catch (SecurityException ok) {
726 jsr166 1.54 return; // Allowed in case test doesn't have privs
727     }
728     assertTrue(p.isShutdown());
729     assertTrue(p.getQueue().isEmpty());
730 jsr166 1.55 if (testImplementationDetails)
731     assertEquals(new HashSet(tasks), new HashSet(queuedTasks));
732 jsr166 1.54 assertEquals(tasks.size(), queuedTasks.size());
733     for (ScheduledFuture task : tasks) {
734     assertFalse(task.isDone());
735     assertFalse(task.isCancelled());
736 dl 1.17 }
737 jsr166 1.54 assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
738     assertTrue(p.isTerminated());
739 dl 1.2 }
740    
741 dl 1.5 /**
742 jsr166 1.58 * By default, periodic tasks are cancelled at shutdown.
743     * By default, delayed tasks keep running after shutdown.
744     * Check that changing the default values work:
745     * - setExecuteExistingDelayedTasksAfterShutdownPolicy
746     * - setContinueExistingPeriodicTasksAfterShutdownPolicy
747     */
748     public void testShutdown_cancellation() throws Exception {
749 jsr166 1.89 final int poolSize = 6;
750 jsr166 1.58 final ScheduledThreadPoolExecutor p
751     = new ScheduledThreadPoolExecutor(poolSize);
752 jsr166 1.89 final BlockingQueue<Runnable> q = p.getQueue();
753 jsr166 1.88 final ThreadLocalRandom rnd = ThreadLocalRandom.current();
754     final boolean effectiveDelayedPolicy;
755     final boolean effectivePeriodicPolicy;
756     final boolean effectiveRemovePolicy;
757    
758     if (rnd.nextBoolean())
759     p.setExecuteExistingDelayedTasksAfterShutdownPolicy(
760     effectiveDelayedPolicy = rnd.nextBoolean());
761     else
762     effectiveDelayedPolicy = true;
763 jsr166 1.58 assertEquals(effectiveDelayedPolicy,
764     p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
765 jsr166 1.88
766     if (rnd.nextBoolean())
767     p.setContinueExistingPeriodicTasksAfterShutdownPolicy(
768     effectivePeriodicPolicy = rnd.nextBoolean());
769     else
770     effectivePeriodicPolicy = false;
771 jsr166 1.58 assertEquals(effectivePeriodicPolicy,
772     p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
773 jsr166 1.88
774     if (rnd.nextBoolean())
775     p.setRemoveOnCancelPolicy(
776     effectiveRemovePolicy = rnd.nextBoolean());
777     else
778     effectiveRemovePolicy = false;
779 jsr166 1.58 assertEquals(effectiveRemovePolicy,
780     p.getRemoveOnCancelPolicy());
781 jsr166 1.88
782 jsr166 1.90 final boolean periodicTasksContinue = effectivePeriodicPolicy && rnd.nextBoolean();
783 jsr166 1.89
784     // Strategy: Wedge the pool with one wave of "blocker" tasks,
785 jsr166 1.90 // then add a second wave that waits in the queue until unblocked.
786 jsr166 1.58 final AtomicInteger ran = new AtomicInteger(0);
787     final CountDownLatch poolBlocked = new CountDownLatch(poolSize);
788     final CountDownLatch unblock = new CountDownLatch(1);
789 jsr166 1.90 final RuntimeException exception = new RuntimeException();
790 jsr166 1.89
791 jsr166 1.90 class Task implements Runnable {
792     public void run() {
793     try {
794     ran.getAndIncrement();
795     poolBlocked.countDown();
796     await(unblock);
797     } catch (Throwable fail) { threadUnexpectedException(fail); }
798 jsr166 1.89 }
799     }
800    
801     class PeriodicTask extends Task {
802     PeriodicTask(int rounds) { this.rounds = rounds; }
803     int rounds;
804 jsr166 1.90 public void run() {
805     if (--rounds == 0) super.run();
806     // throw exception to surely terminate this periodic task,
807     // but in a separate execution and in a detectable way.
808     if (rounds == -1) throw exception;
809 jsr166 1.89 }
810     }
811    
812     Runnable task = new Task();
813    
814     List<Future<?>> immediates = new ArrayList<>();
815     List<Future<?>> delayeds = new ArrayList<>();
816     List<Future<?>> periodics = new ArrayList<>();
817    
818     immediates.add(p.submit(task));
819     delayeds.add(p.schedule(task, 1, MILLISECONDS));
820     for (int rounds : new int[] { 1, 2 }) {
821     periodics.add(p.scheduleAtFixedRate(
822     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
823     periodics.add(p.scheduleWithFixedDelay(
824     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
825     }
826    
827 jsr166 1.86 await(poolBlocked);
828 jsr166 1.58
829 jsr166 1.89 assertEquals(poolSize, ran.get());
830     assertTrue(q.isEmpty());
831    
832     // Add second wave of tasks.
833     immediates.add(p.submit(task));
834     long delay_ms = effectiveDelayedPolicy ? 1 : LONG_DELAY_MS;
835     delayeds.add(p.schedule(task, delay_ms, MILLISECONDS));
836     for (int rounds : new int[] { 1, 2 }) {
837     periodics.add(p.scheduleAtFixedRate(
838     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
839     periodics.add(p.scheduleWithFixedDelay(
840     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
841     }
842    
843     assertEquals(poolSize, q.size());
844     assertEquals(poolSize, ran.get());
845    
846     immediates.forEach(
847     f -> assertTrue(((ScheduledFuture)f).getDelay(NANOSECONDS) <= 0L));
848    
849     Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
850     .forEach(f -> assertFalse(f.isDone()));
851 jsr166 1.22
852 jsr166 1.33 try { p.shutdown(); } catch (SecurityException ok) { return; }
853 jsr166 1.58 assertTrue(p.isShutdown());
854 jsr166 1.89 assertTrue(p.isTerminating());
855 jsr166 1.58 assertFalse(p.isTerminated());
856 jsr166 1.89
857     if (rnd.nextBoolean())
858     assertThrows(
859     RejectedExecutionException.class,
860     () -> p.submit(task),
861     () -> p.schedule(task, 1, SECONDS),
862     () -> p.scheduleAtFixedRate(
863     new PeriodicTask(1), 1, 1, SECONDS),
864     () -> p.scheduleWithFixedDelay(
865     new PeriodicTask(2), 1, 1, SECONDS));
866    
867     assertTrue(q.contains(immediates.get(1)));
868     assertTrue(!effectiveDelayedPolicy
869     ^ q.contains(delayeds.get(1)));
870     assertTrue(!effectivePeriodicPolicy
871     ^ q.containsAll(periodics.subList(4, 8)));
872    
873     immediates.forEach(f -> assertFalse(f.isDone()));
874    
875     assertFalse(delayeds.get(0).isDone());
876     if (effectiveDelayedPolicy)
877     assertFalse(delayeds.get(1).isDone());
878     else
879     assertTrue(delayeds.get(1).isCancelled());
880    
881 jsr166 1.58 if (testImplementationDetails) {
882 jsr166 1.89 if (effectivePeriodicPolicy)
883     periodics.forEach(
884     f -> {
885     assertFalse(f.isDone());
886 jsr166 1.90 if (!periodicTasksContinue)
887     assertTrue(f.cancel(false));
888 jsr166 1.89 });
889     else {
890     periodics.subList(0, 4).forEach(f -> assertFalse(f.isDone()));
891     periodics.subList(4, 8).forEach(f -> assertTrue(f.isCancelled()));
892     }
893 jsr166 1.58 }
894 jsr166 1.89
895 jsr166 1.87 unblock.countDown(); // Release all pool threads
896 jsr166 1.58
897     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
898 jsr166 1.89 assertFalse(p.isTerminating());
899 jsr166 1.33 assertTrue(p.isTerminated());
900 jsr166 1.87
901 jsr166 1.89 assertTrue(q.isEmpty());
902    
903 jsr166 1.90 Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
904     .forEach(f -> assertTrue(f.isDone()));
905    
906 jsr166 1.89 for (Future<?> f : immediates) assertNull(f.get());
907    
908     assertNull(delayeds.get(0).get());
909     if (effectiveDelayedPolicy)
910     assertNull(delayeds.get(1).get());
911     else
912     assertTrue(delayeds.get(1).isCancelled());
913    
914 jsr166 1.90 if (periodicTasksContinue)
915     periodics.forEach(
916     f -> {
917     try { f.get(); }
918     catch (ExecutionException success) {
919     assertSame(exception, success.getCause());
920     }
921     catch (Throwable fail) { threadUnexpectedException(fail); }
922     });
923     else
924     periodics.forEach(f -> assertTrue(f.isCancelled()));
925 jsr166 1.89
926 jsr166 1.90 assertEquals(poolSize + 1
927     + (effectiveDelayedPolicy ? 1 : 0)
928     + (periodicTasksContinue ? 4 : 0),
929     ran.get());
930 jsr166 1.88 }
931 dl 1.1
932 dl 1.10 /**
933     * completed submit of callable returns result
934     */
935 jsr166 1.25 public void testSubmitCallable() throws Exception {
936 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
937     try (PoolCleaner cleaner = cleaner(e)) {
938 dl 1.10 Future<String> future = e.submit(new StringTask());
939     String result = future.get();
940     assertSame(TEST_STRING, result);
941     }
942     }
943    
944     /**
945     * completed submit of runnable returns successfully
946     */
947 jsr166 1.25 public void testSubmitRunnable() throws Exception {
948 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
949     try (PoolCleaner cleaner = cleaner(e)) {
950 dl 1.10 Future<?> future = e.submit(new NoOpRunnable());
951     future.get();
952     assertTrue(future.isDone());
953     }
954     }
955    
956     /**
957     * completed submit of (runnable, result) returns result
958     */
959 jsr166 1.25 public void testSubmitRunnable2() throws Exception {
960 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
961     try (PoolCleaner cleaner = cleaner(e)) {
962 dl 1.10 Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
963     String result = future.get();
964     assertSame(TEST_STRING, result);
965     }
966     }
967    
968     /**
969     * invokeAny(null) throws NPE
970     */
971 jsr166 1.25 public void testInvokeAny1() throws Exception {
972 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
973     try (PoolCleaner cleaner = cleaner(e)) {
974     try {
975     e.invokeAny(null);
976     shouldThrow();
977     } catch (NullPointerException success) {}
978 dl 1.10 }
979     }
980    
981     /**
982     * invokeAny(empty collection) throws IAE
983     */
984 jsr166 1.25 public void testInvokeAny2() throws Exception {
985 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
986     try (PoolCleaner cleaner = cleaner(e)) {
987     try {
988     e.invokeAny(new ArrayList<Callable<String>>());
989     shouldThrow();
990     } catch (IllegalArgumentException success) {}
991 dl 1.10 }
992     }
993    
994     /**
995     * invokeAny(c) throws NPE if c has null elements
996     */
997 jsr166 1.25 public void testInvokeAny3() throws Exception {
998 jsr166 1.29 CountDownLatch latch = new CountDownLatch(1);
999 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1000     try (PoolCleaner cleaner = cleaner(e)) {
1001 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1002 jsr166 1.62 l.add(latchAwaitingStringTask(latch));
1003     l.add(null);
1004     try {
1005     e.invokeAny(l);
1006     shouldThrow();
1007     } catch (NullPointerException success) {}
1008 jsr166 1.25 latch.countDown();
1009 dl 1.10 }
1010     }
1011    
1012     /**
1013     * invokeAny(c) throws ExecutionException if no task completes
1014     */
1015 jsr166 1.25 public void testInvokeAny4() throws Exception {
1016 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1017     try (PoolCleaner cleaner = cleaner(e)) {
1018 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1019 jsr166 1.62 l.add(new NPETask());
1020     try {
1021     e.invokeAny(l);
1022     shouldThrow();
1023     } catch (ExecutionException success) {
1024     assertTrue(success.getCause() instanceof NullPointerException);
1025     }
1026 dl 1.10 }
1027     }
1028    
1029     /**
1030     * invokeAny(c) returns result of some task
1031     */
1032 jsr166 1.25 public void testInvokeAny5() throws Exception {
1033 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1034     try (PoolCleaner cleaner = cleaner(e)) {
1035 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1036 dl 1.10 l.add(new StringTask());
1037     l.add(new StringTask());
1038     String result = e.invokeAny(l);
1039     assertSame(TEST_STRING, result);
1040     }
1041     }
1042    
1043     /**
1044     * invokeAll(null) throws NPE
1045     */
1046 jsr166 1.25 public void testInvokeAll1() throws Exception {
1047 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1048     try (PoolCleaner cleaner = cleaner(e)) {
1049     try {
1050     e.invokeAll(null);
1051     shouldThrow();
1052     } catch (NullPointerException success) {}
1053 dl 1.10 }
1054     }
1055    
1056     /**
1057     * invokeAll(empty collection) returns empty collection
1058     */
1059 jsr166 1.25 public void testInvokeAll2() throws Exception {
1060 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1061     try (PoolCleaner cleaner = cleaner(e)) {
1062 dl 1.10 List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
1063     assertTrue(r.isEmpty());
1064     }
1065     }
1066    
1067     /**
1068     * invokeAll(c) throws NPE if c has null elements
1069     */
1070 jsr166 1.25 public void testInvokeAll3() throws Exception {
1071 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1072     try (PoolCleaner cleaner = cleaner(e)) {
1073 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1074 jsr166 1.62 l.add(new StringTask());
1075     l.add(null);
1076     try {
1077     e.invokeAll(l);
1078     shouldThrow();
1079     } catch (NullPointerException success) {}
1080 dl 1.10 }
1081     }
1082    
1083     /**
1084     * get of invokeAll(c) throws exception on failed task
1085     */
1086 jsr166 1.25 public void testInvokeAll4() throws Exception {
1087 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1088     try (PoolCleaner cleaner = cleaner(e)) {
1089 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1090 jsr166 1.62 l.add(new NPETask());
1091     List<Future<String>> futures = e.invokeAll(l);
1092     assertEquals(1, futures.size());
1093     try {
1094     futures.get(0).get();
1095     shouldThrow();
1096     } catch (ExecutionException success) {
1097     assertTrue(success.getCause() instanceof NullPointerException);
1098     }
1099 dl 1.10 }
1100     }
1101    
1102     /**
1103     * invokeAll(c) returns results of all completed tasks
1104     */
1105 jsr166 1.25 public void testInvokeAll5() throws Exception {
1106 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1107     try (PoolCleaner cleaner = cleaner(e)) {
1108 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1109 dl 1.10 l.add(new StringTask());
1110     l.add(new StringTask());
1111 jsr166 1.29 List<Future<String>> futures = e.invokeAll(l);
1112     assertEquals(2, futures.size());
1113     for (Future<String> future : futures)
1114 jsr166 1.25 assertSame(TEST_STRING, future.get());
1115 dl 1.10 }
1116     }
1117    
1118 dl 1.11 /**
1119     * timed invokeAny(null) throws NPE
1120     */
1121 jsr166 1.25 public void testTimedInvokeAny1() throws Exception {
1122 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1123     try (PoolCleaner cleaner = cleaner(e)) {
1124     try {
1125     e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1126     shouldThrow();
1127     } catch (NullPointerException success) {}
1128 dl 1.11 }
1129     }
1130    
1131     /**
1132     * timed invokeAny(,,null) throws NPE
1133     */
1134 jsr166 1.25 public void testTimedInvokeAnyNullTimeUnit() throws Exception {
1135 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1136     try (PoolCleaner cleaner = cleaner(e)) {
1137 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1138 jsr166 1.62 l.add(new StringTask());
1139     try {
1140     e.invokeAny(l, MEDIUM_DELAY_MS, null);
1141     shouldThrow();
1142     } catch (NullPointerException success) {}
1143 dl 1.11 }
1144     }
1145    
1146     /**
1147     * timed invokeAny(empty collection) throws IAE
1148     */
1149 jsr166 1.25 public void testTimedInvokeAny2() throws Exception {
1150 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1151     try (PoolCleaner cleaner = cleaner(e)) {
1152     try {
1153     e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1154     shouldThrow();
1155     } catch (IllegalArgumentException success) {}
1156 dl 1.11 }
1157     }
1158    
1159     /**
1160     * timed invokeAny(c) throws NPE if c has null elements
1161     */
1162 jsr166 1.25 public void testTimedInvokeAny3() throws Exception {
1163 jsr166 1.29 CountDownLatch latch = new CountDownLatch(1);
1164 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1165     try (PoolCleaner cleaner = cleaner(e)) {
1166 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1167 jsr166 1.62 l.add(latchAwaitingStringTask(latch));
1168     l.add(null);
1169     try {
1170     e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1171     shouldThrow();
1172     } catch (NullPointerException success) {}
1173 jsr166 1.25 latch.countDown();
1174 dl 1.11 }
1175     }
1176    
1177     /**
1178     * timed invokeAny(c) throws ExecutionException if no task completes
1179     */
1180 jsr166 1.25 public void testTimedInvokeAny4() throws Exception {
1181 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1182     try (PoolCleaner cleaner = cleaner(e)) {
1183 jsr166 1.75 long startTime = System.nanoTime();
1184 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1185 jsr166 1.62 l.add(new NPETask());
1186     try {
1187 jsr166 1.75 e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1188 jsr166 1.62 shouldThrow();
1189     } catch (ExecutionException success) {
1190     assertTrue(success.getCause() instanceof NullPointerException);
1191     }
1192 jsr166 1.75 assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1193 dl 1.11 }
1194     }
1195    
1196     /**
1197     * timed invokeAny(c) returns result of some task
1198     */
1199 jsr166 1.25 public void testTimedInvokeAny5() throws Exception {
1200 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1201     try (PoolCleaner cleaner = cleaner(e)) {
1202 jsr166 1.76 long startTime = System.nanoTime();
1203 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1204 dl 1.11 l.add(new StringTask());
1205     l.add(new StringTask());
1206 jsr166 1.76 String result = e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1207 dl 1.11 assertSame(TEST_STRING, result);
1208 jsr166 1.76 assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1209 dl 1.11 }
1210     }
1211    
1212     /**
1213     * timed invokeAll(null) throws NPE
1214     */
1215 jsr166 1.25 public void testTimedInvokeAll1() throws Exception {
1216 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1217     try (PoolCleaner cleaner = cleaner(e)) {
1218     try {
1219     e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1220     shouldThrow();
1221     } catch (NullPointerException success) {}
1222 dl 1.11 }
1223     }
1224    
1225     /**
1226     * timed invokeAll(,,null) throws NPE
1227     */
1228 jsr166 1.25 public void testTimedInvokeAllNullTimeUnit() throws Exception {
1229 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1230     try (PoolCleaner cleaner = cleaner(e)) {
1231 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1232 jsr166 1.62 l.add(new StringTask());
1233     try {
1234     e.invokeAll(l, MEDIUM_DELAY_MS, null);
1235     shouldThrow();
1236     } catch (NullPointerException success) {}
1237 dl 1.11 }
1238     }
1239    
1240     /**
1241     * timed invokeAll(empty collection) returns empty collection
1242     */
1243 jsr166 1.25 public void testTimedInvokeAll2() throws Exception {
1244 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1245     try (PoolCleaner cleaner = cleaner(e)) {
1246     List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(),
1247     MEDIUM_DELAY_MS, MILLISECONDS);
1248 dl 1.11 assertTrue(r.isEmpty());
1249     }
1250     }
1251    
1252     /**
1253     * timed invokeAll(c) throws NPE if c has null elements
1254     */
1255 jsr166 1.25 public void testTimedInvokeAll3() throws Exception {
1256 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1257     try (PoolCleaner cleaner = cleaner(e)) {
1258 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1259 jsr166 1.62 l.add(new StringTask());
1260     l.add(null);
1261     try {
1262     e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1263     shouldThrow();
1264     } catch (NullPointerException success) {}
1265 dl 1.11 }
1266     }
1267    
1268     /**
1269     * get of element of invokeAll(c) throws exception on failed task
1270     */
1271 jsr166 1.25 public void testTimedInvokeAll4() throws Exception {
1272 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1273     try (PoolCleaner cleaner = cleaner(e)) {
1274 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1275 jsr166 1.62 l.add(new NPETask());
1276     List<Future<String>> futures =
1277 jsr166 1.69 e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1278 jsr166 1.62 assertEquals(1, futures.size());
1279     try {
1280     futures.get(0).get();
1281     shouldThrow();
1282     } catch (ExecutionException success) {
1283     assertTrue(success.getCause() instanceof NullPointerException);
1284     }
1285 dl 1.11 }
1286     }
1287    
1288     /**
1289     * timed invokeAll(c) returns results of all completed tasks
1290     */
1291 jsr166 1.25 public void testTimedInvokeAll5() throws Exception {
1292 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1293     try (PoolCleaner cleaner = cleaner(e)) {
1294 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1295 dl 1.11 l.add(new StringTask());
1296     l.add(new StringTask());
1297 jsr166 1.29 List<Future<String>> futures =
1298 jsr166 1.59 e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1299 jsr166 1.29 assertEquals(2, futures.size());
1300     for (Future<String> future : futures)
1301 jsr166 1.25 assertSame(TEST_STRING, future.get());
1302 dl 1.11 }
1303     }
1304    
1305     /**
1306     * timed invokeAll(c) cancels tasks not completed by timeout
1307     */
1308 jsr166 1.25 public void testTimedInvokeAll6() throws Exception {
1309 jsr166 1.74 for (long timeout = timeoutMillis();;) {
1310     final CountDownLatch done = new CountDownLatch(1);
1311     final Callable<String> waiter = new CheckedCallable<String>() {
1312     public String realCall() {
1313     try { done.await(LONG_DELAY_MS, MILLISECONDS); }
1314     catch (InterruptedException ok) {}
1315     return "1"; }};
1316     final ExecutorService p = new ScheduledThreadPoolExecutor(2);
1317     try (PoolCleaner cleaner = cleaner(p, done)) {
1318 jsr166 1.52 List<Callable<String>> tasks = new ArrayList<>();
1319     tasks.add(new StringTask("0"));
1320 jsr166 1.74 tasks.add(waiter);
1321 jsr166 1.52 tasks.add(new StringTask("2"));
1322     long startTime = System.nanoTime();
1323     List<Future<String>> futures =
1324 jsr166 1.74 p.invokeAll(tasks, timeout, MILLISECONDS);
1325 jsr166 1.52 assertEquals(tasks.size(), futures.size());
1326     assertTrue(millisElapsedSince(startTime) >= timeout);
1327     for (Future future : futures)
1328     assertTrue(future.isDone());
1329     assertTrue(futures.get(1).isCancelled());
1330     try {
1331     assertEquals("0", futures.get(0).get());
1332     assertEquals("2", futures.get(2).get());
1333     break;
1334     } catch (CancellationException retryWithLongerTimeout) {
1335     timeout *= 2;
1336     if (timeout >= LONG_DELAY_MS / 2)
1337     fail("expected exactly one task to be cancelled");
1338     }
1339     }
1340 dl 1.11 }
1341     }
1342    
1343 jsr166 1.77 /**
1344     * A fixed delay task with overflowing period should not prevent a
1345     * one-shot task from executing.
1346     * https://bugs.openjdk.java.net/browse/JDK-8051859
1347     */
1348     public void testScheduleWithFixedDelay_overflow() throws Exception {
1349     final CountDownLatch delayedDone = new CountDownLatch(1);
1350     final CountDownLatch immediateDone = new CountDownLatch(1);
1351     final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
1352     try (PoolCleaner cleaner = cleaner(p)) {
1353     final Runnable immediate = new Runnable() { public void run() {
1354     immediateDone.countDown();
1355     }};
1356     final Runnable delayed = new Runnable() { public void run() {
1357     delayedDone.countDown();
1358     p.submit(immediate);
1359     }};
1360     p.scheduleWithFixedDelay(delayed, 0L, Long.MAX_VALUE, SECONDS);
1361     await(delayedDone);
1362     await(immediateDone);
1363     }
1364     }
1365    
1366 dl 1.1 }