11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2007 The Guava Authors 31d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 41d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Licensed under the Apache License, Version 2.0 (the "License"); 51d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * you may not use this file except in compliance with the License. 61d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * You may obtain a copy of the License at 71d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 81d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * http://www.apache.org/licenses/LICENSE-2.0 91d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Unless required by applicable law or agreed to in writing, software 111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * distributed under the License is distributed on an "AS IS" BASIS, 121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * See the License for the specific language governing permissions and 141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * limitations under the License. 151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpackage com.google.common.collect; 181d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 191d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport static java.util.Arrays.asList; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.annotations.GwtCompatible; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.annotations.GwtIncompatible; 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.testing.SerializableTester; 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport junit.framework.TestCase; 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Collection; 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.EnumSet; 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Set; 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/** 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Tests for an {@link EnumMultiset}. 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Jared Levy 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */ 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert@GwtCompatible(emulated = true) 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class EnumMultisetTest extends TestCase { 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert private static enum Color { 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert BLUE, RED, YELLOW, GREEN, WHITE 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testClassCreate() { 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Multiset<Color> ms = EnumMultiset.create(Color.class); 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.RED); 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.YELLOW); 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.RED); 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(0, ms.count(Color.BLUE)); 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(1, ms.count(Color.YELLOW)); 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, ms.count(Color.RED)); 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testCollectionCreate() { 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Multiset<Color> ms = EnumMultiset.create( 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert asList(Color.RED, Color.YELLOW, Color.RED)); 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(0, ms.count(Color.BLUE)); 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(1, ms.count(Color.YELLOW)); 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(2, ms.count(Color.RED)); 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testIllegalCreate() { 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Collection<Color> empty = EnumSet.noneOf(Color.class); 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert try { 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert EnumMultiset.create(empty); 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert fail(); 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } catch (IllegalArgumentException expected) {} 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testToString() { 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Multiset<Color> ms = EnumMultiset.create(Color.class); 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.BLUE, 3); 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.YELLOW, 1); 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.RED, 2); 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals("[BLUE x 3, RED x 2, YELLOW]", ms.toString()); 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert @GwtIncompatible("SerializableTester") 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testSerializable() { 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Multiset<Color> ms = EnumMultiset.create( 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert asList(Color.RED, Color.YELLOW, Color.RED)); 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(ms, SerializableTester.reserialize(ms)); 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testEntrySet() { 841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Multiset<Color> ms = EnumMultiset.create(Color.class); 851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.BLUE, 3); 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.YELLOW, 1); 871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert ms.add(Color.RED, 2); 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Set<Object> uniqueEntries = Sets.newIdentityHashSet(); 901d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert uniqueEntries.addAll(ms.entrySet()); 911d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(3, uniqueEntries.size()); 921d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 931d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 94