/*

 * Copyright 2003 Sun Microsystems, Inc.  All Rights Reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

/*

 * @test

 * @bug     4904140

 * @summary Unit test for EnumMap

 * @author  Josh Bloch

 * @author  Yo Yo Ma

 *

 * @compile -source 1.5 EnumMapBash.java

 * @run main EnumMapBash

 */

import java.util.*;

import java.io.*;

public class EnumMapBash {

    static Random rnd = new Random();

    public static void main(String[] args) {

        bash(Silly31.class);

        bash(Silly32.class);

        bash(Silly33.class);

        bash(Silly63.class);

        bash(Silly64.class);

        bash(Silly65.class);

        bash(Silly127.class);

        bash(Silly128.class);

        bash(Silly129.class);

        bash(Silly500.class);

    }

    private static Enum[] ZERO_LENGTH_ENUM_ARRAY = new Enum[0];

    static <T extends Enum<T>> void bash(Class<T> enumClass) {

        Enum[] universe = enumClass.getEnumConstants();

        int numItr = 100;

        // Linked List test

        for (int i=0; i<numItr; i++) {

            int mapSize = universe.length * 7 / 8;

            // Build the linked list

            Map<T, T> m = new EnumMap<T, T>(enumClass);

            if (!m.isEmpty())

                fail("New instance non empty.");

            Enum[] perm = (Enum[]) universe.clone();

            Collections.shuffle(Arrays.asList(perm));

            T head = (T) perm[0];

            for (int j = 0; j < mapSize; j++)

                m.put((T)perm[j], (T)perm[j + 1]);

            T nil = (T)perm[mapSize];

            if (m.size() != mapSize)

                fail("Size not as expected.");

            Map<T, T> tm = new TreeMap<T, T>(m);

            if (m.hashCode() != tm.hashCode())

                fail("Incorrect hashCode computation.");

            if (!m.toString().equals(tm.toString()))

                fail("Incorrect toString computation.");

            if (!tm.equals(m))

                fail("Incorrect equals (1).");

            if (!m.equals(tm))

                fail("Incorrect equals (2).");

            Map<T, T> m2 = new EnumMap<T, T>(enumClass); m2.putAll(m);

            m2.values().removeAll(m.keySet());

            if (m2.size()!= 1 || !m2.containsValue(nil))

                fail("Collection views test failed.");

            int j=0;

            while (head != nil) {

                if (!m.containsKey(head))

                    fail("Linked list doesn't contain a link.");

                T newHead = m.get(head);

                if (newHead == null)

                    fail("Could not retrieve a link.");

                m.remove(head);

                head = newHead;

                j++;

            }

            if (!m.isEmpty())

                fail("Map nonempty after removing all links.");

            if (j != mapSize)

                fail("Linked list size not as expected.");

        }

        EnumMap<T, T> m = new EnumMap<T, T>(enumClass);

        int mapSize = 0;

        for (int i=0; i<universe.length; i += 2) {

            if (m.put((T)universe[i], (T)universe[i]) != null)

                fail("put returns a non-null value erroenously.");

            mapSize++;

        }

        for (int i=0; i<universe.length; i++)

            if (m.containsValue(universe[i]) != (i%2==0))

                fail("contains value "+i);

        if (m.put((T)universe[0], (T)universe[0]) == null)

            fail("put returns a null value erroenously.");

        Map<T, T>  m2 = m.clone();

        cloneTest(m, m2);

        m2 = new EnumMap<T,T>(enumClass);

        m2.putAll(m);

        cloneTest(m, m2);

        m2 = new EnumMap<T, T>(m);

        cloneTest(m, m2);

        m2 = new EnumMap<T, T>((Map<T, T>) m);

        cloneTest(m, m2);

        if (!m.isEmpty()) {

            m2 = new EnumMap<T, T>(new HashMap<T, T>(m));

            cloneTest(m, m2);

        }

        m2 = deepCopy(m);

        cloneTest(m, m2);

        if (!m.equals(m2))

            fail("Clone not equal to original. (1)");

        if (!m2.equals(m))

            fail("Clone not equal to original. (2)");

        Set<Map.Entry<T,T>> s = m.entrySet(), s2 = m2.entrySet();

        if (!s.equals(s2))

            fail("Clone not equal to original. (3)");

        if (!s2.equals(s))

            fail("Clone not equal to original. (4)");

        if (!s.containsAll(s2))

            fail("Original doesn't contain clone!");

        if (!s2.containsAll(s))

            fail("Clone doesn't contain original!");

        s2.removeAll(s);

        if (!m2.isEmpty()) {

            System.out.println(m2.size());

            System.out.println(m2);

            fail("entrySet().removeAll failed.");

        }

        m2.putAll(m);

        m2.clear();

        if (!m2.isEmpty())

            fail("clear failed.");

        Iterator i = m.entrySet().iterator();

        while(i.hasNext()) {

            i.next();

            i.remove();

        }

        if (!m.isEmpty())

            fail("Iterator.remove() failed");

    }

    // Done inefficiently so as to exercise various functions

    static <K, V> void cloneTest(Map<K, V> m, Map<K, V> clone) {

        if (!m.equals(clone))

            fail("Map not equal to copy.");

        if (!clone.equals(m))

            fail("Copy not equal to map.");

        if (!m.entrySet().containsAll(clone.entrySet()))

            fail("Set does not contain copy.");

        if (!clone.entrySet().containsAll(m.entrySet()))

            fail("Copy does not contain set.");

        if (!m.entrySet().equals(clone.entrySet()))

            fail("Set not equal clone set");

        if (!clone.entrySet().equals(m.entrySet()))

            fail("Clone set not equal set");

    }

    // Utility method to do a deep copy of an object *very slowly* using

    // serialization/deserialization

    static <T> T deepCopy(T oldObj) {

        try {

            ByteArrayOutputStream bos = new ByteArrayOutputStream();

            ObjectOutputStream oos = new ObjectOutputStream(bos);

            oos.writeObject(oldObj);

            oos.flush();

            ByteArrayInputStream bin = new ByteArrayInputStream(

                bos.toByteArray());

            ObjectInputStream ois = new ObjectInputStream(bin);

            return (T) ois.readObject();

        } catch(Exception e) {

            throw new IllegalArgumentException(e.toString());

        }

    }

    static void fail(String s) {

        throw new RuntimeException(s);

    }

    public enum Silly31 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30

    }

    public enum Silly32 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31

    }

    public enum Silly33 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32

    }

    public enum Silly63 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62

    }

    public enum Silly64 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63

    }

    public enum Silly65 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63, e64

    }

    public enum Silly127 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,

        e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,

        e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,

        e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,

        e118, e119, e120, e121, e122, e123, e124, e125, e126

    }

    public enum Silly128 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,

        e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,

        e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,

        e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,

        e118, e119, e120, e121, e122, e123, e124, e125, e126, e127

    }

    public enum Silly129 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,

        e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,

        e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,

        e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,

        e118, e119, e120, e121, e122, e123, e124, e125, e126, e127, e128

    }

    public enum Silly500 {

        e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,

        e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,

        e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,

        e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,

        e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,

        e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,

        e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,

        e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,

        e118, e119, e120, e121, e122, e123, e124, e125, e126, e127, e128, e129,

        e130, e131, e132, e133, e134, e135, e136, e137, e138, e139, e140, e141,

        e142, e143, e144, e145, e146, e147, e148, e149, e150, e151, e152, e153,

        e154, e155, e156, e157, e158, e159, e160, e161, e162, e163, e164, e165,

        e166, e167, e168, e169, e170, e171, e172, e173, e174, e175, e176, e177,

        e178, e179, e180, e181, e182, e183, e184, e185, e186, e187, e188, e189,

        e190, e191, e192, e193, e194, e195, e196, e197, e198, e199, e200, e201,

        e202, e203, e204, e205, e206, e207, e208, e209, e210, e211, e212, e213,

        e214, e215, e216, e217, e218, e219, e220, e221, e222, e223, e224, e225,

        e226, e227, e228, e229, e230, e231, e232, e233, e234, e235, e236, e237,

        e238, e239, e240, e241, e242, e243, e244, e245, e246, e247, e248, e249,

        e250, e251, e252, e253, e254, e255, e256, e257, e258, e259, e260, e261,

        e262, e263, e264, e265, e266, e267, e268, e269, e270, e271, e272, e273,

        e274, e275, e276, e277, e278, e279, e280, e281, e282, e283, e284, e285,

        e286, e287, e288, e289, e290, e291, e292, e293, e294, e295, e296, e297,

        e298, e299, e300, e301, e302, e303, e304, e305, e306, e307, e308, e309,

        e310, e311, e312, e313, e314, e315, e316, e317, e318, e319, e320, e321,

        e322, e323, e324, e325, e326, e327, e328, e329, e330, e331, e332, e333,

        e334, e335, e336, e337, e338, e339, e340, e341, e342, e343, e344, e345,

        e346, e347, e348, e349, e350, e351, e352, e353, e354, e355, e356, e357,

        e358, e359, e360, e361, e362, e363, e364, e365, e366, e367, e368, e369,

        e370, e371, e372, e373, e374, e375, e376, e377, e378, e379, e380, e381,

        e382, e383, e384, e385, e386, e387, e388, e389, e390, e391, e392, e393,

        e394, e395, e396, e397, e398, e399, e400, e401, e402, e403, e404, e405,

        e406, e407, e408, e409, e410, e411, e412, e413, e414, e415, e416, e417,

        e418, e419, e420, e421, e422, e423, e424, e425, e426, e427, e428, e429,

        e430, e431, e432, e433, e434, e435, e436, e437, e438, e439, e440, e441,

        e442, e443, e444, e445, e446, e447, e448, e449, e450, e451, e452, e453,

        e454, e455, e456, e457, e458, e459, e460, e461, e462, e463, e464, e465,

        e466, e467, e468, e469, e470, e471, e472, e473, e474, e475, e476, e477,

        e478, e479, e480, e481, e482, e483, e484, e485, e486, e487, e488, e489,

        e490, e491, e492, e493, e494, e495, e496, e497, e498, e499

    }

}