From 55e6a8bf25eacc0006293870d3b38a7fd4af35ec Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Tue, 29 Oct 2013 07:09:32 +0100 Subject: [PATCH] Memorize all mocked Sones and implement getLocalSones(). --- src/test/java/net/pterodactylus/sone/data/Mocks.java | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java index d67a816..00c1522 100644 --- a/src/test/java/net/pterodactylus/sone/data/Mocks.java +++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java @@ -20,12 +20,14 @@ package net.pterodactylus.sone.data; import static com.google.common.base.Optional.of; import static com.google.common.collect.ArrayListMultimap.create; import static com.google.common.collect.Ordering.from; +import static com.google.common.collect.Sets.newHashSet; import static net.pterodactylus.sone.data.Post.TIME_COMPARATOR; import static org.mockito.Matchers.anyString; import static org.mockito.Matchers.eq; import static org.mockito.Mockito.mock; import static org.mockito.Mockito.when; +import java.util.Collection; import java.util.List; import net.pterodactylus.sone.core.Core; @@ -35,6 +37,7 @@ import net.pterodactylus.sone.database.Database; import net.pterodactylus.sone.database.PostReplyBuilder; import com.google.common.base.Optional; +import com.google.common.collect.FluentIterable; import com.google.common.collect.Multimap; import org.mockito.Matchers; import org.mockito.invocation.InvocationOnMock; @@ -48,12 +51,19 @@ import org.mockito.stubbing.Answer; public class Mocks { private final Multimap sonePosts = create(); + private final Collection sones = newHashSet(); public final Database database; public final Core core; public Mocks() { database = mockDatabase(); core = mockCore(database); + when(core.getLocalSones()).then(new Answer>() { + @Override + public Collection answer(InvocationOnMock invocation) throws Throwable { + return FluentIterable.from(sones).filter(Sone.LOCAL_SONE_FILTER).toList(); + } + }); } private static Core mockCore(Database database) { @@ -106,6 +116,7 @@ public class Mocks { } }); when(sone.toString()).thenReturn(String.format("Sone[%s]", id)); + sones.add(sone); } public Post mockPost(Sone sone, String postId) { -- 2.7.4