ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/ScheduledExecutorTest.java
(Generate patch)

Comparing jsr166/src/test/tck/ScheduledExecutorTest.java (file contents):
Revision 1.4 by dl, Sun Sep 14 20:42:40 2003 UTC vs.
Revision 1.49 by jsr166, Wed Sep 25 07:39:17 2013 UTC

# Line 1 | Line 1
1   /*
2 < * Written by members of JCP JSR-166 Expert Group and released to the
3 < * public domain. Use, modify, and redistribute this code in any way
4 < * without acknowledgement. Other contributors include Andrew Wright,
5 < * Jeffrey Hayes, Pat Fischer, Mike Judd.
2 > * Written by Doug Lea with assistance from members of JCP JSR-166
3 > * Expert Group and released to the public domain, as explained at
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5 > * Other contributors include Andrew Wright, Jeffrey Hayes,
6 > * Pat Fisher, Mike Judd.
7   */
8  
9   import junit.framework.*;
10   import java.util.*;
11   import java.util.concurrent.*;
12 + import static java.util.concurrent.TimeUnit.MILLISECONDS;
13 + import java.util.concurrent.atomic.AtomicInteger;
14  
15   public class ScheduledExecutorTest extends JSR166TestCase {
16      public static void main(String[] args) {
17 <        junit.textui.TestRunner.run (suite());  
17 >        junit.textui.TestRunner.run(suite());
18      }
19      public static Test suite() {
20 <        return new TestSuite(ScheduledExecutorTest.class);
20 >        return new TestSuite(ScheduledExecutorTest.class);
21      }
22  
23 <    static class MyRunnable implements Runnable {
24 <        volatile boolean done = false;
25 <        public void run(){
26 <            try{
27 <                Thread.sleep(SMALL_DELAY_MS);
28 <                done = true;
29 <            } catch(Exception e){
30 <            }
23 >    /**
24 >     * execute successfully executes a runnable
25 >     */
26 >    public void testExecute() throws InterruptedException {
27 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
28 >        final CountDownLatch done = new CountDownLatch(1);
29 >        final Runnable task = new CheckedRunnable() {
30 >            public void realRun() {
31 >                done.countDown();
32 >            }};
33 >        try {
34 >            p.execute(task);
35 >            assertTrue(done.await(SMALL_DELAY_MS, MILLISECONDS));
36 >        } finally {
37 >            joinPool(p);
38 >        }
39 >    }
40 >
41 >    /**
42 >     * delayed schedule of callable successfully executes after delay
43 >     */
44 >    public void testSchedule1() throws Exception {
45 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
46 >        final long startTime = System.nanoTime();
47 >        final CountDownLatch done = new CountDownLatch(1);
48 >        try {
49 >            Callable task = new CheckedCallable<Boolean>() {
50 >                public Boolean realCall() {
51 >                    done.countDown();
52 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
53 >                    return Boolean.TRUE;
54 >                }};
55 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
56 >            assertSame(Boolean.TRUE, f.get());
57 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
58 >            assertTrue(done.await(0L, MILLISECONDS));
59 >        } finally {
60 >            joinPool(p);
61 >        }
62 >    }
63 >
64 >    /**
65 >     * delayed schedule of runnable successfully executes after delay
66 >     */
67 >    public void testSchedule3() throws Exception {
68 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
69 >        final long startTime = System.nanoTime();
70 >        final CountDownLatch done = new CountDownLatch(1);
71 >        try {
72 >            Runnable task = new CheckedRunnable() {
73 >                public void realRun() {
74 >                    done.countDown();
75 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
76 >                }};
77 >            Future f = p.schedule(task, timeoutMillis(), MILLISECONDS);
78 >            await(done);
79 >            assertNull(f.get(LONG_DELAY_MS, MILLISECONDS));
80 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
81 >        } finally {
82 >            joinPool(p);
83 >        }
84 >    }
85 >
86 >    /**
87 >     * scheduleAtFixedRate executes runnable after given initial delay
88 >     */
89 >    public void testSchedule4() throws Exception {
90 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
91 >        final long startTime = System.nanoTime();
92 >        final CountDownLatch done = new CountDownLatch(1);
93 >        try {
94 >            Runnable task = new CheckedRunnable() {
95 >                public void realRun() {
96 >                    done.countDown();
97 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
98 >                }};
99 >            ScheduledFuture f =
100 >                p.scheduleAtFixedRate(task, timeoutMillis(),
101 >                                      LONG_DELAY_MS, MILLISECONDS);
102 >            await(done);
103 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
104 >            f.cancel(true);
105 >        } finally {
106 >            joinPool(p);
107          }
108      }
109  
110 <    static class MyCallable implements Callable {
111 <        volatile boolean done = false;
112 <        public Object call(){
113 <            try{
114 <                Thread.sleep(SMALL_DELAY_MS);
115 <                done = true;
116 <            }catch(Exception e){}
117 <            return Boolean.TRUE;
110 >    /**
111 >     * scheduleWithFixedDelay executes runnable after given initial delay
112 >     */
113 >    public void testSchedule5() throws Exception {
114 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
115 >        final long startTime = System.nanoTime();
116 >        final CountDownLatch done = new CountDownLatch(1);
117 >        try {
118 >            Runnable task = new CheckedRunnable() {
119 >                public void realRun() {
120 >                    done.countDown();
121 >                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
122 >                }};
123 >            ScheduledFuture f =
124 >                p.scheduleWithFixedDelay(task, timeoutMillis(),
125 >                                         LONG_DELAY_MS, MILLISECONDS);
126 >            await(done);
127 >            assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
128 >            f.cancel(true);
129 >        } finally {
130 >            joinPool(p);
131          }
132      }
133  
134 <    public void testExecute(){
135 <        try{
136 <            MyRunnable runnable =new MyRunnable();
137 <            ScheduledExecutor one = new ScheduledExecutor(1);
138 <            one.execute(runnable);
139 <            assertFalse(runnable.done);
140 <            Thread.sleep(SHORT_DELAY_MS);
141 <            one.shutdown();
142 <            try{
143 <                Thread.sleep(MEDIUM_DELAY_MS);
144 <            } catch(InterruptedException e){
145 <                fail("unexpected exception");
134 >    static class RunnableCounter implements Runnable {
135 >        AtomicInteger count = new AtomicInteger(0);
136 >        public void run() { count.getAndIncrement(); }
137 >    }
138 >
139 >    /**
140 >     * scheduleAtFixedRate executes series of tasks at given rate
141 >     */
142 >    public void testFixedRateSequence() throws InterruptedException {
143 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
144 >        try {
145 >            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
146 >                long startTime = System.nanoTime();
147 >                int cycles = 10;
148 >                final CountDownLatch done = new CountDownLatch(cycles);
149 >                Runnable task = new CheckedRunnable() {
150 >                    public void realRun() { done.countDown(); }};
151 >                ScheduledFuture h =
152 >                    p.scheduleAtFixedRate(task, 0, delay, MILLISECONDS);
153 >                done.await();
154 >                h.cancel(true);
155 >                double normalizedTime =
156 >                    (double) millisElapsedSince(startTime) / delay;
157 >                if (normalizedTime >= cycles - 1 &&
158 >                    normalizedTime <= cycles)
159 >                    return;
160              }
161 <            assertTrue(runnable.done);
162 <            one.shutdown();
163 <            joinPool(one);
58 <        }
59 <        catch(Exception e){
60 <            fail("unexpected exception");
61 <        }
62 <        
63 <    }
64 <
65 <    public void testSchedule1(){
66 <        try{
67 <            MyCallable callable = new MyCallable();
68 <            ScheduledExecutor one = new ScheduledExecutor(1);
69 <            Future f = one.schedule(callable, SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
70 <            assertFalse(callable.done);
71 <            Thread.sleep(MEDIUM_DELAY_MS);
72 <            assertTrue(callable.done);
73 <            assertEquals(Boolean.TRUE, f.get());
74 <            one.shutdown();
75 <            joinPool(one);
76 <        }catch(RejectedExecutionException e){}
77 <        catch(Exception e){
78 <            fail("unexpected exception");
79 <        }
80 <    }
81 <
82 <    /**
83 <     *  Another version of schedule, only using Runnable instead of Callable
84 <     */
85 <    public void testSchedule3(){
86 <        try{
87 <            MyRunnable runnable = new MyRunnable();
88 <            ScheduledExecutor one = new ScheduledExecutor(1);
89 <            one.schedule(runnable, SMALL_DELAY_MS, TimeUnit.MILLISECONDS);
90 <            Thread.sleep(SHORT_DELAY_MS);
91 <            assertFalse(runnable.done);
92 <            Thread.sleep(MEDIUM_DELAY_MS);
93 <            assertTrue(runnable.done);
94 <            one.shutdown();
95 <            joinPool(one);
96 <        } catch(Exception e){
97 <            fail("unexpected exception");
98 <        }
99 <    }
100 <    
101 <    /**
102 <     *  The final version of schedule, using both long, TimeUnit and Runnable
103 <     */
104 <    public void testSchedule4(){
105 <        try{
106 <            MyRunnable runnable = new MyRunnable();
107 <            ScheduledExecutor one = new ScheduledExecutor(1);
108 <            one.schedule(runnable, SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
109 <            assertFalse(runnable.done);
110 <            Thread.sleep(MEDIUM_DELAY_MS);
111 <            assertTrue(runnable.done);
112 <            one.shutdown();
113 <            joinPool(one);
114 <        } catch(Exception e){
115 <            fail("unexpected exception");
116 <        }
117 <    }
118 <    
119 <  
120 <    // exception tests
121 <
122 <    /**
123 <     *   schedule(Runnable, long) throws RejectedExecutionException
124 <     *  This occurs on an attempt to schedule a task on a shutdown executor
125 <     */
126 <    public void testSchedule1_RejectedExecutionException(){
127 <        ScheduledExecutor se = new ScheduledExecutor(1);
128 <        try{
129 <            se.shutdown();
130 <            se.schedule(new NoOpRunnable(),
131 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
132 <            fail("shoud throw");
133 <        } catch(RejectedExecutionException success){
161 >            throw new AssertionError("unexpected execution rate");
162 >        } finally {
163 >            joinPool(p);
164          }
165 +    }
166 +
167 +    /**
168 +     * scheduleWithFixedDelay executes series of tasks with given period
169 +     */
170 +    public void testFixedDelaySequence() throws InterruptedException {
171 +        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
172 +        try {
173 +            for (int delay = 1; delay <= LONG_DELAY_MS; delay *= 3) {
174 +                long startTime = System.nanoTime();
175 +                int cycles = 10;
176 +                final CountDownLatch done = new CountDownLatch(cycles);
177 +                Runnable task = new CheckedRunnable() {
178 +                    public void realRun() { done.countDown(); }};
179 +                ScheduledFuture h =
180 +                    p.scheduleWithFixedDelay(task, 0, delay, MILLISECONDS);
181 +                done.await();
182 +                h.cancel(true);
183 +                double normalizedTime =
184 +                    (double) millisElapsedSince(startTime) / delay;
185 +                if (normalizedTime >= cycles - 1 &&
186 +                    normalizedTime <= cycles)
187 +                    return;
188 +            }
189 +            throw new AssertionError("unexpected execution rate");
190 +        } finally {
191 +            joinPool(p);
192 +        }
193 +    }
194 +
195 +    /**
196 +     * execute(null) throws NPE
197 +     */
198 +    public void testExecuteNull() throws InterruptedException {
199 +        ScheduledThreadPoolExecutor se = null;
200 +        try {
201 +            se = new ScheduledThreadPoolExecutor(1);
202 +            se.execute(null);
203 +            shouldThrow();
204 +        } catch (NullPointerException success) {}
205 +
206          joinPool(se);
207 +    }
208  
209 +    /**
210 +     * schedule(null) throws NPE
211 +     */
212 +    public void testScheduleNull() throws InterruptedException {
213 +        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
214 +        try {
215 +            TrackedCallable callable = null;
216 +            Future f = se.schedule(callable, SHORT_DELAY_MS, MILLISECONDS);
217 +            shouldThrow();
218 +        } catch (NullPointerException success) {}
219 +        joinPool(se);
220      }
221  
222      /**
223 <     *   schedule(Callable, long, TimeUnit) throws RejectedExecutionException
141 <     *  This occurs on an attempt to schedule a task on a shutdown executor
223 >     * execute throws RejectedExecutionException if shutdown
224       */
225 <    public void testSchedule2_RejectedExecutionException(){
226 <        ScheduledExecutor se = new ScheduledExecutor(1);
227 <        try{
225 >    public void testSchedule1_RejectedExecutionException() throws InterruptedException {
226 >        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
227 >        try {
228              se.shutdown();
229 <            se.schedule(new NoOpCallable(),
230 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
231 <            fail("should throw");
232 <        } catch(RejectedExecutionException success){
229 >            se.schedule(new NoOpRunnable(),
230 >                        MEDIUM_DELAY_MS, MILLISECONDS);
231 >            shouldThrow();
232 >        } catch (RejectedExecutionException success) {
233 >        } catch (SecurityException ok) {
234          }
235 +
236          joinPool(se);
237      }
238  
239      /**
240 <     *   schedule(Callable, long) throws RejectedExecutionException
157 <     *  This occurs on an attempt to schedule a task on a shutdown executor
240 >     * schedule throws RejectedExecutionException if shutdown
241       */
242 <     public void testSchedule3_RejectedExecutionException(){
243 <         ScheduledExecutor se = new ScheduledExecutor(1);
244 <         try{
242 >    public void testSchedule2_RejectedExecutionException() throws InterruptedException {
243 >        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
244 >        try {
245              se.shutdown();
246              se.schedule(new NoOpCallable(),
247 <                        MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
248 <            fail("should throw");
249 <        } catch(RejectedExecutionException success){
250 <        }
251 <         joinPool(se);
247 >                        MEDIUM_DELAY_MS, MILLISECONDS);
248 >            shouldThrow();
249 >        } catch (RejectedExecutionException success) {
250 >        } catch (SecurityException ok) {
251 >        }
252 >        joinPool(se);
253      }
254  
255      /**
256 <     *   scheduleAtFixedRate(Runnable, long, long, TimeUnit) throws
173 <     *  RejectedExecutionException.
174 <     *  This occurs on an attempt to schedule a task on a shutdown executor
256 >     * schedule callable throws RejectedExecutionException if shutdown
257       */
258 <    public void testScheduleAtFixedRate1_RejectedExecutionException(){
259 <        ScheduledExecutor se = new ScheduledExecutor(1);
260 <        try{
258 >    public void testSchedule3_RejectedExecutionException() throws InterruptedException {
259 >        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
260 >        try {
261              se.shutdown();
262 <            se.scheduleAtFixedRate(new NoOpRunnable(),
263 <                                   MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
264 <            fail("should throw");
265 <        } catch(RejectedExecutionException success){
266 <        }
262 >            se.schedule(new NoOpCallable(),
263 >                        MEDIUM_DELAY_MS, MILLISECONDS);
264 >            shouldThrow();
265 >        } catch (RejectedExecutionException success) {
266 >        } catch (SecurityException ok) {
267 >        }
268          joinPool(se);
269      }
270 <    
270 >
271      /**
272 <     *   scheduleAtFixedRate(Runnable, long, long, TimeUnit) throws
273 <     *  RejectedExecutionException.
274 <     *  This occurs on an attempt to schedule a task on a shutdown executor
275 <     */
276 <    public void testScheduleAtFixedRate2_RejectedExecutionException(){
194 <        ScheduledExecutor se = new ScheduledExecutor(1);
195 <        try{
272 >     * scheduleAtFixedRate throws RejectedExecutionException if shutdown
273 >     */
274 >    public void testScheduleAtFixedRate1_RejectedExecutionException() throws InterruptedException {
275 >        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
276 >        try {
277              se.shutdown();
278              se.scheduleAtFixedRate(new NoOpRunnable(),
279 <                                   1, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
280 <            fail("should throw");
281 <        } catch(RejectedExecutionException success){
282 <        }
279 >                                   MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
280 >            shouldThrow();
281 >        } catch (RejectedExecutionException success) {
282 >        } catch (SecurityException ok) {
283 >        }
284          joinPool(se);
285      }
286  
287      /**
288 <     *   scheduleWithFixedDelay(Runnable, long, long, TimeUnit) throws
289 <     *  RejectedExecutionException.
290 <     *  This occurs on an attempt to schedule a task on a shutdown executor
291 <     */
292 <    public void testScheduleWithFixedDelay1_RejectedExecutionException(){
211 <        ScheduledExecutor se = new ScheduledExecutor(1);
212 <        try{
288 >     * scheduleWithFixedDelay throws RejectedExecutionException if shutdown
289 >     */
290 >    public void testScheduleWithFixedDelay1_RejectedExecutionException() throws InterruptedException {
291 >        ScheduledThreadPoolExecutor se = new ScheduledThreadPoolExecutor(1);
292 >        try {
293              se.shutdown();
294              se.scheduleWithFixedDelay(new NoOpRunnable(),
295 <                                      MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
296 <            fail("should throw");
297 <        } catch(RejectedExecutionException success){
298 <        }
295 >                                      MEDIUM_DELAY_MS, MEDIUM_DELAY_MS, MILLISECONDS);
296 >            shouldThrow();
297 >        } catch (RejectedExecutionException success) {
298 >        } catch (SecurityException ok) {
299 >        }
300          joinPool(se);
301      }
302  
303      /**
304 <     *   scheduleWithFixedDelay(Runnable, long, long, TimeUnit) throws
305 <     *  RejectedExecutionException.
306 <     *  This occurs on an attempt to schedule a task on a shutdown executor
307 <     */
308 <     public void testScheduleWithFixedDelay2_RejectedExecutionException(){
309 <         ScheduledExecutor se = new ScheduledExecutor(1);
310 <        try{
311 <            se.shutdown();
312 <            se.scheduleWithFixedDelay(new NoOpRunnable(),
313 <                                      1, MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
314 <            fail("should throw");
315 <        } catch(RejectedExecutionException success){
316 <        }
317 <        joinPool(se);
304 >     * getActiveCount increases but doesn't overestimate, when a
305 >     * thread becomes active
306 >     */
307 >    public void testGetActiveCount() throws InterruptedException {
308 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
309 >        final CountDownLatch threadStarted = new CountDownLatch(1);
310 >        final CountDownLatch done = new CountDownLatch(1);
311 >        try {
312 >            assertEquals(0, p.getActiveCount());
313 >            p.execute(new CheckedRunnable() {
314 >                public void realRun() throws InterruptedException {
315 >                    threadStarted.countDown();
316 >                    assertEquals(1, p.getActiveCount());
317 >                    done.await();
318 >                }});
319 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
320 >            assertEquals(1, p.getActiveCount());
321 >        } finally {
322 >            done.countDown();
323 >            joinPool(p);
324 >        }
325      }
326  
327      /**
328 <     *   execute throws RejectedExecutionException
329 <     *  This occurs on an attempt to schedule a task on a shutdown executor
328 >     * getCompletedTaskCount increases, but doesn't overestimate,
329 >     * when tasks complete
330       */
331 <    public void testExecute_RejectedExecutionException(){
332 <        ScheduledExecutor se = new ScheduledExecutor(1);
333 <        try{
334 <            se.shutdown();
335 <            se.execute(new NoOpRunnable());
336 <            fail("should throw");
337 <        } catch(RejectedExecutionException success){
338 <        }
339 <        joinPool(se);
331 >    public void testGetCompletedTaskCount() throws InterruptedException {
332 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(2);
333 >        final CountDownLatch threadStarted = new CountDownLatch(1);
334 >        final CountDownLatch threadProceed = new CountDownLatch(1);
335 >        final CountDownLatch threadDone = new CountDownLatch(1);
336 >        try {
337 >            assertEquals(0, p.getCompletedTaskCount());
338 >            p.execute(new CheckedRunnable() {
339 >                public void realRun() throws InterruptedException {
340 >                    threadStarted.countDown();
341 >                    assertEquals(0, p.getCompletedTaskCount());
342 >                    threadProceed.await();
343 >                    threadDone.countDown();
344 >                }});
345 >            await(threadStarted);
346 >            assertEquals(0, p.getCompletedTaskCount());
347 >            threadProceed.countDown();
348 >            threadDone.await();
349 >            long startTime = System.nanoTime();
350 >            while (p.getCompletedTaskCount() != 1) {
351 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
352 >                    fail("timed out");
353 >                Thread.yield();
354 >            }
355 >        } finally {
356 >            joinPool(p);
357 >        }
358 >    }
359 >
360 >    /**
361 >     * getCorePoolSize returns size given in constructor if not otherwise set
362 >     */
363 >    public void testGetCorePoolSize() throws InterruptedException {
364 >        ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
365 >        assertEquals(1, p.getCorePoolSize());
366 >        joinPool(p);
367 >    }
368 >
369 >    /**
370 >     * getLargestPoolSize increases, but doesn't overestimate, when
371 >     * multiple threads active
372 >     */
373 >    public void testGetLargestPoolSize() throws InterruptedException {
374 >        final int THREADS = 3;
375 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(THREADS);
376 >        final CountDownLatch threadsStarted = new CountDownLatch(THREADS);
377 >        final CountDownLatch done = new CountDownLatch(1);
378 >        try {
379 >            assertEquals(0, p.getLargestPoolSize());
380 >            for (int i = 0; i < THREADS; i++)
381 >                p.execute(new CheckedRunnable() {
382 >                    public void realRun() throws InterruptedException {
383 >                        threadsStarted.countDown();
384 >                        done.await();
385 >                        assertEquals(THREADS, p.getLargestPoolSize());
386 >                    }});
387 >            assertTrue(threadsStarted.await(SMALL_DELAY_MS, MILLISECONDS));
388 >            assertEquals(THREADS, p.getLargestPoolSize());
389 >        } finally {
390 >            done.countDown();
391 >            joinPool(p);
392 >            assertEquals(THREADS, p.getLargestPoolSize());
393 >        }
394 >    }
395 >
396 >    /**
397 >     * getPoolSize increases, but doesn't overestimate, when threads
398 >     * become active
399 >     */
400 >    public void testGetPoolSize() throws InterruptedException {
401 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
402 >        final CountDownLatch threadStarted = new CountDownLatch(1);
403 >        final CountDownLatch done = new CountDownLatch(1);
404 >        try {
405 >            assertEquals(0, p.getPoolSize());
406 >            p.execute(new CheckedRunnable() {
407 >                public void realRun() throws InterruptedException {
408 >                    threadStarted.countDown();
409 >                    assertEquals(1, p.getPoolSize());
410 >                    done.await();
411 >                }});
412 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
413 >            assertEquals(1, p.getPoolSize());
414 >        } finally {
415 >            done.countDown();
416 >            joinPool(p);
417 >        }
418 >    }
419 >
420 >    /**
421 >     * getTaskCount increases, but doesn't overestimate, when tasks
422 >     * submitted
423 >     */
424 >    public void testGetTaskCount() throws InterruptedException {
425 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
426 >        final CountDownLatch threadStarted = new CountDownLatch(1);
427 >        final CountDownLatch done = new CountDownLatch(1);
428 >        final int TASKS = 5;
429 >        try {
430 >            assertEquals(0, p.getTaskCount());
431 >            for (int i = 0; i < TASKS; i++)
432 >                p.execute(new CheckedRunnable() {
433 >                    public void realRun() throws InterruptedException {
434 >                        threadStarted.countDown();
435 >                        done.await();
436 >                    }});
437 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
438 >            assertEquals(TASKS, p.getTaskCount());
439 >        } finally {
440 >            done.countDown();
441 >            joinPool(p);
442 >        }
443 >    }
444 >
445 >    /**
446 >     * getThreadFactory returns factory in constructor if not set
447 >     */
448 >    public void testGetThreadFactory() throws InterruptedException {
449 >        ThreadFactory tf = new SimpleThreadFactory();
450 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1, tf);
451 >        assertSame(tf, p.getThreadFactory());
452 >        joinPool(p);
453 >    }
454 >
455 >    /**
456 >     * setThreadFactory sets the thread factory returned by getThreadFactory
457 >     */
458 >    public void testSetThreadFactory() throws InterruptedException {
459 >        ThreadFactory tf = new SimpleThreadFactory();
460 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
461 >        p.setThreadFactory(tf);
462 >        assertSame(tf, p.getThreadFactory());
463 >        joinPool(p);
464 >    }
465 >
466 >    /**
467 >     * setThreadFactory(null) throws NPE
468 >     */
469 >    public void testSetThreadFactoryNull() throws InterruptedException {
470 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
471 >        try {
472 >            p.setThreadFactory(null);
473 >            shouldThrow();
474 >        } catch (NullPointerException success) {
475 >        } finally {
476 >            joinPool(p);
477 >        }
478      }
479  
480      /**
481 <     *   getActiveCount gives correct values
481 >     * isShutdown is false before shutdown, true after
482       */
483 <    public void testGetActiveCount(){
484 <        ScheduledExecutor two = new ScheduledExecutor(2);
485 <        assertEquals(0, two.getActiveCount());
486 <        two.execute(new SmallRunnable());
487 <        try{
262 <            Thread.sleep(SHORT_DELAY_MS);
263 <        } catch(Exception e){
264 <            fail("unexpected exception");
483 >    public void testIsShutdown() {
484 >
485 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
486 >        try {
487 >            assertFalse(p.isShutdown());
488          }
489 <        assertEquals(1, two.getActiveCount());
490 <        joinPool(two);
489 >        finally {
490 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
491 >        }
492 >        assertTrue(p.isShutdown());
493      }
494 <    
494 >
495      /**
496 <     *   getCompleteTaskCount gives correct values
496 >     * isTerminated is false before termination, true after
497       */
498 <    public void testGetCompletedTaskCount(){
499 <        ScheduledExecutor two = new ScheduledExecutor(2);
500 <        assertEquals(0, two.getCompletedTaskCount());
501 <        two.execute(new SmallRunnable());
502 <        try{
503 <            Thread.sleep(MEDIUM_DELAY_MS);
504 <        } catch(Exception e){
505 <            fail("unexpected exception");
498 >    public void testIsTerminated() throws InterruptedException {
499 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
500 >        final CountDownLatch threadStarted = new CountDownLatch(1);
501 >        final CountDownLatch done = new CountDownLatch(1);
502 >        assertFalse(p.isTerminated());
503 >        try {
504 >            p.execute(new CheckedRunnable() {
505 >                public void realRun() throws InterruptedException {
506 >                    assertFalse(p.isTerminated());
507 >                    threadStarted.countDown();
508 >                    done.await();
509 >                }});
510 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
511 >            assertFalse(p.isTerminating());
512 >            done.countDown();
513 >        } finally {
514 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
515          }
516 <        assertEquals(1, two.getCompletedTaskCount());
517 <        joinPool(two);
516 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
517 >        assertTrue(p.isTerminated());
518      }
519 <    
519 >
520      /**
521 <     *   getCorePoolSize gives correct values
521 >     * isTerminating is not true when running or when terminated
522       */
523 <    public void testGetCorePoolSize(){
524 <        ScheduledExecutor one = new ScheduledExecutor(1);
525 <        assertEquals(1, one.getCorePoolSize());
526 <        joinPool(one);
523 >    public void testIsTerminating() throws InterruptedException {
524 >        final ThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
525 >        final CountDownLatch threadStarted = new CountDownLatch(1);
526 >        final CountDownLatch done = new CountDownLatch(1);
527 >        try {
528 >            assertFalse(p.isTerminating());
529 >            p.execute(new CheckedRunnable() {
530 >                public void realRun() throws InterruptedException {
531 >                    assertFalse(p.isTerminating());
532 >                    threadStarted.countDown();
533 >                    done.await();
534 >                }});
535 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
536 >            assertFalse(p.isTerminating());
537 >            done.countDown();
538 >        } finally {
539 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
540 >        }
541 >        assertTrue(p.awaitTermination(LONG_DELAY_MS, MILLISECONDS));
542 >        assertTrue(p.isTerminated());
543 >        assertFalse(p.isTerminating());
544      }
545 <    
545 >
546      /**
547 <     *   getLargestPoolSize gives correct values
547 >     * getQueue returns the work queue, which contains queued tasks
548       */
549 <    public void testGetLargestPoolSize(){
550 <        ScheduledExecutor two = new ScheduledExecutor(2);
551 <        assertEquals(0, two.getLargestPoolSize());
552 <        two.execute(new SmallRunnable());
553 <        two.execute(new SmallRunnable());
554 <        try{
555 <            Thread.sleep(SHORT_DELAY_MS);
556 <        } catch(Exception e){
557 <            fail("unexpected exception");
549 >    public void testGetQueue() throws InterruptedException {
550 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
551 >        final CountDownLatch threadStarted = new CountDownLatch(1);
552 >        final CountDownLatch done = new CountDownLatch(1);
553 >        try {
554 >            ScheduledFuture[] tasks = new ScheduledFuture[5];
555 >            for (int i = 0; i < tasks.length; i++) {
556 >                Runnable r = new CheckedRunnable() {
557 >                    public void realRun() throws InterruptedException {
558 >                        threadStarted.countDown();
559 >                        done.await();
560 >                    }};
561 >                tasks[i] = p.schedule(r, 1, MILLISECONDS);
562 >            }
563 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
564 >            BlockingQueue<Runnable> q = p.getQueue();
565 >            assertTrue(q.contains(tasks[tasks.length - 1]));
566 >            assertFalse(q.contains(tasks[0]));
567 >        } finally {
568 >            done.countDown();
569 >            joinPool(p);
570          }
308        assertEquals(2, two.getLargestPoolSize());
309        joinPool(two);
571      }
572 <    
572 >
573      /**
574 <     *   getPoolSize gives correct values
574 >     * remove(task) removes queued task, and fails to remove active task
575       */
576 <    public void testGetPoolSize(){
577 <        ScheduledExecutor one = new ScheduledExecutor(1);
578 <        assertEquals(0, one.getPoolSize());
579 <        one.execute(new SmallRunnable());
580 <        assertEquals(1, one.getPoolSize());
581 <        joinPool(one);
576 >    public void testRemove() throws InterruptedException {
577 >        final ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
578 >        ScheduledFuture[] tasks = new ScheduledFuture[5];
579 >        final CountDownLatch threadStarted = new CountDownLatch(1);
580 >        final CountDownLatch done = new CountDownLatch(1);
581 >        try {
582 >            for (int i = 0; i < tasks.length; i++) {
583 >                Runnable r = new CheckedRunnable() {
584 >                    public void realRun() throws InterruptedException {
585 >                        threadStarted.countDown();
586 >                        done.await();
587 >                    }};
588 >                tasks[i] = p.schedule(r, 1, MILLISECONDS);
589 >            }
590 >            assertTrue(threadStarted.await(SMALL_DELAY_MS, MILLISECONDS));
591 >            BlockingQueue<Runnable> q = p.getQueue();
592 >            assertFalse(p.remove((Runnable)tasks[0]));
593 >            assertTrue(q.contains((Runnable)tasks[4]));
594 >            assertTrue(q.contains((Runnable)tasks[3]));
595 >            assertTrue(p.remove((Runnable)tasks[4]));
596 >            assertFalse(p.remove((Runnable)tasks[4]));
597 >            assertFalse(q.contains((Runnable)tasks[4]));
598 >            assertTrue(q.contains((Runnable)tasks[3]));
599 >            assertTrue(p.remove((Runnable)tasks[3]));
600 >            assertFalse(q.contains((Runnable)tasks[3]));
601 >        } finally {
602 >            done.countDown();
603 >            joinPool(p);
604 >        }
605      }
606 <    
606 >
607      /**
608 <     *   getTaskCount gives correct values
608 >     * purge eventually removes cancelled tasks from the queue
609       */
610 <    public void testGetTaskCount(){
611 <        ScheduledExecutor one = new ScheduledExecutor(1);
612 <        assertEquals(0, one.getTaskCount());
613 <        for(int i = 0; i < 5; i++)
614 <            one.execute(new SmallRunnable());
615 <        try{
616 <            Thread.sleep(SHORT_DELAY_MS);
617 <        } catch(Exception e){
618 <            fail("unexpected exception");
610 >    public void testPurge() throws InterruptedException {
611 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
612 >        ScheduledFuture[] tasks = new ScheduledFuture[5];
613 >        for (int i = 0; i < tasks.length; i++)
614 >            tasks[i] = p.schedule(new SmallPossiblyInterruptedRunnable(),
615 >                                  LONG_DELAY_MS, MILLISECONDS);
616 >        try {
617 >            int max = tasks.length;
618 >            if (tasks[4].cancel(true)) --max;
619 >            if (tasks[3].cancel(true)) --max;
620 >            // There must eventually be an interference-free point at
621 >            // which purge will not fail. (At worst, when queue is empty.)
622 >            long startTime = System.nanoTime();
623 >            do {
624 >                p.purge();
625 >                long count = p.getTaskCount();
626 >                if (count == max)
627 >                    return;
628 >            } while (millisElapsedSince(startTime) < MEDIUM_DELAY_MS);
629 >            fail("Purge failed to remove cancelled tasks");
630 >        } finally {
631 >            for (ScheduledFuture task : tasks)
632 >                task.cancel(true);
633 >            joinPool(p);
634          }
336        assertEquals(5, one.getTaskCount());
337        joinPool(one);
635      }
636 <    
636 >
637      /**
638 <     *   isShutDown gives correct values
638 >     * shutdownNow returns a list containing tasks that were not run
639       */
640 <    public void testIsShutdown(){
641 <        
642 <        ScheduledExecutor one = new ScheduledExecutor(1);
640 >    public void testShutdownNow() {
641 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
642 >        for (int i = 0; i < 5; i++)
643 >            p.schedule(new SmallPossiblyInterruptedRunnable(),
644 >                       LONG_DELAY_MS, MILLISECONDS);
645          try {
646 <            assertFalse(one.isShutdown());
646 >            List<Runnable> l = p.shutdownNow();
647 >            assertTrue(p.isShutdown());
648 >            assertEquals(5, l.size());
649 >        } catch (SecurityException ok) {
650 >            // Allowed in case test doesn't have privs
651 >        } finally {
652 >            joinPool(p);
653 >        }
654 >    }
655 >
656 >    /**
657 >     * In default setting, shutdown cancels periodic but not delayed
658 >     * tasks at shutdown
659 >     */
660 >    public void testShutdown1() throws InterruptedException {
661 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
662 >        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
663 >        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
664 >
665 >        ScheduledFuture[] tasks = new ScheduledFuture[5];
666 >        for (int i = 0; i < tasks.length; i++)
667 >            tasks[i] = p.schedule(new NoOpRunnable(),
668 >                                  SHORT_DELAY_MS, MILLISECONDS);
669 >        try { p.shutdown(); } catch (SecurityException ok) { return; }
670 >        BlockingQueue<Runnable> q = p.getQueue();
671 >        for (ScheduledFuture task : tasks) {
672 >            assertFalse(task.isDone());
673 >            assertFalse(task.isCancelled());
674 >            assertTrue(q.contains(task));
675 >        }
676 >        assertTrue(p.isShutdown());
677 >        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
678 >        assertTrue(p.isTerminated());
679 >        for (ScheduledFuture task : tasks) {
680 >            assertTrue(task.isDone());
681 >            assertFalse(task.isCancelled());
682 >        }
683 >    }
684 >
685 >    /**
686 >     * If setExecuteExistingDelayedTasksAfterShutdownPolicy is false,
687 >     * delayed tasks are cancelled at shutdown
688 >     */
689 >    public void testShutdown2() throws InterruptedException {
690 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
691 >        p.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
692 >        assertFalse(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
693 >        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
694 >        ScheduledFuture[] tasks = new ScheduledFuture[5];
695 >        for (int i = 0; i < tasks.length; i++)
696 >            tasks[i] = p.schedule(new NoOpRunnable(),
697 >                                  SHORT_DELAY_MS, MILLISECONDS);
698 >        BlockingQueue q = p.getQueue();
699 >        assertEquals(tasks.length, q.size());
700 >        try { p.shutdown(); } catch (SecurityException ok) { return; }
701 >        assertTrue(p.isShutdown());
702 >        assertTrue(q.isEmpty());
703 >        assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
704 >        assertTrue(p.isTerminated());
705 >        for (ScheduledFuture task : tasks) {
706 >            assertTrue(task.isDone());
707 >            assertTrue(task.isCancelled());
708 >        }
709 >    }
710 >
711 >    /**
712 >     * If setContinueExistingPeriodicTasksAfterShutdownPolicy is set false,
713 >     * periodic tasks are cancelled at shutdown
714 >     */
715 >    public void testShutdown3() throws InterruptedException {
716 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
717 >        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
718 >        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
719 >        p.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
720 >        assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
721 >        assertFalse(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
722 >        long initialDelay = LONG_DELAY_MS;
723 >        ScheduledFuture task =
724 >            p.scheduleAtFixedRate(new NoOpRunnable(), initialDelay,
725 >                                  5, MILLISECONDS);
726 >        try { p.shutdown(); } catch (SecurityException ok) { return; }
727 >        assertTrue(p.isShutdown());
728 >        assertTrue(p.getQueue().isEmpty());
729 >        assertTrue(task.isDone());
730 >        assertTrue(task.isCancelled());
731 >        joinPool(p);
732 >    }
733 >
734 >    /**
735 >     * if setContinueExistingPeriodicTasksAfterShutdownPolicy is true,
736 >     * periodic tasks are not cancelled at shutdown
737 >     */
738 >    public void testShutdown4() throws InterruptedException {
739 >        ScheduledThreadPoolExecutor p = new ScheduledThreadPoolExecutor(1);
740 >        final CountDownLatch counter = new CountDownLatch(2);
741 >        try {
742 >            p.setContinueExistingPeriodicTasksAfterShutdownPolicy(true);
743 >            assertTrue(p.getExecuteExistingDelayedTasksAfterShutdownPolicy());
744 >            assertTrue(p.getContinueExistingPeriodicTasksAfterShutdownPolicy());
745 >            final Runnable r = new CheckedRunnable() {
746 >                public void realRun() {
747 >                    counter.countDown();
748 >                }};
749 >            ScheduledFuture task =
750 >                p.scheduleAtFixedRate(r, 1, 1, MILLISECONDS);
751 >            assertFalse(task.isDone());
752 >            assertFalse(task.isCancelled());
753 >            try { p.shutdown(); } catch (SecurityException ok) { return; }
754 >            assertFalse(task.isCancelled());
755 >            assertFalse(p.isTerminated());
756 >            assertTrue(p.isShutdown());
757 >            assertTrue(counter.await(SMALL_DELAY_MS, MILLISECONDS));
758 >            assertFalse(task.isCancelled());
759 >            assertTrue(task.cancel(false));
760 >            assertTrue(task.isDone());
761 >            assertTrue(task.isCancelled());
762 >            assertTrue(p.awaitTermination(SMALL_DELAY_MS, MILLISECONDS));
763 >            assertTrue(p.isTerminated());
764          }
765          finally {
766 <            one.shutdown();
766 >            joinPool(p);
767          }
352        assertTrue(one.isShutdown());
768      }
769  
355        
770      /**
771 <     *   isTerminated gives correct values
358 <     *  Makes sure termination does not take an innapropriate
359 <     *  amount of time
771 >     * completed submit of callable returns result
772       */
773 <    public void testIsTerminated(){
774 <        ScheduledExecutor one = new ScheduledExecutor(1);
773 >    public void testSubmitCallable() throws Exception {
774 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
775          try {
776 <            one.execute(new SmallRunnable());
776 >            Future<String> future = e.submit(new StringTask());
777 >            String result = future.get();
778 >            assertSame(TEST_STRING, result);
779          } finally {
780 <            one.shutdown();
780 >            joinPool(e);
781          }
782 +    }
783 +
784 +    /**
785 +     * completed submit of runnable returns successfully
786 +     */
787 +    public void testSubmitRunnable() throws Exception {
788 +        ExecutorService e = new ScheduledThreadPoolExecutor(2);
789          try {
790 <            assertTrue(one.awaitTermination(LONG_DELAY_MS, TimeUnit.MILLISECONDS));
791 <            assertTrue(one.isTerminated());
792 <        } catch(Exception e){
793 <            fail("unexpected exception");
794 <        }      
790 >            Future<?> future = e.submit(new NoOpRunnable());
791 >            future.get();
792 >            assertTrue(future.isDone());
793 >        } finally {
794 >            joinPool(e);
795 >        }
796      }
797  
798      /**
799 <     *   that purge correctly removes cancelled tasks
378 <     *  from the queue
799 >     * completed submit of (runnable, result) returns result
800       */
801 <    public void testPurge(){
802 <        ScheduledExecutor one = new ScheduledExecutor(1);
803 <        ScheduledCancellable[] tasks = new ScheduledCancellable[5];
804 <        for(int i = 0; i < 5; i++){
805 <            tasks[i] = one.schedule(new SmallRunnable(), 1, TimeUnit.MILLISECONDS);
801 >    public void testSubmitRunnable2() throws Exception {
802 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
803 >        try {
804 >            Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
805 >            String result = future.get();
806 >            assertSame(TEST_STRING, result);
807 >        } finally {
808 >            joinPool(e);
809          }
386        int max = 5;
387        if (tasks[4].cancel(true)) --max;
388        if (tasks[3].cancel(true)) --max;
389        one.purge();
390        long count = one.getTaskCount();
391        assertTrue(count > 0 && count <= max);
392        joinPool(one);
810      }
811  
812      /**
813 <     *   shutDownNow returns a list
397 <     *  containing the correct number of elements
813 >     * invokeAny(null) throws NPE
814       */
815 <    public void testShutDownNow(){
816 <        ScheduledExecutor one = new ScheduledExecutor(1);
817 <        for(int i = 0; i < 5; i++)
818 <            one.schedule(new SmallRunnable(), SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
819 <        List l = one.shutdownNow();
820 <        assertTrue(one.isShutdown());
821 <        assertTrue(l.size() > 0 && l.size() <= 5);
822 <        joinPool(one);
815 >    public void testInvokeAny1() throws Exception {
816 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
817 >        try {
818 >            e.invokeAny(null);
819 >            shouldThrow();
820 >        } catch (NullPointerException success) {
821 >        } finally {
822 >            joinPool(e);
823 >        }
824      }
825  
826 <    public void testShutDown1(){
826 >    /**
827 >     * invokeAny(empty collection) throws IAE
828 >     */
829 >    public void testInvokeAny2() throws Exception {
830 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
831          try {
832 <            ScheduledExecutor one = new ScheduledExecutor(1);
833 <            assertTrue(one.getExecuteExistingDelayedTasksAfterShutdownPolicy());
834 <            assertFalse(one.getContinueExistingPeriodicTasksAfterShutdownPolicy());
832 >            e.invokeAny(new ArrayList<Callable<String>>());
833 >            shouldThrow();
834 >        } catch (IllegalArgumentException success) {
835 >        } finally {
836 >            joinPool(e);
837 >        }
838 >    }
839  
840 <            ScheduledCancellable[] tasks = new ScheduledCancellable[5];
841 <            for(int i = 0; i < 5; i++)
842 <                tasks[i] = one.schedule(new NoOpRunnable(), SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
843 <            one.shutdown();
844 <            BlockingQueue q = one.getQueue();
845 <            for (Iterator it = q.iterator(); it.hasNext();) {
846 <                ScheduledCancellable t = (ScheduledCancellable)it.next();
847 <                assertFalse(t.isCancelled());
848 <            }
849 <            assertTrue(one.isShutdown());
850 <            Thread.sleep(SMALL_DELAY_MS);
851 <            for (int i = 0; i < 5; ++i) {
852 <                assertTrue(tasks[i].isDone());
853 <                assertFalse(tasks[i].isCancelled());
854 <            }
855 <            
840 >    /**
841 >     * invokeAny(c) throws NPE if c has null elements
842 >     */
843 >    public void testInvokeAny3() throws Exception {
844 >        CountDownLatch latch = new CountDownLatch(1);
845 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
846 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
847 >        l.add(latchAwaitingStringTask(latch));
848 >        l.add(null);
849 >        try {
850 >            e.invokeAny(l);
851 >            shouldThrow();
852 >        } catch (NullPointerException success) {
853 >        } finally {
854 >            latch.countDown();
855 >            joinPool(e);
856          }
857 <        catch(Exception ex) {
858 <            fail("unexpected exception");
857 >    }
858 >
859 >    /**
860 >     * invokeAny(c) throws ExecutionException if no task completes
861 >     */
862 >    public void testInvokeAny4() throws Exception {
863 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
864 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
865 >        l.add(new NPETask());
866 >        try {
867 >            e.invokeAny(l);
868 >            shouldThrow();
869 >        } catch (ExecutionException success) {
870 >            assertTrue(success.getCause() instanceof NullPointerException);
871 >        } finally {
872 >            joinPool(e);
873          }
874      }
875  
876 +    /**
877 +     * invokeAny(c) returns result of some task
878 +     */
879 +    public void testInvokeAny5() throws Exception {
880 +        ExecutorService e = new ScheduledThreadPoolExecutor(2);
881 +        try {
882 +            List<Callable<String>> l = new ArrayList<Callable<String>>();
883 +            l.add(new StringTask());
884 +            l.add(new StringTask());
885 +            String result = e.invokeAny(l);
886 +            assertSame(TEST_STRING, result);
887 +        } finally {
888 +            joinPool(e);
889 +        }
890 +    }
891  
892 <    public void testShutDown2(){
892 >    /**
893 >     * invokeAll(null) throws NPE
894 >     */
895 >    public void testInvokeAll1() throws Exception {
896 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
897          try {
898 <            ScheduledExecutor one = new ScheduledExecutor(1);
899 <            one.setExecuteExistingDelayedTasksAfterShutdownPolicy(false);
900 <            ScheduledCancellable[] tasks = new ScheduledCancellable[5];
901 <            for(int i = 0; i < 5; i++)
902 <                tasks[i] = one.schedule(new NoOpRunnable(), SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
445 <            one.shutdown();
446 <            assertTrue(one.isShutdown());
447 <            BlockingQueue q = one.getQueue();
448 <            assertTrue(q.isEmpty());
449 <            Thread.sleep(SMALL_DELAY_MS);
450 <            assertTrue(one.isTerminated());
898 >            e.invokeAll(null);
899 >            shouldThrow();
900 >        } catch (NullPointerException success) {
901 >        } finally {
902 >            joinPool(e);
903          }
904 <        catch(Exception ex) {
905 <            fail("unexpected exception");
904 >    }
905 >
906 >    /**
907 >     * invokeAll(empty collection) returns empty collection
908 >     */
909 >    public void testInvokeAll2() throws Exception {
910 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
911 >        try {
912 >            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
913 >            assertTrue(r.isEmpty());
914 >        } finally {
915 >            joinPool(e);
916          }
917      }
918  
919 +    /**
920 +     * invokeAll(c) throws NPE if c has null elements
921 +     */
922 +    public void testInvokeAll3() throws Exception {
923 +        ExecutorService e = new ScheduledThreadPoolExecutor(2);
924 +        List<Callable<String>> l = new ArrayList<Callable<String>>();
925 +        l.add(new StringTask());
926 +        l.add(null);
927 +        try {
928 +            e.invokeAll(l);
929 +            shouldThrow();
930 +        } catch (NullPointerException success) {
931 +        } finally {
932 +            joinPool(e);
933 +        }
934 +    }
935  
936 <    public void testShutDown3(){
936 >    /**
937 >     * get of invokeAll(c) throws exception on failed task
938 >     */
939 >    public void testInvokeAll4() throws Exception {
940 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
941 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
942 >        l.add(new NPETask());
943 >        List<Future<String>> futures = e.invokeAll(l);
944 >        assertEquals(1, futures.size());
945          try {
946 <            ScheduledExecutor one = new ScheduledExecutor(1);
947 <            one.setContinueExistingPeriodicTasksAfterShutdownPolicy(false);
948 <            ScheduledCancellable task =
949 <                one.scheduleAtFixedRate(new NoOpRunnable(), 5, 5, TimeUnit.MILLISECONDS);
950 <            one.shutdown();
951 <            assertTrue(one.isShutdown());
466 <            BlockingQueue q = one.getQueue();
467 <            assertTrue(q.isEmpty());
468 <            Thread.sleep(SHORT_DELAY_MS);
469 <            assertTrue(one.isTerminated());
946 >            futures.get(0).get();
947 >            shouldThrow();
948 >        } catch (ExecutionException success) {
949 >            assertTrue(success.getCause() instanceof NullPointerException);
950 >        } finally {
951 >            joinPool(e);
952          }
953 <        catch(Exception ex) {
954 <            fail("unexpected exception");
953 >    }
954 >
955 >    /**
956 >     * invokeAll(c) returns results of all completed tasks
957 >     */
958 >    public void testInvokeAll5() throws Exception {
959 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
960 >        try {
961 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
962 >            l.add(new StringTask());
963 >            l.add(new StringTask());
964 >            List<Future<String>> futures = e.invokeAll(l);
965 >            assertEquals(2, futures.size());
966 >            for (Future<String> future : futures)
967 >                assertSame(TEST_STRING, future.get());
968 >        } finally {
969 >            joinPool(e);
970          }
971      }
972  
973 <    public void testShutDown4(){
974 <        ScheduledExecutor one = new ScheduledExecutor(1);
973 >    /**
974 >     * timed invokeAny(null) throws NPE
975 >     */
976 >    public void testTimedInvokeAny1() throws Exception {
977 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
978          try {
979 <            one.setContinueExistingPeriodicTasksAfterShutdownPolicy(true);
980 <            ScheduledCancellable task =
981 <                one.scheduleAtFixedRate(new NoOpRunnable(), 5, 5, TimeUnit.MILLISECONDS);
982 <            assertFalse(task.isCancelled());
983 <            one.shutdown();
984 <            assertFalse(task.isCancelled());
985 <            assertFalse(one.isTerminated());
986 <            assertTrue(one.isShutdown());
987 <            Thread.sleep(SHORT_DELAY_MS);
988 <            assertFalse(task.isCancelled());
989 <            task.cancel(true);
990 <            assertTrue(task.isCancelled());
991 <            Thread.sleep(SHORT_DELAY_MS);
992 <            assertTrue(one.isTerminated());
979 >            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
980 >            shouldThrow();
981 >        } catch (NullPointerException success) {
982 >        } finally {
983 >            joinPool(e);
984 >        }
985 >    }
986 >
987 >    /**
988 >     * timed invokeAny(,,null) throws NPE
989 >     */
990 >    public void testTimedInvokeAnyNullTimeUnit() throws Exception {
991 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
992 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
993 >        l.add(new StringTask());
994 >        try {
995 >            e.invokeAny(l, MEDIUM_DELAY_MS, null);
996 >            shouldThrow();
997 >        } catch (NullPointerException success) {
998 >        } finally {
999 >            joinPool(e);
1000          }
1001 <        catch(Exception ex) {
1002 <            fail("unexpected exception");
1001 >    }
1002 >
1003 >    /**
1004 >     * timed invokeAny(empty collection) throws IAE
1005 >     */
1006 >    public void testTimedInvokeAny2() throws Exception {
1007 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1008 >        try {
1009 >            e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1010 >            shouldThrow();
1011 >        } catch (IllegalArgumentException success) {
1012 >        } finally {
1013 >            joinPool(e);
1014          }
1015 <        finally {
1016 <            one.shutdownNow();
1015 >    }
1016 >
1017 >    /**
1018 >     * timed invokeAny(c) throws NPE if c has null elements
1019 >     */
1020 >    public void testTimedInvokeAny3() throws Exception {
1021 >        CountDownLatch latch = new CountDownLatch(1);
1022 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1023 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
1024 >        l.add(latchAwaitingStringTask(latch));
1025 >        l.add(null);
1026 >        try {
1027 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1028 >            shouldThrow();
1029 >        } catch (NullPointerException success) {
1030 >        } finally {
1031 >            latch.countDown();
1032 >            joinPool(e);
1033 >        }
1034 >    }
1035 >
1036 >    /**
1037 >     * timed invokeAny(c) throws ExecutionException if no task completes
1038 >     */
1039 >    public void testTimedInvokeAny4() throws Exception {
1040 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1041 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
1042 >        l.add(new NPETask());
1043 >        try {
1044 >            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1045 >            shouldThrow();
1046 >        } catch (ExecutionException success) {
1047 >            assertTrue(success.getCause() instanceof NullPointerException);
1048 >        } finally {
1049 >            joinPool(e);
1050 >        }
1051 >    }
1052 >
1053 >    /**
1054 >     * timed invokeAny(c) returns result of some task
1055 >     */
1056 >    public void testTimedInvokeAny5() throws Exception {
1057 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1058 >        try {
1059 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1060 >            l.add(new StringTask());
1061 >            l.add(new StringTask());
1062 >            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
1063 >            assertSame(TEST_STRING, result);
1064 >        } finally {
1065 >            joinPool(e);
1066 >        }
1067 >    }
1068 >
1069 >    /**
1070 >     * timed invokeAll(null) throws NPE
1071 >     */
1072 >    public void testTimedInvokeAll1() throws Exception {
1073 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1074 >        try {
1075 >            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
1076 >            shouldThrow();
1077 >        } catch (NullPointerException success) {
1078 >        } finally {
1079 >            joinPool(e);
1080 >        }
1081 >    }
1082 >
1083 >    /**
1084 >     * timed invokeAll(,,null) throws NPE
1085 >     */
1086 >    public void testTimedInvokeAllNullTimeUnit() throws Exception {
1087 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1088 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
1089 >        l.add(new StringTask());
1090 >        try {
1091 >            e.invokeAll(l, MEDIUM_DELAY_MS, null);
1092 >            shouldThrow();
1093 >        } catch (NullPointerException success) {
1094 >        } finally {
1095 >            joinPool(e);
1096 >        }
1097 >    }
1098 >
1099 >    /**
1100 >     * timed invokeAll(empty collection) returns empty collection
1101 >     */
1102 >    public void testTimedInvokeAll2() throws Exception {
1103 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1104 >        try {
1105 >            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
1106 >            assertTrue(r.isEmpty());
1107 >        } finally {
1108 >            joinPool(e);
1109 >        }
1110 >    }
1111 >
1112 >    /**
1113 >     * timed invokeAll(c) throws NPE if c has null elements
1114 >     */
1115 >    public void testTimedInvokeAll3() throws Exception {
1116 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1117 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
1118 >        l.add(new StringTask());
1119 >        l.add(null);
1120 >        try {
1121 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1122 >            shouldThrow();
1123 >        } catch (NullPointerException success) {
1124 >        } finally {
1125 >            joinPool(e);
1126 >        }
1127 >    }
1128 >
1129 >    /**
1130 >     * get of element of invokeAll(c) throws exception on failed task
1131 >     */
1132 >    public void testTimedInvokeAll4() throws Exception {
1133 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1134 >        List<Callable<String>> l = new ArrayList<Callable<String>>();
1135 >        l.add(new NPETask());
1136 >        List<Future<String>> futures =
1137 >            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1138 >        assertEquals(1, futures.size());
1139 >        try {
1140 >            futures.get(0).get();
1141 >            shouldThrow();
1142 >        } catch (ExecutionException success) {
1143 >            assertTrue(success.getCause() instanceof NullPointerException);
1144 >        } finally {
1145 >            joinPool(e);
1146 >        }
1147 >    }
1148 >
1149 >    /**
1150 >     * timed invokeAll(c) returns results of all completed tasks
1151 >     */
1152 >    public void testTimedInvokeAll5() throws Exception {
1153 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1154 >        try {
1155 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1156 >            l.add(new StringTask());
1157 >            l.add(new StringTask());
1158 >            List<Future<String>> futures =
1159 >                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
1160 >            assertEquals(2, futures.size());
1161 >            for (Future<String> future : futures)
1162 >                assertSame(TEST_STRING, future.get());
1163 >        } finally {
1164 >            joinPool(e);
1165 >        }
1166 >    }
1167 >
1168 >    /**
1169 >     * timed invokeAll(c) cancels tasks not completed by timeout
1170 >     */
1171 >    public void testTimedInvokeAll6() throws Exception {
1172 >        ExecutorService e = new ScheduledThreadPoolExecutor(2);
1173 >        try {
1174 >            List<Callable<String>> l = new ArrayList<Callable<String>>();
1175 >            l.add(new StringTask());
1176 >            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
1177 >            l.add(new StringTask());
1178 >            List<Future<String>> futures =
1179 >                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
1180 >            assertEquals(l.size(), futures.size());
1181 >            for (Future future : futures)
1182 >                assertTrue(future.isDone());
1183 >            assertFalse(futures.get(0).isCancelled());
1184 >            assertTrue(futures.get(1).isCancelled());
1185 >        } finally {
1186 >            joinPool(e);
1187          }
1188      }
1189  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines