1/*
2 * Copyright (C) 2008 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 com.google.common.annotations.GwtCompatible;
20import com.google.common.collect.ImmutableMultimap.Builder;
21import com.google.common.collect.testing.SampleElements;
22import com.google.common.collect.testing.SampleElements.Unhashables;
23import com.google.common.collect.testing.UnhashableObject;
24import com.google.common.testing.EqualsTester;
25
26import junit.framework.TestCase;
27
28import java.util.Arrays;
29import java.util.Map.Entry;
30
31/**
32 * Tests for {@link ImmutableMultimap}.
33 *
34 * @author Jared Levy
35 */
36@GwtCompatible(emulated = true)
37public class ImmutableMultimapTest extends TestCase {
38
39  public void testBuilder_withImmutableEntry() {
40    ImmutableMultimap<String, Integer> multimap = new Builder<String, Integer>()
41        .put(Maps.immutableEntry("one", 1))
42        .build();
43    assertEquals(Arrays.asList(1), multimap.get("one"));
44  }
45
46  public void testBuilder_withImmutableEntryAndNullContents() {
47    Builder<String, Integer> builder = new Builder<String, Integer>();
48    try {
49      builder.put(Maps.immutableEntry("one", (Integer) null));
50      fail();
51    } catch (NullPointerException expected) {
52    }
53    try {
54      builder.put(Maps.immutableEntry((String) null, 1));
55      fail();
56    } catch (NullPointerException expected) {
57    }
58  }
59
60  private static class StringHolder {
61    String string;
62  }
63
64  public void testBuilder_withMutableEntry() {
65    ImmutableMultimap.Builder<String, Integer> builder =
66        new Builder<String, Integer>();
67    final StringHolder holder = new StringHolder();
68    holder.string = "one";
69    Entry<String, Integer> entry = new AbstractMapEntry<String, Integer>() {
70      @Override public String getKey() {
71        return holder.string;
72      }
73      @Override public Integer getValue() {
74        return 1;
75      }
76    };
77
78    builder.put(entry);
79    holder.string = "two";
80    assertEquals(Arrays.asList(1), builder.build().get("one"));
81  }
82
83  // TODO: test ImmutableMultimap builder and factory methods
84
85  public void testCopyOf() {
86    ImmutableSetMultimap<String, String> setMultimap
87        = ImmutableSetMultimap.of("k1", "v1");
88    ImmutableMultimap<String, String> setMultimapCopy
89        = ImmutableMultimap.copyOf(setMultimap);
90    assertSame("copyOf(ImmutableSetMultimap) should not create a new instance",
91        setMultimap, setMultimapCopy);
92
93    ImmutableListMultimap<String, String> listMultimap
94        = ImmutableListMultimap.of("k1", "v1");
95    ImmutableMultimap<String, String> listMultimapCopy
96        = ImmutableMultimap.copyOf(listMultimap);
97    assertSame("copyOf(ImmutableListMultimap) should not create a new instance",
98        listMultimap, listMultimapCopy);
99  }
100
101  public void testUnhashableSingletonValue() {
102    SampleElements<UnhashableObject> unhashables = new Unhashables();
103    Multimap<Integer, UnhashableObject> multimap = ImmutableMultimap.of(
104        0, unhashables.e0);
105    assertEquals(1, multimap.get(0).size());
106    assertTrue(multimap.get(0).contains(unhashables.e0));
107  }
108
109  public void testUnhashableMixedValues() {
110    SampleElements<UnhashableObject> unhashables = new Unhashables();
111    Multimap<Integer, Object> multimap = ImmutableMultimap.<Integer, Object>of(
112        0, unhashables.e0, 2, "hey you", 0, unhashables.e1);
113    assertEquals(2, multimap.get(0).size());
114    assertTrue(multimap.get(0).contains(unhashables.e0));
115    assertTrue(multimap.get(0).contains(unhashables.e1));
116    assertTrue(multimap.get(2).contains("hey you"));
117  }
118
119  public void testEquals() {
120    new EqualsTester()
121        .addEqualityGroup(ImmutableMultimap.of(), ImmutableMultimap.of())
122        .addEqualityGroup(ImmutableMultimap.of(1, "a"), ImmutableMultimap.of(1, "a"))
123        .addEqualityGroup(
124            ImmutableMultimap.of(1, "a", 2, "b"),
125            ImmutableMultimap.of(2, "b", 1, "a"))
126        .testEquals();
127  }
128}
129