X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Ffreenet%2Fwot%2FIdentityChangeEventSenderTest.java;h=8e1cf6afef7dde6fc4dbf0a477e6144a45e23669;hp=fe4f07042708ee9e57973633e88b7923ddb87c00;hb=52f608f5902d6d05dd39599cdb2ed28f2465b4be;hpb=b2a3147c056a532558c9bfe676431f4597b7f8eb

diff --git a/src/test/java/net/pterodactylus/sone/freenet/wot/IdentityChangeEventSenderTest.java b/src/test/java/net/pterodactylus/sone/freenet/wot/IdentityChangeEventSenderTest.java
index fe4f070..8e1cf6a 100644
--- a/src/test/java/net/pterodactylus/sone/freenet/wot/IdentityChangeEventSenderTest.java
+++ b/src/test/java/net/pterodactylus/sone/freenet/wot/IdentityChangeEventSenderTest.java
@@ -25,8 +25,11 @@ import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.verify;
 
+import java.util.Collection;
 import java.util.Collections;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 
 import net.pterodactylus.sone.freenet.wot.event.IdentityAddedEvent;
 import net.pterodactylus.sone.freenet.wot.event.IdentityRemovedEvent;
@@ -34,8 +37,6 @@ import net.pterodactylus.sone.freenet.wot.event.IdentityUpdatedEvent;
 import net.pterodactylus.sone.freenet.wot.event.OwnIdentityAddedEvent;
 import net.pterodactylus.sone.freenet.wot.event.OwnIdentityRemovedEvent;
 
-import com.google.common.collect.ImmutableMultimap;
-import com.google.common.collect.Multimap;
 import com.google.common.eventbus.EventBus;
 import org.junit.Test;
 
@@ -62,7 +63,7 @@ public class IdentityChangeEventSenderTest {
 
 	@Test
 	public void addingAnOwnIdentityIsDetectedAndReportedCorrectly() {
-		Multimap<OwnIdentity, Identity> newIdentities = createNewIdentities();
+		Map<OwnIdentity, Collection<Identity>> newIdentities = createNewIdentities();
 		identityChangeEventSender.detectChanges(newIdentities);
 		verify(eventBus).post(eq(new OwnIdentityRemovedEvent(ownIdentities.get(0))));
 		verify(eventBus).post(eq(new IdentityRemovedEvent(ownIdentities.get(0), identities.get(0))));
@@ -75,22 +76,18 @@ public class IdentityChangeEventSenderTest {
 		verify(eventBus).post(eq(new IdentityUpdatedEvent(ownIdentities.get(1), identities.get(1))));
 	}
 
-	private Multimap<OwnIdentity, Identity> createNewIdentities() {
-		ImmutableMultimap.Builder<OwnIdentity, Identity> oldIdentities = ImmutableMultimap.builder();
-		oldIdentities.put(ownIdentities.get(1), identities.get(3));
-		oldIdentities.put(ownIdentities.get(1), identities.get(2));
-		oldIdentities.put(ownIdentities.get(2), identities.get(1));
-		oldIdentities.put(ownIdentities.get(2), identities.get(2));
-		return oldIdentities.build();
+	private Map<OwnIdentity, Collection<Identity>> createNewIdentities() {
+		Map<OwnIdentity, Collection<Identity>> oldIdentities = new HashMap<OwnIdentity, Collection<Identity>>();
+		oldIdentities.put(ownIdentities.get(1), asList(identities.get(3), identities.get(2)));
+		oldIdentities.put(ownIdentities.get(2), asList(identities.get(1), identities.get(2)));
+		return oldIdentities;
 	}
 
-	private Multimap<OwnIdentity, Identity> createOldIdentities() {
-		ImmutableMultimap.Builder<OwnIdentity, Identity> oldIdentities = ImmutableMultimap.builder();
-		oldIdentities.put(ownIdentities.get(0), identities.get(0));
-		oldIdentities.put(ownIdentities.get(0), identities.get(1));
-		oldIdentities.put(ownIdentities.get(1), identities.get(0));
-		oldIdentities.put(ownIdentities.get(1), identities.get(1));
-		return oldIdentities.build();
+	private Map<OwnIdentity, Collection<Identity>> createOldIdentities() {
+		Map<OwnIdentity, Collection<Identity>> oldIdentities = new HashMap<OwnIdentity, Collection<Identity>>();
+		oldIdentities.put(ownIdentities.get(0), asList(identities.get(0), identities.get(1)));
+		oldIdentities.put(ownIdentities.get(1), asList(identities.get(0), identities.get(1)));
+		return oldIdentities;
 	}
 
 }