test/tck/AbstractExecutorServiceTest.java

/*
 * 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.math.BigInteger;
import java.security.*;

public class AbstractExecutorServiceTest extends JSR166TestCase {
    public static void main(String[] args) {
        junit.textui.TestRunner.run(suite());
    }
    public static Test suite() {
        return new TestSuite(AbstractExecutorServiceTest.class);
    }

    /**
     * A no-frills implementation of AbstractExecutorService, designed
     * to test the submit methods only.
     */
    static class DirectExecutorService extends AbstractExecutorService {
        public void execute(Runnable r) { r.run(); }
        public void shutdown() { shutdown = true; }
        public List<Runnable> shutdownNow() {
            shutdown = true;
            return Collections.EMPTY_LIST;
        }
        public boolean isShutdown() { return shutdown; }
        public boolean isTerminated() { return isShutdown(); }
        public boolean awaitTermination(long timeout, TimeUnit unit) {
            return isShutdown();
        }
        private volatile boolean shutdown = false;
    }

    /**
     * execute(runnable) runs it to completion
     */
    public void testExecuteRunnable() throws Exception {
        ExecutorService e = new DirectExecutorService();
        TrackedShortRunnable task = new TrackedShortRunnable();
        assertFalse(task.done);
        Future<?> future = e.submit(task);
        future.get();
        assertTrue(task.done);
    }


    /**
     * Completed submit(callable) returns result
     */
    public void testSubmitCallable() throws Exception {
        ExecutorService e = new DirectExecutorService();
        Future<String> future = e.submit(new StringTask());
        String result = future.get();
        assertSame(TEST_STRING, result);
    }

    /**
     * Completed submit(runnable) returns successfully
     */
    public void testSubmitRunnable() throws Exception {
        ExecutorService e = new DirectExecutorService();
        Future<?> future = e.submit(new NoOpRunnable());
        future.get();
        assertTrue(future.isDone());
    }

    /**
     * Completed submit(runnable, result) returns result
     */
    public void testSubmitRunnable2() throws Exception {
        ExecutorService e = new DirectExecutorService();
        Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
        String result = future.get();
        assertSame(TEST_STRING, result);
    }


    /**
     * A submitted privileged action runs to completion
     */
    public void testSubmitPrivilegedAction() throws Exception {
        Runnable r = new CheckedRunnable() {
            public void realRun() throws Exception {
                ExecutorService e = new DirectExecutorService();
                Future future = e.submit(Executors.callable(new PrivilegedAction() {
                    public Object run() {
                        return TEST_STRING;
                    }}));

                assertSame(TEST_STRING, future.get());
            }};

        runWithPermissions(r,
                           new RuntimePermission("getClassLoader"),
                           new RuntimePermission("setContextClassLoader"),
                           new RuntimePermission("modifyThread"));
    }

    /**
     * A submitted privileged exception action runs to completion
     */
    public void testSubmitPrivilegedExceptionAction() throws Exception {
        Runnable r = new CheckedRunnable() {
            public void realRun() throws Exception {
                ExecutorService e = new DirectExecutorService();
                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
                    public Object run() {
                        return TEST_STRING;
                    }}));

                assertSame(TEST_STRING, future.get());
            }};

        runWithPermissions(r);
    }

    /**
     * A submitted failed privileged exception action reports exception
     */
    public void testSubmitFailedPrivilegedExceptionAction() throws Exception {
        Runnable r = new CheckedRunnable() {
            public void realRun() throws Exception {
                ExecutorService e = new DirectExecutorService();
                Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
                    public Object run() throws Exception {
                        throw new IndexOutOfBoundsException();
                    }}));

                try {
                    future.get();
                    shouldThrow();
                } catch (ExecutionException success) {
                    assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
                }}};

        runWithPermissions(r);
    }

    /**
     * execute(null runnable) throws NPE
     */
    public void testExecuteNullRunnable() {
        try {
            ExecutorService e = new DirectExecutorService();
            e.submit((Runnable) null);
            shouldThrow();
        } catch (NullPointerException success) {}
    }


    /**
     * submit(null callable) throws NPE
     */
    public void testSubmitNullCallable() {
        try {
            ExecutorService e = new DirectExecutorService();
            e.submit((Callable) null);
            shouldThrow();
        } catch (NullPointerException success) {}
    }

    /**
     * submit(callable).get() throws InterruptedException if interrupted
     */
    public void testInterruptedSubmit() throws InterruptedException {
        final CountDownLatch submitted    = new CountDownLatch(1);
        final CountDownLatch quittingTime = new CountDownLatch(1);
        final ExecutorService p
            = new ThreadPoolExecutor(1,1,60, TimeUnit.SECONDS,
                                     new ArrayBlockingQueue<Runnable>(10));
        final Callable<Void> awaiter = new CheckedCallable<Void>() {
            public Void realCall() throws InterruptedException {
                quittingTime.await();
                return null;
            }};
        try {
            Thread t = new Thread(new CheckedInterruptedRunnable() {
                public void realRun() throws Exception {
                    Future<Void> future = p.submit(awaiter);
                    submitted.countDown();
                    future.get();
                }});
            t.start();
            submitted.await();
            t.interrupt();
            t.join();
        } finally {
            quittingTime.countDown();
            joinPool(p);
        }
    }

    /**
     * get of submit(callable) throws ExecutionException if callable
     * throws exception
     */
    public void testSubmitEE() throws InterruptedException {
        ThreadPoolExecutor p =
            new ThreadPoolExecutor(1, 1,
                                   60, TimeUnit.SECONDS,
                                   new ArrayBlockingQueue<Runnable>(10));

        Callable c = new Callable() {
            public Object call() { return 5/0; }};

        try {
            p.submit(c).get();
            shouldThrow();
        } catch (ExecutionException success) {
            assertTrue(success.getCause() instanceof ArithmeticException);
        }
        joinPool(p);
    }

    /**
     * invokeAny(null) throws NPE
     */
    public void testInvokeAny1() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAny(null);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAny(empty collection) throws IAE
     */
    public void testInvokeAny2() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAny(new ArrayList<Callable<String>>());
            shouldThrow();
        } catch (IllegalArgumentException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAny(c) throws NPE if c has null elements
     */
    public void testInvokeAny3() throws Exception {
        ExecutorService e = new DirectExecutorService();
        List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
        l.add(new Callable<Integer>() {
                  public Integer call() { return 5/0; }});
        l.add(null);
        try {
            e.invokeAny(l);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAny(c) throws ExecutionException if no task in c completes
     */
    public void testInvokeAny4() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new NPETask());
        try {
            e.invokeAny(l);
            shouldThrow();
        } catch (ExecutionException success) {
            assertTrue(success.getCause() instanceof NullPointerException);
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAny(c) returns result of some task in c if at least one completes
     */
    public void testInvokeAny5() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new StringTask());
            l.add(new StringTask());
            String result = e.invokeAny(l);
            assertSame(TEST_STRING, result);
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAll(null) throws NPE
     */
    public void testInvokeAll1() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAll(null);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAll(empty collection) returns empty collection
     */
    public void testInvokeAll2() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
            assertTrue(r.isEmpty());
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAll(c) throws NPE if c has null elements
     */
    public void testInvokeAll3() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new StringTask());
        l.add(null);
        try {
            e.invokeAll(l);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * get of returned element of invokeAll(c) throws exception on failed task
     */
    public void testInvokeAll4() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new NPETask());
            List<Future<String>> futures = e.invokeAll(l);
            assertEquals(1, futures.size());
            try {
                futures.get(0).get();
                shouldThrow();
            } catch (ExecutionException success) {
                assertTrue(success.getCause() instanceof NullPointerException);
            }
        } finally {
            joinPool(e);
        }
    }

    /**
     * invokeAll(c) returns results of all completed tasks in c
     */
    public void testInvokeAll5() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new StringTask());
            l.add(new StringTask());
            List<Future<String>> futures = e.invokeAll(l);
            assertEquals(2, futures.size());
            for (Future<String> future : futures)
                assertSame(TEST_STRING, future.get());
        } finally {
            joinPool(e);
        }
    }


    /**
     * timed invokeAny(null) throws NPE
     */
    public void testTimedInvokeAny1() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAny(null time unit) throws NPE
     */
    public void testTimedInvokeAnyNullTimeUnit() throws Exception {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new StringTask());
        try {
            e.invokeAny(l, MEDIUM_DELAY_MS, null);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAny(empty collection) throws IAE
     */
    public void testTimedInvokeAny2() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (IllegalArgumentException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAny(c) throws NPE if c has null elements
     */
    public void testTimedInvokeAny3() throws Exception {
        ExecutorService e = new DirectExecutorService();
        List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
        l.add(new Callable<Integer>() {
                  public Integer call() { return 5/0; }});
        l.add(null);
        try {
            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAny(c) throws ExecutionException if no task completes
     */
    public void testTimedInvokeAny4() throws Exception {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new NPETask());
        try {
            e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (ExecutionException success) {
            assertTrue(success.getCause() instanceof NullPointerException);
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAny(c) returns result of some task in c
     */
    public void testTimedInvokeAny5() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new StringTask());
            l.add(new StringTask());
            String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
            assertSame(TEST_STRING, result);
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll(null) throws NPE
     */
    public void testTimedInvokeAll1() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        try {
            e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll(null time unit) throws NPE
     */
    public void testTimedInvokeAllNullTimeUnit() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new StringTask());
        try {
            e.invokeAll(l, MEDIUM_DELAY_MS, null);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll(empty collection) returns empty collection
     */
    public void testTimedInvokeAll2() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
            assertTrue(r.isEmpty());
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll(c) throws NPE if c has null elements
     */
    public void testTimedInvokeAll3() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        List<Callable<String>> l = new ArrayList<Callable<String>>();
        l.add(new StringTask());
        l.add(null);
        try {
            e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
            shouldThrow();
        } catch (NullPointerException success) {
        } finally {
            joinPool(e);
        }
    }

    /**
     * get of returned element of invokeAll(c) throws exception on failed task
     */
    public void testTimedInvokeAll4() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new NPETask());
            List<Future<String>> futures =
                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
            assertEquals(1, futures.size());
            try {
                futures.get(0).get();
                shouldThrow();
            } catch (ExecutionException success) {
                assertTrue(success.getCause() instanceof NullPointerException);
            }
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll(c) returns results of all completed tasks in c
     */
    public void testTimedInvokeAll5() throws Exception {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new StringTask());
            l.add(new StringTask());
            List<Future<String>> futures =
                e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
            assertEquals(2, futures.size());
            for (Future<String> future : futures)
                assertSame(TEST_STRING, future.get());
        } finally {
            joinPool(e);
        }
    }

    /**
     * timed invokeAll cancels tasks not completed by timeout
     */
    public void testTimedInvokeAll6() throws InterruptedException {
        ExecutorService e = new DirectExecutorService();
        try {
            List<Callable<String>> l = new ArrayList<Callable<String>>();
            l.add(new StringTask());
            l.add(Executors.callable(possiblyInterruptedRunnable(2 * SHORT_DELAY_MS), TEST_STRING));
            l.add(new StringTask());
            List<Future<String>> futures =
                e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
            assertEquals(3, futures.size());
            Iterator<Future<String>> it = futures.iterator();
            Future<String> f1 = it.next();
            Future<String> f2 = it.next();
            Future<String> f3 = it.next();
            assertTrue(f1.isDone());
            assertFalse(f1.isCancelled());
            assertTrue(f2.isDone());
            assertFalse(f2.isCancelled());
            assertTrue(f3.isDone());
            assertTrue(f3.isCancelled());
        } finally {
            joinPool(e);
        }
    }

}
Revision:	1.29
Committed:	Mon Oct 11 08:30:50 2010 UTC (13 years, 7 months ago) by jsr166
Branch:	MAIN
Changes since 1.28:	+3 -2 lines
Log Message:	optimize testTimedInvokeAll6
#	Content
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.
7	*/
8
9
10	import junit.framework.*;
11	import java.util.*;
12	import java.util.concurrent.*;
13	import static java.util.concurrent.TimeUnit.MILLISECONDS;
14	import java.math.BigInteger;
15	import java.security.*;
16
17	public class AbstractExecutorServiceTest extends JSR166TestCase {
18	public static void main(String[] args) {
19	junit.textui.TestRunner.run(suite());
20	}
21	public static Test suite() {
22	return new TestSuite(AbstractExecutorServiceTest.class);
23	}
24
25	/**
26	* A no-frills implementation of AbstractExecutorService, designed
27	* to test the submit methods only.
28	*/
29	static class DirectExecutorService extends AbstractExecutorService {
30	public void execute(Runnable r) { r.run(); }
31	public void shutdown() { shutdown = true; }
32	public List<Runnable> shutdownNow() {
33	shutdown = true;
34	return Collections.EMPTY_LIST;
35	}
36	public boolean isShutdown() { return shutdown; }
37	public boolean isTerminated() { return isShutdown(); }
38	public boolean awaitTermination(long timeout, TimeUnit unit) {
39	return isShutdown();
40	}
41	private volatile boolean shutdown = false;
42	}
43
44	/**
45	* execute(runnable) runs it to completion
46	*/
47	public void testExecuteRunnable() throws Exception {
48	ExecutorService e = new DirectExecutorService();
49	TrackedShortRunnable task = new TrackedShortRunnable();
50	assertFalse(task.done);
51	Future<?> future = e.submit(task);
52	future.get();
53	assertTrue(task.done);
54	}
55
56
57	/**
58	* Completed submit(callable) returns result
59	*/
60	public void testSubmitCallable() throws Exception {
61	ExecutorService e = new DirectExecutorService();
62	Future<String> future = e.submit(new StringTask());
63	String result = future.get();
64	assertSame(TEST_STRING, result);
65	}
66
67	/**
68	* Completed submit(runnable) returns successfully
69	*/
70	public void testSubmitRunnable() throws Exception {
71	ExecutorService e = new DirectExecutorService();
72	Future<?> future = e.submit(new NoOpRunnable());
73	future.get();
74	assertTrue(future.isDone());
75	}
76
77	/**
78	* Completed submit(runnable, result) returns result
79	*/
80	public void testSubmitRunnable2() throws Exception {
81	ExecutorService e = new DirectExecutorService();
82	Future<String> future = e.submit(new NoOpRunnable(), TEST_STRING);
83	String result = future.get();
84	assertSame(TEST_STRING, result);
85	}
86
87
88	/**
89	* A submitted privileged action runs to completion
90	*/
91	public void testSubmitPrivilegedAction() throws Exception {
92	Runnable r = new CheckedRunnable() {
93	public void realRun() throws Exception {
94	ExecutorService e = new DirectExecutorService();
95	Future future = e.submit(Executors.callable(new PrivilegedAction() {
96	public Object run() {
97	return TEST_STRING;
98	}}));
99
100	assertSame(TEST_STRING, future.get());
101	}};
102
103	runWithPermissions(r,
104	new RuntimePermission("getClassLoader"),
105	new RuntimePermission("setContextClassLoader"),
106	new RuntimePermission("modifyThread"));
107	}
108
109	/**
110	* A submitted privileged exception action runs to completion
111	*/
112	public void testSubmitPrivilegedExceptionAction() throws Exception {
113	Runnable r = new CheckedRunnable() {
114	public void realRun() throws Exception {
115	ExecutorService e = new DirectExecutorService();
116	Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
117	public Object run() {
118	return TEST_STRING;
119	}}));
120
121	assertSame(TEST_STRING, future.get());
122	}};
123
124	runWithPermissions(r);
125	}
126
127	/**
128	* A submitted failed privileged exception action reports exception
129	*/
130	public void testSubmitFailedPrivilegedExceptionAction() throws Exception {
131	Runnable r = new CheckedRunnable() {
132	public void realRun() throws Exception {
133	ExecutorService e = new DirectExecutorService();
134	Future future = e.submit(Executors.callable(new PrivilegedExceptionAction() {
135	public Object run() throws Exception {
136	throw new IndexOutOfBoundsException();
137	}}));
138
139	try {
140	future.get();
141	shouldThrow();
142	} catch (ExecutionException success) {
143	assertTrue(success.getCause() instanceof IndexOutOfBoundsException);
144	}}};
145
146	runWithPermissions(r);
147	}
148
149	/**
150	* execute(null runnable) throws NPE
151	*/
152	public void testExecuteNullRunnable() {
153	try {
154	ExecutorService e = new DirectExecutorService();
155	e.submit((Runnable) null);
156	shouldThrow();
157	} catch (NullPointerException success) {}
158	}
159
160
161	/**
162	* submit(null callable) throws NPE
163	*/
164	public void testSubmitNullCallable() {
165	try {
166	ExecutorService e = new DirectExecutorService();
167	e.submit((Callable) null);
168	shouldThrow();
169	} catch (NullPointerException success) {}
170	}
171
172	/**
173	* submit(callable).get() throws InterruptedException if interrupted
174	*/
175	public void testInterruptedSubmit() throws InterruptedException {
176	final CountDownLatch submitted = new CountDownLatch(1);
177	final CountDownLatch quittingTime = new CountDownLatch(1);
178	final ExecutorService p
179	= new ThreadPoolExecutor(1,1,60, TimeUnit.SECONDS,
180	new ArrayBlockingQueue<Runnable>(10));
181	final Callable<Void> awaiter = new CheckedCallable<Void>() {
182	public Void realCall() throws InterruptedException {
183	quittingTime.await();
184	return null;
185	}};
186	try {
187	Thread t = new Thread(new CheckedInterruptedRunnable() {
188	public void realRun() throws Exception {
189	Future<Void> future = p.submit(awaiter);
190	submitted.countDown();
191	future.get();
192	}});
193	t.start();
194	submitted.await();
195	t.interrupt();
196	t.join();
197	} finally {
198	quittingTime.countDown();
199	joinPool(p);
200	}
201	}
202
203	/**
204	* get of submit(callable) throws ExecutionException if callable
205	* throws exception
206	*/
207	public void testSubmitEE() throws InterruptedException {
208	ThreadPoolExecutor p =
209	new ThreadPoolExecutor(1, 1,
210	60, TimeUnit.SECONDS,
211	new ArrayBlockingQueue<Runnable>(10));
212
213	Callable c = new Callable() {
214	public Object call() { return 5/0; }};
215
216	try {
217	p.submit(c).get();
218	shouldThrow();
219	} catch (ExecutionException success) {
220	assertTrue(success.getCause() instanceof ArithmeticException);
221	}
222	joinPool(p);
223	}
224
225	/**
226	* invokeAny(null) throws NPE
227	*/
228	public void testInvokeAny1() throws Exception {
229	ExecutorService e = new DirectExecutorService();
230	try {
231	e.invokeAny(null);
232	shouldThrow();
233	} catch (NullPointerException success) {
234	} finally {
235	joinPool(e);
236	}
237	}
238
239	/**
240	* invokeAny(empty collection) throws IAE
241	*/
242	public void testInvokeAny2() throws Exception {
243	ExecutorService e = new DirectExecutorService();
244	try {
245	e.invokeAny(new ArrayList<Callable<String>>());
246	shouldThrow();
247	} catch (IllegalArgumentException success) {
248	} finally {
249	joinPool(e);
250	}
251	}
252
253	/**
254	* invokeAny(c) throws NPE if c has null elements
255	*/
256	public void testInvokeAny3() throws Exception {
257	ExecutorService e = new DirectExecutorService();
258	List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
259	l.add(new Callable<Integer>() {
260	public Integer call() { return 5/0; }});
261	l.add(null);
262	try {
263	e.invokeAny(l);
264	shouldThrow();
265	} catch (NullPointerException success) {
266	} finally {
267	joinPool(e);
268	}
269	}
270
271	/**
272	* invokeAny(c) throws ExecutionException if no task in c completes
273	*/
274	public void testInvokeAny4() throws InterruptedException {
275	ExecutorService e = new DirectExecutorService();
276	List<Callable<String>> l = new ArrayList<Callable<String>>();
277	l.add(new NPETask());
278	try {
279	e.invokeAny(l);
280	shouldThrow();
281	} catch (ExecutionException success) {
282	assertTrue(success.getCause() instanceof NullPointerException);
283	} finally {
284	joinPool(e);
285	}
286	}
287
288	/**
289	* invokeAny(c) returns result of some task in c if at least one completes
290	*/
291	public void testInvokeAny5() throws Exception {
292	ExecutorService e = new DirectExecutorService();
293	try {
294	List<Callable<String>> l = new ArrayList<Callable<String>>();
295	l.add(new StringTask());
296	l.add(new StringTask());
297	String result = e.invokeAny(l);
298	assertSame(TEST_STRING, result);
299	} finally {
300	joinPool(e);
301	}
302	}
303
304	/**
305	* invokeAll(null) throws NPE
306	*/
307	public void testInvokeAll1() throws InterruptedException {
308	ExecutorService e = new DirectExecutorService();
309	try {
310	e.invokeAll(null);
311	shouldThrow();
312	} catch (NullPointerException success) {
313	} finally {
314	joinPool(e);
315	}
316	}
317
318	/**
319	* invokeAll(empty collection) returns empty collection
320	*/
321	public void testInvokeAll2() throws InterruptedException {
322	ExecutorService e = new DirectExecutorService();
323	try {
324	List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>());
325	assertTrue(r.isEmpty());
326	} finally {
327	joinPool(e);
328	}
329	}
330
331	/**
332	* invokeAll(c) throws NPE if c has null elements
333	*/
334	public void testInvokeAll3() throws InterruptedException {
335	ExecutorService e = new DirectExecutorService();
336	List<Callable<String>> l = new ArrayList<Callable<String>>();
337	l.add(new StringTask());
338	l.add(null);
339	try {
340	e.invokeAll(l);
341	shouldThrow();
342	} catch (NullPointerException success) {
343	} finally {
344	joinPool(e);
345	}
346	}
347
348	/**
349	* get of returned element of invokeAll(c) throws exception on failed task
350	*/
351	public void testInvokeAll4() throws Exception {
352	ExecutorService e = new DirectExecutorService();
353	try {
354	List<Callable<String>> l = new ArrayList<Callable<String>>();
355	l.add(new NPETask());
356	List<Future<String>> futures = e.invokeAll(l);
357	assertEquals(1, futures.size());
358	try {
359	futures.get(0).get();
360	shouldThrow();
361	} catch (ExecutionException success) {
362	assertTrue(success.getCause() instanceof NullPointerException);
363	}
364	} finally {
365	joinPool(e);
366	}
367	}
368
369	/**
370	* invokeAll(c) returns results of all completed tasks in c
371	*/
372	public void testInvokeAll5() throws Exception {
373	ExecutorService e = new DirectExecutorService();
374	try {
375	List<Callable<String>> l = new ArrayList<Callable<String>>();
376	l.add(new StringTask());
377	l.add(new StringTask());
378	List<Future<String>> futures = e.invokeAll(l);
379	assertEquals(2, futures.size());
380	for (Future<String> future : futures)
381	assertSame(TEST_STRING, future.get());
382	} finally {
383	joinPool(e);
384	}
385	}
386
387
388	/**
389	* timed invokeAny(null) throws NPE
390	*/
391	public void testTimedInvokeAny1() throws Exception {
392	ExecutorService e = new DirectExecutorService();
393	try {
394	e.invokeAny(null, MEDIUM_DELAY_MS, MILLISECONDS);
395	shouldThrow();
396	} catch (NullPointerException success) {
397	} finally {
398	joinPool(e);
399	}
400	}
401
402	/**
403	* timed invokeAny(null time unit) throws NPE
404	*/
405	public void testTimedInvokeAnyNullTimeUnit() throws Exception {
406	ExecutorService e = new DirectExecutorService();
407	List<Callable<String>> l = new ArrayList<Callable<String>>();
408	l.add(new StringTask());
409	try {
410	e.invokeAny(l, MEDIUM_DELAY_MS, null);
411	shouldThrow();
412	} catch (NullPointerException success) {
413	} finally {
414	joinPool(e);
415	}
416	}
417
418	/**
419	* timed invokeAny(empty collection) throws IAE
420	*/
421	public void testTimedInvokeAny2() throws Exception {
422	ExecutorService e = new DirectExecutorService();
423	try {
424	e.invokeAny(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
425	shouldThrow();
426	} catch (IllegalArgumentException success) {
427	} finally {
428	joinPool(e);
429	}
430	}
431
432	/**
433	* timed invokeAny(c) throws NPE if c has null elements
434	*/
435	public void testTimedInvokeAny3() throws Exception {
436	ExecutorService e = new DirectExecutorService();
437	List<Callable<Integer>> l = new ArrayList<Callable<Integer>>();
438	l.add(new Callable<Integer>() {
439	public Integer call() { return 5/0; }});
440	l.add(null);
441	try {
442	e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
443	shouldThrow();
444	} catch (NullPointerException success) {
445	} finally {
446	joinPool(e);
447	}
448	}
449
450	/**
451	* timed invokeAny(c) throws ExecutionException if no task completes
452	*/
453	public void testTimedInvokeAny4() throws Exception {
454	ExecutorService e = new DirectExecutorService();
455	List<Callable<String>> l = new ArrayList<Callable<String>>();
456	l.add(new NPETask());
457	try {
458	e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
459	shouldThrow();
460	} catch (ExecutionException success) {
461	assertTrue(success.getCause() instanceof NullPointerException);
462	} finally {
463	joinPool(e);
464	}
465	}
466
467	/**
468	* timed invokeAny(c) returns result of some task in c
469	*/
470	public void testTimedInvokeAny5() throws Exception {
471	ExecutorService e = new DirectExecutorService();
472	try {
473	List<Callable<String>> l = new ArrayList<Callable<String>>();
474	l.add(new StringTask());
475	l.add(new StringTask());
476	String result = e.invokeAny(l, MEDIUM_DELAY_MS, MILLISECONDS);
477	assertSame(TEST_STRING, result);
478	} finally {
479	joinPool(e);
480	}
481	}
482
483	/**
484	* timed invokeAll(null) throws NPE
485	*/
486	public void testTimedInvokeAll1() throws InterruptedException {
487	ExecutorService e = new DirectExecutorService();
488	try {
489	e.invokeAll(null, MEDIUM_DELAY_MS, MILLISECONDS);
490	shouldThrow();
491	} catch (NullPointerException success) {
492	} finally {
493	joinPool(e);
494	}
495	}
496
497	/**
498	* timed invokeAll(null time unit) throws NPE
499	*/
500	public void testTimedInvokeAllNullTimeUnit() throws InterruptedException {
501	ExecutorService e = new DirectExecutorService();
502	List<Callable<String>> l = new ArrayList<Callable<String>>();
503	l.add(new StringTask());
504	try {
505	e.invokeAll(l, MEDIUM_DELAY_MS, null);
506	shouldThrow();
507	} catch (NullPointerException success) {
508	} finally {
509	joinPool(e);
510	}
511	}
512
513	/**
514	* timed invokeAll(empty collection) returns empty collection
515	*/
516	public void testTimedInvokeAll2() throws InterruptedException {
517	ExecutorService e = new DirectExecutorService();
518	try {
519	List<Future<String>> r = e.invokeAll(new ArrayList<Callable<String>>(), MEDIUM_DELAY_MS, MILLISECONDS);
520	assertTrue(r.isEmpty());
521	} finally {
522	joinPool(e);
523	}
524	}
525
526	/**
527	* timed invokeAll(c) throws NPE if c has null elements
528	*/
529	public void testTimedInvokeAll3() throws InterruptedException {
530	ExecutorService e = new DirectExecutorService();
531	List<Callable<String>> l = new ArrayList<Callable<String>>();
532	l.add(new StringTask());
533	l.add(null);
534	try {
535	e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
536	shouldThrow();
537	} catch (NullPointerException success) {
538	} finally {
539	joinPool(e);
540	}
541	}
542
543	/**
544	* get of returned element of invokeAll(c) throws exception on failed task
545	*/
546	public void testTimedInvokeAll4() throws Exception {
547	ExecutorService e = new DirectExecutorService();
548	try {
549	List<Callable<String>> l = new ArrayList<Callable<String>>();
550	l.add(new NPETask());
551	List<Future<String>> futures =
552	e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
553	assertEquals(1, futures.size());
554	try {
555	futures.get(0).get();
556	shouldThrow();
557	} catch (ExecutionException success) {
558	assertTrue(success.getCause() instanceof NullPointerException);
559	}
560	} finally {
561	joinPool(e);
562	}
563	}
564
565	/**
566	* timed invokeAll(c) returns results of all completed tasks in c
567	*/
568	public void testTimedInvokeAll5() throws Exception {
569	ExecutorService e = new DirectExecutorService();
570	try {
571	List<Callable<String>> l = new ArrayList<Callable<String>>();
572	l.add(new StringTask());
573	l.add(new StringTask());
574	List<Future<String>> futures =
575	e.invokeAll(l, MEDIUM_DELAY_MS, MILLISECONDS);
576	assertEquals(2, futures.size());
577	for (Future<String> future : futures)
578	assertSame(TEST_STRING, future.get());
579	} finally {
580	joinPool(e);
581	}
582	}
583
584	/**
585	* timed invokeAll cancels tasks not completed by timeout
586	*/
587	public void testTimedInvokeAll6() throws InterruptedException {
588	ExecutorService e = new DirectExecutorService();
589	try {
590	List<Callable<String>> l = new ArrayList<Callable<String>>();
591	l.add(new StringTask());
592	l.add(Executors.callable(possiblyInterruptedRunnable(2 * SHORT_DELAY_MS), TEST_STRING));
593	l.add(new StringTask());
594	List<Future<String>> futures =
595	e.invokeAll(l, SHORT_DELAY_MS, MILLISECONDS);
596	assertEquals(3, futures.size());
597	Iterator<Future<String>> it = futures.iterator();
598	Future<String> f1 = it.next();
599	Future<String> f2 = it.next();
600	Future<String> f3 = it.next();
601	assertTrue(f1.isDone());
602	assertFalse(f1.isCancelled());
603	assertTrue(f2.isDone());
604	assertFalse(f2.isCancelled());
605	assertTrue(f3.isDone());
606	assertTrue(f3.isCancelled());
607	} finally {
608	joinPool(e);
609	}
610	}
611
612	}