X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=d5c2c08267872ab85f48ed796f490c0c2fc7b240;hb=1be3820cd48951abbcfd5a7f95b82d44aa0e016a;hp=8720312a22943c85837022d5b07523a172aca5c2;hpb=253c9c1552e471513d34ae48c156a367b417b827;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 8720312..d5c2c08 100644
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -84,13 +84,8 @@ public class Mocks {
 		return new SoneMocker(id);
 	}
 
-	public Post mockPost(Sone sone, String postId) {
-		Post post = mock(Post.class);
-		when(post.getId()).thenReturn(postId);
-		when(post.getSone()).thenReturn(sone);
-		when(database.getPost(eq(postId))).thenReturn(of(post));
-		sonePosts.put(sone, post);
-		return post;
+	public PostMocker mockPost(Sone sone, String postId) {
+		return new PostMocker(postId, sone);
 	}
 
 	public PostReply mockPostReply(Sone sone, String replyId) {
@@ -148,4 +143,25 @@ public class Mocks {
 
 	}
 
+	public class PostMocker {
+
+		private final Post post = mock(Post.class);
+		private final String id;
+		private final Sone sone;
+
+		public PostMocker(String id, Sone sone) {
+			this.id = id;
+			this.sone = sone;
+		}
+
+		public Post create() {
+			when(post.getId()).thenReturn(id);
+			when(post.getSone()).thenReturn(sone);
+			when(database.getPost(eq(id))).thenReturn(of(post));
+			sonePosts.put(sone, post);
+			return post;
+		}
+
+	}
+
 }