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.4 by dl, Sun Sep 14 20:42:40 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 <    public void testConstructor(){
96 >    /**
97 >     * Creating a future with a null callable throws NPE
98 >     */
99 >    public void testConstructor() {
100          try {
101              FutureTask task = new FutureTask(null);
102 <            fail("should throw");
103 <        }
38 <        catch(NullPointerException success) {
39 <        }
102 >            shouldThrow();
103 >        } catch (NullPointerException success) {}
104      }
105  
106 <    public void testConstructor2(){
106 >    /**
107 >     * creating a future with null runnable fails
108 >     */
109 >    public void testConstructor2() {
110          try {
111              FutureTask task = new FutureTask(null, Boolean.TRUE);
112 <            fail("should throw");
113 <        }
47 <        catch(NullPointerException success) {
48 <        }
112 >            shouldThrow();
113 >        } catch (NullPointerException success) {}
114      }
115  
116 <    public void testIsDone(){
117 <        FutureTask task = new FutureTask( new NoOpCallable());
118 <        task.run();
119 <        assertTrue(task.isDone());
120 <        assertFalse(task.isCancelled());
116 >    /**
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 >        checkCompletedNormally(task, Boolean.TRUE);
124      }
125  
126 <    public void testReset(){
127 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
128 <        task.run();
129 <        assertTrue(task.isDone());
130 <        assertTrue(task.reset());
126 >    /**
127 >     * runAndReset of a non-cancelled task succeeds
128 >     */
129 >    public void testRunAndReset() {
130 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
131 >        assertTrue(task.runAndReset());
132 >        checkNotDone(task);
133      }
134  
135 +    /**
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());
70 <        assertTrue(task.isCancelled());
71 <        assertFalse(task.reset());
72 <    }
73 <
74 <    public void testSetDone() {
75 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
76 <        task.setDone();
77 <        assertTrue(task.isDone());
78 <        assertFalse(task.isCancelled());
141 >        assertFalse(task.runAndReset());
142 >        checkCancelled(task);
143      }
144  
81    public void testSetCancelled() {
82        MyFutureTask task = new MyFutureTask(new NoOpCallable());
83        assertTrue(task.cancel(false));
84        task.setCancelled();
85        assertTrue(task.isDone());
86        assertTrue(task.isCancelled());
87    }
145  
146 <    public void testSet() {
147 <        MyFutureTask task = new MyFutureTask(new NoOpCallable());
146 >    /**
147 >     * setting value causes get to return it
148 >     */
149 >    public void testSet() throws Exception {
150 >        PublicFutureTask task = new PublicFutureTask(new NoOpCallable());
151          task.set(one);
152 <        try {
153 <            assertEquals(task.get(), one);
94 <        }
95 <        catch(Exception e) {
96 <            fail("unexpected exception");
97 <        }
152 >        assertSame(task.get(), one);
153 >        checkCompletedNormally(task, one);
154      }
155  
156 <    public void testSetException() {
156 >    /**
157 >     * setException causes get to throw ExecutionException
158 >     */
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 <            fail("should throw");
166 <        }
167 <        catch(ExecutionException ee) {
168 <            Throwable cause = ee.getCause();
110 <            assertEquals(cause, nse);
111 <        }
112 <        catch(Exception e) {
113 <            fail("unexpected exception");
165 >            shouldThrow();
166 >        } catch (ExecutionException success) {
167 >            assertSame(success.getCause(), nse);
168 >            checkCompletedAbnormally(task, nse);
169          }
170      }
171  
172 +    /**
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());
122 <        assertTrue(task.isCancelled());
178 >        task.run();
179 >        checkCancelled(task);
180      }
181  
182 +    /**
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());
130 <        assertTrue(task.isCancelled());
188 >        task.run();
189 >        checkCancelled(task);
190      }
191  
192 +    /**
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());
138 <        assertFalse(task.isCancelled());
199 >        checkCompletedNormally(task, Boolean.TRUE);
200      }
201  
202 <    public void testCancelInterrupt(){
203 <        FutureTask task = new FutureTask( new Callable() {
204 <                public Object call() {
205 <                    try {
206 <                        Thread.sleep(MEDIUM_DELAY_MS);
207 <                        threadFail("should throw");
202 >    /**
203 >     * cancel(true) interrupts a running task
204 >     */
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());
159 <            assertTrue(task.isCancelled());
160 <        } catch(InterruptedException e){
161 <            fail("unexpected exception");
162 <        }
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 <    public void testCancelNoInterrupt(){
260 <        FutureTask task = new FutureTask( new Callable() {
261 <                public Object call() {
169 <                    try {
170 <                        Thread.sleep(MEDIUM_DELAY_MS);
171 <                    }
172 <                    catch (InterruptedException success) {
173 <                        threadFail("should not interrupt");
174 <                    }
259 >        final FutureTask task =
260 >            new FutureTask(new CheckedCallable<Object>() {
261 >                public Object realCall() throws InterruptedException {
262                      return Boolean.TRUE;
263 <                } });
264 <        Thread t = new  Thread(task);
265 <        t.start();
266 <        
267 <        try{
268 <            Thread.sleep(SHORT_DELAY_MS);
269 <            assertTrue(task.cancel(false));
270 <            t.join();
271 <            assertTrue(task.isDone());
272 <            assertTrue(task.isCancelled());
273 <        } catch(InterruptedException e){
274 <            fail("unexpected exception");
275 <        }
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 <    public void testGet1() {
280 <        final FutureTask ft = new FutureTask(new Callable(){
281 <                public Object call(){
282 <                    try{
283 <                        Thread.sleep(MEDIUM_DELAY_MS);
284 <                    } catch(InterruptedException e){
285 <                        threadFail("unexpected exception");
286 <                    }
279 >    /**
280 >     * set in one thread causes get in another thread to retrieve value
281 >     */
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 <        });
291 <        Thread t = new Thread(new Runnable(){
292 <                public void run(){
293 <                    try{
294 <                        ft.get();
295 <                    } catch(Exception e){
296 <                        threadFail("unexpected exception");
297 <                    }
298 <                }
299 <            });
300 <        try{
301 <            assertFalse(ft.isDone());
302 <            assertFalse(ft.isCancelled());
303 <            t.start();
304 <            Thread.sleep(SHORT_DELAY_MS);
305 <            ft.run();
306 <            t.join();
307 <            assertTrue(ft.isDone());
308 <            assertFalse(ft.isCancelled());
309 <        } catch(InterruptedException e){
310 <            fail("unexpected exception");
311 <
312 <        }      
313 <    }
225 <
226 <    public void testTimedGet1() {
227 <        final FutureTask ft = new FutureTask(new Callable(){
228 <                public Object call(){
229 <                    try{
230 <                        Thread.sleep(MEDIUM_DELAY_MS);
231 <                    } catch(InterruptedException e){
232 <                        threadFail("unexpected exception");
233 <                    }
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 >     * run in one thread causes timed get in another thread to retrieve value
307 >     */
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(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
321 <                    } catch(TimeoutException success) {
322 <                    } catch(Exception e){
323 <                        threadFail("unexpected exception");
324 <                    }
325 <                }
326 <            });
327 <        try{
328 <            assertFalse(ft.isDone());
329 <            assertFalse(ft.isCancelled());
250 <            t.start();
251 <            ft.run();
252 <            t.join();
253 <            assertTrue(ft.isDone());
254 <            assertFalse(ft.isCancelled());
255 <        } catch(InterruptedException e){
256 <            fail("unexpected exception");
257 <            
258 <        }      
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 +     * 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 <    public void testGet_Cancellation(){
339 <        final FutureTask ft = new FutureTask(new Callable(){
340 <                public Object call(){
265 <                    try{
266 <                        Thread.sleep(MEDIUM_DELAY_MS);
267 <                    } catch(InterruptedException e){
268 <                        threadFail("unexpected exception");
269 <                    }
338 >        final PublicFutureTask task =
339 >            new PublicFutureTask(new CheckedCallable<Object>() {
340 >                public Object realCall() throws InterruptedException {
341                      return Boolean.TRUE;
342 <                }
343 <            });
344 <        try {
345 <            Thread.sleep(SHORT_DELAY_MS);
346 <            Thread t = new Thread(new Runnable(){
347 <                    public void run(){
348 <                        try{
349 <                            ft.get();
350 <                            threadFail("should throw");
351 <                        } catch(CancellationException success){
352 <                        }
353 <                        catch(Exception e){
354 <                            threadFail("unexpected exception");
355 <                        }
356 <                    }
286 <                });
287 <            t.start();
288 <            ft.cancel(true);
289 <            t.join();
290 <        } catch(InterruptedException success){
291 <            fail("unexpected exception");
292 <        }
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 <    public void testGet_Cancellation2(){
360 <        final FutureTask ft = new FutureTask(new Callable(){
361 <                public Object call(){
362 <                    try{
363 <                        Thread.sleep(SHORT_DELAY_MS);
364 <                    } catch(InterruptedException e) {
365 <                        threadFail("unexpected exception");
366 <                    }
367 <                    return Boolean.TRUE;
368 <                }
369 <            });
370 <        try{
371 <            Thread.sleep(SHORT_DELAY_MS);
372 <            Thread t = new Thread(new Runnable(){
373 <                    public void run(){
374 <                        try{
375 <                            ft.get(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
376 <                            threadFail("should throw");
377 <                        } catch(CancellationException success) {}
378 <                        catch(Exception e){
379 <                            threadFail("unexpected exception");
380 <                        }
381 <                    }
382 <                });
383 <            t.start();
384 <            Thread.sleep(SHORT_DELAY_MS);
385 <            ft.cancel(true);
322 <            Thread.sleep(SHORT_DELAY_MS);
323 <            t.join();
324 <        } catch(InterruptedException ie){
325 <            fail("unexpected exception");
326 <        }
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 <    public void testGet_ExecutionException(){
389 <        final FutureTask ft = new FutureTask(new Callable(){
390 <                public Object call(){
391 <                    int i = 5/0;
392 <                    return Boolean.TRUE;
393 <                }
394 <            });
395 <        try{
396 <            ft.run();
397 <            ft.get();
398 <            fail("should throw");
399 <        } catch(ExecutionException success){
400 <        }
401 <        catch(Exception e){
402 <            fail("unexpected exception");
403 <        }
404 <    }
405 <  
406 <    public void testTimedGet_ExecutionException2(){
407 <        final FutureTask ft = new FutureTask(new Callable(){
408 <                public Object call(){
409 <                    int i = 5/0;
410 <                    return Boolean.TRUE;
411 <                }
412 <            });
413 <        try{
414 <            ft.run();
415 <            ft.get(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
416 <            fail("should throw");
417 <        } catch(ExecutionException success) {
418 <        } catch(TimeoutException success) { } // unlikely but OK
419 <        catch(Exception e){
420 <            fail("unexpected exception");
421 <        }
422 <    }
423 <      
424 <
425 <    public void testGet_InterruptedException(){
426 <        final FutureTask ft = new FutureTask(new NoOpCallable());
427 <        Thread t = new Thread(new Runnable(){
369 <                public void run(){                  
370 <                    try{
371 <                        ft.get();
372 <                        threadFail("should throw");
373 <                    } catch(InterruptedException success){
374 <                    } catch(Exception e){
375 <                        threadFail("unexpected exception");
376 <                    }
377 <                }
378 <            });
388 >    /**
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 >        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 >
417 >
418 >    /**
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 <            t.start();
430 <            Thread.sleep(SHORT_DELAY_MS);
431 <            t.interrupt();
432 <            t.join();
433 <        } catch(Exception e){
385 <            fail("unexpected exception");
429 >            task.get();
430 >            shouldThrow();
431 >        } catch (ExecutionException success) {
432 >            assertTrue(success.getCause() instanceof ArithmeticException);
433 >            checkCompletedAbnormally(task, success.getCause());
434          }
435      }
436  
437 <    public void testTimedGet_InterruptedException2(){
438 <        final FutureTask ft = new FutureTask(new NoOpCallable());
439 <        Thread t = new Thread(new Runnable(){
440 <                public void run(){                  
441 <                    try{
442 <                        ft.get(LONG_DELAY_MS,TimeUnit.MILLISECONDS);
443 <                        threadFail("should throw");
444 <                    } catch(InterruptedException success){}
445 <                    catch(Exception e){
446 <                        threadFail("unexpected exception");
447 <                    }
448 <                }
449 <            });
450 <        try {
451 <            t.start();
452 <            Thread.sleep(SHORT_DELAY_MS);
405 <            t.interrupt();
406 <            t.join();
407 <        } catch(Exception e){
408 <            fail("unexpected exception");
437 >    /**
438 >     * A runtime exception in task causes timed get to throw ExecutionException
439 >     */
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 >            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 <    public void testGet_TimeoutException(){
457 <        try{
458 <            FutureTask ft = new FutureTask(new NoOpCallable());
459 <            ft.get(1,TimeUnit.MILLISECONDS);
460 <            fail("should throw");
461 <        } catch(TimeoutException success){}
462 <        catch(Exception success){
463 <            fail("unexpected exception");
464 <        }
455 >
456 >
457 >    /**
458 >     * Interrupting a waiting get causes it to throw InterruptedException
459 >     */
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 >     * Interrupting a waiting timed get causes it to throw InterruptedException
477 >     */
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 >
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