11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/* 21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2011 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 com.google.common.annotations.GwtCompatible; 201d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport junit.framework.TestCase; 221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.EnumMap; 241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Map; 251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Set; 261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert@GwtCompatible 281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class WellBehavedMapTest extends TestCase { 291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert enum Foo { 301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert X, Y, Z, T 311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testEntrySet_contain() { 341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert WellBehavedMap<Foo, Integer> map = WellBehavedMap.wrap( 351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new EnumMap<Foo, Integer>(Foo.class)); 361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert map.putAll(ImmutableMap.of(Foo.X, 1, Foo.Y, 2, Foo.Z, 3)); 371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // testing with the exact entry 391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(map.entrySet().contains(Maps.immutableEntry(Foo.X, 1))); 401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(map.entrySet().contains(Maps.immutableEntry(Foo.Y, new Integer(2)))); 411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // testing an entry with a contained key, but not the same value 431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(map.entrySet().contains(Maps.immutableEntry(Foo.X, 5))); 441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // testing a non-existent key 461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(map.entrySet().contains(Maps.immutableEntry(Foo.T, 0))); 471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testEntry_setValue() { 501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert WellBehavedMap<Foo, Integer> map = WellBehavedMap.wrap( 511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new EnumMap<Foo, Integer>(Foo.class)); 521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert map.putAll(ImmutableMap.of(Foo.X, 1, Foo.Y, 2, Foo.Z, 3)); 531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert for (Map.Entry<Foo, Integer> entry : map.entrySet()) { 551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert entry.setValue(entry.getValue() + 5); 561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(ImmutableMap.of(Foo.X, 6, Foo.Y, 7, Foo.Z, 8), map); 591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testEntriesAreMutableAndConsistent() { 621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert WellBehavedMap<Foo, Integer> map = WellBehavedMap.wrap( 631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new EnumMap<Foo, Integer>(Foo.class)); 641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert map.putAll(ImmutableMap.of(Foo.X, 1)); 651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Map.Entry<Foo, Integer> entry1 = Iterables.getOnlyElement(map.entrySet()); 671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Map.Entry<Foo, Integer> entry2 = Iterables.getOnlyElement(map.entrySet()); 681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // the entries are constructed and forgotten, thus different 701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertNotSame(entry1, entry2); 711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Set<Map.Entry<Foo, Integer>> entrySet = map.entrySet(); 731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(entrySet.contains(entry1)); 751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(entrySet.contains(entry2)); 761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // mutating entry 781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert entry1.setValue(2); 791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // entry2 is also modified 811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertEquals(entry1.getValue(), entry2.getValue()); 821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // and both are still contained in the set 841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(entrySet.contains(entry1)); 851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(entrySet.contains(entry2)); 861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert public void testEntrySet_remove() { 891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert WellBehavedMap<Foo, Integer> map = WellBehavedMap.wrap( 901d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert new EnumMap<Foo, Integer>(Foo.class)); 911d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert map.putAll(ImmutableMap.of(Foo.X, 1, Foo.Y, 2, Foo.Z, 3)); 921d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Set<Map.Entry<Foo, Integer>> entrySet = map.entrySet(); 931d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 941d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // removing an existing entry, verifying consistency 951d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert Map.Entry<Foo, Integer> entry = Maps.immutableEntry(Foo.Y, 2); 961d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertTrue(entrySet.remove(entry)); 971d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(map.containsKey(Foo.Y)); 981d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertNull(map.get(Foo.Y)); 991d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(entrySet.contains(entry)); 1001d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1011d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // we didn't have that entry, not removed 1021d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(entrySet.remove(Maps.immutableEntry(Foo.T, 4))); 1031d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert 1041d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert // we didn't have that entry, only <Z, 3>, must not remove 1051d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert assertFalse(entrySet.remove(Maps.immutableEntry(Foo.Z, 5))); 1061d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert } 1071d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert} 108