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

Comparing jsr166/src/test/tck/FutureTaskTest.java (file contents):
Revision 1.5 by dl, Sat Sep 20 18:20:07 2003 UTC vs.
Revision 1.25 by dl, Fri May 6 11:22:07 2011 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.concurrent.*;
11 + import static java.util.concurrent.TimeUnit.MILLISECONDS;
12 + import static java.util.concurrent.TimeUnit.SECONDS;
13   import java.util.*;
14  
15   public class FutureTaskTest extends JSR166TestCase {
16  
17      public static void main(String[] args) {
18 <        junit.textui.TestRunner.run (suite());  
18 >        junit.textui.TestRunner.run(suite());
19      }
20      public static Test suite() {
21 <        return new TestSuite(FutureTaskTest.class);
21 >        return new TestSuite(FutureTaskTest.class);
22 >    }
23 >
24 >    void checkNotDone(Future<?> f) {
25 >        assertFalse(f.isDone());
26 >        assertFalse(f.isCancelled());
27 >    }
28 >
29 >    <T> void checkCompletedNormally(Future<T> f, T expected) {
30 >        assertTrue(f.isDone());
31 >        assertFalse(f.isCancelled());
32 >
33 >        try {
34 >            assertSame(expected, f.get());
35 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
36 >        try {
37 >            assertSame(expected, f.get(5L, SECONDS));
38 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
39 >
40 >        assertFalse(f.cancel(false));
41 >        assertFalse(f.cancel(true));
42 >    }
43 >
44 >    void checkCancelled(Future<?> f) {
45 >        assertTrue(f.isDone());
46 >        assertTrue(f.isCancelled());
47 >
48 >        try {
49 >            f.get();
50 >            shouldThrow();
51 >        } catch (CancellationException success) {
52 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
53 >
54 >        try {
55 >            f.get(5L, SECONDS);
56 >            shouldThrow();
57 >        } catch (CancellationException success) {
58 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
59 >
60 >        assertFalse(f.cancel(false));
61 >        assertFalse(f.cancel(true));
62 >    }
63 >
64 >    void checkCompletedAbnormally(Future<?> f, Throwable t) {
65 >        assertTrue(f.isDone());
66 >        assertFalse(f.isCancelled());
67 >
68 >        try {
69 >            f.get();
70 >            shouldThrow();
71 >        } catch (ExecutionException success) {
72 >            assertSame(t, success.getCause());
73 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
74 >
75 >        try {
76 >            f.get(5L, SECONDS);
77 >            shouldThrow();
78 >        } catch (ExecutionException success) {
79 >            assertSame(t, success.getCause());
80 >        } catch (Throwable fail) { threadUnexpectedException(fail); }
81 >
82 >        assertFalse(f.cancel(false));
83 >        assertFalse(f.cancel(true));
84      }
85  
86      /**
87       * Subclass to expose protected methods
88       */
89 <    static class MyFutureTask extends FutureTask {
90 <        public MyFutureTask(Callable r) { super(r); }
91 <        public boolean reset() { return super.reset(); }
27 <        public void setCancelled() { super.setCancelled(); }
28 <        public void setDone() { super.setDone(); }
89 >    static class PublicFutureTask extends FutureTask {
90 >        public PublicFutureTask(Callable r) { super(r); }
91 >        public boolean runAndReset() { return super.runAndReset(); }
92          public void set(Object x) { super.set(x); }
93          public void setException(Throwable t) { super.setException(t); }
94      }
95  
96      /**
97 <     *
97 >     * Creating a future with a null callable throws NPE
98       */
99      public void testConstructor() {
100          try {
101              FutureTask task = new FutureTask(null);
102              shouldThrow();
103 <        }
41 <        catch(NullPointerException success) {
42 <        }
103 >        } catch (NullPointerException success) {}
104      }
105  
106      /**
107 <     *
107 >     * creating a future with null runnable fails
108       */
109      public void testConstructor2() {
110          try {
111              FutureTask task = new FutureTask(null, Boolean.TRUE);
112              shouldThrow();
113 <        }
53 <        catch(NullPointerException success) {
54 <        }
113 >        } catch (NullPointerException success) {}
114      }
115  
116      /**
117 <     *
117 >     * isDone is true when a task completes
118       */
119      public void testIsDone() {
120 <        FutureTask task = new FutureTask( new NoOpCallable());
121 <        task.run();
122 <        assertTrue(task.isDone());
123 <        assertFalse(task.isCancelled());
120 >        FutureTask task = new FutureTask(new NoOpCallable());
121 >        task.run();
122 >        assertTrue(task.isDone());
123 >        checkCompletedNormally(task, Boolean.TRUE);
124      }
125  
126      /**
127 <     *
127 >     * runAndReset of a non-cancelled task succeeds
128       */
129 <    public void testReset() {
130 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
131 <        task.run();
132 <        assertTrue(task.isDone());
74 <        assertTrue(task.reset());
129 >    public void testRunAndReset() {
130 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
131 >        assertTrue(task.runAndReset());
132 >        checkNotDone(task);
133      }
134  
135      /**
136 <     *
136 >     * runAndReset after cancellation fails
137       */
138      public void testResetAfterCancel() {
139 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
139 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
140          assertTrue(task.cancel(false));
141 <        task.run();
142 <        assertTrue(task.isDone());
85 <        assertTrue(task.isCancelled());
86 <        assertFalse(task.reset());
141 >        assertFalse(task.runAndReset());
142 >        checkCancelled(task);
143      }
144  
89    /**
90     *
91     */
92    public void testSetDone() {
93        MyFutureTask task = new MyFutureTask(new NoOpCallable());
94        task.setDone();
95        assertTrue(task.isDone());
96        assertFalse(task.isCancelled());
97    }
145  
146      /**
147 <     *
147 >     * setting value causes get to return it
148       */
149 <    public void testSetCancelled() {
150 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
104 <        assertTrue(task.cancel(false));
105 <        task.setCancelled();
106 <        assertTrue(task.isDone());
107 <        assertTrue(task.isCancelled());
108 <    }
109 <
110 <    /**
111 <     *
112 <     */
113 <    public void testSet() {
114 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
149 >    public void testSet() throws Exception {
150 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
151          task.set(one);
152 <        try {
153 <            assertEquals(task.get(), one);
118 <        }
119 <        catch(Exception e) {
120 <            unexpectedException();
121 <        }
152 >        assertSame(task.get(), one);
153 >        checkCompletedNormally(task, one);
154      }
155  
156      /**
157 <     *
157 >     * setException causes get to throw ExecutionException
158       */
159 <    public void testSetException() {
159 >    public void testSetException() throws Exception {
160          Exception nse = new NoSuchElementException();
161 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
161 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
162          task.setException(nse);
163          try {
164              Object x = task.get();
165              shouldThrow();
166 <        }
167 <        catch(ExecutionException ee) {
168 <            Throwable cause = ee.getCause();
137 <            assertEquals(cause, nse);
138 <        }
139 <        catch(Exception e) {
140 <            unexpectedException();
166 >        } catch (ExecutionException success) {
167 >            assertSame(success.getCause(), nse);
168 >            checkCompletedAbnormally(task, nse);
169          }
170      }
171  
172      /**
173 <     *
173 >     * Cancelling before running succeeds
174       */
175      public void testCancelBeforeRun() {
176 <        FutureTask task = new FutureTask( new NoOpCallable());
176 >        FutureTask task = new FutureTask(new NoOpCallable());
177          assertTrue(task.cancel(false));
178 <        task.run();
179 <        assertTrue(task.isDone());
152 <        assertTrue(task.isCancelled());
178 >        task.run();
179 >        checkCancelled(task);
180      }
181  
182      /**
183 <     *
183 >     * Cancel(true) before run succeeds
184       */
185      public void testCancelBeforeRun2() {
186 <        FutureTask task = new FutureTask( new NoOpCallable());
186 >        FutureTask task = new FutureTask(new NoOpCallable());
187          assertTrue(task.cancel(true));
188 <        task.run();
189 <        assertTrue(task.isDone());
163 <        assertTrue(task.isCancelled());
188 >        task.run();
189 >        checkCancelled(task);
190      }
191  
192      /**
193 <     *
193 >     * cancel of a completed task fails
194       */
195      public void testCancelAfterRun() {
196 <        FutureTask task = new FutureTask( new NoOpCallable());
197 <        task.run();
196 >        FutureTask task = new FutureTask(new NoOpCallable());
197 >        task.run();
198          assertFalse(task.cancel(false));
199 <        assertTrue(task.isDone());
174 <        assertFalse(task.isCancelled());
199 >        checkCompletedNormally(task, Boolean.TRUE);
200      }
201  
202      /**
203 <     *
203 >     * cancel(true) interrupts a running task
204       */
205 <    public void testCancelInterrupt() {
206 <        FutureTask task = new FutureTask( new Callable() {
207 <                public Object call() {
208 <                    try {
209 <                        Thread.sleep(MEDIUM_DELAY_MS);
210 <                        threadShouldThrow();
205 >    public void testCancelInterrupt() throws InterruptedException {
206 >        final CountDownLatch threadStarted = new CountDownLatch(1);
207 >        final FutureTask task =
208 >            new FutureTask(new CheckedCallable<Object>() {
209 >                public Object realCall() {
210 >                    threadStarted.countDown();
211 >                    long t0 = System.nanoTime();
212 >                    for (;;) {
213 >                        if (Thread.interrupted())
214 >                            return Boolean.TRUE;
215 >                        if (millisElapsedSince(t0) > MEDIUM_DELAY_MS)
216 >                            fail("interrupt not delivered");
217 >                        Thread.yield();
218                      }
219 <                    catch (InterruptedException success) {}
219 >                }});
220 >
221 >        Thread t = newStartedThread(task);
222 >        threadStarted.await();
223 >        assertTrue(task.cancel(true));
224 >        checkCancelled(task);
225 >        awaitTermination(t, MEDIUM_DELAY_MS);
226 >        checkCancelled(task);
227 >    }
228 >
229 >    /**
230 >     * cancel(false) does not interrupt a running task
231 >     */
232 >    public void testCancelNoInterrupt() throws InterruptedException {
233 >        final CountDownLatch threadStarted = new CountDownLatch(1);
234 >        final CountDownLatch cancelled = new CountDownLatch(1);
235 >        final FutureTask<Boolean> task =
236 >            new FutureTask<Boolean>(new CheckedCallable<Boolean>() {
237 >                public Boolean realCall() throws InterruptedException {
238 >                    threadStarted.countDown();
239 >                    cancelled.await(MEDIUM_DELAY_MS, MILLISECONDS);
240 >                    assertFalse(Thread.interrupted());
241                      return Boolean.TRUE;
242 <                } });
243 <        Thread t = new  Thread(task);
244 <        t.start();
245 <        
246 <        try {
247 <            Thread.sleep(SHORT_DELAY_MS);
248 <            assertTrue(task.cancel(true));
249 <            t.join();
250 <            assertTrue(task.isDone());
198 <            assertTrue(task.isCancelled());
199 <        } catch(InterruptedException e){
200 <            unexpectedException();
201 <        }
242 >                }});
243 >
244 >        Thread t = newStartedThread(task);
245 >        threadStarted.await();
246 >        assertTrue(task.cancel(false));
247 >        checkCancelled(task);
248 >        cancelled.countDown();
249 >        awaitTermination(t, MEDIUM_DELAY_MS);
250 >        checkCancelled(task);
251      }
252  
253 +    /**
254 +     * run in one thread causes get in another thread to retrieve value
255 +     */
256 +    public void testGetRun() throws InterruptedException {
257 +        final CountDownLatch threadStarted = new CountDownLatch(1);
258 +
259 +        final FutureTask task =
260 +            new FutureTask(new CheckedCallable<Object>() {
261 +                public Object realCall() throws InterruptedException {
262 +                    return Boolean.TRUE;
263 +                }});
264 +
265 +        Thread t = newStartedThread(new CheckedRunnable() {
266 +            public void realRun() throws Exception {
267 +                threadStarted.countDown();
268 +                assertSame(Boolean.TRUE, task.get());
269 +            }});
270 +
271 +        threadStarted.await();
272 +        checkNotDone(task);
273 +        assertTrue(t.isAlive());
274 +        task.run();
275 +        checkCompletedNormally(task, Boolean.TRUE);
276 +        awaitTermination(t, MEDIUM_DELAY_MS);
277 +    }
278  
279      /**
280 <     *
280 >     * set in one thread causes get in another thread to retrieve value
281       */
282 <    public void testCancelNoInterrupt() {
283 <        FutureTask task = new FutureTask( new Callable() {
284 <                public Object call() {
285 <                    try {
286 <                        Thread.sleep(MEDIUM_DELAY_MS);
287 <                    }
214 <                    catch (InterruptedException success) {
215 <                        threadFail("should not interrupt");
216 <                    }
282 >    public void testGetSet() throws InterruptedException {
283 >        final CountDownLatch threadStarted = new CountDownLatch(1);
284 >
285 >        final PublicFutureTask task =
286 >            new PublicFutureTask(new CheckedCallable<Object>() {
287 >                public Object realCall() throws InterruptedException {
288                      return Boolean.TRUE;
289 <                } });
290 <        Thread t = new  Thread(task);
291 <        t.start();
292 <        
293 <        try {
294 <            Thread.sleep(SHORT_DELAY_MS);
295 <            assertTrue(task.cancel(false));
296 <            t.join();
297 <            assertTrue(task.isDone());
298 <            assertTrue(task.isCancelled());
299 <        } catch(InterruptedException e){
300 <            unexpectedException();
301 <        }
289 >                }});
290 >
291 >        Thread t = newStartedThread(new CheckedRunnable() {
292 >            public void realRun() throws Exception {
293 >                threadStarted.countDown();
294 >                assertSame(Boolean.FALSE, task.get());
295 >            }});
296 >
297 >        threadStarted.await();
298 >        checkNotDone(task);
299 >        assertTrue(t.isAlive());
300 >        task.set(Boolean.FALSE);
301 >        checkCompletedNormally(task, Boolean.FALSE);
302 >        awaitTermination(t, MEDIUM_DELAY_MS);
303      }
304  
305      /**
306 <     *
306 >     * run in one thread causes timed get in another thread to retrieve value
307       */
308 <    public void testGet1() {
309 <        final FutureTask ft = new FutureTask(new Callable() {
310 <                public Object call() {
311 <                    try {
312 <                        Thread.sleep(MEDIUM_DELAY_MS);
313 <                    } catch(InterruptedException e){
242 <                        threadUnexpectedException();
243 <                    }
308 >    public void testTimedGetRun() throws InterruptedException {
309 >        final CountDownLatch threadStarted = new CountDownLatch(1);
310 >
311 >        final FutureTask task =
312 >            new FutureTask(new CheckedCallable<Object>() {
313 >                public Object realCall() throws InterruptedException {
314                      return Boolean.TRUE;
315 <                }
316 <        });
317 <        Thread t = new Thread(new Runnable() {
318 <                public void run() {
319 <                    try {
320 <                        ft.get();
321 <                    } catch(Exception e){
322 <                        threadUnexpectedException();
323 <                    }
324 <                }
325 <            });
326 <        try {
327 <            assertFalse(ft.isDone());
328 <            assertFalse(ft.isCancelled());
329 <            t.start();
260 <            Thread.sleep(SHORT_DELAY_MS);
261 <            ft.run();
262 <            t.join();
263 <            assertTrue(ft.isDone());
264 <            assertFalse(ft.isCancelled());
265 <        } catch(InterruptedException e){
266 <            unexpectedException();
267 <
268 <        }      
315 >                }});
316 >
317 >        Thread t = newStartedThread(new CheckedRunnable() {
318 >            public void realRun() throws Exception {
319 >                threadStarted.countDown();
320 >                assertSame(Boolean.TRUE,
321 >                           task.get(MEDIUM_DELAY_MS, MILLISECONDS));
322 >            }});
323 >
324 >        threadStarted.await();
325 >        checkNotDone(task);
326 >        assertTrue(t.isAlive());
327 >        task.run();
328 >        checkCompletedNormally(task, Boolean.TRUE);
329 >        awaitTermination(t, MEDIUM_DELAY_MS);
330      }
331  
332      /**
333 <     *
334 <     */
335 <    public void testTimedGet1() {
336 <        final FutureTask ft = new FutureTask(new Callable() {
337 <                public Object call() {
338 <                    try {
339 <                        Thread.sleep(MEDIUM_DELAY_MS);
340 <                    } catch(InterruptedException e){
280 <                        threadUnexpectedException();
281 <                    }
333 >     * set in one thread causes timed get in another thread to retrieve value
334 >     */
335 >    public void testTimedGetSet() throws InterruptedException {
336 >        final CountDownLatch threadStarted = new CountDownLatch(1);
337 >
338 >        final PublicFutureTask task =
339 >            new PublicFutureTask(new CheckedCallable<Object>() {
340 >                public Object realCall() throws InterruptedException {
341                      return Boolean.TRUE;
342 <                }
343 <            });
344 <        Thread t = new Thread(new Runnable() {
345 <                public void run() {
346 <                    try {
347 <                        ft.get(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
348 <                    } catch(TimeoutException success) {
349 <                    } catch(Exception e){
350 <                        threadUnexpectedException();
351 <                    }
352 <                }
353 <            });
354 <        try {
355 <            assertFalse(ft.isDone());
356 <            assertFalse(ft.isCancelled());
298 <            t.start();
299 <            ft.run();
300 <            t.join();
301 <            assertTrue(ft.isDone());
302 <            assertFalse(ft.isCancelled());
303 <        } catch(InterruptedException e){
304 <            unexpectedException();
305 <            
306 <        }      
342 >                }});
343 >
344 >        Thread t = newStartedThread(new CheckedRunnable() {
345 >            public void realRun() throws Exception {
346 >                threadStarted.countDown();
347 >                assertSame(Boolean.FALSE,
348 >                           task.get(MEDIUM_DELAY_MS, MILLISECONDS));
349 >            }});
350 >
351 >        threadStarted.await();
352 >        checkNotDone(task);
353 >        assertTrue(t.isAlive());
354 >        task.set(Boolean.FALSE);
355 >        checkCompletedNormally(task, Boolean.FALSE);
356 >        awaitTermination(t, MEDIUM_DELAY_MS);
357      }
358  
359 +    /**
360 +     * Cancelling a task causes timed get in another thread to throw
361 +     * CancellationException
362 +     */
363 +    public void testTimedGet_Cancellation() throws InterruptedException {
364 +        final CountDownLatch threadStarted = new CountDownLatch(2);
365 +        final FutureTask task =
366 +            new FutureTask(new CheckedInterruptedCallable<Object>() {
367 +                public Object realCall() throws InterruptedException {
368 +                    threadStarted.countDown();
369 +                    delay(LONG_DELAY_MS);
370 +                    return Boolean.TRUE;
371 +                }});
372 +
373 +        Thread t1 = new ThreadShouldThrow(CancellationException.class) {
374 +            public void realRun() throws Exception {
375 +                threadStarted.countDown();
376 +                task.get(MEDIUM_DELAY_MS, MILLISECONDS);
377 +            }};
378 +        Thread t2 = new Thread(task);
379 +        t1.start();
380 +        t2.start();
381 +        threadStarted.await();
382 +        task.cancel(true);
383 +        awaitTermination(t1, MEDIUM_DELAY_MS);
384 +        awaitTermination(t2, MEDIUM_DELAY_MS);
385 +        checkCancelled(task);
386 +    }
387  
388      /**
389 <     *
390 <     */
391 <    public void testGet_Cancellation() {
392 <        final FutureTask ft = new FutureTask(new Callable() {
393 <                public Object call() {
394 <                    try {
395 <                        Thread.sleep(MEDIUM_DELAY_MS);
396 <                    } catch(InterruptedException e){
397 <                        threadUnexpectedException();
398 <                    }
389 >     * Cancelling a task causes get in another thread to throw
390 >     * CancellationException
391 >     */
392 >    public void testGet_Cancellation() throws InterruptedException {
393 >        final CountDownLatch threadStarted = new CountDownLatch(2);
394 >        final FutureTask task =
395 >            new FutureTask(new CheckedInterruptedCallable<Object>() {
396 >                public Object realCall() throws InterruptedException {
397 >                    threadStarted.countDown();
398 >                    delay(LONG_DELAY_MS);
399                      return Boolean.TRUE;
400 <                }
401 <            });
402 <        try {
403 <            Thread.sleep(SHORT_DELAY_MS);
404 <            Thread t = new Thread(new Runnable() {
405 <                    public void run() {
406 <                        try {
407 <                            ft.get();
408 <                            threadShouldThrow();
409 <                        } catch(CancellationException success){
410 <                        }
411 <                        catch(Exception e){
412 <                            threadUnexpectedException();
413 <                        }
414 <                    }
337 <                });
338 <            t.start();
339 <            ft.cancel(true);
340 <            t.join();
341 <        } catch(InterruptedException success){
342 <            unexpectedException();
343 <        }
400 >                }});
401 >
402 >        Thread t1 = new ThreadShouldThrow(CancellationException.class) {
403 >            public void realRun() throws Exception {
404 >                threadStarted.countDown();
405 >                task.get();
406 >            }};
407 >        Thread t2 = new Thread(task);
408 >        t1.start();
409 >        t2.start();
410 >        threadStarted.await();
411 >        task.cancel(true);
412 >        awaitTermination(t1, MEDIUM_DELAY_MS);
413 >        awaitTermination(t2, MEDIUM_DELAY_MS);
414 >        checkCancelled(task);
415      }
416 <    
416 >
417 >
418      /**
419 <     *
420 <     */
421 <    public void testGet_Cancellation2() {
422 <        final FutureTask ft = new FutureTask(new Callable() {
423 <                public Object call() {
424 <                    try {
425 <                        Thread.sleep(SHORT_DELAY_MS);
426 <                    } catch(InterruptedException e) {
427 <                        threadUnexpectedException();
428 <                    }
429 <                    return Boolean.TRUE;
430 <                }
431 <            });
432 <        try {
433 <            Thread.sleep(SHORT_DELAY_MS);
362 <            Thread t = new Thread(new Runnable() {
363 <                    public void run() {
364 <                        try {
365 <                            ft.get(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
366 <                            threadShouldThrow();
367 <                        } catch(CancellationException success) {}
368 <                        catch(Exception e){
369 <                            threadUnexpectedException();
370 <                        }
371 <                    }
372 <                });
373 <            t.start();
374 <            Thread.sleep(SHORT_DELAY_MS);
375 <            ft.cancel(true);
376 <            Thread.sleep(SHORT_DELAY_MS);
377 <            t.join();
378 <        } catch(InterruptedException ie){
379 <            unexpectedException();
419 >     * A runtime exception in task causes get to throw ExecutionException
420 >     */
421 >    public void testGet_ExecutionException() throws InterruptedException {
422 >        final FutureTask task = new FutureTask(new Callable() {
423 >            public Object call() {
424 >                return 5/0;
425 >            }});
426 >
427 >        task.run();
428 >        try {
429 >            task.get();
430 >            shouldThrow();
431 >        } catch (ExecutionException success) {
432 >            assertTrue(success.getCause() instanceof ArithmeticException);
433 >            checkCompletedAbnormally(task, success.getCause());
434          }
435      }
436  
437      /**
438 <     *
438 >     * A runtime exception in task causes timed get to throw ExecutionException
439       */
440 <    public void testGet_ExecutionException() {
441 <        final FutureTask ft = new FutureTask(new Callable() {
442 <                public Object call() {
443 <                    int i = 5/0;
444 <                    return Boolean.TRUE;
445 <                }
446 <            });
393 <        try {
394 <            ft.run();
395 <            ft.get();
396 <            shouldThrow();
397 <        } catch(ExecutionException success){
398 <        }
399 <        catch(Exception e){
400 <            unexpectedException();
401 <        }
402 <    }
403 <  
404 <    /**
405 <     *
406 <     */
407 <    public void testTimedGet_ExecutionException2() {
408 <        final FutureTask ft = new FutureTask(new Callable() {
409 <                public Object call() {
410 <                    int i = 5/0;
411 <                    return Boolean.TRUE;
412 <                }
413 <            });
414 <        try {
415 <            ft.run();
416 <            ft.get(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
417 <            shouldThrow();
418 <        } catch(ExecutionException success) {
419 <        } catch(TimeoutException success) { } // unlikely but OK
420 <        catch(Exception e){
421 <            unexpectedException();
422 <        }
423 <    }
424 <      
425 <
426 <    /**
427 <     *
428 <     */
429 <    public void testGet_InterruptedException() {
430 <        final FutureTask ft = new FutureTask(new NoOpCallable());
431 <        Thread t = new Thread(new Runnable() {
432 <                public void run() {                
433 <                    try {
434 <                        ft.get();
435 <                        threadShouldThrow();
436 <                    } catch(InterruptedException success){
437 <                    } catch(Exception e){
438 <                        threadUnexpectedException();
439 <                    }
440 <                }
441 <            });
440 >    public void testTimedGet_ExecutionException2() throws Exception {
441 >        final FutureTask task = new FutureTask(new Callable() {
442 >            public Object call() {
443 >                return 5/0;
444 >            }});
445 >
446 >        task.run();
447          try {
448 <            t.start();
449 <            Thread.sleep(SHORT_DELAY_MS);
450 <            t.interrupt();
451 <            t.join();
452 <        } catch(Exception e){
448 <            unexpectedException();
448 >            task.get(SHORT_DELAY_MS, MILLISECONDS);
449 >            shouldThrow();
450 >        } catch (ExecutionException success) {
451 >            assertTrue(success.getCause() instanceof ArithmeticException);
452 >            checkCompletedAbnormally(task, success.getCause());
453          }
454      }
455  
456 +
457      /**
458 <     *
458 >     * Interrupting a waiting get causes it to throw InterruptedException
459       */
460 <    public void testTimedGet_InterruptedException2() {
461 <        final FutureTask ft = new FutureTask(new NoOpCallable());
462 <        Thread t = new Thread(new Runnable() {
463 <                public void run() {                
464 <                    try {
465 <                        ft.get(LONG_DELAY_MS,TimeUnit.MILLISECONDS);
466 <                        threadShouldThrow();
467 <                    } catch(InterruptedException success){}
468 <                    catch(Exception e){
469 <                        threadUnexpectedException();
470 <                    }
471 <                }
472 <            });
468 <        try {
469 <            t.start();
470 <            Thread.sleep(SHORT_DELAY_MS);
471 <            t.interrupt();
472 <            t.join();
473 <        } catch(Exception e){
474 <            unexpectedException();
475 <        }
460 >    public void testGet_InterruptedException() throws InterruptedException {
461 >        final CountDownLatch threadStarted = new CountDownLatch(1);
462 >        final FutureTask task = new FutureTask(new NoOpCallable());
463 >        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
464 >            public void realRun() throws Exception {
465 >                threadStarted.countDown();
466 >                task.get();
467 >            }});
468 >
469 >        threadStarted.await();
470 >        t.interrupt();
471 >        awaitTermination(t, MEDIUM_DELAY_MS);
472 >        checkNotDone(task);
473      }
474 <    
474 >
475      /**
476 <     *
476 >     * Interrupting a waiting timed get causes it to throw InterruptedException
477       */
478 <    public void testGet_TimeoutException() {
479 <        try {
480 <            FutureTask ft = new FutureTask(new NoOpCallable());
481 <            ft.get(1,TimeUnit.MILLISECONDS);
482 <            shouldThrow();
483 <        } catch(TimeoutException success){}
484 <        catch(Exception success){
485 <            unexpectedException();
486 <        }
478 >    public void testTimedGet_InterruptedException2() throws InterruptedException {
479 >        final CountDownLatch threadStarted = new CountDownLatch(1);
480 >        final FutureTask task = new FutureTask(new NoOpCallable());
481 >        Thread t = newStartedThread(new CheckedInterruptedRunnable() {
482 >            public void realRun() throws Exception {
483 >                threadStarted.countDown();
484 >                task.get(LONG_DELAY_MS, MILLISECONDS);
485 >            }});
486 >
487 >        threadStarted.await();
488 >        t.interrupt();
489 >        awaitTermination(t, MEDIUM_DELAY_MS);
490 >        checkNotDone(task);
491      }
492 <    
492 >
493 >    /**
494 >     * A timed out timed get throws TimeoutException
495 >     */
496 >    public void testGet_TimeoutException() throws Exception {
497 >        try {
498 >            FutureTask task = new FutureTask(new NoOpCallable());
499 >            task.get(1, MILLISECONDS);
500 >            shouldThrow();
501 >        } catch (TimeoutException success) {}
502 >    }
503 >
504   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines