X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=10d40a194b1d61ee46e80ae2440d81d895088751;hb=0c6789e2989672a6d08ed2c2184e978d7b1a266a;hp=a1c1a71246a07b1ca7a475a7995c91b21aec80ed;hpb=420547fb85667091ab92cc99c8e82e868239eca3;p=Sone.git

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
--- 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;