11d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/*
21d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Copyright (C) 2008 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 Bringertimport com.google.common.collect.ImmutableMultimap.Builder;
211d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.SampleElements;
221d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.SampleElements.Unhashables;
231d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport com.google.common.collect.testing.UnhashableObject;
241d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
251d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport junit.framework.TestCase;
261d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
271d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Arrays;
281d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertimport java.util.Map.Entry;
291d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
301d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert/**
311d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * Tests for {@link ImmutableMultimap}.
321d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert *
331d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert * @author Jared Levy
341d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert */
351d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert@GwtCompatible
361d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringertpublic class ImmutableMultimapTest extends TestCase {
371d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
381d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testBuilder_withImmutableEntry() {
391d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableMultimap<String, Integer> multimap = new Builder<String, Integer>()
401d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        .put(Maps.immutableEntry("one", 1))
411d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        .build();
421d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertEquals(Arrays.asList(1), multimap.get("one"));
431d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
441d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
451d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testBuilder_withImmutableEntryAndNullContents() {
461d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    Builder<String, Integer> builder = new Builder<String, Integer>();
471d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    try {
481d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      builder.put(Maps.immutableEntry("one", (Integer) null));
491d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      fail();
501d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    } catch (NullPointerException expected) {
511d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    }
521d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    try {
531d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      builder.put(Maps.immutableEntry((String) null, 1));
541d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      fail();
551d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    } catch (NullPointerException expected) {
561d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    }
571d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
581d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
591d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  private static class StringHolder {
601d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    String string;
611d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
621d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
631d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testBuilder_withMutableEntry() {
641d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableMultimap.Builder<String, Integer> builder =
651d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        new Builder<String, Integer>();
661d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    final StringHolder holder = new StringHolder();
671d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    holder.string = "one";
681d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    Entry<String, Integer> entry = new AbstractMapEntry<String, Integer>() {
691d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      @Override public String getKey() {
701d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        return holder.string;
711d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      }
721d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      @Override public Integer getValue() {
731d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        return 1;
741d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert      }
751d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    };
761d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
771d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    builder.put(entry);
781d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    holder.string = "two";
791d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertEquals(Arrays.asList(1), builder.build().get("one"));
801d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
811d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
821d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  // TODO: test ImmutableMultimap builder and factory methods
831d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
841d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testCopyOf() {
851d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableSetMultimap<String, String> setMultimap
861d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        = ImmutableSetMultimap.of("k1", "v1");
871d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableMultimap<String, String> setMultimapCopy
881d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        = ImmutableMultimap.copyOf(setMultimap);
891d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertSame("copyOf(ImmutableSetMultimap) should not create a new instance",
901d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        setMultimap, setMultimapCopy);
911d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
921d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableListMultimap<String, String> listMultimap
931d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        = ImmutableListMultimap.of("k1", "v1");
941d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    ImmutableMultimap<String, String> listMultimapCopy
951d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        = ImmutableMultimap.copyOf(listMultimap);
961d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertSame("copyOf(ImmutableListMultimap) should not create a new instance",
971d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        listMultimap, listMultimapCopy);
981d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
991d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
1001d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testUnhashableSingletonValue() {
1011d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    SampleElements<UnhashableObject> unhashables = new Unhashables();
1021d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    Multimap<Integer, UnhashableObject> multimap = ImmutableMultimap.of(
1031d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        0, unhashables.e0);
1041d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertEquals(1, multimap.get(0).size());
1051d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertTrue(multimap.get(0).contains(unhashables.e0));
1061d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
1071d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert
1081d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  public void testUnhashableMixedValues() {
1091d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    SampleElements<UnhashableObject> unhashables = new Unhashables();
1101d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    Multimap<Integer, Object> multimap = ImmutableMultimap.<Integer, Object>of(
1111d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert        0, unhashables.e0, 2, "hey you", 0, unhashables.e1);
1121d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertEquals(2, multimap.get(0).size());
1131d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertTrue(multimap.get(0).contains(unhashables.e0));
1141d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertTrue(multimap.get(0).contains(unhashables.e1));
1151d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert    assertTrue(multimap.get(2).contains("hey you"));
1161d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert  }
1171d580d0f6ee4f21eb309ba7b509d2c6d671c4044Bjorn Bringert}
118