--- jsr166/src/test/tck/AtomicStampedReferenceTest.java 2009/11/02 20:28:31 1.8 +++ jsr166/src/test/tck/AtomicStampedReferenceTest.java 2014/12/31 19:21:20 1.22 @@ -1,17 +1,19 @@ /* * Written by Doug Lea with assistance from members of JCP JSR-166 * Expert Group and released to the public domain, as explained at - * http://creativecommons.org/licenses/publicdomain + * http://creativecommons.org/publicdomain/zero/1.0/ * Other contributors include Andrew Wright, Jeffrey Hayes, * Pat Fisher, Mike Judd. */ -import junit.framework.*; -import java.util.concurrent.atomic.*; +import java.util.concurrent.atomic.AtomicStampedReference; -public class AtomicStampedReferenceTest extends JSR166TestCase{ - public static void main (String[] args) { - junit.textui.TestRunner.run (suite()); +import junit.framework.Test; +import junit.framework.TestSuite; + +public class AtomicStampedReferenceTest extends JSR166TestCase { + public static void main(String[] args) { + junit.textui.TestRunner.run(suite()); } public static Test suite() { return new TestSuite(AtomicStampedReferenceTest.class); @@ -20,48 +22,47 @@ public class AtomicStampedReferenceTest /** * constructor initializes to given reference and stamp */ - public void testConstructor(){ + public void testConstructor() { AtomicStampedReference ai = new AtomicStampedReference(one, 0); - assertEquals(one,ai.getReference()); - assertEquals(0, ai.getStamp()); + assertSame(one, ai.getReference()); + assertEquals(0, ai.getStamp()); AtomicStampedReference a2 = new AtomicStampedReference(null, 1); - assertNull(a2.getReference()); - assertEquals(1, a2.getStamp()); - + assertNull(a2.getReference()); + assertEquals(1, a2.getStamp()); } /** - * get returns the last values of reference and stamp set + * get returns the last values of reference and stamp set */ - public void testGetSet(){ + public void testGetSet() { int[] mark = new int[1]; AtomicStampedReference ai = new AtomicStampedReference(one, 0); - assertEquals(one,ai.getReference()); - assertEquals(0, ai.getStamp()); - assertEquals(one, ai.get(mark)); + assertSame(one, ai.getReference()); + assertEquals(0, ai.getStamp()); + assertSame(one, ai.get(mark)); assertEquals(0, mark[0]); - ai.set(two, 0); - assertEquals(two,ai.getReference()); - assertEquals(0, ai.getStamp()); - assertEquals(two, ai.get(mark)); + ai.set(two, 0); + assertSame(two, ai.getReference()); + assertEquals(0, ai.getStamp()); + assertSame(two, ai.get(mark)); assertEquals(0, mark[0]); - ai.set(one, 1); - assertEquals(one,ai.getReference()); - assertEquals(1, ai.getStamp()); - assertEquals(one, ai.get(mark)); - assertEquals(1,mark[0]); + ai.set(one, 1); + assertSame(one, ai.getReference()); + assertEquals(1, ai.getStamp()); + assertSame(one, ai.get(mark)); + assertEquals(1, mark[0]); } /** - * attemptStamp succeeds in single thread + * attemptStamp succeeds in single thread */ - public void testAttemptStamp(){ + public void testAttemptStamp() { int[] mark = new int[1]; AtomicStampedReference ai = new AtomicStampedReference(one, 0); assertEquals(0, ai.getStamp()); assertTrue(ai.attemptStamp(one, 1)); - assertEquals(1, ai.getStamp()); - assertEquals(one, ai.get(mark)); + assertEquals(1, ai.getStamp()); + assertSame(one, ai.get(mark)); assertEquals(1, mark[0]); } @@ -69,90 +70,84 @@ public class AtomicStampedReferenceTest * compareAndSet succeeds in changing values if equal to expected reference * and stamp else fails */ - public void testCompareAndSet(){ + public void testCompareAndSet() { int[] mark = new int[1]; AtomicStampedReference ai = new AtomicStampedReference(one, 0); - assertEquals(one, ai.get(mark)); + assertSame(one, ai.get(mark)); assertEquals(0, ai.getStamp()); - assertEquals(0, mark[0]); + assertEquals(0, mark[0]); assertTrue(ai.compareAndSet(one, two, 0, 0)); - assertEquals(two, ai.get(mark)); - assertEquals(0, mark[0]); + assertSame(two, ai.get(mark)); + assertEquals(0, mark[0]); assertTrue(ai.compareAndSet(two, m3, 0, 1)); - assertEquals(m3, ai.get(mark)); - assertEquals(1, mark[0]); + assertSame(m3, ai.get(mark)); + assertEquals(1, mark[0]); assertFalse(ai.compareAndSet(two, m3, 1, 1)); - assertEquals(m3, ai.get(mark)); - assertEquals(1, mark[0]); + assertSame(m3, ai.get(mark)); + assertEquals(1, mark[0]); } /** * compareAndSet in one thread enables another waiting for reference value * to succeed */ - public void testCompareAndSetInMultipleThreads() { + public void testCompareAndSetInMultipleThreads() throws Exception { final AtomicStampedReference ai = new AtomicStampedReference(one, 0); - Thread t = new Thread(new Runnable() { - public void run() { - while(!ai.compareAndSet(two, three, 0, 0)) Thread.yield(); - }}); - try { - t.start(); - assertTrue(ai.compareAndSet(one, two, 0, 0)); - t.join(LONG_DELAY_MS); - assertFalse(t.isAlive()); - assertEquals(ai.getReference(), three); - assertEquals(ai.getStamp(), 0); - } - catch(Exception e) { - unexpectedException(); - } + Thread t = new Thread(new CheckedRunnable() { + public void realRun() { + while (!ai.compareAndSet(two, three, 0, 0)) + Thread.yield(); + }}); + + t.start(); + assertTrue(ai.compareAndSet(one, two, 0, 0)); + t.join(LONG_DELAY_MS); + assertFalse(t.isAlive()); + assertSame(three, ai.getReference()); + assertEquals(0, ai.getStamp()); } /** * compareAndSet in one thread enables another waiting for stamp value * to succeed */ - public void testCompareAndSetInMultipleThreads2() { + public void testCompareAndSetInMultipleThreads2() throws Exception { final AtomicStampedReference ai = new AtomicStampedReference(one, 0); - Thread t = new Thread(new Runnable() { - public void run() { - while(!ai.compareAndSet(one, one, 1, 2)) Thread.yield(); - }}); - try { - t.start(); - assertTrue(ai.compareAndSet(one, one, 0, 1)); - t.join(LONG_DELAY_MS); - assertFalse(t.isAlive()); - assertEquals(ai.getReference(), one); - assertEquals(ai.getStamp(), 2); - } - catch(Exception e) { - unexpectedException(); - } + Thread t = new Thread(new CheckedRunnable() { + public void realRun() { + while (!ai.compareAndSet(one, one, 1, 2)) + Thread.yield(); + }}); + + t.start(); + assertTrue(ai.compareAndSet(one, one, 0, 1)); + t.join(LONG_DELAY_MS); + assertFalse(t.isAlive()); + assertSame(one, ai.getReference()); + assertEquals(2, ai.getStamp()); } /** * repeated weakCompareAndSet succeeds in changing values when equal * to expected */ - public void testWeakCompareAndSet(){ + public void testWeakCompareAndSet() { int[] mark = new int[1]; AtomicStampedReference ai = new AtomicStampedReference(one, 0); - assertEquals(one, ai.get(mark)); - assertEquals(0, ai.getStamp ()); - assertEquals(0, mark[0]); - - while(!ai.weakCompareAndSet(one, two, 0, 0)); - assertEquals(two, ai.get(mark)); - assertEquals(0, mark[0]); - - while(!ai.weakCompareAndSet(two, m3, 0, 1)); - assertEquals(m3, ai.get(mark)); - assertEquals(1, mark[0]); + assertSame(one, ai.get(mark)); + assertEquals(0, ai.getStamp()); + assertEquals(0, mark[0]); + + do {} while (!ai.weakCompareAndSet(one, two, 0, 0)); + assertSame(two, ai.get(mark)); + assertEquals(0, mark[0]); + + do {} while (!ai.weakCompareAndSet(two, m3, 0, 1)); + assertSame(m3, ai.get(mark)); + assertEquals(1, mark[0]); } }