From: David ‘Bombe’ Roden Date: Sun, 27 Oct 2013 21:59:00 +0000 (+0100) Subject: Add method to mock a remote Sone. X-Git-Url: https://git.pterodactylus.net/?a=commitdiff_plain;h=36f8de0f94fcf2d69baf6924f2b4c8952f7593e8;p=Sone.git Add method to mock a remote Sone. --- diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java index 10d40a1..ba029b4 100644 --- a/src/test/java/net/pterodactylus/sone/data/Mocks.java +++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java @@ -32,6 +32,7 @@ import net.pterodactylus.sone.database.PostReplyBuilder; import com.google.common.base.Optional; import org.mockito.ArgumentCaptor; +import org.mockito.Matchers; import org.mockito.invocation.InvocationOnMock; import org.mockito.stubbing.Answer; @@ -74,6 +75,18 @@ public class Mocks { return sone; } + public static Sone mockRemoteSone(Core core, final String id) { + Sone sone = mock(Sone.class); + when(sone.getId()).thenReturn(id); + when(sone.isLocal()).thenReturn(false); + final Database database = core.getDatabase(); + when(sone.newPostBuilder()).thenReturn(new DefaultPostBuilder(database, id)); + when(sone.newPostReplyBuilder(Matchers.anyObject())).thenThrow(IllegalStateException.class); + when(core.getSone(eq(id))).thenReturn(of(sone)); + when(database.getSone(eq(id))).thenReturn(of(sone)); + return sone; + } + public static Post mockPost(Core core, Sone sone, String postId) { Post post = mock(Post.class); when(post.getId()).thenReturn(postId);