X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fdata%2FMocks.java;h=845f7c37cc977ec1b57845799e31ec797c248926;hp=51fa239373467cf258e26b6bf1de37325b9910d6;hb=0e12efbf4845abffa937fd4b94caf24b1cfe05e1;hpb=58e71083db32980d249c61fa6ddc7645256576e8

diff --git a/src/test/java/net/pterodactylus/sone/data/Mocks.java b/src/test/java/net/pterodactylus/sone/data/Mocks.java
index 51fa239..845f7c3 100644
--- a/src/test/java/net/pterodactylus/sone/data/Mocks.java
+++ b/src/test/java/net/pterodactylus/sone/data/Mocks.java
@@ -52,6 +52,7 @@ import net.pterodactylus.sone.database.Database;
 import net.pterodactylus.sone.database.PostBuilder.PostCreated;
 import net.pterodactylus.sone.database.PostReplyBuilder;
 import net.pterodactylus.sone.fcp.FcpInterface.FullAccessRequired;
+import net.pterodactylus.sone.database.PostReplyBuilder.PostReplyCreated;
 import net.pterodactylus.sone.utils.IntegerRangePredicate;
 import net.pterodactylus.sone.web.WebInterface;
 import net.pterodactylus.sone.web.page.FreenetRequest;
@@ -136,6 +137,17 @@ public class Mocks {
 				sonePosts.put(post.getSone(), post);
 			}
 		}));
+		when(core.postReplyCreated()).then(new Answer<Optional<PostReplyCreated>>() {
+			@Override
+			public Optional<PostReplyCreated> answer(InvocationOnMock invocation) throws Throwable {
+				return Optional.<PostReplyCreated>of(new PostReplyCreated() {
+					@Override
+					public void postReplyCreated(PostReply postReply) {
+						postReplies.put(postReply.getPost().get(), postReply);
+					}
+				});
+			}
+		});
 		Options options = createOptions();
 		when(core.getPreferences()).thenReturn(new Preferences(options));
 		when(database.getDirectedPosts(anyString())).then(new Answer<Collection<Post>>() {