ViewVC Help
View File | Revision Log | Show Annotations | Download File | Root Listing
root/jsr166/jsr166/src/test/tck/PhaserTest.java
Revision: 1.46
Committed: Sat May 13 23:05:02 2017 UTC (6 years, 11 months ago) by jsr166
Branch: MAIN
Changes since 1.45: +5 -5 lines
Log Message:
claw back some millis using assertThreadBlocks

File Contents

# User Rev Content
1 dl 1.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 jsr166 1.31 * http://creativecommons.org/publicdomain/zero/1.0/
5 dl 1.1 * Other contributors include John Vint
6     */
7    
8 jsr166 1.39 import static java.util.concurrent.TimeUnit.MILLISECONDS;
9    
10 dl 1.1 import java.util.ArrayList;
11     import java.util.List;
12 jsr166 1.39 import java.util.concurrent.CountDownLatch;
13 jsr166 1.32 import java.util.concurrent.Phaser;
14 jsr166 1.33 import java.util.concurrent.TimeoutException;
15 jsr166 1.32 import java.util.concurrent.atomic.AtomicInteger;
16 dl 1.1
17 jsr166 1.39 import junit.framework.Test;
18     import junit.framework.TestSuite;
19    
20 dl 1.1 public class PhaserTest extends JSR166TestCase {
21    
22     public static void main(String[] args) {
23 jsr166 1.40 main(suite(), args);
24 dl 1.1 }
25    
26     public static Test suite() {
27     return new TestSuite(PhaserTest.class);
28     }
29    
30 jsr166 1.21 private static final int maxParties = 65535;
31 jsr166 1.22
32 jsr166 1.28 /** Checks state of unterminated phaser. */
33 jsr166 1.19 protected void assertState(Phaser phaser,
34     int phase, int parties, int unarrived) {
35     assertEquals(phase, phaser.getPhase());
36     assertEquals(parties, phaser.getRegisteredParties());
37     assertEquals(unarrived, phaser.getUnarrivedParties());
38     assertEquals(parties - unarrived, phaser.getArrivedParties());
39 jsr166 1.28 assertFalse(phaser.isTerminated());
40 jsr166 1.19 }
41    
42     /** Checks state of terminated phaser. */
43 jsr166 1.28 protected void assertTerminated(Phaser phaser, int maxPhase, int parties) {
44 jsr166 1.19 assertTrue(phaser.isTerminated());
45 jsr166 1.25 int expectedPhase = maxPhase + Integer.MIN_VALUE;
46 dl 1.26 assertEquals(expectedPhase, phaser.getPhase());
47     assertEquals(parties, phaser.getRegisteredParties());
48 jsr166 1.25 assertEquals(expectedPhase, phaser.register());
49     assertEquals(expectedPhase, phaser.arrive());
50     assertEquals(expectedPhase, phaser.arriveAndDeregister());
51 jsr166 1.19 }
52    
53 jsr166 1.25 protected void assertTerminated(Phaser phaser, int maxPhase) {
54 jsr166 1.28 assertTerminated(phaser, maxPhase, 0);
55 jsr166 1.19 }
56    
57 dl 1.1 /**
58     * Empty constructor builds a new Phaser with no parent, no registered
59     * parties and initial phase number of 0
60     */
61 jsr166 1.21 public void testConstructorDefaultValues() {
62 dl 1.1 Phaser phaser = new Phaser();
63     assertNull(phaser.getParent());
64 jsr166 1.21 assertEquals(0, phaser.getRegisteredParties());
65 dl 1.1 assertEquals(0, phaser.getArrivedParties());
66 jsr166 1.21 assertEquals(0, phaser.getUnarrivedParties());
67 dl 1.1 assertEquals(0, phaser.getPhase());
68     }
69    
70     /**
71 jsr166 1.21 * Constructing with a negative number of parties throws
72     * IllegalArgumentException
73 dl 1.1 */
74 jsr166 1.21 public void testConstructorNegativeParties() {
75 dl 1.1 try {
76     new Phaser(-1);
77 jsr166 1.6 shouldThrow();
78 jsr166 1.8 } catch (IllegalArgumentException success) {}
79 dl 1.1 }
80    
81     /**
82 jsr166 1.21 * Constructing with a negative number of parties throws
83     * IllegalArgumentException
84 dl 1.1 */
85 jsr166 1.21 public void testConstructorNegativeParties2() {
86     try {
87     new Phaser(new Phaser(), -1);
88     shouldThrow();
89     } catch (IllegalArgumentException success) {}
90 dl 1.1 }
91    
92     /**
93 jsr166 1.21 * Constructing with a number of parties > 65535 throws
94     * IllegalArgumentException
95 dl 1.1 */
96 jsr166 1.21 public void testConstructorPartiesExceedsLimit() {
97     new Phaser(maxParties);
98 dl 1.1 try {
99 jsr166 1.21 new Phaser(maxParties + 1);
100     shouldThrow();
101     } catch (IllegalArgumentException success) {}
102    
103     new Phaser(new Phaser(), maxParties);
104     try {
105     new Phaser(new Phaser(), maxParties + 1);
106 dl 1.1 shouldThrow();
107 jsr166 1.8 } catch (IllegalArgumentException success) {}
108 dl 1.1 }
109    
110     /**
111 jsr166 1.21 * The parent provided to the constructor should be returned from
112     * a later call to getParent
113     */
114     public void testConstructor3() {
115     Phaser parent = new Phaser();
116     assertSame(parent, new Phaser(parent).getParent());
117     assertNull(new Phaser(null).getParent());
118     }
119    
120     /**
121 dl 1.1 * The parent being input into the parameter should equal the original
122     * parent when being returned
123     */
124     public void testConstructor5() {
125     Phaser parent = new Phaser();
126 jsr166 1.21 assertSame(parent, new Phaser(parent, 0).getParent());
127     assertNull(new Phaser(null, 0).getParent());
128 dl 1.1 }
129    
130     /**
131 jsr166 1.21 * register() will increment the number of unarrived parties by
132     * one and not affect its arrived parties
133 dl 1.1 */
134     public void testRegister1() {
135     Phaser phaser = new Phaser();
136 jsr166 1.19 assertState(phaser, 0, 0, 0);
137     assertEquals(0, phaser.register());
138     assertState(phaser, 0, 1, 1);
139 dl 1.1 }
140    
141     /**
142 jsr166 1.4 * Registering more than 65536 parties causes IllegalStateException
143 dl 1.1 */
144     public void testRegister2() {
145     Phaser phaser = new Phaser(0);
146 jsr166 1.19 assertState(phaser, 0, 0, 0);
147     assertEquals(0, phaser.bulkRegister(maxParties - 10));
148     assertState(phaser, 0, maxParties - 10, maxParties - 10);
149     for (int i = 0; i < 10; i++) {
150     assertState(phaser, 0, maxParties - 10 + i, maxParties - 10 + i);
151     assertEquals(0, phaser.register());
152 dl 1.1 }
153 jsr166 1.19 assertState(phaser, 0, maxParties, maxParties);
154 dl 1.1 try {
155     phaser.register();
156     shouldThrow();
157 jsr166 1.8 } catch (IllegalStateException success) {}
158 jsr166 1.21
159     try {
160     phaser.bulkRegister(Integer.MAX_VALUE);
161     shouldThrow();
162     } catch (IllegalStateException success) {}
163 jsr166 1.30
164     assertEquals(0, phaser.bulkRegister(0));
165     assertState(phaser, 0, maxParties, maxParties);
166 dl 1.1 }
167    
168     /**
169 jsr166 1.34 * register() correctly returns the current barrier phase number
170     * when invoked
171 dl 1.1 */
172     public void testRegister3() {
173     Phaser phaser = new Phaser();
174     assertEquals(0, phaser.register());
175 jsr166 1.19 assertEquals(0, phaser.arrive());
176 dl 1.1 assertEquals(1, phaser.register());
177 jsr166 1.19 assertState(phaser, 1, 2, 2);
178 dl 1.1 }
179    
180     /**
181 jsr166 1.34 * register causes the next arrive to not increment the phase
182     * rather retain the phase number
183 dl 1.1 */
184     public void testRegister4() {
185     Phaser phaser = new Phaser(1);
186 jsr166 1.19 assertEquals(0, phaser.arrive());
187     assertEquals(1, phaser.register());
188     assertEquals(1, phaser.arrive());
189     assertState(phaser, 1, 2, 1);
190 dl 1.1 }
191    
192     /**
193 jsr166 1.34 * register on a subphaser that is currently empty succeeds, even
194     * in the presence of another non-empty subphaser
195     */
196 dl 1.35 public void testRegisterEmptySubPhaser() {
197 jsr166 1.34 Phaser root = new Phaser();
198     Phaser child1 = new Phaser(root, 1);
199     Phaser child2 = new Phaser(root, 0);
200     assertEquals(0, child2.register());
201 jsr166 1.36 assertState(root, 0, 2, 2);
202     assertState(child1, 0, 1, 1);
203     assertState(child2, 0, 1, 1);
204 jsr166 1.34 assertEquals(0, child2.arriveAndDeregister());
205 jsr166 1.36 assertState(root, 0, 1, 1);
206     assertState(child1, 0, 1, 1);
207     assertState(child2, 0, 0, 0);
208 jsr166 1.34 assertEquals(0, child2.register());
209     assertEquals(0, child2.arriveAndDeregister());
210 jsr166 1.36 assertState(root, 0, 1, 1);
211     assertState(child1, 0, 1, 1);
212 jsr166 1.34 assertState(child2, 0, 0, 0);
213 jsr166 1.36 assertEquals(0, child1.arriveAndDeregister());
214     assertTerminated(root, 1);
215     assertTerminated(child1, 1);
216     assertTerminated(child2, 1);
217 jsr166 1.34 }
218    
219     /**
220 dl 1.1 * Invoking bulkRegister with a negative parameter throws an
221 jsr166 1.4 * IllegalArgumentException
222 dl 1.1 */
223     public void testBulkRegister1() {
224     try {
225     new Phaser().bulkRegister(-1);
226     shouldThrow();
227 jsr166 1.8 } catch (IllegalArgumentException success) {}
228 dl 1.1 }
229    
230     /**
231 jsr166 1.34 * bulkRegister should correctly record the number of unarrived
232     * parties with the number of parties being registered
233 dl 1.1 */
234     public void testBulkRegister2() {
235     Phaser phaser = new Phaser();
236 jsr166 1.30 assertEquals(0, phaser.bulkRegister(0));
237     assertState(phaser, 0, 0, 0);
238 jsr166 1.19 assertEquals(0, phaser.bulkRegister(20));
239     assertState(phaser, 0, 20, 20);
240 dl 1.1 }
241    
242     /**
243 jsr166 1.4 * Registering with a number of parties greater than or equal to 1<<16
244     * throws IllegalStateException.
245 dl 1.1 */
246     public void testBulkRegister3() {
247 jsr166 1.19 assertEquals(0, new Phaser().bulkRegister((1 << 16) - 1));
248    
249 dl 1.1 try {
250     new Phaser().bulkRegister(1 << 16);
251     shouldThrow();
252 jsr166 1.8 } catch (IllegalStateException success) {}
253 jsr166 1.19
254     try {
255     new Phaser(2).bulkRegister((1 << 16) - 2);
256     shouldThrow();
257     } catch (IllegalStateException success) {}
258 dl 1.1 }
259    
260     /**
261     * the phase number increments correctly when tripping the barrier
262     */
263     public void testPhaseIncrement1() {
264     for (int size = 1; size < nine; size++) {
265     final Phaser phaser = new Phaser(size);
266     for (int index = 0; index <= (1 << size); index++) {
267     int phase = phaser.arrive();
268     assertTrue(index % size == 0 ? (index / size) == phase : index - (phase * size) > 0);
269     }
270     }
271     }
272    
273     /**
274 jsr166 1.19 * arrive() on a registered phaser increments phase.
275 dl 1.1 */
276     public void testArrive1() {
277     Phaser phaser = new Phaser(1);
278 jsr166 1.19 assertState(phaser, 0, 1, 1);
279     assertEquals(0, phaser.arrive());
280     assertState(phaser, 1, 1, 1);
281     }
282    
283     /**
284     * arriveAndDeregister does not wait for others to arrive at barrier
285     */
286 jsr166 1.33 public void testArriveAndDeregister() {
287 jsr166 1.19 final Phaser phaser = new Phaser(1);
288     for (int i = 0; i < 10; i++) {
289     assertState(phaser, 0, 1, 1);
290     assertEquals(0, phaser.register());
291     assertState(phaser, 0, 2, 2);
292     assertEquals(0, phaser.arriveAndDeregister());
293     assertState(phaser, 0, 1, 1);
294     }
295     assertEquals(0, phaser.arriveAndDeregister());
296 jsr166 1.25 assertTerminated(phaser, 1);
297 dl 1.1 }
298    
299     /**
300 jsr166 1.7 * arriveAndDeregister does not wait for others to arrive at barrier
301 dl 1.1 */
302 jsr166 1.33 public void testArrive2() {
303 jsr166 1.19 final Phaser phaser = new Phaser();
304     assertEquals(0, phaser.register());
305 jsr166 1.44 List<Thread> threads = new ArrayList<>();
306 jsr166 1.19 for (int i = 0; i < 10; i++) {
307     assertEquals(0, phaser.register());
308 jsr166 1.7 threads.add(newStartedThread(new CheckedRunnable() {
309 jsr166 1.33 public void realRun() {
310 jsr166 1.19 assertEquals(0, phaser.arriveAndDeregister());
311 jsr166 1.7 }}));
312 jsr166 1.19 }
313 dl 1.1
314 jsr166 1.7 for (Thread thread : threads)
315 jsr166 1.33 awaitTermination(thread);
316 jsr166 1.19 assertState(phaser, 0, 1, 1);
317     assertEquals(0, phaser.arrive());
318     assertState(phaser, 1, 1, 1);
319 dl 1.1 }
320    
321     /**
322 jsr166 1.3 * arrive() returns a negative number if the Phaser is terminated
323 dl 1.1 */
324     public void testArrive3() {
325     Phaser phaser = new Phaser(1);
326     phaser.forceTermination();
327 jsr166 1.28 assertTerminated(phaser, 0, 1);
328 dl 1.24 assertEquals(0, phaser.getPhase() + Integer.MIN_VALUE);
329 dl 1.1 assertTrue(phaser.arrive() < 0);
330 jsr166 1.19 assertTrue(phaser.register() < 0);
331     assertTrue(phaser.arriveAndDeregister() < 0);
332     assertTrue(phaser.awaitAdvance(1) < 0);
333     assertTrue(phaser.getPhase() < 0);
334 dl 1.1 }
335    
336     /**
337     * arriveAndDeregister() throws IllegalStateException if number of
338 jsr166 1.3 * registered or unarrived parties would become negative
339 dl 1.1 */
340     public void testArriveAndDeregister1() {
341 jsr166 1.41 Phaser phaser = new Phaser();
342 dl 1.1 try {
343     phaser.arriveAndDeregister();
344     shouldThrow();
345 jsr166 1.8 } catch (IllegalStateException success) {}
346 dl 1.1 }
347    
348     /**
349 jsr166 1.19 * arriveAndDeregister reduces the number of arrived parties
350 dl 1.1 */
351 jsr166 1.19 public void testArriveAndDeregister2() {
352 dl 1.1 final Phaser phaser = new Phaser(1);
353 jsr166 1.19 assertEquals(0, phaser.register());
354     assertEquals(0, phaser.arrive());
355     assertState(phaser, 0, 2, 1);
356     assertEquals(0, phaser.arriveAndDeregister());
357     assertState(phaser, 1, 1, 1);
358 dl 1.1 }
359    
360     /**
361 jsr166 1.19 * arriveAndDeregister arrives at the barrier on a phaser with a parent and
362 dl 1.1 * when a deregistration occurs and causes the phaser to have zero parties
363     * its parent will be deregistered as well
364     */
365 jsr166 1.19 public void testArriveAndDeregister3() {
366 dl 1.1 Phaser parent = new Phaser();
367 jsr166 1.19 Phaser child = new Phaser(parent);
368     assertState(child, 0, 0, 0);
369 dl 1.23 assertState(parent, 0, 0, 0);
370 jsr166 1.19 assertEquals(0, child.register());
371     assertState(child, 0, 1, 1);
372     assertState(parent, 0, 1, 1);
373     assertEquals(0, child.arriveAndDeregister());
374 jsr166 1.25 assertTerminated(child, 1);
375     assertTerminated(parent, 1);
376 dl 1.1 }
377    
378     /**
379     * arriveAndDeregister deregisters one party from its parent when
380 jsr166 1.19 * the number of parties of child is zero after deregistration
381 dl 1.1 */
382 jsr166 1.19 public void testArriveAndDeregister4() {
383 dl 1.1 Phaser parent = new Phaser();
384 jsr166 1.19 Phaser child = new Phaser(parent);
385     assertEquals(0, parent.register());
386     assertEquals(0, child.register());
387     assertState(child, 0, 1, 1);
388     assertState(parent, 0, 2, 2);
389     assertEquals(0, child.arriveAndDeregister());
390     assertState(child, 0, 0, 0);
391     assertState(parent, 0, 1, 1);
392 dl 1.1 }
393    
394     /**
395     * arriveAndDeregister deregisters one party from its parent when
396     * the number of parties of root is nonzero after deregistration.
397     */
398     public void testArriveAndDeregister5() {
399 jsr166 1.19 Phaser root = new Phaser();
400     Phaser parent = new Phaser(root);
401 dl 1.1 Phaser child = new Phaser(parent);
402 dl 1.23 assertState(root, 0, 0, 0);
403     assertState(parent, 0, 0, 0);
404 jsr166 1.19 assertState(child, 0, 0, 0);
405     assertEquals(0, child.register());
406     assertState(root, 0, 1, 1);
407     assertState(parent, 0, 1, 1);
408     assertState(child, 0, 1, 1);
409     assertEquals(0, child.arriveAndDeregister());
410 jsr166 1.25 assertTerminated(child, 1);
411     assertTerminated(parent, 1);
412     assertTerminated(root, 1);
413 dl 1.1 }
414    
415     /**
416     * arriveAndDeregister returns the phase in which it leaves the
417     * phaser in after deregistration
418     */
419 jsr166 1.33 public void testArriveAndDeregister6() {
420 dl 1.1 final Phaser phaser = new Phaser(2);
421 jsr166 1.7 Thread t = newStartedThread(new CheckedRunnable() {
422 jsr166 1.9 public void realRun() {
423 jsr166 1.19 assertEquals(0, phaser.arrive());
424 jsr166 1.7 }});
425 jsr166 1.19 assertEquals(1, phaser.arriveAndAwaitAdvance());
426     assertState(phaser, 1, 2, 2);
427     assertEquals(1, phaser.arriveAndDeregister());
428     assertState(phaser, 1, 1, 1);
429     assertEquals(1, phaser.arriveAndDeregister());
430 jsr166 1.25 assertTerminated(phaser, 2);
431 jsr166 1.33 awaitTermination(t);
432 dl 1.1 }
433    
434     /**
435     * awaitAdvance succeeds upon advance
436     */
437     public void testAwaitAdvance1() {
438     final Phaser phaser = new Phaser(1);
439 jsr166 1.19 assertEquals(0, phaser.arrive());
440     assertEquals(1, phaser.awaitAdvance(0));
441 dl 1.1 }
442    
443     /**
444     * awaitAdvance with a negative parameter will return without affecting the
445     * phaser
446     */
447     public void testAwaitAdvance2() {
448 jsr166 1.6 Phaser phaser = new Phaser();
449 jsr166 1.19 assertTrue(phaser.awaitAdvance(-1) < 0);
450     assertState(phaser, 0, 0, 0);
451 dl 1.1 }
452    
453     /**
454 jsr166 1.33 * awaitAdvanceInterruptibly blocks interruptibly
455     */
456     public void testAwaitAdvanceInterruptibly_interruptible() throws InterruptedException {
457     final Phaser phaser = new Phaser(1);
458     final CountDownLatch pleaseInterrupt = new CountDownLatch(2);
459    
460     Thread t1 = newStartedThread(new CheckedRunnable() {
461     public void realRun() {
462     Thread.currentThread().interrupt();
463     try {
464     phaser.awaitAdvanceInterruptibly(0);
465     shouldThrow();
466     } catch (InterruptedException success) {}
467     assertFalse(Thread.interrupted());
468    
469     pleaseInterrupt.countDown();
470     try {
471     phaser.awaitAdvanceInterruptibly(0);
472     shouldThrow();
473     } catch (InterruptedException success) {}
474     assertFalse(Thread.interrupted());
475     }});
476    
477     Thread t2 = newStartedThread(new CheckedRunnable() {
478     public void realRun() throws TimeoutException {
479     Thread.currentThread().interrupt();
480     try {
481     phaser.awaitAdvanceInterruptibly(0, 2*LONG_DELAY_MS, MILLISECONDS);
482     shouldThrow();
483     } catch (InterruptedException success) {}
484     assertFalse(Thread.interrupted());
485    
486     pleaseInterrupt.countDown();
487     try {
488     phaser.awaitAdvanceInterruptibly(0, 2*LONG_DELAY_MS, MILLISECONDS);
489     shouldThrow();
490     } catch (InterruptedException success) {}
491     assertFalse(Thread.interrupted());
492     }});
493    
494     await(pleaseInterrupt);
495     assertState(phaser, 0, 1, 1);
496     assertThreadsStayAlive(t1, t2);
497     t1.interrupt();
498     t2.interrupt();
499     awaitTermination(t1);
500     awaitTermination(t2);
501     assertState(phaser, 0, 1, 1);
502     assertEquals(0, phaser.arrive());
503     assertState(phaser, 1, 1, 1);
504     }
505    
506     /**
507 jsr166 1.20 * awaitAdvance continues waiting if interrupted before waiting
508 dl 1.1 */
509 jsr166 1.33 public void testAwaitAdvanceAfterInterrupt() {
510 dl 1.1 final Phaser phaser = new Phaser();
511 jsr166 1.19 assertEquals(0, phaser.register());
512 jsr166 1.33 final CountDownLatch pleaseArrive = new CountDownLatch(1);
513 dl 1.1
514 jsr166 1.7 Thread t = newStartedThread(new CheckedRunnable() {
515 jsr166 1.33 public void realRun() {
516 jsr166 1.20 Thread.currentThread().interrupt();
517 jsr166 1.19 assertEquals(0, phaser.register());
518 jsr166 1.20 assertEquals(0, phaser.arrive());
519 jsr166 1.33 pleaseArrive.countDown();
520 jsr166 1.20 assertTrue(Thread.currentThread().isInterrupted());
521     assertEquals(1, phaser.awaitAdvance(0));
522 jsr166 1.33 assertTrue(Thread.interrupted());
523 jsr166 1.5 }});
524 jsr166 1.20
525 jsr166 1.33 await(pleaseArrive);
526 jsr166 1.46 assertThreadBlocks(t, Thread.State.WAITING);
527 jsr166 1.20 assertEquals(0, phaser.arrive());
528 jsr166 1.33 awaitTermination(t);
529 jsr166 1.20
530     Thread.currentThread().interrupt();
531     assertEquals(1, phaser.awaitAdvance(0));
532     assertTrue(Thread.interrupted());
533     }
534    
535     /**
536 jsr166 1.33 * awaitAdvance continues waiting if interrupted while waiting
537 jsr166 1.20 */
538 jsr166 1.33 public void testAwaitAdvanceBeforeInterrupt() {
539 jsr166 1.20 final Phaser phaser = new Phaser();
540     assertEquals(0, phaser.register());
541 jsr166 1.33 final CountDownLatch pleaseArrive = new CountDownLatch(1);
542 jsr166 1.20
543     Thread t = newStartedThread(new CheckedRunnable() {
544 jsr166 1.33 public void realRun() {
545 jsr166 1.20 assertEquals(0, phaser.register());
546     assertEquals(0, phaser.arrive());
547     assertFalse(Thread.currentThread().isInterrupted());
548 jsr166 1.33 pleaseArrive.countDown();
549 jsr166 1.20 assertEquals(1, phaser.awaitAdvance(0));
550 jsr166 1.33 assertTrue(Thread.interrupted());
551 jsr166 1.20 }});
552    
553 jsr166 1.33 await(pleaseArrive);
554 jsr166 1.46 assertThreadBlocks(t, Thread.State.WAITING);
555 jsr166 1.7 t.interrupt();
556 jsr166 1.19 assertEquals(0, phaser.arrive());
557 jsr166 1.33 awaitTermination(t);
558 jsr166 1.20
559     Thread.currentThread().interrupt();
560     assertEquals(1, phaser.awaitAdvance(0));
561     assertTrue(Thread.interrupted());
562     }
563    
564     /**
565     * arriveAndAwaitAdvance continues waiting if interrupted before waiting
566     */
567 jsr166 1.33 public void testArriveAndAwaitAdvanceAfterInterrupt() {
568 jsr166 1.20 final Phaser phaser = new Phaser();
569     assertEquals(0, phaser.register());
570 jsr166 1.45 final CountDownLatch pleaseArrive = new CountDownLatch(1);
571 jsr166 1.20
572     Thread t = newStartedThread(new CheckedRunnable() {
573 jsr166 1.33 public void realRun() {
574 jsr166 1.20 Thread.currentThread().interrupt();
575     assertEquals(0, phaser.register());
576 jsr166 1.45 pleaseArrive.countDown();
577 jsr166 1.20 assertTrue(Thread.currentThread().isInterrupted());
578     assertEquals(1, phaser.arriveAndAwaitAdvance());
579 jsr166 1.45 assertTrue(Thread.interrupted());
580 jsr166 1.20 }});
581    
582 jsr166 1.45 await(pleaseArrive);
583 jsr166 1.46 assertThreadBlocks(t, Thread.State.WAITING);
584 jsr166 1.20 Thread.currentThread().interrupt();
585     assertEquals(1, phaser.arriveAndAwaitAdvance());
586     assertTrue(Thread.interrupted());
587 jsr166 1.33 awaitTermination(t);
588 jsr166 1.20 }
589    
590     /**
591     * arriveAndAwaitAdvance continues waiting if interrupted while waiting
592     */
593 jsr166 1.33 public void testArriveAndAwaitAdvanceBeforeInterrupt() {
594 jsr166 1.20 final Phaser phaser = new Phaser();
595     assertEquals(0, phaser.register());
596 jsr166 1.33 final CountDownLatch pleaseInterrupt = new CountDownLatch(1);
597 jsr166 1.20
598     Thread t = newStartedThread(new CheckedRunnable() {
599 jsr166 1.33 public void realRun() {
600 jsr166 1.20 assertEquals(0, phaser.register());
601     assertFalse(Thread.currentThread().isInterrupted());
602 jsr166 1.33 pleaseInterrupt.countDown();
603 jsr166 1.20 assertEquals(1, phaser.arriveAndAwaitAdvance());
604 jsr166 1.45 assertTrue(Thread.interrupted());
605 jsr166 1.20 }});
606    
607 jsr166 1.33 await(pleaseInterrupt);
608 jsr166 1.46 assertThreadBlocks(t, Thread.State.WAITING);
609 jsr166 1.20 t.interrupt();
610     Thread.currentThread().interrupt();
611     assertEquals(1, phaser.arriveAndAwaitAdvance());
612     assertTrue(Thread.interrupted());
613 jsr166 1.33 awaitTermination(t);
614 dl 1.1 }
615    
616     /**
617     * awaitAdvance atomically waits for all parties within the same phase to
618     * complete before continuing
619     */
620 jsr166 1.33 public void testAwaitAdvance4() {
621 jsr166 1.5 final Phaser phaser = new Phaser(4);
622 jsr166 1.19 final AtomicInteger count = new AtomicInteger(0);
623 jsr166 1.44 List<Thread> threads = new ArrayList<>();
624 jsr166 1.19 for (int i = 0; i < 4; i++)
625 jsr166 1.7 threads.add(newStartedThread(new CheckedRunnable() {
626 jsr166 1.9 public void realRun() {
627 jsr166 1.19 for (int k = 0; k < 3; k++) {
628 jsr166 1.42 assertEquals(2 * k + 1, phaser.arriveAndAwaitAdvance());
629 jsr166 1.19 count.incrementAndGet();
630 jsr166 1.42 assertEquals(2 * k + 1, phaser.arrive());
631     assertEquals(2 * k + 2, phaser.awaitAdvance(2 * k + 1));
632     assertEquals(4 * (k + 1), count.get());
633 jsr166 1.19 }}}));
634    
635 jsr166 1.5 for (Thread thread : threads)
636 jsr166 1.33 awaitTermination(thread);
637 dl 1.1 }
638    
639     /**
640     * awaitAdvance returns the current phase
641     */
642 jsr166 1.33 public void testAwaitAdvance5() {
643 dl 1.1 final Phaser phaser = new Phaser(1);
644 jsr166 1.19 assertEquals(1, phaser.awaitAdvance(phaser.arrive()));
645     assertEquals(1, phaser.getPhase());
646     assertEquals(1, phaser.register());
647 jsr166 1.44 List<Thread> threads = new ArrayList<>();
648 jsr166 1.7 for (int i = 0; i < 8; i++) {
649 jsr166 1.18 final CountDownLatch latch = new CountDownLatch(1);
650     final boolean goesFirst = ((i & 1) == 0);
651 jsr166 1.7 threads.add(newStartedThread(new CheckedRunnable() {
652 jsr166 1.33 public void realRun() {
653 jsr166 1.18 if (goesFirst)
654     latch.countDown();
655     else
656 jsr166 1.33 await(latch);
657 dl 1.1 phaser.arrive();
658 jsr166 1.7 }}));
659 jsr166 1.18 if (goesFirst)
660 jsr166 1.33 await(latch);
661 jsr166 1.18 else
662     latch.countDown();
663 jsr166 1.19 assertEquals(i + 2, phaser.awaitAdvance(phaser.arrive()));
664     assertEquals(i + 2, phaser.getPhase());
665 dl 1.1 }
666 jsr166 1.7 for (Thread thread : threads)
667 jsr166 1.33 awaitTermination(thread);
668 dl 1.1 }
669    
670     /**
671 jsr166 1.29 * awaitAdvance returns the current phase in child phasers
672     */
673     public void testAwaitAdvanceTieredPhaser() throws Exception {
674     final Phaser parent = new Phaser();
675 jsr166 1.44 final List<Phaser> zeroPartyChildren = new ArrayList<>(3);
676     final List<Phaser> onePartyChildren = new ArrayList<>(3);
677 jsr166 1.29 for (int i = 0; i < 3; i++) {
678     zeroPartyChildren.add(new Phaser(parent, 0));
679     onePartyChildren.add(new Phaser(parent, 1));
680     }
681 jsr166 1.44 final List<Phaser> phasers = new ArrayList<>();
682 jsr166 1.29 phasers.addAll(zeroPartyChildren);
683     phasers.addAll(onePartyChildren);
684     phasers.add(parent);
685     for (Phaser phaser : phasers) {
686     assertEquals(-42, phaser.awaitAdvance(-42));
687     assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42));
688 jsr166 1.43 assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42, MEDIUM_DELAY_MS, MILLISECONDS));
689 jsr166 1.29 }
690    
691     for (Phaser child : onePartyChildren)
692     assertEquals(0, child.arrive());
693     for (Phaser phaser : phasers) {
694     assertEquals(-42, phaser.awaitAdvance(-42));
695     assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42));
696 jsr166 1.43 assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42, MEDIUM_DELAY_MS, MILLISECONDS));
697 jsr166 1.29 assertEquals(1, phaser.awaitAdvance(0));
698     assertEquals(1, phaser.awaitAdvanceInterruptibly(0));
699 jsr166 1.43 assertEquals(1, phaser.awaitAdvanceInterruptibly(0, MEDIUM_DELAY_MS, MILLISECONDS));
700 jsr166 1.29 }
701    
702     for (Phaser child : onePartyChildren)
703     assertEquals(1, child.arrive());
704     for (Phaser phaser : phasers) {
705     assertEquals(-42, phaser.awaitAdvance(-42));
706     assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42));
707 jsr166 1.43 assertEquals(-42, phaser.awaitAdvanceInterruptibly(-42, MEDIUM_DELAY_MS, MILLISECONDS));
708 jsr166 1.29 assertEquals(2, phaser.awaitAdvance(0));
709     assertEquals(2, phaser.awaitAdvanceInterruptibly(0));
710 jsr166 1.43 assertEquals(2, phaser.awaitAdvanceInterruptibly(0, MEDIUM_DELAY_MS, MILLISECONDS));
711 jsr166 1.29 assertEquals(2, phaser.awaitAdvance(1));
712     assertEquals(2, phaser.awaitAdvanceInterruptibly(1));
713 jsr166 1.43 assertEquals(2, phaser.awaitAdvanceInterruptibly(1, MEDIUM_DELAY_MS, MILLISECONDS));
714 jsr166 1.29 }
715     }
716    
717     /**
718 dl 1.1 * awaitAdvance returns when the phaser is externally terminated
719     */
720 jsr166 1.33 public void testAwaitAdvance6() {
721 dl 1.1 final Phaser phaser = new Phaser(3);
722 jsr166 1.33 final CountDownLatch pleaseForceTermination = new CountDownLatch(2);
723 jsr166 1.44 final List<Thread> threads = new ArrayList<>();
724 jsr166 1.18 for (int i = 0; i < 2; i++) {
725     Runnable r = new CheckedRunnable() {
726     public void realRun() {
727 jsr166 1.19 assertEquals(0, phaser.arrive());
728 jsr166 1.33 pleaseForceTermination.countDown();
729 jsr166 1.19 assertTrue(phaser.awaitAdvance(0) < 0);
730 jsr166 1.18 assertTrue(phaser.isTerminated());
731 jsr166 1.19 assertTrue(phaser.getPhase() < 0);
732 dl 1.24 assertEquals(0, phaser.getPhase() + Integer.MIN_VALUE);
733 jsr166 1.19 assertEquals(3, phaser.getRegisteredParties());
734 jsr166 1.18 }};
735     threads.add(newStartedThread(r));
736     }
737 jsr166 1.33 await(pleaseForceTermination);
738 dl 1.1 phaser.forceTermination();
739 jsr166 1.27 assertTrue(phaser.isTerminated());
740 dl 1.24 assertEquals(0, phaser.getPhase() + Integer.MIN_VALUE);
741 jsr166 1.18 for (Thread thread : threads)
742 jsr166 1.33 awaitTermination(thread);
743 jsr166 1.19 assertEquals(3, phaser.getRegisteredParties());
744 dl 1.1 }
745    
746     /**
747     * arriveAndAwaitAdvance throws IllegalStateException with no
748     * unarrived parties
749     */
750     public void testArriveAndAwaitAdvance1() {
751 jsr166 1.41 Phaser phaser = new Phaser();
752 dl 1.1 try {
753     phaser.arriveAndAwaitAdvance();
754     shouldThrow();
755 jsr166 1.8 } catch (IllegalStateException success) {}
756 dl 1.1 }
757    
758     /**
759     * arriveAndAwaitAdvance waits for all threads to arrive, the
760     * number of arrived parties is the same number that is accounted
761     * for when the main thread awaitsAdvance
762     */
763 jsr166 1.33 public void testArriveAndAwaitAdvance3() {
764 dl 1.1 final Phaser phaser = new Phaser(1);
765 jsr166 1.19 final int THREADS = 3;
766 jsr166 1.33 final CountDownLatch pleaseArrive = new CountDownLatch(THREADS);
767 jsr166 1.44 final List<Thread> threads = new ArrayList<>();
768 jsr166 1.19 for (int i = 0; i < THREADS; i++)
769 jsr166 1.7 threads.add(newStartedThread(new CheckedRunnable() {
770 jsr166 1.33 public void realRun() {
771 jsr166 1.19 assertEquals(0, phaser.register());
772 jsr166 1.33 pleaseArrive.countDown();
773 jsr166 1.19 assertEquals(1, phaser.arriveAndAwaitAdvance());
774     }}));
775    
776 jsr166 1.33 await(pleaseArrive);
777     long startTime = System.nanoTime();
778 jsr166 1.19 while (phaser.getArrivedParties() < THREADS)
779     Thread.yield();
780     assertEquals(THREADS, phaser.getArrivedParties());
781 jsr166 1.33 assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
782     for (Thread thread : threads)
783 jsr166 1.46 assertThreadBlocks(thread, Thread.State.WAITING);
784 jsr166 1.19 for (Thread thread : threads)
785     assertTrue(thread.isAlive());
786     assertState(phaser, 0, THREADS + 1, 1);
787 dl 1.12 phaser.arriveAndAwaitAdvance();
788 jsr166 1.7 for (Thread thread : threads)
789 jsr166 1.33 awaitTermination(thread);
790 jsr166 1.19 assertState(phaser, 1, THREADS + 1, THREADS + 1);
791 dl 1.1 }
792    
793     }