import junit.framework.TestCase;
import org.hibernate.util.SerializationHelper;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class TestSerialize extends TestCase implements Serializable {
    public void testSerializeManyToMany() throws Exception {
        DataObjectXref1 dataObjectXref1 = new DataObjectXref1(new Integer(100));
        DataObjectXref2 dataObjectXref2 = new DataObjectXref2(new Integer(200));

        dataObjectXref1.addXref2(dataObjectXref2);
        dataObjectXref2.addXref1(dataObjectXref1);

        assertTrue(dataObjectXref1.getXref2s().contains(dataObjectXref2));
        assertTrue(dataObjectXref2.getXref1s().contains(dataObjectXref1));

        //pull the xref2 from the set, it's the same reference but it's to illustrate a point further down
        DataObjectXref2 xref2ReferenceFromXref1Set = ((DataObjectXref2) dataObjectXref1.getXref2s().toArray()[0]);
        assertTrue(xref2ReferenceFromXref1Set.getXref1s().contains(dataObjectXref1));

        //clone the first xref object
        DataObjectXref1 clonedDataObjectXref1 = (DataObjectXref1) SerializationHelper.clone(dataObjectXref1);

        //see if the second xref object is still contained in the first's xref2 set
        assertTrue(clonedDataObjectXref1.getXref2s().contains(dataObjectXref2));

        //pull the cloned xref2 reference out of the collection of the cloned xref1
        xref2ReferenceFromXref1Set = ((DataObjectXref2) clonedDataObjectXref1.getXref2s().toArray()[0]);

        //the cloned xref2 reference cannot find the xref1 object by use of contains (nor will it via add() or remove())
        assertFalse(xref2ReferenceFromXref1Set.getXref1s().contains(clonedDataObjectXref1));

        //verify that the xref1 reference actually does exist in the set
        //by using a List it iterates through the list and calls equals() on each element
        List itReallyDoesExist = new ArrayList(xref2ReferenceFromXref1Set.getXref1s());
        assertTrue(itReallyDoesExist.contains(clonedDataObjectXref1));
    }


    private class DataObjectXref1 implements Serializable {
        private Integer someBusinessKey;
        private Set xref2s;

        public DataObjectXref1(Integer key) {
            this.someBusinessKey = key;
            xref2s = new HashSet();
        }

        public void addXref2(DataObjectXref2 xref2) {
            xref2s.add(xref2);
        }

        public Set getXref2s() {
            return xref2s;
        }

        public boolean equals(Object value) {
            if (this == value) {
                return true;
            }
            if (!(value instanceof DataObjectXref1)) {
                return false;
            }

            final DataObjectXref1 dataObjectXref1 = (DataObjectXref1) value;

            if (someBusinessKey != null ? !someBusinessKey.equals(dataObjectXref1.someBusinessKey) : dataObjectXref1.someBusinessKey != null) {
                return false;
            }

            return true;
        }

        public int hashCode() {
            int hashCode = (someBusinessKey != null ? someBusinessKey.hashCode() : 0);
            return hashCode;
        }
    }

    private class DataObjectXref2 implements Serializable {
        private Integer someBusinessKey;
        private Set xref1s;

        public DataObjectXref2(Integer key) {
            this.someBusinessKey = key;
            this.xref1s = new HashSet();
        }

        public void addXref1(DataObjectXref1 xref1) {
            xref1s.add(xref1);
        }

        public Set getXref1s() {
            return xref1s;
        }

        public boolean equals(Object value) {
            if (this == value) {
                return true;
            }
            if (!(value instanceof DataObjectXref2)) {
                return false;
            }

            final DataObjectXref2 dataObjectXref2 = (DataObjectXref2) value;

            if (someBusinessKey != null ? !someBusinessKey.equals(dataObjectXref2.someBusinessKey) : dataObjectXref2.someBusinessKey != null) {
                return false;
            }

            return true;
        }

        public int hashCode() {
            int hashCode = (someBusinessKey != null ? someBusinessKey.hashCode() : 0);
            return hashCode;
        }
    }
}