--- jsr166/src/test/tck/ExecutorCompletionServiceTest.java	2003/12/23 19:40:24	1.1
+++ jsr166/src/test/tck/ExecutorCompletionServiceTest.java	2009/11/21 02:33:20	1.9
@@ -1,20 +1,23 @@
 /*
- * Written by members of JCP JSR-166 Expert Group and released to the
- * public domain. Use, modify, and redistribute this code in any way
- * without acknowledgement. Other contributors include Andrew Wright,
- * Jeffrey Hayes, Pat Fischer, Mike Judd.
+ * Written by Doug Lea with assistance from members of JCP JSR-166
+ * Expert Group and released to the public domain, as explained at
+ * http://creativecommons.org/licenses/publicdomain
+ * Other contributors include Andrew Wright, Jeffrey Hayes,
+ * Pat Fisher, Mike Judd.
  */
 
 
 import junit.framework.*;
 import java.util.*;
 import java.util.concurrent.*;
+import static java.util.concurrent.TimeUnit.MILLISECONDS;
+import java.util.concurrent.atomic.*;
 import java.math.BigInteger;
 import java.security.*;
 
-public class ExecutorCompletionServiceTest extends JSR166TestCase{
+public class ExecutorCompletionServiceTest extends JSR166TestCase {
     public static void main(String[] args) {
-        junit.textui.TestRunner.run (suite());  
+        junit.textui.TestRunner.run (suite());
     }
     public static Test suite() {
         return new TestSuite(ExecutorCompletionServiceTest.class);
@@ -23,7 +26,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * Creating a new ECS with null Executor throw NPE
-     */ 
+     */
     public void testConstructorNPE() {
         try {
             ExecutorCompletionService ecs = new ExecutorCompletionService(null);
@@ -34,7 +37,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * Creating a new ECS with null queue throw NPE
-     */ 
+     */
     public void testConstructorNPE2() {
         try {
             ExecutorService e = Executors.newCachedThreadPool();
@@ -46,7 +49,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * Submitting a null callable throws NPE
-     */ 
+     */
     public void testSubmitNPE() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -62,7 +65,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * Submitting a null runnable throws NPE
-     */ 
+     */
     public void testSubmitNPE2() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -78,7 +81,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * A taken submitted task is completed
-     */ 
+     */
     public void testTake() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -86,7 +89,7 @@ public class ExecutorCompletionServiceTe
             Callable c = new StringTask();
             ecs.submit(c);
             Future f = ecs.take();
-            assert(f.isDone());
+            assertTrue(f.isDone());
         } catch (Exception ex) {
             unexpectedException();
         } finally {
@@ -96,7 +99,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * Take returns the same future object returned by submit
-     */ 
+     */
     public void testTake2() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -114,7 +117,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * If poll returns non-null, the returned task is completed
-     */ 
+     */
     public void testPoll1() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -126,7 +129,7 @@ public class ExecutorCompletionServiceTe
             for (;;) {
                 Future f = ecs.poll();
                 if (f != null) {
-                    assert(f.isDone());
+                    assertTrue(f.isDone());
                     break;
                 }
             }
@@ -139,7 +142,7 @@ public class ExecutorCompletionServiceTe
 
     /**
      * If timed poll returns non-null, the returned task is completed
-     */ 
+     */
     public void testPoll2() {
         ExecutorService e = Executors.newCachedThreadPool();
         ExecutorCompletionService ecs = new ExecutorCompletionService(e);
@@ -147,14 +150,85 @@ public class ExecutorCompletionServiceTe
             assertNull(ecs.poll());
             Callable c = new StringTask();
             ecs.submit(c);
-            Future f = ecs.poll(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
-            if (f != null) 
-                assert(f.isDone());
+            Future f = ecs.poll(SHORT_DELAY_MS, MILLISECONDS);
+            if (f != null)
+                assertTrue(f.isDone());
         } catch (Exception ex) {
             unexpectedException();
         } finally {
             joinPool(e);
         }
     }
+     /**
+      * Submitting to underlying AES that overrides newTaskFor(Callable)
+      * returns and eventually runs Future returned by newTaskFor.
+      */
+     public void testNewTaskForCallable() {
+         final AtomicBoolean done = new AtomicBoolean(false);
+         class MyCallableFuture<V> extends FutureTask<V> {
+             MyCallableFuture(Callable<V> c) { super(c); }
+             protected void done() { done.set(true); }
+         }
+         ExecutorService e = new ThreadPoolExecutor(
+                                 1, 1, 30L, TimeUnit.SECONDS,
+                                 new ArrayBlockingQueue<Runnable>(1)) {
+             protected <T> RunnableFuture<T> newTaskFor(Callable<T> c) {
+                 return new MyCallableFuture<T>(c);
+             }
+         };
+         ExecutorCompletionService<String> ecs =
+             new ExecutorCompletionService<String>(e);
+         try {
+             assertNull(ecs.poll());
+             Callable<String> c = new StringTask();
+             Future f1 = ecs.submit(c);
+             assertTrue("submit must return MyCallableFuture",
+                        f1 instanceof MyCallableFuture);
+             Future f2 = ecs.take();
+             assertSame("submit and take must return same objects", f1, f2);
+             assertTrue("completed task must have set done", done.get());
+         } catch (Exception ex) {
+             unexpectedException();
+         } finally {
+             joinPool(e);
+         }
+     }
+
+     /**
+      * Submitting to underlying AES that overrides newTaskFor(Runnable,T)
+      * returns and eventually runs Future returned by newTaskFor.
+      */
+     public void testNewTaskForRunnable() {
+         final AtomicBoolean done = new AtomicBoolean(false);
+         class MyRunnableFuture<V> extends FutureTask<V> {
+             MyRunnableFuture(Runnable t, V r) { super(t, r); }
+             protected void done() { done.set(true); }
+         }
+         ExecutorService e = new ThreadPoolExecutor(
+                                 1, 1, 30L, TimeUnit.SECONDS,
+                                 new ArrayBlockingQueue<Runnable>(1)) {
+             protected <T> RunnableFuture<T> newTaskFor(Runnable t, T r) {
+                 return new MyRunnableFuture<T>(t, r);
+             }
+         };
+         ExecutorCompletionService<String> ecs =
+             new ExecutorCompletionService<String>(e);
+         try {
+             assertNull(ecs.poll());
+             Runnable r = new NoOpRunnable();
+             Future f1 = ecs.submit(r, null);
+             assertTrue("submit must return MyRunnableFuture",
+                        f1 instanceof MyRunnableFuture);
+             Future f2 = ecs.take();
+             assertSame("submit and take must return same objects", f1, f2);
+             assertTrue("completed task must have set done", done.get());
+         } catch (Exception ex) {
+             unexpectedException();
+         } finally {
+             joinPool(e);
+         }
+     }
+
+
 
 }