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

Comparing jsr166/src/test/tck/AbstractExecutorServiceTest.java (file contents):
Revision 1.27 by jsr166, Sat Oct 9 19:30:34 2010 UTC vs.
Revision 1.39 by jsr166, Sun Oct 4 06:05:53 2015 UTC

# Line 1 | Line 1
1   /*
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/licenses/publicdomain
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5   * Other contributors include Andrew Wright, Jeffrey Hayes,
6   * Pat Fisher, Mike Judd.
7   */
8  
9
10 import junit.framework.*;
11 import java.util.*;
12 import java.util.concurrent.*;
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 < import java.math.BigInteger;
11 < import java.security.*;
10 >
11 > import java.security.PrivilegedAction;
12 > import java.security.PrivilegedExceptionAction;
13 > import java.util.ArrayList;
14 > import java.util.Collections;
15 > import java.util.List;
16 > import java.util.concurrent.AbstractExecutorService;
17 > import java.util.concurrent.ArrayBlockingQueue;
18 > import java.util.concurrent.Callable;
19 > import java.util.concurrent.CancellationException;
20 > import java.util.concurrent.CountDownLatch;
21 > import java.util.concurrent.ExecutionException;
22 > import java.util.concurrent.Executors;
23 > import java.util.concurrent.ExecutorService;
24 > import java.util.concurrent.Future;
25 > import java.util.concurrent.ThreadPoolExecutor;
26 > import java.util.concurrent.TimeUnit;
27 > import java.util.concurrent.atomic.AtomicBoolean;
28 >
29 > import junit.framework.Test;
30 > import junit.framework.TestSuite;
31  
32   public class AbstractExecutorServiceTest extends JSR166TestCase {
33      public static void main(String[] args) {
34 <        junit.textui.TestRunner.run(suite());
34 >        main(suite(), args);
35      }
36      public static Test suite() {
37          return new TestSuite(AbstractExecutorServiceTest.class);
# Line 46 | Line 61 | public class AbstractExecutorServiceTest
61       */
62      public void testExecuteRunnable() throws Exception {
63          ExecutorService e = new DirectExecutorService();
64 <        TrackedShortRunnable task = new TrackedShortRunnable();
65 <        assertFalse(task.done);
66 <        Future<?> future = e.submit(task);
67 <        future.get();
68 <        assertTrue(task.done);
64 >        final AtomicBoolean done = new AtomicBoolean(false);
65 >        Future<?> future = e.submit(new CheckedRunnable() {
66 >            public void realRun() {
67 >                done.set(true);
68 >            }});
69 >        assertNull(future.get());
70 >        assertNull(future.get(0, MILLISECONDS));
71 >        assertTrue(done.get());
72 >        assertTrue(future.isDone());
73 >        assertFalse(future.isCancelled());
74      }
75  
56
76      /**
77       * Completed submit(callable) returns result
78       */
# Line 84 | Line 103 | public class AbstractExecutorServiceTest
103          assertSame(TEST_STRING, result);
104      }
105  
87
106      /**
107       * A submitted privileged action runs to completion
108       */
# Line 150 | Line 168 | public class AbstractExecutorServiceTest
168       * execute(null runnable) throws NPE
169       */
170      public void testExecuteNullRunnable() {
171 +        ExecutorService e = new DirectExecutorService();
172          try {
154            ExecutorService e = new DirectExecutorService();
173              e.submit((Runnable) null);
174              shouldThrow();
175          } catch (NullPointerException success) {}
176      }
177  
160
178      /**
179       * submit(null callable) throws NPE
180       */
181      public void testSubmitNullCallable() {
182 +        ExecutorService e = new DirectExecutorService();
183          try {
166            ExecutorService e = new DirectExecutorService();
184              e.submit((Callable) null);
185              shouldThrow();
186          } catch (NullPointerException success) {}
187      }
188  
189      /**
173     * submit(runnable) throws RejectedExecutionException if
174     * executor is saturated.
175     */
176    public void testExecute1() {
177        ThreadPoolExecutor p =
178            new ThreadPoolExecutor(1, 1,
179                                   60, TimeUnit.SECONDS,
180                                   new ArrayBlockingQueue<Runnable>(1));
181        try {
182            for (int i = 0; i < 2; ++i)
183                p.submit(new MediumRunnable());
184            for (int i = 0; i < 2; ++i) {
185                try {
186                    p.submit(new MediumRunnable());
187                    shouldThrow();
188                } catch (RejectedExecutionException success) {}
189            }
190        } finally {
191            joinPool(p);
192        }
193    }
194
195    /**
196     * submit(callable) throws RejectedExecutionException
197     * if executor is saturated.
198     */
199    public void testExecute2() {
200        ThreadPoolExecutor p =
201            new ThreadPoolExecutor(1, 1,
202                                   60, TimeUnit.SECONDS,
203                                   new ArrayBlockingQueue<Runnable>(1));
204        try {
205            for (int i = 0; i < 2; ++i)
206                p.submit(new MediumRunnable());
207            for (int i = 0; i < 2; ++i) {
208                try {
209                    p.submit(new SmallCallable());
210                    shouldThrow();
211                } catch (RejectedExecutionException success) {}
212            }
213        } finally {
214            joinPool(p);
215        }
216    }
217
218
219    /**
190       * submit(callable).get() throws InterruptedException if interrupted
191       */
192      public void testInterruptedSubmit() throws InterruptedException {
# Line 248 | Line 218 | public class AbstractExecutorServiceTest
218      }
219  
220      /**
251     * get of submitted callable throws InterruptedException if callable
252     * interrupted
253     */
254    public void testSubmitIE() throws InterruptedException {
255        final ThreadPoolExecutor p =
256            new ThreadPoolExecutor(1, 1,
257                                   60, TimeUnit.SECONDS,
258                                   new ArrayBlockingQueue<Runnable>(10));
259
260        Thread t = new Thread(new CheckedInterruptedRunnable() {
261            public void realRun() throws Exception {
262                p.submit(new SmallCallable()).get();
263            }});
264
265        t.start();
266        Thread.sleep(SHORT_DELAY_MS);
267        t.interrupt();
268        t.join();
269        joinPool(p);
270    }
271
272    /**
221       * get of submit(callable) throws ExecutionException if callable
222       * throws exception
223       */
# Line 280 | Line 228 | public class AbstractExecutorServiceTest
228                                     new ArrayBlockingQueue<Runnable>(10));
229  
230          Callable c = new Callable() {
231 <            public Object call() { return 5/0; }};
231 >            public Object call() { throw new ArithmeticException(); }};
232  
233          try {
234              p.submit(c).get();
# Line 294 | Line 242 | public class AbstractExecutorServiceTest
242      /**
243       * invokeAny(null) throws NPE
244       */
245 <    public void testInvokeAny1()
298 <        throws InterruptedException, ExecutionException {
245 >    public void testInvokeAny1() throws Exception {
246          ExecutorService e = new DirectExecutorService();
247          try {
248              e.invokeAny(null);
# Line 309 | Line 256 | public class AbstractExecutorServiceTest
256      /**
257       * invokeAny(empty collection) throws IAE
258       */
259 <    public void testInvokeAny2()
313 <        throws InterruptedException, ExecutionException {
259 >    public void testInvokeAny2() throws Exception {
260          ExecutorService e = new DirectExecutorService();
261          try {
262              e.invokeAny(new ArrayList<Callable<String>>());
# Line 326 | Line 272 | public class AbstractExecutorServiceTest
272       */
273      public void testInvokeAny3() throws Exception {
274          ExecutorService e = new DirectExecutorService();
275 <        List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
276 <        l.add(new Callable<Integer>() {
277 <                  public Integer call() { return 5/0; }});
275 >        List<Callable<Long>> l = new ArrayList<Callable<Long>>();
276 >        l.add(new Callable<Long>() {
277 >            public Long call() { throw new ArithmeticException(); }});
278          l.add(null);
279          try {
280              e.invokeAny(l);
# Line 455 | Line 401 | public class AbstractExecutorServiceTest
401          }
402      }
403  
458
404      /**
405       * timed invokeAny(null) throws NPE
406       */
# Line 505 | Line 450 | public class AbstractExecutorServiceTest
450       */
451      public void testTimedInvokeAny3() throws Exception {
452          ExecutorService e = new DirectExecutorService();
453 <        List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
454 <        l.add(new Callable<Integer>() {
455 <                  public Integer call() { return 5/0; }});
453 >        List<Callable<Long>> l = new ArrayList<Callable<Long>>();
454 >        l.add(new Callable<Long>() {
455 >            public Long call() { throw new ArithmeticException(); }});
456          l.add(null);
457          try {
458              e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
# Line 643 | Line 588 | public class AbstractExecutorServiceTest
588              l.add(new StringTask());
589              l.add(new StringTask());
590              List<Future<String>> futures =
591 <                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
591 >                e.invokeAll(l, LONG_DELAY_MS, MILLISECONDS);
592              assertEquals(2, futures.size());
593              for (Future<String> future : futures)
594                  assertSame(TEST_STRING, future.get());
# Line 655 | Line 600 | public class AbstractExecutorServiceTest
600      /**
601       * timed invokeAll cancels tasks not completed by timeout
602       */
603 <    public void testTimedInvokeAll6() throws InterruptedException {
603 >    public void testTimedInvokeAll6() throws Exception {
604          ExecutorService e = new DirectExecutorService();
605          try {
606 <            List<Callable<String>> l = new ArrayList<Callable<String>>();
607 <            l.add(new StringTask());
608 <            l.add(Executors.callable(new MediumPossiblyInterruptedRunnable(), TEST_STRING));
609 <            l.add(new StringTask());
610 <            List<Future<String>> futures =
611 <                e.invokeAll(l, SMALL_DELAY_MS, MILLISECONDS);
612 <            assertEquals(3, futures.size());
613 <            Iterator<Future<String>> it = futures.iterator();
614 <            Future<String> f1 = it.next();
615 <            Future<String> f2 = it.next();
616 <            Future<String> f3 = it.next();
617 <            assertTrue(f1.isDone());
618 <            assertFalse(f1.isCancelled());
619 <            assertTrue(f2.isDone());
620 <            assertTrue(f3.isDone());
621 <            assertTrue(f3.isCancelled());
606 >            for (long timeout = timeoutMillis();;) {
607 >                List<Callable<String>> tasks = new ArrayList<>();
608 >                tasks.add(new StringTask("0"));
609 >                tasks.add(Executors.callable(possiblyInterruptedRunnable(timeout),
610 >                                             TEST_STRING));
611 >                tasks.add(new StringTask("2"));
612 >                long startTime = System.nanoTime();
613 >                List<Future<String>> futures =
614 >                    e.invokeAll(tasks, timeout, MILLISECONDS);
615 >                assertEquals(tasks.size(), futures.size());
616 >                assertTrue(millisElapsedSince(startTime) >= timeout);
617 >                for (Future future : futures)
618 >                    assertTrue(future.isDone());
619 >                try {
620 >                    assertEquals("0", futures.get(0).get());
621 >                    assertEquals(TEST_STRING, futures.get(1).get());
622 >                } catch (CancellationException retryWithLongerTimeout) {
623 >                    // unusual delay before starting second task
624 >                    timeout *= 2;
625 >                    if (timeout >= LONG_DELAY_MS / 2)
626 >                        fail("expected exactly one task to be cancelled");
627 >                    continue;
628 >                }
629 >                assertTrue(futures.get(2).isCancelled());
630 >                break;
631 >            }
632          } finally {
633              joinPool(e);
634          }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines