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

Comparing jsr166/src/test/tck/SemaphoreTest.java (file contents):
Revision 1.2 by dl, Sun Sep 7 20:39:11 2003 UTC vs.
Revision 1.32 by jsr166, Wed Dec 31 19:05:43 2014 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.*;
11 < import java.io.*;
9 > import static java.util.concurrent.TimeUnit.MILLISECONDS;
10  
11 < public class SemaphoreTest extends TestCase{
11 > import java.util.Collection;
12 > import java.util.concurrent.CountDownLatch;
13 > import java.util.concurrent.Semaphore;
14  
15 + import junit.framework.AssertionFailedError;
16 + import junit.framework.Test;
17 + import junit.framework.TestSuite;
18 +
19 + public class SemaphoreTest extends JSR166TestCase {
20      public static void main(String[] args) {
21 <        junit.textui.TestRunner.run (suite());  
21 >        junit.textui.TestRunner.run(suite());
22      }
18    
23      public static Test suite() {
24 <        return new TestSuite(SemaphoreTest.class);
24 >        return new TestSuite(SemaphoreTest.class);
25      }
26  
27 <    private static long SHORT_DELAY_MS = 100;
28 <    private static long MEDIUM_DELAY_MS = 1000;
29 <    private static long LONG_DELAY_MS = 10000;
27 >    /**
28 >     * Subclass to expose protected methods
29 >     */
30 >    static class PublicSemaphore extends Semaphore {
31 >        PublicSemaphore(int permits) { super(permits); }
32 >        PublicSemaphore(int permits, boolean fair) { super(permits, fair); }
33 >        public Collection<Thread> getQueuedThreads() {
34 >            return super.getQueuedThreads();
35 >        }
36 >        public boolean hasQueuedThread(Thread t) {
37 >            return super.getQueuedThreads().contains(t);
38 >        }
39 >        public void reducePermits(int reduction) {
40 >            super.reducePermits(reduction);
41 >        }
42 >    }
43  
44 +    /**
45 +     * A runnable calling acquire
46 +     */
47 +    class InterruptibleLockRunnable extends CheckedRunnable {
48 +        final Semaphore lock;
49 +        InterruptibleLockRunnable(Semaphore s) { lock = s; }
50 +        public void realRun() {
51 +            try {
52 +                lock.acquire();
53 +            }
54 +            catch (InterruptedException ignored) {}
55 +        }
56 +    }
57  
58 <    public void testConstructor1() {
59 <        Semaphore s = new Semaphore(0);
60 <        assertEquals(0, s.availablePermits());
58 >    /**
59 >     * A runnable calling acquire that expects to be interrupted
60 >     */
61 >    class InterruptedLockRunnable extends CheckedInterruptedRunnable {
62 >        final Semaphore lock;
63 >        InterruptedLockRunnable(Semaphore s) { lock = s; }
64 >        public void realRun() throws InterruptedException {
65 >            lock.acquire();
66 >        }
67      }
68  
69 <    public void testConstructor2() {
70 <        Semaphore s = new Semaphore(-1);
71 <        assertEquals(-1, s.availablePermits());
69 >    /**
70 >     * Spin-waits until s.hasQueuedThread(t) becomes true.
71 >     */
72 >    void waitForQueuedThread(PublicSemaphore s, Thread t) {
73 >        long startTime = System.nanoTime();
74 >        while (!s.hasQueuedThread(t)) {
75 >            if (millisElapsedSince(startTime) > LONG_DELAY_MS)
76 >                throw new AssertionFailedError("timed out");
77 >            Thread.yield();
78 >        }
79 >        assertTrue(s.hasQueuedThreads());
80 >        assertTrue(t.isAlive());
81 >    }
82 >
83 >    /**
84 >     * Spin-waits until s.hasQueuedThreads() becomes true.
85 >     */
86 >    void waitForQueuedThreads(Semaphore s) {
87 >        long startTime = System.nanoTime();
88 >        while (!s.hasQueuedThreads()) {
89 >            if (millisElapsedSince(startTime) > LONG_DELAY_MS)
90 >                throw new AssertionFailedError("timed out");
91 >            Thread.yield();
92 >        }
93 >    }
94 >
95 >    enum AcquireMethod {
96 >        acquire() {
97 >            void acquire(Semaphore s) throws InterruptedException {
98 >                s.acquire();
99 >            }
100 >        },
101 >        acquireN() {
102 >            void acquire(Semaphore s, int permits) throws InterruptedException {
103 >                s.acquire(permits);
104 >            }
105 >        },
106 >        acquireUninterruptibly() {
107 >            void acquire(Semaphore s) {
108 >                s.acquireUninterruptibly();
109 >            }
110 >        },
111 >        acquireUninterruptiblyN() {
112 >            void acquire(Semaphore s, int permits) {
113 >                s.acquireUninterruptibly(permits);
114 >            }
115 >        },
116 >        tryAcquire() {
117 >            void acquire(Semaphore s) {
118 >                assertTrue(s.tryAcquire());
119 >            }
120 >        },
121 >        tryAcquireN() {
122 >            void acquire(Semaphore s, int permits) {
123 >                assertTrue(s.tryAcquire(permits));
124 >            }
125 >        },
126 >        tryAcquireTimed() {
127 >            void acquire(Semaphore s) throws InterruptedException {
128 >                assertTrue(s.tryAcquire(2 * LONG_DELAY_MS, MILLISECONDS));
129 >            }
130 >        },
131 >        tryAcquireTimedN {
132 >            void acquire(Semaphore s, int permits) throws InterruptedException {
133 >                assertTrue(s.tryAcquire(permits, 2 * LONG_DELAY_MS, MILLISECONDS));
134 >            }
135 >        };
136 >
137 >        // Intentionally meta-circular
138 >
139 >        /** Acquires 1 permit. */
140 >        void acquire(Semaphore s) throws InterruptedException {
141 >            acquire(s, 1);
142 >        }
143 >        /** Acquires the given number of permits. */
144 >        void acquire(Semaphore s, int permits) throws InterruptedException {
145 >            for (int i = 0; i < permits; i++)
146 >                acquire(s);
147 >        }
148 >    }
149 >
150 >    /**
151 >     * Zero, negative, and positive initial values are allowed in constructor
152 >     */
153 >    public void testConstructor()      { testConstructor(false); }
154 >    public void testConstructor_fair() { testConstructor(true); }
155 >    public void testConstructor(boolean fair) {
156 >        for (int permits : new int[] { -42, -1, 0, 1, 42 }) {
157 >            Semaphore s = new Semaphore(permits, fair);
158 >            assertEquals(permits, s.availablePermits());
159 >            assertEquals(fair, s.isFair());
160 >        }
161 >    }
162 >
163 >    /**
164 >     * Constructor without fairness argument behaves as nonfair
165 >     */
166 >    public void testConstructorDefaultsToNonFair() {
167 >        for (int permits : new int[] { -42, -1, 0, 1, 42 }) {
168 >            Semaphore s = new Semaphore(permits);
169 >            assertEquals(permits, s.availablePermits());
170 >            assertFalse(s.isFair());
171 >        }
172      }
173  
174 <    public void testTryAcquireInSameThread() {
175 <        Semaphore s = new Semaphore(2);
174 >    /**
175 >     * tryAcquire succeeds when sufficient permits, else fails
176 >     */
177 >    public void testTryAcquireInSameThread()      { testTryAcquireInSameThread(false); }
178 >    public void testTryAcquireInSameThread_fair() { testTryAcquireInSameThread(true); }
179 >    public void testTryAcquireInSameThread(boolean fair) {
180 >        Semaphore s = new Semaphore(2, fair);
181          assertEquals(2, s.availablePermits());
182          assertTrue(s.tryAcquire());
183          assertTrue(s.tryAcquire());
184          assertEquals(0, s.availablePermits());
185          assertFalse(s.tryAcquire());
186 +        assertFalse(s.tryAcquire());
187 +        assertEquals(0, s.availablePermits());
188      }
189  
190 <    public void testAcquireReleaseInSameThread(){
191 <        Semaphore s = new Semaphore(1);
192 <        try {
193 <            s.acquire();
194 <            s.release();
195 <            s.acquire();
196 <            s.release();
197 <            s.acquire();
198 <            s.release();
199 <            s.acquire();
200 <            s.release();
201 <            s.acquire();
202 <            s.release();
203 <            assertEquals(1, s.availablePermits());
204 <        } catch( InterruptedException e){
205 <            fail("unexpected exception");
206 <        }
190 >    /**
191 >     * timed tryAcquire times out
192 >     */
193 >    public void testTryAcquire_timeout()      { testTryAcquire_timeout(false); }
194 >    public void testTryAcquire_timeout_fair() { testTryAcquire_timeout(true); }
195 >    public void testTryAcquire_timeout(boolean fair) {
196 >        Semaphore s = new Semaphore(0, fair);
197 >        long startTime = System.nanoTime();
198 >        try { assertFalse(s.tryAcquire(timeoutMillis(), MILLISECONDS)); }
199 >        catch (InterruptedException e) { threadUnexpectedException(e); }
200 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
201 >    }
202 >
203 >    /**
204 >     * timed tryAcquire(N) times out
205 >     */
206 >    public void testTryAcquireN_timeout()      { testTryAcquireN_timeout(false); }
207 >    public void testTryAcquireN_timeout_fair() { testTryAcquireN_timeout(true); }
208 >    public void testTryAcquireN_timeout(boolean fair) {
209 >        Semaphore s = new Semaphore(2, fair);
210 >        long startTime = System.nanoTime();
211 >        try { assertFalse(s.tryAcquire(3, timeoutMillis(), MILLISECONDS)); }
212 >        catch (InterruptedException e) { threadUnexpectedException(e); }
213 >        assertTrue(millisElapsedSince(startTime) >= timeoutMillis());
214 >    }
215 >
216 >    /**
217 >     * acquire(), acquire(N), timed tryAcquired, timed tryAcquire(N)
218 >     * are interruptible
219 >     */
220 >    public void testInterruptible_acquire()               { testInterruptible(false, AcquireMethod.acquire); }
221 >    public void testInterruptible_acquire_fair()          { testInterruptible(true,  AcquireMethod.acquire); }
222 >    public void testInterruptible_acquireN()              { testInterruptible(false, AcquireMethod.acquireN); }
223 >    public void testInterruptible_acquireN_fair()         { testInterruptible(true,  AcquireMethod.acquireN); }
224 >    public void testInterruptible_tryAcquireTimed()       { testInterruptible(false, AcquireMethod.tryAcquireTimed); }
225 >    public void testInterruptible_tryAcquireTimed_fair()  { testInterruptible(true,  AcquireMethod.tryAcquireTimed); }
226 >    public void testInterruptible_tryAcquireTimedN()      { testInterruptible(false, AcquireMethod.tryAcquireTimedN); }
227 >    public void testInterruptible_tryAcquireTimedN_fair() { testInterruptible(true,  AcquireMethod.tryAcquireTimedN); }
228 >    public void testInterruptible(boolean fair, final AcquireMethod acquirer) {
229 >        final PublicSemaphore s = new PublicSemaphore(0, fair);
230 >        final Semaphore pleaseInterrupt = new Semaphore(0, fair);
231 >        Thread t = newStartedThread(new CheckedRunnable() {
232 >            public void realRun() {
233 >                // Interrupt before acquire
234 >                Thread.currentThread().interrupt();
235 >                try {
236 >                    acquirer.acquire(s);
237 >                    shouldThrow();
238 >                } catch (InterruptedException success) {}
239 >
240 >                // Interrupt during acquire
241 >                try {
242 >                    acquirer.acquire(s);
243 >                    shouldThrow();
244 >                } catch (InterruptedException success) {}
245 >
246 >                // Interrupt before acquire(N)
247 >                Thread.currentThread().interrupt();
248 >                try {
249 >                    acquirer.acquire(s, 3);
250 >                    shouldThrow();
251 >                } catch (InterruptedException success) {}
252 >
253 >                pleaseInterrupt.release();
254 >
255 >                // Interrupt during acquire(N)
256 >                try {
257 >                    acquirer.acquire(s, 3);
258 >                    shouldThrow();
259 >                } catch (InterruptedException success) {}
260 >            }});
261 >
262 >        waitForQueuedThread(s, t);
263 >        t.interrupt();
264 >        await(pleaseInterrupt);
265 >        waitForQueuedThread(s, t);
266 >        t.interrupt();
267 >        awaitTermination(t);
268 >    }
269 >
270 >    /**
271 >     * acquireUninterruptibly(), acquireUninterruptibly(N) are
272 >     * uninterruptible
273 >     */
274 >    public void testUninterruptible_acquireUninterruptibly()       { testUninterruptible(false, AcquireMethod.acquireUninterruptibly); }
275 >    public void testUninterruptible_acquireUninterruptibly_fair()  { testUninterruptible(true,  AcquireMethod.acquireUninterruptibly); }
276 >    public void testUninterruptible_acquireUninterruptiblyN()      { testUninterruptible(false, AcquireMethod.acquireUninterruptiblyN); }
277 >    public void testUninterruptible_acquireUninterruptiblyN_fair() { testUninterruptible(true,  AcquireMethod.acquireUninterruptiblyN); }
278 >    public void testUninterruptible(boolean fair, final AcquireMethod acquirer) {
279 >        final PublicSemaphore s = new PublicSemaphore(0, fair);
280 >        final Semaphore pleaseInterrupt = new Semaphore(-1, fair);
281 >
282 >        Thread t1 = newStartedThread(new CheckedRunnable() {
283 >            public void realRun() throws InterruptedException {
284 >                // Interrupt before acquire
285 >                pleaseInterrupt.release();
286 >                Thread.currentThread().interrupt();
287 >                acquirer.acquire(s);
288 >                assertTrue(Thread.interrupted());
289 >            }});
290 >
291 >        Thread t2 = newStartedThread(new CheckedRunnable() {
292 >            public void realRun() throws InterruptedException {
293 >                // Interrupt during acquire
294 >                pleaseInterrupt.release();
295 >                acquirer.acquire(s);
296 >                assertTrue(Thread.interrupted());
297 >            }});
298 >
299 >        await(pleaseInterrupt);
300 >        waitForQueuedThread(s, t1);
301 >        waitForQueuedThread(s, t2);
302 >        t2.interrupt();
303 >
304 >        assertThreadStaysAlive(t1);
305 >        assertTrue(t2.isAlive());
306 >
307 >        s.release(2);
308 >
309 >        awaitTermination(t1);
310 >        awaitTermination(t2);
311 >    }
312 >
313 >    /**
314 >     * hasQueuedThreads reports whether there are waiting threads
315 >     */
316 >    public void testHasQueuedThreads()      { testHasQueuedThreads(false); }
317 >    public void testHasQueuedThreads_fair() { testHasQueuedThreads(true); }
318 >    public void testHasQueuedThreads(boolean fair) {
319 >        final PublicSemaphore lock = new PublicSemaphore(1, fair);
320 >        assertFalse(lock.hasQueuedThreads());
321 >        lock.acquireUninterruptibly();
322 >        Thread t1 = newStartedThread(new InterruptedLockRunnable(lock));
323 >        waitForQueuedThread(lock, t1);
324 >        assertTrue(lock.hasQueuedThreads());
325 >        Thread t2 = newStartedThread(new InterruptibleLockRunnable(lock));
326 >        waitForQueuedThread(lock, t2);
327 >        assertTrue(lock.hasQueuedThreads());
328 >        t1.interrupt();
329 >        awaitTermination(t1);
330 >        assertTrue(lock.hasQueuedThreads());
331 >        lock.release();
332 >        awaitTermination(t2);
333 >        assertFalse(lock.hasQueuedThreads());
334 >    }
335 >
336 >    /**
337 >     * getQueueLength reports number of waiting threads
338 >     */
339 >    public void testGetQueueLength()      { testGetQueueLength(false); }
340 >    public void testGetQueueLength_fair() { testGetQueueLength(true); }
341 >    public void testGetQueueLength(boolean fair) {
342 >        final PublicSemaphore lock = new PublicSemaphore(1, fair);
343 >        assertEquals(0, lock.getQueueLength());
344 >        lock.acquireUninterruptibly();
345 >        Thread t1 = newStartedThread(new InterruptedLockRunnable(lock));
346 >        waitForQueuedThread(lock, t1);
347 >        assertEquals(1, lock.getQueueLength());
348 >        Thread t2 = newStartedThread(new InterruptibleLockRunnable(lock));
349 >        waitForQueuedThread(lock, t2);
350 >        assertEquals(2, lock.getQueueLength());
351 >        t1.interrupt();
352 >        awaitTermination(t1);
353 >        assertEquals(1, lock.getQueueLength());
354 >        lock.release();
355 >        awaitTermination(t2);
356 >        assertEquals(0, lock.getQueueLength());
357 >    }
358 >
359 >    /**
360 >     * getQueuedThreads includes waiting threads
361 >     */
362 >    public void testGetQueuedThreads()      { testGetQueuedThreads(false); }
363 >    public void testGetQueuedThreads_fair() { testGetQueuedThreads(true); }
364 >    public void testGetQueuedThreads(boolean fair) {
365 >        final PublicSemaphore lock = new PublicSemaphore(1, fair);
366 >        assertTrue(lock.getQueuedThreads().isEmpty());
367 >        lock.acquireUninterruptibly();
368 >        assertTrue(lock.getQueuedThreads().isEmpty());
369 >        Thread t1 = newStartedThread(new InterruptedLockRunnable(lock));
370 >        waitForQueuedThread(lock, t1);
371 >        assertTrue(lock.getQueuedThreads().contains(t1));
372 >        Thread t2 = newStartedThread(new InterruptibleLockRunnable(lock));
373 >        waitForQueuedThread(lock, t2);
374 >        assertTrue(lock.getQueuedThreads().contains(t1));
375 >        assertTrue(lock.getQueuedThreads().contains(t2));
376 >        t1.interrupt();
377 >        awaitTermination(t1);
378 >        assertFalse(lock.getQueuedThreads().contains(t1));
379 >        assertTrue(lock.getQueuedThreads().contains(t2));
380 >        lock.release();
381 >        awaitTermination(t2);
382 >        assertTrue(lock.getQueuedThreads().isEmpty());
383 >    }
384 >
385 >    /**
386 >     * drainPermits reports and removes given number of permits
387 >     */
388 >    public void testDrainPermits()      { testDrainPermits(false); }
389 >    public void testDrainPermits_fair() { testDrainPermits(true); }
390 >    public void testDrainPermits(boolean fair) {
391 >        Semaphore s = new Semaphore(0, fair);
392 >        assertEquals(0, s.availablePermits());
393 >        assertEquals(0, s.drainPermits());
394 >        s.release(10);
395 >        assertEquals(10, s.availablePermits());
396 >        assertEquals(10, s.drainPermits());
397 >        assertEquals(0, s.availablePermits());
398 >        assertEquals(0, s.drainPermits());
399      }
400  
401 <    public void testAcquireUninterruptiblyReleaseInSameThread(){
402 <        Semaphore s = new Semaphore(1);
401 >    /**
402 >     * release(-N) throws IllegalArgumentException
403 >     */
404 >    public void testReleaseIAE()      { testReleaseIAE(false); }
405 >    public void testReleaseIAE_fair() { testReleaseIAE(true); }
406 >    public void testReleaseIAE(boolean fair) {
407 >        Semaphore s = new Semaphore(10, fair);
408          try {
409 <            s.acquireUninterruptibly();
410 <            s.release();
411 <            s.acquireUninterruptibly();
72 <            s.release();
73 <            s.acquireUninterruptibly();
74 <            s.release();
75 <            s.acquireUninterruptibly();
76 <            s.release();
77 <            s.acquireUninterruptibly();
78 <            s.release();
79 <            assertEquals(1, s.availablePermits());
80 <        } finally {
81 <        }
409 >            s.release(-1);
410 >            shouldThrow();
411 >        } catch (IllegalArgumentException success) {}
412      }
413  
414 <
415 <    public void testAcquireReleaseInDifferentThreads() {
416 <        final Semaphore s = new Semaphore(1);
417 <        Thread t = new Thread(new Runnable(){
418 <                public void run(){
419 <                    try{
420 <                        s.acquire();
91 <                        s.release();
92 <                        s.release();
93 <                        s.acquire();
94 <                    }catch(InterruptedException ie){
95 <                        fail("unexpected exception");
96 <                    }
97 <                }
98 <            });
99 <        t.start();
414 >    /**
415 >     * reducePermits(-N) throws IllegalArgumentException
416 >     */
417 >    public void testReducePermitsIAE()      { testReducePermitsIAE(false); }
418 >    public void testReducePermitsIAE_fair() { testReducePermitsIAE(true); }
419 >    public void testReducePermitsIAE(boolean fair) {
420 >        PublicSemaphore s = new PublicSemaphore(10, fair);
421          try {
422 <            s.release();
423 <            s.release();
424 <            s.acquire();
425 <            s.acquire();
426 <            t.join();
427 <        } catch( InterruptedException e){
428 <            fail("unexpected exception");
429 <        }
422 >            s.reducePermits(-1);
423 >            shouldThrow();
424 >        } catch (IllegalArgumentException success) {}
425 >    }
426 >
427 >    /**
428 >     * reducePermits reduces number of permits
429 >     */
430 >    public void testReducePermits()      { testReducePermits(false); }
431 >    public void testReducePermits_fair() { testReducePermits(true); }
432 >    public void testReducePermits(boolean fair) {
433 >        PublicSemaphore s = new PublicSemaphore(10, fair);
434 >        assertEquals(10, s.availablePermits());
435 >        s.reducePermits(0);
436 >        assertEquals(10, s.availablePermits());
437 >        s.reducePermits(1);
438 >        assertEquals(9, s.availablePermits());
439 >        s.reducePermits(10);
440 >        assertEquals(-1, s.availablePermits());
441 >        s.reducePermits(10);
442 >        assertEquals(-11, s.availablePermits());
443 >        s.reducePermits(0);
444 >        assertEquals(-11, s.availablePermits());
445      }
446  
447 <    public void testTimedAcquireReleaseInSameThread(){
448 <        Semaphore s = new Semaphore(1);
447 >    /**
448 >     * a reserialized semaphore has same number of permits and
449 >     * fairness, but no queued threads
450 >     */
451 >    public void testSerialization()      { testSerialization(false); }
452 >    public void testSerialization_fair() { testSerialization(true); }
453 >    public void testSerialization(boolean fair) {
454          try {
455 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
456 <            s.release();
457 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
117 <            s.release();
118 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
119 <            s.release();
120 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
121 <            s.release();
122 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
455 >            Semaphore s = new Semaphore(3, fair);
456 >            s.acquire();
457 >            s.acquire();
458              s.release();
459 <            assertEquals(1, s.availablePermits());
460 <        } catch( InterruptedException e){
461 <            fail("unexpected exception");
462 <        }
459 >
460 >            Semaphore clone = serialClone(s);
461 >            assertEquals(fair, s.isFair());
462 >            assertEquals(fair, clone.isFair());
463 >            assertEquals(2, s.availablePermits());
464 >            assertEquals(2, clone.availablePermits());
465 >            clone.acquire();
466 >            clone.acquire();
467 >            clone.release();
468 >            assertEquals(2, s.availablePermits());
469 >            assertEquals(1, clone.availablePermits());
470 >
471 >            s = new Semaphore(0, fair);
472 >            Thread t = newStartedThread(new InterruptibleLockRunnable(s));
473 >            waitForQueuedThreads(s);
474 >            clone = serialClone(s);
475 >            assertEquals(fair, s.isFair());
476 >            assertEquals(fair, clone.isFair());
477 >            assertEquals(0, s.availablePermits());
478 >            assertEquals(0, clone.availablePermits());
479 >            assertTrue(s.hasQueuedThreads());
480 >            assertFalse(clone.hasQueuedThreads());
481 >            s.release();
482 >            awaitTermination(t);
483 >            assertFalse(s.hasQueuedThreads());
484 >            assertFalse(clone.hasQueuedThreads());
485 >        } catch (InterruptedException e) { threadUnexpectedException(e); }
486 >    }
487 >
488 >    /**
489 >     * tryAcquire(n) succeeds when sufficient permits, else fails
490 >     */
491 >    public void testTryAcquireNInSameThread()      { testTryAcquireNInSameThread(false); }
492 >    public void testTryAcquireNInSameThread_fair() { testTryAcquireNInSameThread(true); }
493 >    public void testTryAcquireNInSameThread(boolean fair) {
494 >        Semaphore s = new Semaphore(2, fair);
495 >        assertEquals(2, s.availablePermits());
496 >        assertFalse(s.tryAcquire(3));
497 >        assertEquals(2, s.availablePermits());
498 >        assertTrue(s.tryAcquire(2));
499 >        assertEquals(0, s.availablePermits());
500 >        assertFalse(s.tryAcquire(1));
501 >        assertFalse(s.tryAcquire(2));
502 >        assertEquals(0, s.availablePermits());
503      }
504  
505 <    public void testTimedAcquireReleaseInDifferentThreads() {
506 <        final Semaphore s = new Semaphore(1);
507 <        Thread t = new Thread(new Runnable(){
508 <                public void run(){
509 <                    try{
510 <                        s.release();
511 <                        assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
512 <                        s.release();
513 <                        assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
514 <
515 <                    }catch(InterruptedException ie){
516 <                        fail("unexpected exception");
517 <                    }
518 <                }
519 <            });
520 <        t.start();
521 <        try {
522 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
523 <            s.release();
524 <            assertTrue(s.tryAcquire(SHORT_DELAY_MS, TimeUnit.MILLISECONDS));
525 <            s.release();
526 <            t.join();
527 <        } catch( InterruptedException e){
528 <            fail("unexpected exception");
505 >    /**
506 >     * acquire succeeds if permits available
507 >     */
508 >    public void testReleaseAcquireSameThread_acquire()       { testReleaseAcquireSameThread(false, AcquireMethod.acquire); }
509 >    public void testReleaseAcquireSameThread_acquire_fair()  { testReleaseAcquireSameThread(true, AcquireMethod.acquire); }
510 >    public void testReleaseAcquireSameThread_acquireN()      { testReleaseAcquireSameThread(false, AcquireMethod.acquireN); }
511 >    public void testReleaseAcquireSameThread_acquireN_fair() { testReleaseAcquireSameThread(true, AcquireMethod.acquireN); }
512 >    public void testReleaseAcquireSameThread_acquireUninterruptibly()       { testReleaseAcquireSameThread(false, AcquireMethod.acquireUninterruptibly); }
513 >    public void testReleaseAcquireSameThread_acquireUninterruptibly_fair()  { testReleaseAcquireSameThread(true, AcquireMethod.acquireUninterruptibly); }
514 >    public void testReleaseAcquireSameThread_acquireUninterruptiblyN()      { testReleaseAcquireSameThread(false, AcquireMethod.acquireUninterruptibly); }
515 >    public void testReleaseAcquireSameThread_acquireUninterruptiblyN_fair() { testReleaseAcquireSameThread(true, AcquireMethod.acquireUninterruptibly); }
516 >    public void testReleaseAcquireSameThread_tryAcquire()       { testReleaseAcquireSameThread(false, AcquireMethod.tryAcquire); }
517 >    public void testReleaseAcquireSameThread_tryAcquire_fair()  { testReleaseAcquireSameThread(true, AcquireMethod.tryAcquire); }
518 >    public void testReleaseAcquireSameThread_tryAcquireN()      { testReleaseAcquireSameThread(false, AcquireMethod.tryAcquireN); }
519 >    public void testReleaseAcquireSameThread_tryAcquireN_fair() { testReleaseAcquireSameThread(true, AcquireMethod.tryAcquireN); }
520 >    public void testReleaseAcquireSameThread_tryAcquireTimed()       { testReleaseAcquireSameThread(false, AcquireMethod.tryAcquireTimed); }
521 >    public void testReleaseAcquireSameThread_tryAcquireTimed_fair()  { testReleaseAcquireSameThread(true, AcquireMethod.tryAcquireTimed); }
522 >    public void testReleaseAcquireSameThread_tryAcquireTimedN()      { testReleaseAcquireSameThread(false, AcquireMethod.tryAcquireTimedN); }
523 >    public void testReleaseAcquireSameThread_tryAcquireTimedN_fair() { testReleaseAcquireSameThread(true, AcquireMethod.tryAcquireTimedN); }
524 >    public void testReleaseAcquireSameThread(boolean fair,
525 >                                             final AcquireMethod acquirer) {
526 >        Semaphore s = new Semaphore(1, fair);
527 >        for (int i = 1; i < 6; i++) {
528 >            s.release(i);
529 >            assertEquals(1 + i, s.availablePermits());
530 >            try {
531 >                acquirer.acquire(s, i);
532 >            } catch (InterruptedException e) { threadUnexpectedException(e); }
533 >            assertEquals(1, s.availablePermits());
534          }
535      }
536  
537 <    public void testAcquire_InterruptedException(){
538 <        final Semaphore s = new Semaphore(0);
539 <        Thread t = new Thread(new Runnable(){
540 <                public void run(){
541 <                    try{
542 <                        s.acquire();
543 <                        fail("should throw");
544 <                    }catch(InterruptedException success){}
545 <                }
546 <            });
547 <        t.start();
548 <        try{
549 <            Thread.sleep(SHORT_DELAY_MS);
550 <            t.interrupt();
551 <            t.join();
552 <        } catch(InterruptedException e){
553 <            fail("unexpected exception");
554 <        }
555 <    }
556 <    
557 <    public void testTryAcquire_InterruptedException(){
558 <        final Semaphore s = new Semaphore(0);
559 <        Thread t = new Thread(new Runnable(){
560 <                public void run(){
561 <                    try{
562 <                        s.tryAcquire(MEDIUM_DELAY_MS, TimeUnit.MILLISECONDS);
563 <                        fail("should throw");
564 <                    }catch(InterruptedException success){
565 <                    }
566 <                }
567 <            });
568 <        t.start();
569 <        try{
570 <            Thread.sleep(SHORT_DELAY_MS);
571 <            t.interrupt();
572 <            t.join();
573 <        } catch(InterruptedException e){
574 <            fail("unexpected exception");
537 >    /**
538 >     * release in one thread enables acquire in another thread
539 >     */
540 >    public void testReleaseAcquireDifferentThreads_acquire()       { testReleaseAcquireDifferentThreads(false, AcquireMethod.acquire); }
541 >    public void testReleaseAcquireDifferentThreads_acquire_fair()  { testReleaseAcquireDifferentThreads(true, AcquireMethod.acquire); }
542 >    public void testReleaseAcquireDifferentThreads_acquireN()      { testReleaseAcquireDifferentThreads(false, AcquireMethod.acquireN); }
543 >    public void testReleaseAcquireDifferentThreads_acquireN_fair() { testReleaseAcquireDifferentThreads(true, AcquireMethod.acquireN); }
544 >    public void testReleaseAcquireDifferentThreads_acquireUninterruptibly()       { testReleaseAcquireDifferentThreads(false, AcquireMethod.acquireUninterruptibly); }
545 >    public void testReleaseAcquireDifferentThreads_acquireUninterruptibly_fair()  { testReleaseAcquireDifferentThreads(true, AcquireMethod.acquireUninterruptibly); }
546 >    public void testReleaseAcquireDifferentThreads_acquireUninterruptiblyN()      { testReleaseAcquireDifferentThreads(false, AcquireMethod.acquireUninterruptibly); }
547 >    public void testReleaseAcquireDifferentThreads_acquireUninterruptiblyN_fair() { testReleaseAcquireDifferentThreads(true, AcquireMethod.acquireUninterruptibly); }
548 >    public void testReleaseAcquireDifferentThreads_tryAcquireTimed()       { testReleaseAcquireDifferentThreads(false, AcquireMethod.tryAcquireTimed); }
549 >    public void testReleaseAcquireDifferentThreads_tryAcquireTimed_fair()  { testReleaseAcquireDifferentThreads(true, AcquireMethod.tryAcquireTimed); }
550 >    public void testReleaseAcquireDifferentThreads_tryAcquireTimedN()      { testReleaseAcquireDifferentThreads(false, AcquireMethod.tryAcquireTimedN); }
551 >    public void testReleaseAcquireDifferentThreads_tryAcquireTimedN_fair() { testReleaseAcquireDifferentThreads(true, AcquireMethod.tryAcquireTimedN); }
552 >    public void testReleaseAcquireDifferentThreads(boolean fair,
553 >                                                   final AcquireMethod acquirer) {
554 >        final Semaphore s = new Semaphore(0, fair);
555 >        final int rounds = 4;
556 >        long startTime = System.nanoTime();
557 >        Thread t = newStartedThread(new CheckedRunnable() {
558 >            public void realRun() throws InterruptedException {
559 >                for (int i = 0; i < rounds; i++) {
560 >                    assertFalse(s.hasQueuedThreads());
561 >                    if (i % 2 == 0)
562 >                        acquirer.acquire(s);
563 >                    else
564 >                        acquirer.acquire(s, 3);
565 >                }}});
566 >
567 >        for (int i = 0; i < rounds; i++) {
568 >            while (! (s.availablePermits() == 0 && s.hasQueuedThreads()))
569 >                Thread.yield();
570 >            assertTrue(t.isAlive());
571 >            if (i % 2 == 0)
572 >                s.release();
573 >            else
574 >                s.release(3);
575          }
576 +        awaitTermination(t);
577 +        assertEquals(0, s.availablePermits());
578 +        assertTrue(millisElapsedSince(startTime) < LONG_DELAY_MS);
579      }
580  
581 <    public void testSerialization() {
582 <        Semaphore l = new Semaphore(3);
583 <
584 <        try {
585 <            l.acquire();
586 <            l.release();
587 <            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
588 <            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
589 <            out.writeObject(l);
590 <            out.close();
591 <
592 <            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
593 <            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
594 <            Semaphore r = (Semaphore) in.readObject();
595 <            assertEquals(3, r.availablePermits());
596 <            r.acquire();
597 <            r.release();
598 <        } catch(Exception e){
599 <            e.printStackTrace();
600 <            fail("unexpected exception");
601 <        }
581 >    /**
582 >     * fair locks are strictly FIFO
583 >     */
584 >    public void testFairLocksFifo() {
585 >        final PublicSemaphore s = new PublicSemaphore(1, true);
586 >        final CountDownLatch pleaseRelease = new CountDownLatch(1);
587 >        Thread t1 = newStartedThread(new CheckedRunnable() {
588 >            public void realRun() throws InterruptedException {
589 >                // Will block; permits are available, but not three
590 >                s.acquire(3);
591 >            }});
592 >
593 >        waitForQueuedThreads(s);
594 >
595 >        Thread t2 = newStartedThread(new CheckedRunnable() {
596 >            public void realRun() throws InterruptedException {
597 >                // Will fail, even though 1 permit is available
598 >                assertFalse(s.tryAcquire(0L, MILLISECONDS));
599 >                assertFalse(s.tryAcquire(1, 0L, MILLISECONDS));
600 >
601 >                // untimed tryAcquire will barge and succeed
602 >                assertTrue(s.tryAcquire());
603 >                s.release(2);
604 >                assertTrue(s.tryAcquire(2));
605 >                s.release();
606 >
607 >                pleaseRelease.countDown();
608 >                // Will queue up behind t1, even though 1 permit is available
609 >                s.acquire();
610 >            }});
611 >
612 >        await(pleaseRelease);
613 >        waitForQueuedThread(s, t2);
614 >        s.release(2);
615 >        awaitTermination(t1);
616 >        assertTrue(t2.isAlive());
617 >        s.release();
618 >        awaitTermination(t2);
619 >   }
620 >
621 >    /**
622 >     * toString indicates current number of permits
623 >     */
624 >    public void testToString()      { testToString(false); }
625 >    public void testToString_fair() { testToString(true); }
626 >    public void testToString(boolean fair) {
627 >        PublicSemaphore s = new PublicSemaphore(0, fair);
628 >        assertTrue(s.toString().contains("Permits = 0"));
629 >        s.release();
630 >        assertTrue(s.toString().contains("Permits = 1"));
631 >        s.release(2);
632 >        assertTrue(s.toString().contains("Permits = 3"));
633 >        s.reducePermits(5);
634 >        assertTrue(s.toString().contains("Permits = -2"));
635      }
636  
637   }

Diff Legend

Removed lines
+ Added lines
< Changed lines
> Changed lines