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.2 by dl, Sat Dec 27 19:26:43 2003 UTC vs.
Revision 1.16 by jsr166, Sun May 29 06:50:55 2011 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
5 < * Other contributors include Andrew Wright, Jeffrey Hayes,
6 < * Pat Fisher, Mike Judd.
4 > * http://creativecommons.org/publicdomain/zero/1.0/
5 > * Other contributors include Andrew Wright, Jeffrey Hayes,
6 > * Pat Fisher, Mike Judd.
7   */
8  
9
9   import junit.framework.*;
10   import java.util.*;
11   import java.util.concurrent.*;
12 + import static java.util.concurrent.TimeUnit.MILLISECONDS;
13 + import java.util.concurrent.atomic.*;
14   import java.math.BigInteger;
15   import java.security.*;
16  
17 < public class ExecutorCompletionServiceTest extends JSR166TestCase{
17 > public class ExecutorCompletionServiceTest extends JSR166TestCase {
18      public static void main(String[] args) {
19 <        junit.textui.TestRunner.run (suite());  
19 >        junit.textui.TestRunner.run(suite());
20      }
21      public static Test suite() {
22          return new TestSuite(ExecutorCompletionServiceTest.class);
23      }
24  
24
25      /**
26       * Creating a new ECS with null Executor throw NPE
27 <     */
27 >     */
28      public void testConstructorNPE() {
29          try {
30              ExecutorCompletionService ecs = new ExecutorCompletionService(null);
31              shouldThrow();
32 <        } catch (NullPointerException success) {
33 <        }
32 >        } catch (NullPointerException success) {}
33      }
34  
35      /**
36       * Creating a new ECS with null queue throw NPE
37 <     */
37 >     */
38      public void testConstructorNPE2() {
39          try {
40              ExecutorService e = Executors.newCachedThreadPool();
41              ExecutorCompletionService ecs = new ExecutorCompletionService(e, null);
42              shouldThrow();
43 <        } catch (NullPointerException success) {
45 <        }
43 >        } catch (NullPointerException success) {}
44      }
45  
46      /**
47       * Submitting a null callable throws NPE
48 <     */
48 >     */
49      public void testSubmitNPE() {
50          ExecutorService e = Executors.newCachedThreadPool();
51          ExecutorCompletionService ecs = new ExecutorCompletionService(e);
# Line 63 | Line 61 | public class ExecutorCompletionServiceTe
61  
62      /**
63       * Submitting a null runnable throws NPE
64 <     */
64 >     */
65      public void testSubmitNPE2() {
66          ExecutorService e = Executors.newCachedThreadPool();
67          ExecutorCompletionService ecs = new ExecutorCompletionService(e);
# Line 79 | Line 77 | public class ExecutorCompletionServiceTe
77  
78      /**
79       * A taken submitted task is completed
80 <     */
81 <    public void testTake() {
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 <            assert(f.isDone());
91 <        } catch (Exception ex) {
92 <            unexpectedException();
88 >            assertTrue(f.isDone());
89          } finally {
90              joinPool(e);
91          }
# Line 97 | Line 93 | public class ExecutorCompletionServiceTe
93  
94      /**
95       * Take returns the same future object returned by submit
96 <     */
97 <    public void testTake2() {
96 >     */
97 >    public void testTake2() throws InterruptedException {
98          ExecutorService e = Executors.newCachedThreadPool();
99          ExecutorCompletionService ecs = new ExecutorCompletionService(e);
100          try {
# Line 106 | Line 102 | public class ExecutorCompletionServiceTe
102              Future f1 = ecs.submit(c);
103              Future f2 = ecs.take();
104              assertSame(f1, f2);
109        } catch (Exception ex) {
110            unexpectedException();
105          } finally {
106              joinPool(e);
107          }
# Line 115 | Line 109 | public class ExecutorCompletionServiceTe
109  
110      /**
111       * If poll returns non-null, the returned task is completed
112 <     */
113 <    public void testPoll1() {
112 >     */
113 >    public void testPoll1() throws Exception {
114          ExecutorService e = Executors.newCachedThreadPool();
115          ExecutorCompletionService ecs = new ExecutorCompletionService(e);
116          try {
117              assertNull(ecs.poll());
118              Callable c = new StringTask();
119              ecs.submit(c);
120 <            Thread.sleep(SHORT_DELAY_MS);
121 <            for (;;) {
122 <                Future f = ecs.poll();
123 <                if (f != null) {
124 <                    assert(f.isDone());
125 <                    break;
126 <                }
120 >
121 >            long startTime = System.nanoTime();
122 >            Future f;
123 >            while ((f = ecs.poll()) == null) {
124 >                if (millisElapsedSince(startTime) > LONG_DELAY_MS)
125 >                    fail("timed out");
126 >                Thread.yield();
127              }
128 <        } catch (Exception ex) {
129 <            unexpectedException();
128 >            assertTrue(f.isDone());
129 >            assertSame(TEST_STRING, f.get());
130          } finally {
131              joinPool(e);
132          }
# Line 140 | Line 134 | public class ExecutorCompletionServiceTe
134  
135      /**
136       * If timed poll returns non-null, the returned task is completed
137 <     */
138 <    public void testPoll2() {
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, TimeUnit.MILLISECONDS);
146 <            if (f != null)
147 <                assert(f.isDone());
148 <        } catch (Exception ex) {
149 <            unexpectedException();
145 >            Future f = ecs.poll(SHORT_DELAY_MS, MILLISECONDS);
146 >            if (f != null)
147 >                assertTrue(f.isDone());
148 >        } finally {
149 >            joinPool(e);
150 >        }
151 >    }
152 >
153 >    /**
154 >     * Submitting to underlying AES that overrides newTaskFor(Callable)
155 >     * returns and eventually runs Future returned by newTaskFor.
156 >     */
157 >    public void testNewTaskForCallable() throws InterruptedException {
158 >        final AtomicBoolean done = new AtomicBoolean(false);
159 >        class MyCallableFuture<V> extends FutureTask<V> {
160 >            MyCallableFuture(Callable<V> c) { super(c); }
161 >            protected void done() { done.set(true); }
162 >        }
163 >        ExecutorService e = new ThreadPoolExecutor(
164 >                                 1, 1, 30L, TimeUnit.SECONDS,
165 >                                 new ArrayBlockingQueue<Runnable>(1)) {
166 >            protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
167 >                return new MyCallableFuture<T>(c);
168 >            }};
169 >        ExecutorCompletionService<String> ecs =
170 >            new ExecutorCompletionService<String>(e);
171 >        try {
172 >            assertNull(ecs.poll());
173 >            Callable<String> c = new StringTask();
174 >            Future f1 = ecs.submit(c);
175 >            assertTrue("submit must return MyCallableFuture",
176 >                       f1 instanceof MyCallableFuture);
177 >            Future f2 = ecs.take();
178 >            assertSame("submit and take must return same objects", f1, f2);
179 >            assertTrue("completed task must have set done", done.get());
180 >        } finally {
181 >            joinPool(e);
182 >        }
183 >    }
184 >
185 >    /**
186 >     * Submitting to underlying AES that overrides newTaskFor(Runnable,T)
187 >     * returns and eventually runs Future returned by newTaskFor.
188 >     */
189 >    public void testNewTaskForRunnable() throws InterruptedException {
190 >        final AtomicBoolean done = new AtomicBoolean(false);
191 >        class MyRunnableFuture<V> extends FutureTask<V> {
192 >            MyRunnableFuture(Runnable t, V r) { super(t, r); }
193 >            protected void done() { done.set(true); }
194 >        }
195 >        ExecutorService e = new ThreadPoolExecutor(
196 >                                 1, 1, 30L, TimeUnit.SECONDS,
197 >                                 new ArrayBlockingQueue<Runnable>(1)) {
198 >            protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
199 >                return new MyRunnableFuture<T>(t, r);
200 >            }};
201 >        ExecutorCompletionService<String> ecs =
202 >            new ExecutorCompletionService<String>(e);
203 >        try {
204 >            assertNull(ecs.poll());
205 >            Runnable r = new NoOpRunnable();
206 >            Future f1 = ecs.submit(r, null);
207 >            assertTrue("submit must return MyRunnableFuture",
208 >                       f1 instanceof MyRunnableFuture);
209 >            Future f2 = ecs.take();
210 >            assertSame("submit and take must return same objects", f1, f2);
211 >            assertTrue("completed task must have set done", done.get());
212          } finally {
213              joinPool(e);
214          }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines