| /* |
| * 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 |
| } |
| |
| } |
