/*
* 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();
}
}