From 010c6b3ace491d44472d113babea6bfda4732407 Mon Sep 17 00:00:00 2001 From: Grzegorz Piwowarek Date: Mon, 4 Jul 2016 13:30:34 +0200 Subject: [PATCH] Refactor multimap example --- .../assertj/introduction/AssertJGuavaTest.java | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/assertj/src/test/java/com/baeldung/assertj/introduction/AssertJGuavaTest.java b/assertj/src/test/java/com/baeldung/assertj/introduction/AssertJGuavaTest.java index 95faaec3d0..9c51af3ee9 100644 --- a/assertj/src/test/java/com/baeldung/assertj/introduction/AssertJGuavaTest.java +++ b/assertj/src/test/java/com/baeldung/assertj/introduction/AssertJGuavaTest.java @@ -2,16 +2,17 @@ package com.baeldung.assertj.introduction; import com.google.common.base.Optional; import com.google.common.collect.HashBasedTable; -import com.google.common.collect.ImmutableMap; +import com.google.common.collect.Multimap; import com.google.common.collect.Multimaps; import com.google.common.collect.Range; -import com.google.common.collect.SetMultimap; +import com.google.common.collect.Sets; import com.google.common.collect.TreeRangeMap; import com.google.common.io.Files; import org.assertj.guava.data.MapEntry; import org.junit.Test; import java.io.File; +import java.util.HashMap; import static org.assertj.guava.api.Assertions.assertThat; import static org.assertj.guava.api.Assertions.entry; @@ -30,12 +31,15 @@ public class AssertJGuavaTest { @Test public void givenMultimap_whenVerifying_thenCorrect() throws Exception { - final SetMultimap mmap = Multimaps.forMap(ImmutableMap.of(1, "one")); + final Multimap mmap = Multimaps.newMultimap(new HashMap<>(), Sets::newHashSet); + mmap.put(1, "one"); + mmap.put(1, "1"); assertThat(mmap) - .hasSize(1) + .hasSize(2) .containsKeys(1) - .contains(entry(1, "one")); + .contains(entry(1, "one")) + .contains(entry(1, "1")); } @Test