ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/ScheduledExecutorTest.java
Revision: 1.91
Committed: Wed Mar 29 16:53:20 2017 UTC (7 years, 1 month ago) by jsr166
Branch: MAIN
Changes since 1.90: +13 -12 lines
Log Message:
fix flake when testImplementationDetails is false

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 jsr166 1.91 assertEquals(poolSize, p.getActiveCount());
831 jsr166 1.89 assertTrue(q.isEmpty());
832    
833     // Add second wave of tasks.
834     immediates.add(p.submit(task));
835     long delay_ms = effectiveDelayedPolicy ? 1 : LONG_DELAY_MS;
836     delayeds.add(p.schedule(task, delay_ms, MILLISECONDS));
837     for (int rounds : new int[] { 1, 2 }) {
838     periodics.add(p.scheduleAtFixedRate(
839     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
840     periodics.add(p.scheduleWithFixedDelay(
841     new PeriodicTask(rounds), 1, 1, MILLISECONDS));
842     }
843    
844     assertEquals(poolSize, q.size());
845     assertEquals(poolSize, ran.get());
846    
847     immediates.forEach(
848     f -> assertTrue(((ScheduledFuture)f).getDelay(NANOSECONDS) <= 0L));
849    
850     Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
851     .forEach(f -> assertFalse(f.isDone()));
852 jsr166 1.22
853 jsr166 1.33 try { p.shutdown(); } catch (SecurityException ok) { return; }
854 jsr166 1.58 assertTrue(p.isShutdown());
855 jsr166 1.89 assertTrue(p.isTerminating());
856 jsr166 1.58 assertFalse(p.isTerminated());
857 jsr166 1.89
858     if (rnd.nextBoolean())
859     assertThrows(
860     RejectedExecutionException.class,
861     () -> p.submit(task),
862     () -> p.schedule(task, 1, SECONDS),
863     () -> p.scheduleAtFixedRate(
864     new PeriodicTask(1), 1, 1, SECONDS),
865     () -> p.scheduleWithFixedDelay(
866     new PeriodicTask(2), 1, 1, SECONDS));
867    
868     assertTrue(q.contains(immediates.get(1)));
869     assertTrue(!effectiveDelayedPolicy
870     ^ q.contains(delayeds.get(1)));
871     assertTrue(!effectivePeriodicPolicy
872     ^ q.containsAll(periodics.subList(4, 8)));
873    
874     immediates.forEach(f -> assertFalse(f.isDone()));
875    
876     assertFalse(delayeds.get(0).isDone());
877     if (effectiveDelayedPolicy)
878     assertFalse(delayeds.get(1).isDone());
879     else
880     assertTrue(delayeds.get(1).isCancelled());
881    
882 jsr166 1.91 if (effectivePeriodicPolicy)
883     periodics.forEach(
884     f -> {
885     assertFalse(f.isDone());
886     if (!periodicTasksContinue) {
887     assertTrue(f.cancel(false));
888     assertTrue(f.isCancelled());
889     }
890     });
891     else {
892     periodics.subList(0, 4).forEach(f -> assertFalse(f.isDone()));
893     periodics.subList(4, 8).forEach(f -> assertTrue(f.isCancelled()));
894 jsr166 1.58 }
895 jsr166 1.89
896 jsr166 1.87 unblock.countDown(); // Release all pool threads
897 jsr166 1.58
898     assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
899 jsr166 1.89 assertFalse(p.isTerminating());
900 jsr166 1.33 assertTrue(p.isTerminated());
901 jsr166 1.87
902 jsr166 1.89 assertTrue(q.isEmpty());
903    
904 jsr166 1.90 Stream.of(immediates, delayeds, periodics).flatMap(c -> c.stream())
905     .forEach(f -> assertTrue(f.isDone()));
906    
907 jsr166 1.89 for (Future<?> f : immediates) assertNull(f.get());
908    
909     assertNull(delayeds.get(0).get());
910     if (effectiveDelayedPolicy)
911     assertNull(delayeds.get(1).get());
912     else
913     assertTrue(delayeds.get(1).isCancelled());
914    
915 jsr166 1.90 if (periodicTasksContinue)
916     periodics.forEach(
917     f -> {
918     try { f.get(); }
919     catch (ExecutionException success) {
920     assertSame(exception, success.getCause());
921     }
922     catch (Throwable fail) { threadUnexpectedException(fail); }
923     });
924     else
925     periodics.forEach(f -> assertTrue(f.isCancelled()));
926 jsr166 1.89
927 jsr166 1.90 assertEquals(poolSize + 1
928     + (effectiveDelayedPolicy ? 1 : 0)
929     + (periodicTasksContinue ? 4 : 0),
930     ran.get());
931 jsr166 1.88 }
932 dl 1.1
933 dl 1.10 /**
934     * completed submit of callable returns result
935     */
936 jsr166 1.25 public void testSubmitCallable() throws Exception {
937 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
938     try (PoolCleaner cleaner = cleaner(e)) {
939 dl 1.10 Future<String> future = e.submit(new StringTask());
940     String result = future.get();
941     assertSame(TEST_STRING, result);
942     }
943     }
944    
945     /**
946     * completed submit of runnable returns successfully
947     */
948 jsr166 1.25 public void testSubmitRunnable() throws Exception {
949 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
950     try (PoolCleaner cleaner = cleaner(e)) {
951 dl 1.10 Future<?> future = e.submit(new NoOpRunnable());
952     future.get();
953     assertTrue(future.isDone());
954     }
955     }
956    
957     /**
958     * completed submit of (runnable, result) returns result
959     */
960 jsr166 1.25 public void testSubmitRunnable2() throws Exception {
961 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
962     try (PoolCleaner cleaner = cleaner(e)) {
963 dl 1.10 Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
964     String result = future.get();
965     assertSame(TEST_STRING, result);
966     }
967     }
968    
969     /**
970     * invokeAny(null) throws NPE
971     */
972 jsr166 1.25 public void testInvokeAny1() throws Exception {
973 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
974     try (PoolCleaner cleaner = cleaner(e)) {
975     try {
976     e.invokeAny(null);
977     shouldThrow();
978     } catch (NullPointerException success) {}
979 dl 1.10 }
980     }
981    
982     /**
983     * invokeAny(empty collection) throws IAE
984     */
985 jsr166 1.25 public void testInvokeAny2() throws Exception {
986 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
987     try (PoolCleaner cleaner = cleaner(e)) {
988     try {
989     e.invokeAny(new ArrayList<Callable<String>>());
990     shouldThrow();
991     } catch (IllegalArgumentException success) {}
992 dl 1.10 }
993     }
994    
995     /**
996     * invokeAny(c) throws NPE if c has null elements
997     */
998 jsr166 1.25 public void testInvokeAny3() throws Exception {
999 jsr166 1.29 CountDownLatch latch = new CountDownLatch(1);
1000 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1001     try (PoolCleaner cleaner = cleaner(e)) {
1002 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1003 jsr166 1.62 l.add(latchAwaitingStringTask(latch));
1004     l.add(null);
1005     try {
1006     e.invokeAny(l);
1007     shouldThrow();
1008     } catch (NullPointerException success) {}
1009 jsr166 1.25 latch.countDown();
1010 dl 1.10 }
1011     }
1012    
1013     /**
1014     * invokeAny(c) throws ExecutionException if no task completes
1015     */
1016 jsr166 1.25 public void testInvokeAny4() throws Exception {
1017 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1018     try (PoolCleaner cleaner = cleaner(e)) {
1019 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1020 jsr166 1.62 l.add(new NPETask());
1021     try {
1022     e.invokeAny(l);
1023     shouldThrow();
1024     } catch (ExecutionException success) {
1025     assertTrue(success.getCause() instanceof NullPointerException);
1026     }
1027 dl 1.10 }
1028     }
1029    
1030     /**
1031     * invokeAny(c) returns result of some task
1032     */
1033 jsr166 1.25 public void testInvokeAny5() throws Exception {
1034 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1035     try (PoolCleaner cleaner = cleaner(e)) {
1036 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1037 dl 1.10 l.add(new StringTask());
1038     l.add(new StringTask());
1039     String result = e.invokeAny(l);
1040     assertSame(TEST_STRING, result);
1041     }
1042     }
1043    
1044     /**
1045     * invokeAll(null) throws NPE
1046     */
1047 jsr166 1.25 public void testInvokeAll1() throws Exception {
1048 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1049     try (PoolCleaner cleaner = cleaner(e)) {
1050     try {
1051     e.invokeAll(null);
1052     shouldThrow();
1053     } catch (NullPointerException success) {}
1054 dl 1.10 }
1055     }
1056    
1057     /**
1058     * invokeAll(empty collection) returns empty collection
1059     */
1060 jsr166 1.25 public void testInvokeAll2() throws Exception {
1061 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1062     try (PoolCleaner cleaner = cleaner(e)) {
1063 dl 1.10 List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
1064     assertTrue(r.isEmpty());
1065     }
1066     }
1067    
1068     /**
1069     * invokeAll(c) throws NPE if c has null elements
1070     */
1071 jsr166 1.25 public void testInvokeAll3() throws Exception {
1072 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1073     try (PoolCleaner cleaner = cleaner(e)) {
1074 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1075 jsr166 1.62 l.add(new StringTask());
1076     l.add(null);
1077     try {
1078     e.invokeAll(l);
1079     shouldThrow();
1080     } catch (NullPointerException success) {}
1081 dl 1.10 }
1082     }
1083    
1084     /**
1085     * get of invokeAll(c) throws exception on failed task
1086     */
1087 jsr166 1.25 public void testInvokeAll4() throws Exception {
1088 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1089     try (PoolCleaner cleaner = cleaner(e)) {
1090 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1091 jsr166 1.62 l.add(new NPETask());
1092     List<Future<String>> futures = e.invokeAll(l);
1093     assertEquals(1, futures.size());
1094     try {
1095     futures.get(0).get();
1096     shouldThrow();
1097     } catch (ExecutionException success) {
1098     assertTrue(success.getCause() instanceof NullPointerException);
1099     }
1100 dl 1.10 }
1101     }
1102    
1103     /**
1104     * invokeAll(c) returns results of all completed tasks
1105     */
1106 jsr166 1.25 public void testInvokeAll5() throws Exception {
1107 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1108     try (PoolCleaner cleaner = cleaner(e)) {
1109 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1110 dl 1.10 l.add(new StringTask());
1111     l.add(new StringTask());
1112 jsr166 1.29 List<Future<String>> futures = e.invokeAll(l);
1113     assertEquals(2, futures.size());
1114     for (Future<String> future : futures)
1115 jsr166 1.25 assertSame(TEST_STRING, future.get());
1116 dl 1.10 }
1117     }
1118    
1119 dl 1.11 /**
1120     * timed invokeAny(null) throws NPE
1121     */
1122 jsr166 1.25 public void testTimedInvokeAny1() throws Exception {
1123 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1124     try (PoolCleaner cleaner = cleaner(e)) {
1125     try {
1126     e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
1127     shouldThrow();
1128     } catch (NullPointerException success) {}
1129 dl 1.11 }
1130     }
1131    
1132     /**
1133     * timed invokeAny(,,null) throws NPE
1134     */
1135 jsr166 1.25 public void testTimedInvokeAnyNullTimeUnit() throws Exception {
1136 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1137     try (PoolCleaner cleaner = cleaner(e)) {
1138 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1139 jsr166 1.62 l.add(new StringTask());
1140     try {
1141     e.invokeAny(l, MEDIUM_DELAY_MS, null);
1142     shouldThrow();
1143     } catch (NullPointerException success) {}
1144 dl 1.11 }
1145     }
1146    
1147     /**
1148     * timed invokeAny(empty collection) throws IAE
1149     */
1150 jsr166 1.25 public void testTimedInvokeAny2() throws Exception {
1151 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1152     try (PoolCleaner cleaner = cleaner(e)) {
1153     try {
1154     e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1155     shouldThrow();
1156     } catch (IllegalArgumentException success) {}
1157 dl 1.11 }
1158     }
1159    
1160     /**
1161     * timed invokeAny(c) throws NPE if c has null elements
1162     */
1163 jsr166 1.25 public void testTimedInvokeAny3() throws Exception {
1164 jsr166 1.29 CountDownLatch latch = new CountDownLatch(1);
1165 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1166     try (PoolCleaner cleaner = cleaner(e)) {
1167 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1168 jsr166 1.62 l.add(latchAwaitingStringTask(latch));
1169     l.add(null);
1170     try {
1171     e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1172     shouldThrow();
1173     } catch (NullPointerException success) {}
1174 jsr166 1.25 latch.countDown();
1175 dl 1.11 }
1176     }
1177    
1178     /**
1179     * timed invokeAny(c) throws ExecutionException if no task completes
1180     */
1181 jsr166 1.25 public void testTimedInvokeAny4() throws Exception {
1182 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1183     try (PoolCleaner cleaner = cleaner(e)) {
1184 jsr166 1.75 long startTime = System.nanoTime();
1185 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1186 jsr166 1.62 l.add(new NPETask());
1187     try {
1188 jsr166 1.75 e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1189 jsr166 1.62 shouldThrow();
1190     } catch (ExecutionException success) {
1191     assertTrue(success.getCause() instanceof NullPointerException);
1192     }
1193 jsr166 1.75 assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1194 dl 1.11 }
1195     }
1196    
1197     /**
1198     * timed invokeAny(c) returns result of some task
1199     */
1200 jsr166 1.25 public void testTimedInvokeAny5() throws Exception {
1201 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1202     try (PoolCleaner cleaner = cleaner(e)) {
1203 jsr166 1.76 long startTime = System.nanoTime();
1204 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1205 dl 1.11 l.add(new StringTask());
1206     l.add(new StringTask());
1207 jsr166 1.76 String result = e.invokeAny(l, LONG_DELAY_MS, MILLISECONDS);
1208 dl 1.11 assertSame(TEST_STRING, result);
1209 jsr166 1.76 assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
1210 dl 1.11 }
1211     }
1212    
1213     /**
1214     * timed invokeAll(null) throws NPE
1215     */
1216 jsr166 1.25 public void testTimedInvokeAll1() throws Exception {
1217 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1218     try (PoolCleaner cleaner = cleaner(e)) {
1219     try {
1220     e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1221     shouldThrow();
1222     } catch (NullPointerException success) {}
1223 dl 1.11 }
1224     }
1225    
1226     /**
1227     * timed invokeAll(,,null) throws NPE
1228     */
1229 jsr166 1.25 public void testTimedInvokeAllNullTimeUnit() throws Exception {
1230 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1231     try (PoolCleaner cleaner = cleaner(e)) {
1232 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1233 jsr166 1.62 l.add(new StringTask());
1234     try {
1235     e.invokeAll(l, MEDIUM_DELAY_MS, null);
1236     shouldThrow();
1237     } catch (NullPointerException success) {}
1238 dl 1.11 }
1239     }
1240    
1241     /**
1242     * timed invokeAll(empty collection) returns empty collection
1243     */
1244 jsr166 1.25 public void testTimedInvokeAll2() throws Exception {
1245 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1246     try (PoolCleaner cleaner = cleaner(e)) {
1247     List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(),
1248     MEDIUM_DELAY_MS, MILLISECONDS);
1249 dl 1.11 assertTrue(r.isEmpty());
1250     }
1251     }
1252    
1253     /**
1254     * timed invokeAll(c) throws NPE if c has null elements
1255     */
1256 jsr166 1.25 public void testTimedInvokeAll3() throws Exception {
1257 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1258     try (PoolCleaner cleaner = cleaner(e)) {
1259 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1260 jsr166 1.62 l.add(new StringTask());
1261     l.add(null);
1262     try {
1263     e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1264     shouldThrow();
1265     } catch (NullPointerException success) {}
1266 dl 1.11 }
1267     }
1268    
1269     /**
1270     * get of element of invokeAll(c) throws exception on failed task
1271     */
1272 jsr166 1.25 public void testTimedInvokeAll4() throws Exception {
1273 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1274     try (PoolCleaner cleaner = cleaner(e)) {
1275 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1276 jsr166 1.62 l.add(new NPETask());
1277     List<Future<String>> futures =
1278 jsr166 1.69 e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1279 jsr166 1.62 assertEquals(1, futures.size());
1280     try {
1281     futures.get(0).get();
1282     shouldThrow();
1283     } catch (ExecutionException success) {
1284     assertTrue(success.getCause() instanceof NullPointerException);
1285     }
1286 dl 1.11 }
1287     }
1288    
1289     /**
1290     * timed invokeAll(c) returns results of all completed tasks
1291     */
1292 jsr166 1.25 public void testTimedInvokeAll5() throws Exception {
1293 jsr166 1.62 final ExecutorService e = new ScheduledThreadPoolExecutor(2);
1294     try (PoolCleaner cleaner = cleaner(e)) {
1295 jsr166 1.83 List<Callable<String>> l = new ArrayList<>();
1296 dl 1.11 l.add(new StringTask());
1297     l.add(new StringTask());
1298 jsr166 1.29 List<Future<String>> futures =
1299 jsr166 1.59 e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
1300 jsr166 1.29 assertEquals(2, futures.size());
1301     for (Future<String> future : futures)
1302 jsr166 1.25 assertSame(TEST_STRING, future.get());
1303 dl 1.11 }
1304     }
1305    
1306     /**
1307     * timed invokeAll(c) cancels tasks not completed by timeout
1308     */
1309 jsr166 1.25 public void testTimedInvokeAll6() throws Exception {
1310 jsr166 1.74 for (long timeout = timeoutMillis();;) {
1311     final CountDownLatch done = new CountDownLatch(1);
1312     final Callable<String> waiter = new CheckedCallable<String>() {
1313     public String realCall() {
1314     try { done.await(LONG_DELAY_MS, MILLISECONDS); }
1315     catch (InterruptedException ok) {}
1316     return "1"; }};
1317     final ExecutorService p = new ScheduledThreadPoolExecutor(2);
1318     try (PoolCleaner cleaner = cleaner(p, done)) {
1319 jsr166 1.52 List<Callable<String>> tasks = new ArrayList<>();
1320     tasks.add(new StringTask("0"));
1321 jsr166 1.74 tasks.add(waiter);
1322 jsr166 1.52 tasks.add(new StringTask("2"));
1323     long startTime = System.nanoTime();
1324     List<Future<String>> futures =
1325 jsr166 1.74 p.invokeAll(tasks, timeout, MILLISECONDS);
1326 jsr166 1.52 assertEquals(tasks.size(), futures.size());
1327     assertTrue(millisElapsedSince(startTime) >= timeout);
1328     for (Future future : futures)
1329     assertTrue(future.isDone());
1330     assertTrue(futures.get(1).isCancelled());
1331     try {
1332     assertEquals("0", futures.get(0).get());
1333     assertEquals("2", futures.get(2).get());
1334     break;
1335     } catch (CancellationException retryWithLongerTimeout) {
1336     timeout *= 2;
1337     if (timeout >= LONG_DELAY_MS / 2)
1338     fail("expected exactly one task to be cancelled");
1339     }
1340     }
1341 dl 1.11 }
1342     }
1343    
1344 jsr166 1.77 /**
1345     * A fixed delay task with overflowing period should not prevent a
1346     * one-shot task from executing.
1347     * https://bugs.openjdk.java.net/browse/JDK-8051859
1348     */
1349     public void testScheduleWithFixedDelay_overflow() throws Exception {
1350     final CountDownLatch delayedDone = new CountDownLatch(1);
1351     final CountDownLatch immediateDone = new CountDownLatch(1);
1352     final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
1353     try (PoolCleaner cleaner = cleaner(p)) {
1354     final Runnable immediate = new Runnable() { public void run() {
1355     immediateDone.countDown();
1356     }};
1357     final Runnable delayed = new Runnable() { public void run() {
1358     delayedDone.countDown();
1359     p.submit(immediate);
1360     }};
1361     p.scheduleWithFixedDelay(delayed, 0L, Long.MAX_VALUE, SECONDS);
1362     await(delayedDone);
1363     await(immediateDone);
1364     }
1365     }
1366    
1367 dl 1.1 }