test/tck/CyclicBarrierTest.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/publicdomain/zero/1.0/
 * Other contributors include Andrew Wright, Jeffrey Hayes,
 * Pat Fisher, Mike Judd.
 */

import static java.util.concurrent.TimeUnit.MILLISECONDS;

import java.util.concurrent.BrokenBarrierException;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.CyclicBarrier;
import java.util.concurrent.TimeoutException;
import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicInteger;

import junit.framework.Test;
import junit.framework.TestSuite;

public class CyclicBarrierTest extends JSR166TestCase {
    public static void main(String[] args) {
        main(suite(), args);
    }
    public static Test suite() {
        return new TestSuite(CyclicBarrierTest.class);
    }

    /**
     * Spin-waits till the number of waiters == numberOfWaiters.
     */
    void awaitNumberWaiting(CyclicBarrier barrier, int numberOfWaiters) {
        long startTime = System.nanoTime();
        while (barrier.getNumberWaiting() != numberOfWaiters) {
            if (millisElapsedSince(startTime) > LONG_DELAY_MS)
                fail("timed out");
            Thread.yield();
        }
    }

    /**
     * Creating with negative parties throws IllegalArgumentException
     */
    public void testConstructor1() {
        try {
            new CyclicBarrier(-1, (Runnable)null);
            shouldThrow();
        } catch (IllegalArgumentException success) {}
    }

    /**
     * Creating with negative parties and no action throws
     * IllegalArgumentException
     */
    public void testConstructor2() {
        try {
            new CyclicBarrier(-1);
            shouldThrow();
        } catch (IllegalArgumentException success) {}
    }

    /**
     * getParties returns the number of parties given in constructor
     */
    public void testGetParties() {
        CyclicBarrier b = new CyclicBarrier(2);
        assertEquals(2, b.getParties());
        assertEquals(0, b.getNumberWaiting());
    }

    /**
     * A 1-party barrier triggers after single await
     */
    public void testSingleParty() throws Exception {
        CyclicBarrier b = new CyclicBarrier(1);
        assertEquals(1, b.getParties());
        assertEquals(0, b.getNumberWaiting());
        b.await();
        b.await();
        assertEquals(0, b.getNumberWaiting());
    }

    /**
     * The supplied barrier action is run at barrier
     */
    public void testBarrierAction() throws Exception {
        final AtomicInteger count = new AtomicInteger(0);
        final Runnable incCount = new Runnable() { public void run() {
            count.getAndIncrement(); }};
        CyclicBarrier b = new CyclicBarrier(1, incCount);
        assertEquals(1, b.getParties());
        assertEquals(0, b.getNumberWaiting());
        b.await();
        b.await();
        assertEquals(0, b.getNumberWaiting());
        assertEquals(2, count.get());
    }

    /**
     * A 2-party/thread barrier triggers after both threads invoke await
     */
    public void testTwoParties() throws Exception {
        final CyclicBarrier b = new CyclicBarrier(2);
        Thread t = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                b.await();
                b.await();
                b.await();
                b.await();
            }});

        b.await();
        b.await();
        b.await();
        b.await();
        awaitTermination(t);
    }

    /**
     * An interruption in one party causes others waiting in await to
     * throw BrokenBarrierException
     */
    public void testAwait1_Interrupted_BrokenBarrier() {
        final CyclicBarrier c = new CyclicBarrier(3);
        final CountDownLatch pleaseInterrupt = new CountDownLatch(2);
        Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
            public void realRun() throws Exception {
                pleaseInterrupt.countDown();
                c.await();
            }};
        Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
            public void realRun() throws Exception {
                pleaseInterrupt.countDown();
                c.await();
            }};

        t1.start();
        t2.start();
        await(pleaseInterrupt);
        t1.interrupt();
        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * An interruption in one party causes others waiting in timed await to
     * throw BrokenBarrierException
     */
    public void testAwait2_Interrupted_BrokenBarrier() throws Exception {
        final CyclicBarrier c = new CyclicBarrier(3);
        final CountDownLatch pleaseInterrupt = new CountDownLatch(2);
        Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
            public void realRun() throws Exception {
                pleaseInterrupt.countDown();
                c.await(LONG_DELAY_MS, MILLISECONDS);
            }};
        Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
            public void realRun() throws Exception {
                pleaseInterrupt.countDown();
                c.await(LONG_DELAY_MS, MILLISECONDS);
            }};

        t1.start();
        t2.start();
        await(pleaseInterrupt);
        t1.interrupt();
        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * A timeout in timed await throws TimeoutException
     */
    public void testAwait3_TimeoutException() throws InterruptedException {
        final CyclicBarrier c = new CyclicBarrier(2);
        Thread t = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                long startTime = System.nanoTime();
                try {
                    c.await(timeoutMillis(), MILLISECONDS);
                    shouldThrow();
                } catch (TimeoutException success) {}
                assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
            }});

        awaitTermination(t);
    }

    /**
     * A timeout in one party causes others waiting in timed await to
     * throw BrokenBarrierException
     */
    public void testAwait4_Timeout_BrokenBarrier() throws InterruptedException {
        final CyclicBarrier c = new CyclicBarrier(3);
        Thread t1 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                try {
                    c.await(LONG_DELAY_MS, MILLISECONDS);
                    shouldThrow();
                } catch (BrokenBarrierException success) {}
            }});
        Thread t2 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                awaitNumberWaiting(c, 1);
                long startTime = System.nanoTime();
                try {
                    c.await(timeoutMillis(), MILLISECONDS);
                    shouldThrow();
                } catch (TimeoutException success) {}
                assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
            }});

        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * A timeout in one party causes others waiting in await to
     * throw BrokenBarrierException
     */
    public void testAwait5_Timeout_BrokenBarrier() throws InterruptedException {
        final CyclicBarrier c = new CyclicBarrier(3);
        Thread t1 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                try {
                    c.await();
                    shouldThrow();
                } catch (BrokenBarrierException success) {}
            }});
        Thread t2 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                awaitNumberWaiting(c, 1);
                long startTime = System.nanoTime();
                try {
                    c.await(timeoutMillis(), MILLISECONDS);
                    shouldThrow();
                } catch (TimeoutException success) {}
                assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
            }});

        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * A reset of an active barrier causes waiting threads to throw
     * BrokenBarrierException
     */
    public void testReset_BrokenBarrier() throws InterruptedException {
        final CyclicBarrier c = new CyclicBarrier(3);
        final CountDownLatch pleaseReset = new CountDownLatch(2);
        Thread t1 = new ThreadShouldThrow(BrokenBarrierException.class) {
            public void realRun() throws Exception {
                pleaseReset.countDown();
                c.await();
            }};
        Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
            public void realRun() throws Exception {
                pleaseReset.countDown();
                c.await();
            }};

        t1.start();
        t2.start();
        await(pleaseReset);

        awaitNumberWaiting(c, 2);
        c.reset();
        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * A reset before threads enter barrier does not throw
     * BrokenBarrierException
     */
    public void testReset_NoBrokenBarrier() throws Exception {
        final CyclicBarrier c = new CyclicBarrier(3);
        c.reset();

        Thread t1 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                c.await();
            }});
        Thread t2 = newStartedThread(new CheckedRunnable() {
            public void realRun() throws Exception {
                c.await();
            }});

        c.await();
        awaitTermination(t1);
        awaitTermination(t2);
    }

    /**
     * All threads block while a barrier is broken.
     */
    public void testReset_Leakage() throws InterruptedException {
        final CyclicBarrier c = new CyclicBarrier(2);
        final AtomicBoolean done = new AtomicBoolean();
        Thread t = newStartedThread(new CheckedRunnable() {
            public void realRun() {
                while (!done.get()) {
                    try {
                        while (c.isBroken())
                            c.reset();

                        c.await();
                        shouldThrow();
                    }
                    catch (BrokenBarrierException ok) {}
                    catch (InterruptedException ok) {}
                }}});

        for (int i = 0; i < 4; i++) {
            delay(timeoutMillis());
            t.interrupt();
        }
        done.set(true);
        t.interrupt();
        awaitTermination(t);
    }

    /**
     * Reset of a non-broken barrier does not break barrier
     */
    public void testResetWithoutBreakage() throws Exception {
        final CyclicBarrier barrier = new CyclicBarrier(3);
        for (int i = 0; i < 3; i++) {
            final CyclicBarrier start = new CyclicBarrier(3);
            Thread t1 = newStartedThread(new CheckedRunnable() {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }});

            Thread t2 = newStartedThread(new CheckedRunnable() {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }});

            start.await();
            barrier.await();
            awaitTermination(t1);
            awaitTermination(t2);
            assertFalse(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
            if (i == 1) barrier.reset();
            assertFalse(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
        }
    }

    /**
     * Reset of a barrier after interruption reinitializes it.
     */
    public void testResetAfterInterrupt() throws Exception {
        final CyclicBarrier barrier = new CyclicBarrier(3);
        for (int i = 0; i < 2; i++) {
            final CyclicBarrier start = new CyclicBarrier(3);
            Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }};

            Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }};

            t1.start();
            t2.start();
            start.await();
            t1.interrupt();
            awaitTermination(t1);
            awaitTermination(t2);
            assertTrue(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
            barrier.reset();
            assertFalse(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
        }
    }

    /**
     * Reset of a barrier after timeout reinitializes it.
     */
    public void testResetAfterTimeout() throws Exception {
        final CyclicBarrier barrier = new CyclicBarrier(3);
        for (int i = 0; i < 2; i++) {
            assertEquals(0, barrier.getNumberWaiting());
            Thread t1 = newStartedThread(new CheckedRunnable() {
                public void realRun() throws Exception {
                    try {
                        barrier.await();
                        shouldThrow();
                    } catch (BrokenBarrierException success) {}
                }});
            Thread t2 = newStartedThread(new CheckedRunnable() {
                public void realRun() throws Exception {
                    awaitNumberWaiting(barrier, 1);
                    long startTime = System.nanoTime();
                    try {
                        barrier.await(timeoutMillis(), MILLISECONDS);
                        shouldThrow();
                    } catch (TimeoutException success) {}
                    assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
                }});

            awaitTermination(t1);
            awaitTermination(t2);
            assertEquals(0, barrier.getNumberWaiting());
            assertTrue(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
            barrier.reset();
            assertFalse(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
        }
    }

    /**
     * Reset of a barrier after a failed command reinitializes it.
     */
    public void testResetAfterCommandException() throws Exception {
        final CyclicBarrier barrier =
            new CyclicBarrier(3, new Runnable() {
                    public void run() {
                        throw new NullPointerException(); }});
        for (int i = 0; i < 2; i++) {
            final CyclicBarrier start = new CyclicBarrier(3);
            Thread t1 = new ThreadShouldThrow(BrokenBarrierException.class) {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }};

            Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
                public void realRun() throws Exception {
                    start.await();
                    barrier.await();
                }};

            t1.start();
            t2.start();
            start.await();
            awaitNumberWaiting(barrier, 2);
            try {
                barrier.await();
                shouldThrow();
            } catch (NullPointerException success) {}
            awaitTermination(t1);
            awaitTermination(t2);
            assertTrue(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
            barrier.reset();
            assertFalse(barrier.isBroken());
            assertEquals(0, barrier.getNumberWaiting());
        }
    }
}
Revision:	1.29
Committed:	Mon May 29 22:44:26 2017 UTC (6 years, 11 months ago) by jsr166
Branch:	MAIN
Changes since 1.28:	+3 -2 lines
Log Message:	more timeout handling rework; remove most uses of MEDIUM_DELAY_MS; randomize timeouts and TimeUnits; write out IAE and ISE
#	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/publicdomain/zero/1.0/
5	* Other contributors include Andrew Wright, Jeffrey Hayes,
6	* Pat Fisher, Mike Judd.
7	*/
8
9	import static java.util.concurrent.TimeUnit.MILLISECONDS;
10
11	import java.util.concurrent.BrokenBarrierException;
12	import java.util.concurrent.CountDownLatch;
13	import java.util.concurrent.CyclicBarrier;
14	import java.util.concurrent.TimeoutException;
15	import java.util.concurrent.atomic.AtomicBoolean;
16	import java.util.concurrent.atomic.AtomicInteger;
17
18	import junit.framework.Test;
19	import junit.framework.TestSuite;
20
21	public class CyclicBarrierTest extends JSR166TestCase {
22	public static void main(String[] args) {
23	main(suite(), args);
24	}
25	public static Test suite() {
26	return new TestSuite(CyclicBarrierTest.class);
27	}
28
29	/**
30	* Spin-waits till the number of waiters == numberOfWaiters.
31	*/
32	void awaitNumberWaiting(CyclicBarrier barrier, int numberOfWaiters) {
33	long startTime = System.nanoTime();
34	while (barrier.getNumberWaiting() != numberOfWaiters) {
35	if (millisElapsedSince(startTime) > LONG_DELAY_MS)
36	fail("timed out");
37	Thread.yield();
38	}
39	}
40
41	/**
42	* Creating with negative parties throws IllegalArgumentException
43	*/
44	public void testConstructor1() {
45	try {
46	new CyclicBarrier(-1, (Runnable)null);
47	shouldThrow();
48	} catch (IllegalArgumentException success) {}
49	}
50
51	/**
52	* Creating with negative parties and no action throws
53	* IllegalArgumentException
54	*/
55	public void testConstructor2() {
56	try {
57	new CyclicBarrier(-1);
58	shouldThrow();
59	} catch (IllegalArgumentException success) {}
60	}
61
62	/**
63	* getParties returns the number of parties given in constructor
64	*/
65	public void testGetParties() {
66	CyclicBarrier b = new CyclicBarrier(2);
67	assertEquals(2, b.getParties());
68	assertEquals(0, b.getNumberWaiting());
69	}
70
71	/**
72	* A 1-party barrier triggers after single await
73	*/
74	public void testSingleParty() throws Exception {
75	CyclicBarrier b = new CyclicBarrier(1);
76	assertEquals(1, b.getParties());
77	assertEquals(0, b.getNumberWaiting());
78	b.await();
79	b.await();
80	assertEquals(0, b.getNumberWaiting());
81	}
82
83	/**
84	* The supplied barrier action is run at barrier
85	*/
86	public void testBarrierAction() throws Exception {
87	final AtomicInteger count = new AtomicInteger(0);
88	final Runnable incCount = new Runnable() { public void run() {
89	count.getAndIncrement(); }};
90	CyclicBarrier b = new CyclicBarrier(1, incCount);
91	assertEquals(1, b.getParties());
92	assertEquals(0, b.getNumberWaiting());
93	b.await();
94	b.await();
95	assertEquals(0, b.getNumberWaiting());
96	assertEquals(2, count.get());
97	}
98
99	/**
100	* A 2-party/thread barrier triggers after both threads invoke await
101	*/
102	public void testTwoParties() throws Exception {
103	final CyclicBarrier b = new CyclicBarrier(2);
104	Thread t = newStartedThread(new CheckedRunnable() {
105	public void realRun() throws Exception {
106	b.await();
107	b.await();
108	b.await();
109	b.await();
110	}});
111
112	b.await();
113	b.await();
114	b.await();
115	b.await();
116	awaitTermination(t);
117	}
118
119	/**
120	* An interruption in one party causes others waiting in await to
121	* throw BrokenBarrierException
122	*/
123	public void testAwait1_Interrupted_BrokenBarrier() {
124	final CyclicBarrier c = new CyclicBarrier(3);
125	final CountDownLatch pleaseInterrupt = new CountDownLatch(2);
126	Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
127	public void realRun() throws Exception {
128	pleaseInterrupt.countDown();
129	c.await();
130	}};
131	Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
132	public void realRun() throws Exception {
133	pleaseInterrupt.countDown();
134	c.await();
135	}};
136
137	t1.start();
138	t2.start();
139	await(pleaseInterrupt);
140	t1.interrupt();
141	awaitTermination(t1);
142	awaitTermination(t2);
143	}
144
145	/**
146	* An interruption in one party causes others waiting in timed await to
147	* throw BrokenBarrierException
148	*/
149	public void testAwait2_Interrupted_BrokenBarrier() throws Exception {
150	final CyclicBarrier c = new CyclicBarrier(3);
151	final CountDownLatch pleaseInterrupt = new CountDownLatch(2);
152	Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
153	public void realRun() throws Exception {
154	pleaseInterrupt.countDown();
155	c.await(LONG_DELAY_MS, MILLISECONDS);
156	}};
157	Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
158	public void realRun() throws Exception {
159	pleaseInterrupt.countDown();
160	c.await(LONG_DELAY_MS, MILLISECONDS);
161	}};
162
163	t1.start();
164	t2.start();
165	await(pleaseInterrupt);
166	t1.interrupt();
167	awaitTermination(t1);
168	awaitTermination(t2);
169	}
170
171	/**
172	* A timeout in timed await throws TimeoutException
173	*/
174	public void testAwait3_TimeoutException() throws InterruptedException {
175	final CyclicBarrier c = new CyclicBarrier(2);
176	Thread t = newStartedThread(new CheckedRunnable() {
177	public void realRun() throws Exception {
178	long startTime = System.nanoTime();
179	try {
180	c.await(timeoutMillis(), MILLISECONDS);
181	shouldThrow();
182	} catch (TimeoutException success) {}
183	assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
184	}});
185
186	awaitTermination(t);
187	}
188
189	/**
190	* A timeout in one party causes others waiting in timed await to
191	* throw BrokenBarrierException
192	*/
193	public void testAwait4_Timeout_BrokenBarrier() throws InterruptedException {
194	final CyclicBarrier c = new CyclicBarrier(3);
195	Thread t1 = newStartedThread(new CheckedRunnable() {
196	public void realRun() throws Exception {
197	try {
198	c.await(LONG_DELAY_MS, MILLISECONDS);
199	shouldThrow();
200	} catch (BrokenBarrierException success) {}
201	}});
202	Thread t2 = newStartedThread(new CheckedRunnable() {
203	public void realRun() throws Exception {
204	awaitNumberWaiting(c, 1);
205	long startTime = System.nanoTime();
206	try {
207	c.await(timeoutMillis(), MILLISECONDS);
208	shouldThrow();
209	} catch (TimeoutException success) {}
210	assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
211	}});
212
213	awaitTermination(t1);
214	awaitTermination(t2);
215	}
216
217	/**
218	* A timeout in one party causes others waiting in await to
219	* throw BrokenBarrierException
220	*/
221	public void testAwait5_Timeout_BrokenBarrier() throws InterruptedException {
222	final CyclicBarrier c = new CyclicBarrier(3);
223	Thread t1 = newStartedThread(new CheckedRunnable() {
224	public void realRun() throws Exception {
225	try {
226	c.await();
227	shouldThrow();
228	} catch (BrokenBarrierException success) {}
229	}});
230	Thread t2 = newStartedThread(new CheckedRunnable() {
231	public void realRun() throws Exception {
232	awaitNumberWaiting(c, 1);
233	long startTime = System.nanoTime();
234	try {
235	c.await(timeoutMillis(), MILLISECONDS);
236	shouldThrow();
237	} catch (TimeoutException success) {}
238	assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
239	}});
240
241	awaitTermination(t1);
242	awaitTermination(t2);
243	}
244
245	/**
246	* A reset of an active barrier causes waiting threads to throw
247	* BrokenBarrierException
248	*/
249	public void testReset_BrokenBarrier() throws InterruptedException {
250	final CyclicBarrier c = new CyclicBarrier(3);
251	final CountDownLatch pleaseReset = new CountDownLatch(2);
252	Thread t1 = new ThreadShouldThrow(BrokenBarrierException.class) {
253	public void realRun() throws Exception {
254	pleaseReset.countDown();
255	c.await();
256	}};
257	Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
258	public void realRun() throws Exception {
259	pleaseReset.countDown();
260	c.await();
261	}};
262
263	t1.start();
264	t2.start();
265	await(pleaseReset);
266
267	awaitNumberWaiting(c, 2);
268	c.reset();
269	awaitTermination(t1);
270	awaitTermination(t2);
271	}
272
273	/**
274	* A reset before threads enter barrier does not throw
275	* BrokenBarrierException
276	*/
277	public void testReset_NoBrokenBarrier() throws Exception {
278	final CyclicBarrier c = new CyclicBarrier(3);
279	c.reset();
280
281	Thread t1 = newStartedThread(new CheckedRunnable() {
282	public void realRun() throws Exception {
283	c.await();
284	}});
285	Thread t2 = newStartedThread(new CheckedRunnable() {
286	public void realRun() throws Exception {
287	c.await();
288	}});
289
290	c.await();
291	awaitTermination(t1);
292	awaitTermination(t2);
293	}
294
295	/**
296	* All threads block while a barrier is broken.
297	*/
298	public void testReset_Leakage() throws InterruptedException {
299	final CyclicBarrier c = new CyclicBarrier(2);
300	final AtomicBoolean done = new AtomicBoolean();
301	Thread t = newStartedThread(new CheckedRunnable() {
302	public void realRun() {
303	while (!done.get()) {
304	try {
305	while (c.isBroken())
306	c.reset();
307
308	c.await();
309	shouldThrow();
310	}
311	catch (BrokenBarrierException ok) {}
312	catch (InterruptedException ok) {}
313	}}});
314
315	for (int i = 0; i < 4; i++) {
316	delay(timeoutMillis());
317	t.interrupt();
318	}
319	done.set(true);
320	t.interrupt();
321	awaitTermination(t);
322	}
323
324	/**
325	* Reset of a non-broken barrier does not break barrier
326	*/
327	public void testResetWithoutBreakage() throws Exception {
328	final CyclicBarrier barrier = new CyclicBarrier(3);
329	for (int i = 0; i < 3; i++) {
330	final CyclicBarrier start = new CyclicBarrier(3);
331	Thread t1 = newStartedThread(new CheckedRunnable() {
332	public void realRun() throws Exception {
333	start.await();
334	barrier.await();
335	}});
336
337	Thread t2 = newStartedThread(new CheckedRunnable() {
338	public void realRun() throws Exception {
339	start.await();
340	barrier.await();
341	}});
342
343	start.await();
344	barrier.await();
345	awaitTermination(t1);
346	awaitTermination(t2);
347	assertFalse(barrier.isBroken());
348	assertEquals(0, barrier.getNumberWaiting());
349	if (i == 1) barrier.reset();
350	assertFalse(barrier.isBroken());
351	assertEquals(0, barrier.getNumberWaiting());
352	}
353	}
354
355	/**
356	* Reset of a barrier after interruption reinitializes it.
357	*/
358	public void testResetAfterInterrupt() throws Exception {
359	final CyclicBarrier barrier = new CyclicBarrier(3);
360	for (int i = 0; i < 2; i++) {
361	final CyclicBarrier start = new CyclicBarrier(3);
362	Thread t1 = new ThreadShouldThrow(InterruptedException.class) {
363	public void realRun() throws Exception {
364	start.await();
365	barrier.await();
366	}};
367
368	Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
369	public void realRun() throws Exception {
370	start.await();
371	barrier.await();
372	}};
373
374	t1.start();
375	t2.start();
376	start.await();
377	t1.interrupt();
378	awaitTermination(t1);
379	awaitTermination(t2);
380	assertTrue(barrier.isBroken());
381	assertEquals(0, barrier.getNumberWaiting());
382	barrier.reset();
383	assertFalse(barrier.isBroken());
384	assertEquals(0, barrier.getNumberWaiting());
385	}
386	}
387
388	/**
389	* Reset of a barrier after timeout reinitializes it.
390	*/
391	public void testResetAfterTimeout() throws Exception {
392	final CyclicBarrier barrier = new CyclicBarrier(3);
393	for (int i = 0; i < 2; i++) {
394	assertEquals(0, barrier.getNumberWaiting());
395	Thread t1 = newStartedThread(new CheckedRunnable() {
396	public void realRun() throws Exception {
397	try {
398	barrier.await();
399	shouldThrow();
400	} catch (BrokenBarrierException success) {}
401	}});
402	Thread t2 = newStartedThread(new CheckedRunnable() {
403	public void realRun() throws Exception {
404	awaitNumberWaiting(barrier, 1);
405	long startTime = System.nanoTime();
406	try {
407	barrier.await(timeoutMillis(), MILLISECONDS);
408	shouldThrow();
409	} catch (TimeoutException success) {}
410	assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
411	}});
412
413	awaitTermination(t1);
414	awaitTermination(t2);
415	assertEquals(0, barrier.getNumberWaiting());
416	assertTrue(barrier.isBroken());
417	assertEquals(0, barrier.getNumberWaiting());
418	barrier.reset();
419	assertFalse(barrier.isBroken());
420	assertEquals(0, barrier.getNumberWaiting());
421	}
422	}
423
424	/**
425	* Reset of a barrier after a failed command reinitializes it.
426	*/
427	public void testResetAfterCommandException() throws Exception {
428	final CyclicBarrier barrier =
429	new CyclicBarrier(3, new Runnable() {
430	public void run() {
431	throw new NullPointerException(); }});
432	for (int i = 0; i < 2; i++) {
433	final CyclicBarrier start = new CyclicBarrier(3);
434	Thread t1 = new ThreadShouldThrow(BrokenBarrierException.class) {
435	public void realRun() throws Exception {
436	start.await();
437	barrier.await();
438	}};
439
440	Thread t2 = new ThreadShouldThrow(BrokenBarrierException.class) {
441	public void realRun() throws Exception {
442	start.await();
443	barrier.await();
444	}};
445
446	t1.start();
447	t2.start();
448	start.await();
449	awaitNumberWaiting(barrier, 2);
450	try {
451	barrier.await();
452	shouldThrow();
453	} catch (NullPointerException success) {}
454	awaitTermination(t1);
455	awaitTermination(t2);
456	assertTrue(barrier.isBroken());
457	assertEquals(0, barrier.getNumberWaiting());
458	barrier.reset();
459	assertFalse(barrier.isBroken());
460	assertEquals(0, barrier.getNumberWaiting());
461	}
462	}
463	}