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

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines