Require a Sone when mocking a post.

[Sone.git] / src / test / java / net / pterodactylus / sone / data / Mocks.java

diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java
index a1c1a71..10d40a1 100644 (file)
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -74,9 +74,10 @@ public class Mocks {
                return sone;
        }
 
-       public static Post mockPost(Core core, String postId) {
+       public static Post mockPost(Core core, Sone sone, String postId) {
                Post post = mock(Post.class);
                when(post.getId()).thenReturn(postId);
+               when(post.getSone()).thenReturn(sone);
                Database database = core.getDatabase();
                when(database.getPost(eq(postId))).thenReturn(of(post));
                return post;