Add test for delete profile field 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 5d43a1e..afa9b17 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/ajax/JsonPageTest.kt
@@ -8,6 +8,7 @@ import net.pterodactylus.sone.core.ElementLoader
 import net.pterodactylus.sone.core.LinkedElement
 import net.pterodactylus.sone.data.Post
 import net.pterodactylus.sone.data.PostReply
 import net.pterodactylus.sone.core.LinkedElement
 import net.pterodactylus.sone.data.Post
 import net.pterodactylus.sone.data.PostReply

+import net.pterodactylus.sone.data.Profile

 import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.data.Sone.SoneStatus
 import net.pterodactylus.sone.data.Sone.SoneStatus.idle
 import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.data.Sone.SoneStatus
 import net.pterodactylus.sone.data.Sone.SoneStatus.idle


@@ -51,6 +52,7 @@ abstract class JsonPageTest(
        protected val freenetRequest = mock<FreenetRequest>()
        protected val httpRequest = mock<HTTPRequest>()
        protected val currentSone = deepMock<Sone>()
        protected val freenetRequest = mock<FreenetRequest>()
        protected val httpRequest = mock<HTTPRequest>()
        protected val currentSone = deepMock<Sone>()

+       protected val profile = Profile(currentSone)

 
        private val requestHeaders = mutableMapOf<String, String>()
        private val requestParameters = mutableMapOf<String, String>()
 
        private val requestHeaders = mutableMapOf<String, String>()
        private val requestParameters = mutableMapOf<String, String>()


@@ -116,6 +118,11 @@ abstract class JsonPageTest(
                whenever(httpRequest.isPartSet(anyString())).thenAnswer { it.getArgument(0) in requestParts }
        }
 
                whenever(httpRequest.isPartSet(anyString())).thenAnswer { it.getArgument(0) in requestParts }
        }
 

+       @Before
+       fun setupProfile() {
+               whenever(currentSone.profile).thenReturn(profile)
+       }
+

        protected val JsonReturnObject.error get() = if (this is JsonErrorReturnObject) this.error else null
 
        protected fun Sone.mock(id: String, name: String, local: Boolean = false, time: Long, status: SoneStatus = idle) = apply {
        protected val JsonReturnObject.error get() = if (this is JsonErrorReturnObject) this.error else null
 
        protected fun Sone.mock(id: String, name: String, local: Boolean = false, time: Long, status: SoneStatus = idle) = apply {