/* * Copyright (c) 2003, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA * or visit www.oracle.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 */ 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); } 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 } }