From 36f8de0f94fcf2d69baf6924f2b4c8952f7593e8 Mon Sep 17 00:00:00 2001 From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?= Date: Sun, 27 Oct 2013 22:59:00 +0100 Subject: [PATCH] Add method to mock a remote Sone. --- src/test/java/net/pterodactylus/sone/data/Mocks.java | 13 +++++++++++++ 1 file changed, 13 insertions(+) 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); -- 2.7.4