X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FJsonPageTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fajax%2FJsonPageTest.kt;h=afa9b17039e24df775301ff3d92e5e6952bfedf1;hp=5d43a1edb71d532f447e146332d2b5bf7bfb747c;hb=4588d890e2d7eae8f9ca727a4435ad388c172615;hpb=6cac5513ec98c44bc574a87111bec2a2524464bd

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
--- 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.data.Profile
 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 profile = Profile(currentSone)
 
 	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 }
 	}
 
+	@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 {