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

Comparing jsr166/src/test/tck/CyclicBarrierTest.java (file contents):
Revision 1.1 by dl, Sun Aug 31 19:24:54 2003 UTC vs.
Revision 1.29 by jsr166, Mon May 29 22:44:26 2017 UTC

# Line 1 | Line 1
1   /*
2 < * Written by members of JCP JSR-166 Expert Group and released to the
3 < * public domain. Use, modify, and redistribute this code in any way
4 < * without acknowledgement. Other contributors include Andrew Wright,
5 < * Jeffrey Hayes, Pat Fischer, Mike Judd.
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 junit.framework.*;
9 < import java.util.*;
10 < import java.util.concurrent.*;
9 > import static java.util.concurrent.TimeUnit.MILLISECONDS;
10  
11 < public class CyclicBarrierTest extends TestCase{
12 <    
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 <        junit.textui.TestRunner.run (suite());  
23 >        main(suite(), args);
24      }
17    
18    
25      public static Test suite() {
26 <        return new TestSuite(CyclicBarrierTest.class);
26 >        return new TestSuite(CyclicBarrierTest.class);
27      }
22    
23    private static long SHORT_DELAY_MS = 100;
24    private static long MEDIUM_DELAY_MS = 1000;
25    private static long LONG_DELAY_MS = 10000;
28  
29 <    public void testConstructor1(){
30 <        try{
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 <            fail("should throw");
48 <        } catch(IllegalArgumentException e){}
47 >            shouldThrow();
48 >        } catch (IllegalArgumentException success) {}
49      }
50  
51 <    public void testConstructor2(){
52 <        try{
51 >    /**
52 >     * Creating with negative parties and no action throws
53 >     * IllegalArgumentException
54 >     */
55 >    public void testConstructor2() {
56 >        try {
57              new CyclicBarrier(-1);
58 <            fail("should throw");
59 <        } catch(IllegalArgumentException e){}
58 >            shouldThrow();
59 >        } catch (IllegalArgumentException success) {}
60      }
61  
62 <    public void testConstructor3(){
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());
67 >        assertEquals(2, b.getParties());
68          assertEquals(0, b.getNumberWaiting());
69      }
70  
71 <    public void testSingleParty() {
72 <        try {
73 <            CyclicBarrier b = new CyclicBarrier(1);
74 <            assertEquals(1, b.getParties());
75 <            assertEquals(0, b.getNumberWaiting());
76 <            b.await();
77 <            b.await();
78 <            assertEquals(0, b.getNumberWaiting());
79 <        }
80 <        catch(Exception e) {
57 <            fail("unexpected exception");
58 <        }
59 <    }
60 <    
61 <    private volatile int countAction;
62 <    private class MyAction implements Runnable {
63 <        public void run() { ++countAction; }
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 <    public void testBarrierAction() {
84 <        try {
85 <            countAction = 0;
86 <            CyclicBarrier b = new CyclicBarrier(1, new MyAction());
87 <            assertEquals(1, b.getParties());
88 <            assertEquals(0, b.getNumberWaiting());
89 <            b.await();
90 <            b.await();
91 <            assertEquals(0, b.getNumberWaiting());
92 <            assertEquals(countAction, 2);
93 <        }
94 <        catch(Exception e) {
95 <            fail("unexpected exception");
96 <        }
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 <    public void testTwoParties(){
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 = new Thread(new Runnable() {
105 <                public void run(){
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 <                        b.await();
306 <                        b.await();
90 <                        b.await();
91 <                        b.await();
92 <                    } catch(Exception e){
93 <                        fail("unexpected exception");
94 <                    }}});
305 >                        while (c.isBroken())
306 >                            c.reset();
307  
308 <        try {
309 <            t.start();
310 <            b.await();
311 <            b.await();
312 <            b.await();
313 <            b.await();
314 <            t.join();
315 <        } catch(Exception e){
316 <            fail("unexpected exception");
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  
109    public void testAwait1_Interrupted_BrokenBarrier(){
110        final CyclicBarrier c = new CyclicBarrier(3);
111        Thread t1 = new Thread(new Runnable() {
112                public void run(){
113                    try{
114                        c.await();
115                        fail("should throw");
116                    } catch(InterruptedException success){}                
117                    catch(Exception b){
118                        fail("should throw IE");
119                    }
120                }
121            });
122        Thread t2 = new Thread(new Runnable(){
123                public void run(){
124                    try{
125                        c.await();
126                        fail("should throw");                        
127                    } catch(BrokenBarrierException success){
128                    } catch(Exception i){
129                        fail("should throw BBE");
130                    }
131                }
132            });
133        try {
374              t1.start();
375              t2.start();
376 <            Thread.sleep(SHORT_DELAY_MS);
376 >            start.await();
377              t1.interrupt();
378 <            t1.join();
379 <            t2.join();
380 <        } catch(InterruptedException e){
381 <            fail("unexpected exception");
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 <    public void testAwait2_Interrupted_BrokenBarrier(){
389 <      final CyclicBarrier c = new CyclicBarrier(3);
390 <        Thread t1 = new Thread(new Runnable() {
391 <                public void run(){
392 <                    try{
393 <                        c.await(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
394 <                        fail("should throw");
395 <                    } catch(InterruptedException success){
396 <                    } catch(Exception b){
397 <                        fail("should throw IE");
398 <                    }
399 <                }
400 <            });
401 <        Thread t2 = new Thread(new Runnable(){
402 <                public void run(){
403 <                    try{
404 <                        c.await(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
405 <                        fail("should throw");                        
406 <                    } catch(BrokenBarrierException success){
407 <                    } catch(Exception i){
408 <                        fail("should throw BBE");
409 <                    }
410 <                }
411 <            });
412 <        try {
413 <            t1.start();
414 <            t2.start();
415 <            Thread.sleep(SHORT_DELAY_MS);
416 <            t1.interrupt();
417 <            t1.join();
418 <            t2.join();
419 <        } catch(InterruptedException e){
420 <            fail("unexpected exception");
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 <    public void testAwait3_TimeOutException(){
425 <        final CyclicBarrier c = new CyclicBarrier(2);
426 <        Thread t = new Thread(new Runnable() {
427 <                public void run(){
428 <                    try{
429 <                        c.await(SHORT_DELAY_MS, TimeUnit.MILLISECONDS);
430 <                        fail("should throw");
431 <                    } catch(TimeoutException success){
432 <                    } catch(Exception b){
433 <                        fail("should throw TOE");
434 <                        
435 <                    }
436 <                }
437 <            });
438 <        try {
439 <            t.start();
440 <            t.join();
441 <        } catch(InterruptedException e){
442 <            fail("unexpected exception");
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      }
202    
463   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines