1/*
2 * Copyright (C) 2007 The Guava Authors
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.google.common.collect;
18
19import static java.util.Arrays.asList;
20
21import com.google.common.annotations.GwtCompatible;
22import com.google.common.annotations.GwtIncompatible;
23import com.google.common.testing.SerializableTester;
24
25import junit.framework.TestCase;
26
27import java.util.Collection;
28import java.util.EnumSet;
29import java.util.Set;
30
31/**
32 * Tests for an {@link EnumMultiset}.
33 *
34 * @author Jared Levy
35 */
36@GwtCompatible(emulated = true)
37public class EnumMultisetTest extends TestCase {
38  private static enum Color {
39    BLUE, RED, YELLOW, GREEN, WHITE
40  }
41
42  public void testClassCreate() {
43    Multiset<Color> ms = EnumMultiset.create(Color.class);
44    ms.add(Color.RED);
45    ms.add(Color.YELLOW);
46    ms.add(Color.RED);
47    assertEquals(0, ms.count(Color.BLUE));
48    assertEquals(1, ms.count(Color.YELLOW));
49    assertEquals(2, ms.count(Color.RED));
50  }
51
52  public void testCollectionCreate() {
53    Multiset<Color> ms = EnumMultiset.create(
54        asList(Color.RED, Color.YELLOW, Color.RED));
55    assertEquals(0, ms.count(Color.BLUE));
56    assertEquals(1, ms.count(Color.YELLOW));
57    assertEquals(2, ms.count(Color.RED));
58  }
59
60  public void testIllegalCreate() {
61    Collection<Color> empty = EnumSet.noneOf(Color.class);
62    try {
63      EnumMultiset.create(empty);
64      fail();
65    } catch (IllegalArgumentException expected) {}
66  }
67
68  public void testToString() {
69    Multiset<Color> ms = EnumMultiset.create(Color.class);
70    ms.add(Color.BLUE, 3);
71    ms.add(Color.YELLOW, 1);
72    ms.add(Color.RED, 2);
73    assertEquals("[BLUE x 3, RED x 2, YELLOW]", ms.toString());
74  }
75
76  @GwtIncompatible("SerializableTester")
77  public void testSerializable() {
78    Multiset<Color> ms = EnumMultiset.create(
79        asList(Color.RED, Color.YELLOW, Color.RED));
80    assertEquals(ms, SerializableTester.reserialize(ms));
81  }
82
83  public void testEntrySet() {
84    Multiset<Color> ms = EnumMultiset.create(Color.class);
85    ms.add(Color.BLUE, 3);
86    ms.add(Color.YELLOW, 1);
87    ms.add(Color.RED, 2);
88
89    Set<Object> uniqueEntries = Sets.newIdentityHashSet();
90    uniqueEntries.addAll(ms.entrySet());
91    assertEquals(3, uniqueEntries.size());
92  }
93}
94