Add test for delete reply ajax page

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / ajax / JsonPageTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
index afa9b17..cdb68a3 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
@@ -61,6 +61,7 @@ abstract class JsonPageTest(
        private val remoteSones = mutableMapOf<String, Sone>()
        private val posts = mutableMapOf<String, Post>()
        private val newPosts = mutableMapOf<String, Post>()
+       private val replies = mutableMapOf<String, PostReply>()
        private val newReplies = mutableMapOf<String, PostReply>()
        private val linkedElements = mutableMapOf<String, LinkedElement>()
        private val notifications = mutableListOf<Notification>()
@@ -81,6 +82,7 @@ abstract class JsonPageTest(
                whenever(core.getSone(anyString())).thenAnswer { (localSones + remoteSones)[it.getArgument(0)].asOptional() }
                whenever(core.getLocalSone(anyString())).thenAnswer { localSones[it[0]] }
                whenever(core.getPost(anyString())).thenAnswer { (posts + newPosts)[it[0]].asOptional() }
+               whenever(core.getPostReply(anyString())).then { replies[it[0]].asOptional() }
        }
 
        @Before
@@ -176,6 +178,10 @@ abstract class JsonPageTest(
                                whenever(this.recipientId).thenReturn(recipientId.asOptional())
                        }.also { newPosts[id] = it }
 
+       protected fun addReply(id: String, reply: PostReply) {
+               replies[id] = reply
+       }
+
        protected fun addNewReply(id: String, soneId: String, postId: String, postSoneId: String) {
                newReplies[id] = mock<PostReply>().apply {
                        whenever(this.id).thenReturn(id)