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

Comparing jsr166/src/test/tck/ExecutorCompletionServiceTest.java (file contents):
Revision 1.16 by jsr166, Sun May 29 06:50:55 2011 UTC vs.
Revision 1.27 by jsr166, Wed Aug 24 22:22:39 2016 UTC

# Line 6 | Line 6
6   * Pat Fisher, Mike Judd.
7   */
8  
9 import junit.framework.*;
10 import java.util.*;
11 import java.util.concurrent.*;
9   import static java.util.concurrent.TimeUnit.MILLISECONDS;
10 < import java.util.concurrent.atomic.*;
11 < import java.math.BigInteger;
12 < import java.security.*;
10 >
11 > import java.util.concurrent.ArrayBlockingQueue;
12 > import java.util.concurrent.Callable;
13 > import java.util.concurrent.CompletionService;
14 > import java.util.concurrent.CountDownLatch;
15 > import java.util.concurrent.ExecutionException;
16 > import java.util.concurrent.ExecutorCompletionService;
17 > import java.util.concurrent.ExecutorService;
18 > import java.util.concurrent.Future;
19 > import java.util.concurrent.FutureTask;
20 > import java.util.concurrent.RunnableFuture;
21 > import java.util.concurrent.ThreadPoolExecutor;
22 > import java.util.concurrent.TimeUnit;
23 > import java.util.concurrent.atomic.AtomicBoolean;
24 >
25 > import junit.framework.Test;
26 > import junit.framework.TestSuite;
27  
28   public class ExecutorCompletionServiceTest extends JSR166TestCase {
29      public static void main(String[] args) {
30 <        junit.textui.TestRunner.run(suite());
30 >        main(suite(), args);
31      }
32      public static Test suite() {
33          return new TestSuite(ExecutorCompletionServiceTest.class);
34      }
35  
36      /**
37 <     * Creating a new ECS with null Executor throw NPE
37 >     * new ExecutorCompletionService(null) throws NullPointerException
38       */
39      public void testConstructorNPE() {
40          try {
41 <            ExecutorCompletionService ecs = new ExecutorCompletionService(null);
41 >            new ExecutorCompletionService(null);
42              shouldThrow();
43          } catch (NullPointerException success) {}
44      }
45  
46      /**
47 <     * Creating a new ECS with null queue throw NPE
47 >     * new ExecutorCompletionService(e, null) throws NullPointerException
48       */
49      public void testConstructorNPE2() {
50          try {
51 <            ExecutorService e = Executors.newCachedThreadPool();
41 <            ExecutorCompletionService ecs = new ExecutorCompletionService(e, null);
51 >            new ExecutorCompletionService(cachedThreadPool, null);
52              shouldThrow();
53          } catch (NullPointerException success) {}
54      }
55  
56      /**
57 <     * Submitting a null callable throws NPE
57 >     * ecs.submit(null) throws NullPointerException
58       */
59 <    public void testSubmitNPE() {
60 <        ExecutorService e = Executors.newCachedThreadPool();
51 <        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
59 >    public void testSubmitNullCallable() {
60 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
61          try {
62 <            Callable c = null;
54 <            ecs.submit(c);
62 >            cs.submit((Callable) null);
63              shouldThrow();
64 <        } catch (NullPointerException success) {
57 <        } finally {
58 <            joinPool(e);
59 <        }
64 >        } catch (NullPointerException success) {}
65      }
66  
67      /**
68 <     * Submitting a null runnable throws NPE
68 >     * ecs.submit(null, val) throws NullPointerException
69       */
70 <    public void testSubmitNPE2() {
71 <        ExecutorService e = Executors.newCachedThreadPool();
67 <        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
70 >    public void testSubmitNullRunnable() {
71 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
72          try {
73 <            Runnable r = null;
70 <            ecs.submit(r, Boolean.TRUE);
73 >            cs.submit((Runnable) null, Boolean.TRUE);
74              shouldThrow();
75 <        } catch (NullPointerException success) {
73 <        } finally {
74 <            joinPool(e);
75 <        }
75 >        } catch (NullPointerException success) {}
76      }
77  
78      /**
79       * A taken submitted task is completed
80       */
81 <    public void testTake() throws InterruptedException {
82 <        ExecutorService e = Executors.newCachedThreadPool();
83 <        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
84 <        try {
85 <            Callable c = new StringTask();
86 <            ecs.submit(c);
87 <            Future f = ecs.take();
88 <            assertTrue(f.isDone());
89 <        } finally {
90 <            joinPool(e);
91 <        }
81 >    public void testTake()
82 >        throws InterruptedException, ExecutionException {
83 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
84 >        cs.submit(new StringTask());
85 >        Future f = cs.take();
86 >        assertTrue(f.isDone());
87 >        assertSame(TEST_STRING, f.get());
88      }
89  
90      /**
91       * Take returns the same future object returned by submit
92       */
93      public void testTake2() throws InterruptedException {
94 <        ExecutorService e = Executors.newCachedThreadPool();
95 <        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
96 <        try {
97 <            Callable c = new StringTask();
98 <            Future f1 = ecs.submit(c);
99 <            Future f2 = ecs.take();
100 <            assertSame(f1, f2);
101 <        } finally {
102 <            joinPool(e);
103 <        }
104 <    }
105 <
106 <    /**
107 <     * If poll returns non-null, the returned task is completed
108 <     */
109 <    public void testPoll1() throws Exception {
110 <        ExecutorService e = Executors.newCachedThreadPool();
111 <        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
112 <        try {
113 <            assertNull(ecs.poll());
114 <            Callable c = new StringTask();
115 <            ecs.submit(c);
116 <
117 <            long startTime = System.nanoTime();
118 <            Future f;
119 <            while ((f = ecs.poll()) == null) {
120 <                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
121 <                    fail("timed out");
122 <                Thread.yield();
94 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
95 >        Future f1 = cs.submit(new StringTask());
96 >        Future f2 = cs.take();
97 >        assertSame(f1, f2);
98 >    }
99 >
100 >    /**
101 >     * poll returns non-null when the returned task is completed
102 >     */
103 >    public void testPoll1()
104 >        throws InterruptedException, ExecutionException {
105 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
106 >        assertNull(cs.poll());
107 >        cs.submit(new StringTask());
108 >
109 >        long startTime = System.nanoTime();
110 >        Future f;
111 >        while ((f = cs.poll()) == null) {
112 >            if (millisElapsedSince(startTime) > LONG_DELAY_MS)
113 >                fail("timed out");
114 >            Thread.yield();
115 >        }
116 >        assertTrue(f.isDone());
117 >        assertSame(TEST_STRING, f.get());
118 >    }
119 >
120 >    /**
121 >     * timed poll returns non-null when the returned task is completed
122 >     */
123 >    public void testPoll2()
124 >        throws InterruptedException, ExecutionException {
125 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
126 >        assertNull(cs.poll());
127 >        cs.submit(new StringTask());
128 >
129 >        long startTime = System.nanoTime();
130 >        Future f;
131 >        while ((f = cs.poll(SHORT_DELAY_MS, MILLISECONDS)) == null) {
132 >            if (millisElapsedSince(startTime) > LONG_DELAY_MS)
133 >                fail("timed out");
134 >            Thread.yield();
135 >        }
136 >        assertTrue(f.isDone());
137 >        assertSame(TEST_STRING, f.get());
138 >    }
139 >
140 >    /**
141 >     * poll returns null before the returned task is completed
142 >     */
143 >    public void testPollReturnsNull()
144 >        throws InterruptedException, ExecutionException {
145 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
146 >        final CountDownLatch proceed = new CountDownLatch(1);
147 >        cs.submit(new Callable() { public String call() throws Exception {
148 >            proceed.await();
149 >            return TEST_STRING;
150 >        }});
151 >        assertNull(cs.poll());
152 >        assertNull(cs.poll(0L, MILLISECONDS));
153 >        assertNull(cs.poll(Long.MIN_VALUE, MILLISECONDS));
154 >        long startTime = System.nanoTime();
155 >        assertNull(cs.poll(timeoutMillis(), MILLISECONDS));
156 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
157 >        proceed.countDown();
158 >        assertSame(TEST_STRING, cs.take().get());
159 >    }
160 >
161 >    /**
162 >     * successful and failed tasks are both returned
163 >     */
164 >    public void testTaskAssortment()
165 >        throws InterruptedException, ExecutionException {
166 >        CompletionService cs = new ExecutorCompletionService(cachedThreadPool);
167 >        ArithmeticException ex = new ArithmeticException();
168 >        for (int i = 0; i < 2; i++) {
169 >            cs.submit(new StringTask());
170 >            cs.submit(callableThrowing(ex));
171 >            cs.submit(runnableThrowing(ex), null);
172 >        }
173 >        int normalCompletions = 0;
174 >        int exceptionalCompletions = 0;
175 >        for (int i = 0; i < 3 * 2; i++) {
176 >            try {
177 >                if (cs.take().get() == TEST_STRING)
178 >                    normalCompletions++;
179 >            }
180 >            catch (ExecutionException expected) {
181 >                assertTrue(expected.getCause() instanceof ArithmeticException);
182 >                exceptionalCompletions++;
183              }
128            assertTrue(f.isDone());
129            assertSame(TEST_STRING, f.get());
130        } finally {
131            joinPool(e);
132        }
133    }
134
135    /**
136     * If timed poll returns non-null, the returned task is completed
137     */
138    public void testPoll2() throws InterruptedException {
139        ExecutorService e = Executors.newCachedThreadPool();
140        ExecutorCompletionService ecs = new ExecutorCompletionService(e);
141        try {
142            assertNull(ecs.poll());
143            Callable c = new StringTask();
144            ecs.submit(c);
145            Future f = ecs.poll(SHORT_DELAY_MS, MILLISECONDS);
146            if (f != null)
147                assertTrue(f.isDone());
148        } finally {
149            joinPool(e);
184          }
185 +        assertEquals(2 * 1, normalCompletions);
186 +        assertEquals(2 * 2, exceptionalCompletions);
187 +        assertNull(cs.poll());
188      }
189  
190      /**
# Line 158 | Line 195 | public class ExecutorCompletionServiceTe
195          final AtomicBoolean done = new AtomicBoolean(false);
196          class MyCallableFuture<V> extends FutureTask<V> {
197              MyCallableFuture(Callable<V> c) { super(c); }
198 <            protected void done() { done.set(true); }
198 >            @Override protected void done() { done.set(true); }
199          }
200 <        ExecutorService e = new ThreadPoolExecutor(
201 <                                 1, 1, 30L, TimeUnit.SECONDS,
202 <                                 new ArrayBlockingQueue<Runnable>(1)) {
203 <            protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
204 <                return new MyCallableFuture<T>(c);
205 <            }};
206 <        ExecutorCompletionService<String> ecs =
207 <            new ExecutorCompletionService<String>(e);
208 <        try {
209 <            assertNull(ecs.poll());
200 >        final ExecutorService e =
201 >            new ThreadPoolExecutor(1, 1,
202 >                                   30L, TimeUnit.SECONDS,
203 >                                   new ArrayBlockingQueue<Runnable>(1)) {
204 >                protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
205 >                    return new MyCallableFuture<T>(c);
206 >                }};
207 >        CompletionService<String> cs = new ExecutorCompletionService<>(e);
208 >        try (PoolCleaner cleaner = cleaner(e)) {
209 >            assertNull(cs.poll());
210              Callable<String> c = new StringTask();
211 <            Future f1 = ecs.submit(c);
211 >            Future f1 = cs.submit(c);
212              assertTrue("submit must return MyCallableFuture",
213                         f1 instanceof MyCallableFuture);
214 <            Future f2 = ecs.take();
214 >            Future f2 = cs.take();
215              assertSame("submit and take must return same objects", f1, f2);
216              assertTrue("completed task must have set done", done.get());
180        } finally {
181            joinPool(e);
217          }
218      }
219  
# Line 190 | Line 225 | public class ExecutorCompletionServiceTe
225          final AtomicBoolean done = new AtomicBoolean(false);
226          class MyRunnableFuture<V> extends FutureTask<V> {
227              MyRunnableFuture(Runnable t, V r) { super(t, r); }
228 <            protected void done() { done.set(true); }
228 >            @Override protected void done() { done.set(true); }
229          }
230 <        ExecutorService e = new ThreadPoolExecutor(
231 <                                 1, 1, 30L, TimeUnit.SECONDS,
232 <                                 new ArrayBlockingQueue<Runnable>(1)) {
233 <            protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
234 <                return new MyRunnableFuture<T>(t, r);
235 <            }};
236 <        ExecutorCompletionService<String> ecs =
237 <            new ExecutorCompletionService<String>(e);
238 <        try {
239 <            assertNull(ecs.poll());
230 >        final ExecutorService e =
231 >            new ThreadPoolExecutor(1, 1,
232 >                                   30L, TimeUnit.SECONDS,
233 >                                   new ArrayBlockingQueue<Runnable>(1)) {
234 >                protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
235 >                    return new MyRunnableFuture<T>(t, r);
236 >                }};
237 >        CompletionService<String> cs = new ExecutorCompletionService<>(e);
238 >        try (PoolCleaner cleaner = cleaner(e)) {
239 >            assertNull(cs.poll());
240              Runnable r = new NoOpRunnable();
241 <            Future f1 = ecs.submit(r, null);
241 >            Future f1 = cs.submit(r, null);
242              assertTrue("submit must return MyRunnableFuture",
243                         f1 instanceof MyRunnableFuture);
244 <            Future f2 = ecs.take();
244 >            Future f2 = cs.take();
245              assertSame("submit and take must return same objects", f1, f2);
246              assertTrue("completed task must have set done", done.get());
212        } finally {
213            joinPool(e);
247          }
248      }
249  

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines