X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;fp=src%2Ftest%2Fjava%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FWebPageTest.java;h=b38121f9ef8a1cc1bac64bdbb11fa563db0dd05a;hb=fa90abc436a8f44118b601843b7f5a801c29ce79;hp=16610e47420c0b1b563b05dd77e27587182b8dd7;hpb=9824d45ada252d673a259029052f5ac4eecc5420;p=Sone.git

diff --git a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java
index 16610e4..b38121f 100644
--- a/src/test/java/net/pterodactylus/sone/web/WebPageTest.java
+++ b/src/test/java/net/pterodactylus/sone/web/WebPageTest.java
@@ -30,6 +30,7 @@ import net.pterodactylus.sone.core.UpdateChecker;
 import net.pterodactylus.sone.data.Album;
 import net.pterodactylus.sone.data.Image;
 import net.pterodactylus.sone.data.Post;
+import net.pterodactylus.sone.data.PostReply;
 import net.pterodactylus.sone.data.Sone;
 import net.pterodactylus.sone.data.SoneOptions.DefaultSoneOptions;
 import net.pterodactylus.sone.data.TemporaryImage;
@@ -210,6 +211,10 @@ public abstract class WebPageTest {
 		when(core.getPost(postId)).thenReturn(Optional.fromNullable(post));
 	}
 
+	protected void addPostReply(String postReplyId, PostReply postReply) {
+		when(core.getPostReply(postReplyId)).thenReturn(Optional.fromNullable(postReply));
+	}
+
 	protected void addSone(String soneId, Sone sone) {
 		sones.put(soneId, sone);
 	}