1 /*
2  * Copyright (c) 2003, 2004, Oracle and/or its affiliates. All rights reserved.
3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4  *
5  * This code is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 only, as
7  * published by the Free Software Foundation.
8  *
9  * This code is distributed in the hope that it will be useful, but WITHOUT
10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12  * version 2 for more details (a copy is included in the LICENSE file that
13  * accompanied this code).
14  *
15  * You should have received a copy of the GNU General Public License version
16  * 2 along with this work; if not, write to the Free Software Foundation,
17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18  *
19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20  * or visit www.oracle.com if you need additional information or have any
21  * questions.
22  */
23 
24 /*
25  * @test
26  * @bug     4904135 4923181
27  * @summary Unit test for EnumSet
28  * @author  Josh Bloch
29  * @author  Neal Gafter
30  * @author  Yo Ma Ma
31  * @key randomness
32  */
33 
34 import java.util.*;
35 import java.io.*;
36 
37 public class EnumSetBash {
38     static Random rnd = new Random();
39 
main(String[] args)40     public static void main(String[] args) {
41         bash(Silly0.class);
42         bash(Silly1.class);
43         bash(Silly31.class);
44         bash(Silly32.class);
45         bash(Silly33.class);
46         bash(Silly63.class);
47         bash(Silly64.class);
48         bash(Silly65.class);
49         bash(Silly127.class);
50         bash(Silly128.class);
51         bash(Silly129.class);
52         bash(Silly500.class);
53     }
54 
bash(Class<T> enumClass)55     static <T extends Enum<T>> void bash(Class<T> enumClass) {
56         Enum[] universe = EnumSet.allOf(enumClass).toArray(new Enum[0]);
57         int numItr = 1000;
58 
59         for (int i=0; i<numItr; i++) {
60             EnumSet<T> s1 = EnumSet.noneOf(enumClass);
61             EnumSet<T> s2 = clone(s1, enumClass);
62             AddRandoms(s1, universe);
63             AddRandoms(s2, universe);
64 
65             EnumSet<T> intersection = clone(s1, enumClass);
66             intersection.retainAll(s2);
67             EnumSet<T> diff1 = clone(s1, enumClass); diff1.removeAll(s2);
68             EnumSet<T> diff2 = clone(s2, enumClass); diff2.removeAll(s1);
69             EnumSet<T> union = clone(s1, enumClass); union.addAll(s2);
70 
71             if (diff1.removeAll(diff2))
72                 fail("Set algebra identity 2 failed");
73             if (diff1.removeAll(intersection))
74                 fail("Set algebra identity 3 failed");
75             if (diff2.removeAll(diff1))
76                 fail("Set algebra identity 4 failed");
77             if (diff2.removeAll(intersection))
78                 fail("Set algebra identity 5 failed");
79             if (intersection.removeAll(diff1))
80                 fail("Set algebra identity 6 failed");
81             if (intersection.removeAll(diff1))
82                 fail("Set algebra identity 7 failed");
83 
84             intersection.addAll(diff1); intersection.addAll(diff2);
85             if (!intersection.equals(union))
86                 fail("Set algebra identity 1 failed");
87 
88             if (new HashSet<T>(union).hashCode() != union.hashCode())
89                 fail("Incorrect hashCode computation.");
90 
91             Iterator e = union.iterator();
92             while (e.hasNext())
93                 if (!intersection.remove(e.next()))
94                     fail("Couldn't remove element from copy.");
95             if (!intersection.isEmpty())
96                 fail("Copy nonempty after deleting all elements.");
97 
98             e = union.iterator();
99             while (e.hasNext()) {
100                 Object o = e.next();
101                 if (!union.contains(o))
102                     fail("Set doesn't contain one of its elements.");
103                 e.remove();
104                 if (union.contains(o))
105                     fail("Set contains element after deletion.");
106             }
107             if (!union.isEmpty())
108                 fail("Set nonempty after deleting all elements.");
109 
110             s1.clear();
111             if (!s1.isEmpty())
112                 fail("Set nonempty after clear.");
113         }
114     }
115 
116     // Done inefficiently so as to exercise various functions
clone(EnumSet<E> s, Class<E> cl)117     static <E extends Enum<E>> EnumSet<E> clone(EnumSet<E> s, Class<E> cl) {
118         EnumSet<E> clone = null;
119         int method = rnd.nextInt(6);
120         switch(method) {
121             case 0:
122                 clone = s.clone();
123                 break;
124             case 1:
125                 clone = EnumSet.noneOf(cl);
126                 Collection arrayList = (Collection)Arrays.asList(s.toArray());
127                 clone.addAll((Collection<E>)arrayList);
128                 break;
129             case 2:
130                 clone = EnumSet.copyOf(s);
131                 break;
132             case 3:
133                 clone = EnumSet.copyOf((Collection<E>)s);
134                 break;
135             case 4:
136                 if (s.isEmpty())
137                     clone = EnumSet.copyOf((Collection<E>)s);
138                 else
139                     clone = EnumSet.copyOf((Collection<E>)(Collection)
140                                            Arrays.asList(s.toArray()));
141                 break;
142             case 5:
143                 clone = (EnumSet<E>) deepCopy(s);
144         }
145         if (!s.equals(clone))
146             fail("Set not equal to copy. " + method);
147         if (!s.containsAll(clone))
148             fail("Set does not contain copy. " + method);
149         if (!clone.containsAll(s))
150             fail("Copy does not contain set. " + method);
151         return clone;
152     }
153 
154     // Utility method to do a deep copy of an object *very slowly* using
155     // serialization/deserialization
deepCopy(T oldObj)156     static <T> T deepCopy(T oldObj) {
157         try {
158             ByteArrayOutputStream bos = new ByteArrayOutputStream();
159             ObjectOutputStream oos = new ObjectOutputStream(bos);
160             oos.writeObject(oldObj);
161             oos.flush();
162             ByteArrayInputStream bin = new ByteArrayInputStream(
163                 bos.toByteArray());
164             ObjectInputStream ois = new ObjectInputStream(bin);
165             return (T) ois.readObject();
166         } catch(Exception e) {
167             throw new IllegalArgumentException(e.toString());
168         }
169     }
170 
AddRandoms(EnumSet<T> s, Enum[] universe)171     static <T extends Enum<T>> void AddRandoms(EnumSet<T> s, Enum[] universe) {
172         for (int i=0; i < universe.length * 2 / 3; i++) {
173             T e = (T) universe[rnd.nextInt(universe.length)];
174 
175             boolean prePresent = s.contains(e);
176             int preSize = s.size();
177             boolean added = s.add(e);
178             if (!s.contains(e))
179                 fail ("Element not present after addition.");
180             if (added == prePresent)
181                 fail ("added == alreadyPresent");
182             int postSize = s.size();
183             if (added && preSize == postSize)
184                 fail ("Add returned true, but size didn't change.");
185             if (!added && preSize != postSize)
186                 fail ("Add returned false, but size changed.");
187         }
188     }
189 
fail(String s)190     static void fail(String s) {
191         throw new RuntimeException(s);
192     }
193 
194     public enum Silly0 { };
195 
196     public enum Silly1 { e1 }
197 
198     public enum Silly31 {
199         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
200         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30
201     }
202 
203     public enum Silly32 {
204         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
205         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31
206     }
207 
208     public enum Silly33 {
209         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
210         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
211         e32
212     }
213 
214     public enum Silly63 {
215         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
216         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
217         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
218         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
219         e62
220     }
221 
222     public enum Silly64 {
223         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
224         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
225         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
226         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
227         e62, e63
228     }
229 
230     public enum Silly65 {
231         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
232         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
233         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
234         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
235         e62, e63, e64
236     }
237 
238     public enum Silly127 {
239         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
240         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
241         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
242         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
243         e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,
244         e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,
245         e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,
246         e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,
247         e118, e119, e120, e121, e122, e123, e124, e125, e126
248     }
249 
250     public enum Silly128 {
251         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
252         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
253         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
254         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
255         e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,
256         e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,
257         e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,
258         e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,
259         e118, e119, e120, e121, e122, e123, e124, e125, e126, e127
260     }
261 
262     public enum Silly129 {
263         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
264         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
265         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
266         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
267         e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,
268         e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,
269         e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,
270         e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,
271         e118, e119, e120, e121, e122, e123, e124, e125, e126, e127, e128
272     }
273 
274     public enum Silly500 {
275         e0, e1, e2, e3, e4, e5, e6, e7, e8, e9, e10, e11, e12, e13, e14, e15, e16,
276         e17, e18, e19, e20, e21, e22, e23, e24, e25, e26, e27, e28, e29, e30, e31,
277         e32, e33, e34, e35, e36, e37, e38, e39, e40, e41, e42, e43, e44, e45, e46,
278         e47, e48, e49, e50, e51, e52, e53, e54, e55, e56, e57, e58, e59, e60, e61,
279         e62, e63, e64, e65, e66, e67, e68, e69, e70, e71, e72, e73, e74, e75, e76,
280         e77, e78, e79, e80, e81, e82, e83, e84, e85, e86, e87, e88, e89, e90, e91,
281         e92, e93, e94, e95, e96, e97, e98, e99, e100, e101, e102, e103, e104, e105,
282         e106, e107, e108, e109, e110, e111, e112, e113, e114, e115, e116, e117,
283         e118, e119, e120, e121, e122, e123, e124, e125, e126, e127, e128, e129,
284         e130, e131, e132, e133, e134, e135, e136, e137, e138, e139, e140, e141,
285         e142, e143, e144, e145, e146, e147, e148, e149, e150, e151, e152, e153,
286         e154, e155, e156, e157, e158, e159, e160, e161, e162, e163, e164, e165,
287         e166, e167, e168, e169, e170, e171, e172, e173, e174, e175, e176, e177,
288         e178, e179, e180, e181, e182, e183, e184, e185, e186, e187, e188, e189,
289         e190, e191, e192, e193, e194, e195, e196, e197, e198, e199, e200, e201,
290         e202, e203, e204, e205, e206, e207, e208, e209, e210, e211, e212, e213,
291         e214, e215, e216, e217, e218, e219, e220, e221, e222, e223, e224, e225,
292         e226, e227, e228, e229, e230, e231, e232, e233, e234, e235, e236, e237,
293         e238, e239, e240, e241, e242, e243, e244, e245, e246, e247, e248, e249,
294         e250, e251, e252, e253, e254, e255, e256, e257, e258, e259, e260, e261,
295         e262, e263, e264, e265, e266, e267, e268, e269, e270, e271, e272, e273,
296         e274, e275, e276, e277, e278, e279, e280, e281, e282, e283, e284, e285,
297         e286, e287, e288, e289, e290, e291, e292, e293, e294, e295, e296, e297,
298         e298, e299, e300, e301, e302, e303, e304, e305, e306, e307, e308, e309,
299         e310, e311, e312, e313, e314, e315, e316, e317, e318, e319, e320, e321,
300         e322, e323, e324, e325, e326, e327, e328, e329, e330, e331, e332, e333,
301         e334, e335, e336, e337, e338, e339, e340, e341, e342, e343, e344, e345,
302         e346, e347, e348, e349, e350, e351, e352, e353, e354, e355, e356, e357,
303         e358, e359, e360, e361, e362, e363, e364, e365, e366, e367, e368, e369,
304         e370, e371, e372, e373, e374, e375, e376, e377, e378, e379, e380, e381,
305         e382, e383, e384, e385, e386, e387, e388, e389, e390, e391, e392, e393,
306         e394, e395, e396, e397, e398, e399, e400, e401, e402, e403, e404, e405,
307         e406, e407, e408, e409, e410, e411, e412, e413, e414, e415, e416, e417,
308         e418, e419, e420, e421, e422, e423, e424, e425, e426, e427, e428, e429,
309         e430, e431, e432, e433, e434, e435, e436, e437, e438, e439, e440, e441,
310         e442, e443, e444, e445, e446, e447, e448, e449, e450, e451, e452, e453,
311         e454, e455, e456, e457, e458, e459, e460, e461, e462, e463, e464, e465,
312         e466, e467, e468, e469, e470, e471, e472, e473, e474, e475, e476, e477,
313         e478, e479, e480, e481, e482, e483, e484, e485, e486, e487, e488, e489,
314         e490, e491, e492, e493, e494, e495, e496, e497, e498, e499
315     }
316 
317 }
318