| /* |
| * Copyright (C) 2008 The Guava Authors |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| package com.google.common.collect; |
| |
| import com.google.common.annotations.GwtCompatible; |
| import com.google.common.collect.ImmutableMultimap.Builder; |
| import com.google.common.collect.testing.SampleElements; |
| import com.google.common.collect.testing.SampleElements.Unhashables; |
| import com.google.common.collect.testing.UnhashableObject; |
| import com.google.common.testing.EqualsTester; |
| |
| import junit.framework.TestCase; |
| |
| import java.util.Arrays; |
| import java.util.Map.Entry; |
| |
| /** |
| * Tests for {@link ImmutableMultimap}. |
| * |
| * @author Jared Levy |
| */ |
| @GwtCompatible(emulated = true) |
| public class ImmutableMultimapTest extends TestCase { |
| |
| public void testBuilder_withImmutableEntry() { |
| ImmutableMultimap<String, Integer> multimap = new Builder<String, Integer>() |
| .put(Maps.immutableEntry("one", 1)) |
| .build(); |
| assertEquals(Arrays.asList(1), multimap.get("one")); |
| } |
| |
| public void testBuilder_withImmutableEntryAndNullContents() { |
| Builder<String, Integer> builder = new Builder<String, Integer>(); |
| try { |
| builder.put(Maps.immutableEntry("one", (Integer) null)); |
| fail(); |
| } catch (NullPointerException expected) { |
| } |
| try { |
| builder.put(Maps.immutableEntry((String) null, 1)); |
| fail(); |
| } catch (NullPointerException expected) { |
| } |
| } |
| |
| private static class StringHolder { |
| String string; |
| } |
| |
| public void testBuilder_withMutableEntry() { |
| ImmutableMultimap.Builder<String, Integer> builder = |
| new Builder<String, Integer>(); |
| final StringHolder holder = new StringHolder(); |
| holder.string = "one"; |
| Entry<String, Integer> entry = new AbstractMapEntry<String, Integer>() { |
| @Override public String getKey() { |
| return holder.string; |
| } |
| @Override public Integer getValue() { |
| return 1; |
| } |
| }; |
| |
| builder.put(entry); |
| holder.string = "two"; |
| assertEquals(Arrays.asList(1), builder.build().get("one")); |
| } |
| |
| // TODO: test ImmutableMultimap builder and factory methods |
| |
| public void testCopyOf() { |
| ImmutableSetMultimap<String, String> setMultimap |
| = ImmutableSetMultimap.of("k1", "v1"); |
| ImmutableMultimap<String, String> setMultimapCopy |
| = ImmutableMultimap.copyOf(setMultimap); |
| assertSame("copyOf(ImmutableSetMultimap) should not create a new instance", |
| setMultimap, setMultimapCopy); |
| |
| ImmutableListMultimap<String, String> listMultimap |
| = ImmutableListMultimap.of("k1", "v1"); |
| ImmutableMultimap<String, String> listMultimapCopy |
| = ImmutableMultimap.copyOf(listMultimap); |
| assertSame("copyOf(ImmutableListMultimap) should not create a new instance", |
| listMultimap, listMultimapCopy); |
| } |
| |
| public void testUnhashableSingletonValue() { |
| SampleElements<UnhashableObject> unhashables = new Unhashables(); |
| Multimap<Integer, UnhashableObject> multimap = ImmutableMultimap.of( |
| 0, unhashables.e0); |
| assertEquals(1, multimap.get(0).size()); |
| assertTrue(multimap.get(0).contains(unhashables.e0)); |
| } |
| |
| public void testUnhashableMixedValues() { |
| SampleElements<UnhashableObject> unhashables = new Unhashables(); |
| Multimap<Integer, Object> multimap = ImmutableMultimap.<Integer, Object>of( |
| 0, unhashables.e0, 2, "hey you", 0, unhashables.e1); |
| assertEquals(2, multimap.get(0).size()); |
| assertTrue(multimap.get(0).contains(unhashables.e0)); |
| assertTrue(multimap.get(0).contains(unhashables.e1)); |
| assertTrue(multimap.get(2).contains("hey you")); |
| } |
| |
| public void testEquals() { |
| new EqualsTester() |
| .addEqualityGroup(ImmutableMultimap.of(), ImmutableMultimap.of()) |
| .addEqualityGroup(ImmutableMultimap.of(1, "a"), ImmutableMultimap.of(1, "a")) |
| .addEqualityGroup( |
| ImmutableMultimap.of(1, "a", 2, "b"), |
| ImmutableMultimap.of(2, "b", 1, "a")) |
| .testEquals(); |
| } |
| } |
| |