--- jsr166/src/test/tck/AtomicReferenceTest.java	2003/08/31 19:24:53	1.1
+++ jsr166/src/test/tck/AtomicReferenceTest.java	2005/05/25 14:27:37	1.8
@@ -1,14 +1,16 @@
 /*
- * Written by members of JCP JSR-166 Expert Group and released to the
- * public domain. Use, modify, and redistribute this code in any way
- * without acknowledgement. Other contributors include Andrew Wright,
- * Jeffrey Hayes, Pat Fischer, Mike Judd.
+ * 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
+ * Other contributors include Andrew Wright, Jeffrey Hayes, 
+ * Pat Fisher, Mike Judd. 
  */
 
 import junit.framework.*;
 import java.util.concurrent.atomic.*;
+import java.io.*;
 
-public class AtomicReferenceTest extends TestCase {
+public class AtomicReferenceTest extends JSR166TestCase {
     public static void main (String[] args) {
         junit.textui.TestRunner.run (suite());
     }
@@ -16,25 +18,25 @@ public class AtomicReferenceTest extends
         return new TestSuite(AtomicReferenceTest.class);
     }
 
-    static final Integer zero = new Integer(0);
-    static final Integer one = new Integer(1);
-    static final Integer two = new Integer(2);
-    static final Integer m3  = new Integer(-3);
-    static final Integer m4 = new Integer(-4);
-    static final Integer m5 = new Integer(-5);
-    static final Integer seven = new Integer(7);
-    static final Integer m10 = new Integer(-10);
-
+    /**
+     * constructor initializes to given value
+     */
     public void testConstructor(){
         AtomicReference ai = new AtomicReference(one);
 	assertEquals(one,ai.get());
     }
 
+    /**
+     * default constructed initializes to null
+     */
     public void testConstructor2(){
         AtomicReference ai = new AtomicReference();
 	assertNull(ai.get());
     }
 
+    /**
+     * get returns the last value set
+     */
     public void testGetSet(){
         AtomicReference ai = new AtomicReference(one);
 	assertEquals(one,ai.get());
@@ -42,8 +44,23 @@ public class AtomicReferenceTest extends
 	assertEquals(two,ai.get());
 	ai.set(m3);
 	assertEquals(m3,ai.get());
-	
     }
+
+    /**
+     * get returns the last value lazySet in same thread
+     */
+    public void testGetLazySet(){
+        AtomicReference ai = new AtomicReference(one);
+	assertEquals(one,ai.get());
+	ai.lazySet(two);
+	assertEquals(two,ai.get());
+	ai.lazySet(m3);
+	assertEquals(m3,ai.get());
+    }
+
+    /**
+     * compareAndSet succeeds in changing value if equal to expected else fails
+     */
     public void testCompareAndSet(){
         AtomicReference ai = new AtomicReference(one);
 	assertTrue(ai.compareAndSet(one,two));
@@ -55,6 +72,32 @@ public class AtomicReferenceTest extends
 	assertEquals(seven,ai.get());
     }
 
+    /**
+     * compareAndSet in one thread enables another waiting for value
+     * to succeed
+     */
+    public void testCompareAndSetInMultipleThreads() {
+        final AtomicReference ai = new AtomicReference(one);
+        Thread t = new Thread(new Runnable() {
+                public void run() {
+                    while(!ai.compareAndSet(two, three)) Thread.yield();
+                }});
+        try {
+            t.start();
+            assertTrue(ai.compareAndSet(one, two));
+            t.join(LONG_DELAY_MS);
+            assertFalse(t.isAlive());
+            assertEquals(ai.get(), three);
+        }
+        catch(Exception e) {
+            unexpectedException();
+        }
+    }
+
+    /**
+     * repeated weakCompareAndSet succeeds in changing value when equal
+     * to expected 
+     */
     public void testWeakCompareAndSet(){
         AtomicReference ai = new AtomicReference(one);
 	while(!ai.weakCompareAndSet(one,two));
@@ -64,6 +107,9 @@ public class AtomicReferenceTest extends
 	assertEquals(seven,ai.get());
     }
 
+    /**
+     * getAndSet returns previous value and sets to given value
+     */
     public void testGetAndSet(){
         AtomicReference ai = new AtomicReference(one);
 	assertEquals(one,ai.getAndSet(zero));
@@ -71,5 +117,37 @@ public class AtomicReferenceTest extends
 	assertEquals(m10,ai.getAndSet(one));
     }
 
+    /**
+     * a deserialized serialized atomic holds same value
+     */
+    public void testSerialization() {
+        AtomicReference l = new AtomicReference();
+
+        try {
+            l.set(one);
+            ByteArrayOutputStream bout = new ByteArrayOutputStream(10000);
+            ObjectOutputStream out = new ObjectOutputStream(new BufferedOutputStream(bout));
+            out.writeObject(l);
+            out.close();
+
+            ByteArrayInputStream bin = new ByteArrayInputStream(bout.toByteArray());
+            ObjectInputStream in = new ObjectInputStream(new BufferedInputStream(bin));
+            AtomicReference r = (AtomicReference) in.readObject();
+            assertEquals(l.get(), r.get());
+        } catch(Exception e){
+            unexpectedException();
+        }
+    }
+
+    /**
+     * toString returns current value.
+     */ 
+    public void testToString() {
+        AtomicReference<Integer> ai = new AtomicReference<Integer>(one); 
+        assertEquals(ai.toString(), one.toString());
+        ai.set(two);
+        assertEquals(ai.toString(), two.toString());
+    }
+
 }