1 package net.pterodactylus.sone.web.pages
3 import com.google.common.eventbus.EventBus
4 import freenet.clients.http.ToadletContext
5 import freenet.support.api.HTTPRequest
6 import net.pterodactylus.sone.core.Preferences
7 import net.pterodactylus.sone.data.Post
8 import net.pterodactylus.sone.data.Sone
9 import net.pterodactylus.sone.test.deepMock
10 import net.pterodactylus.sone.test.get
11 import net.pterodactylus.sone.test.mock
12 import net.pterodactylus.sone.test.whenever
13 import net.pterodactylus.sone.utils.asOptional
14 import net.pterodactylus.sone.web.WebInterface
15 import net.pterodactylus.sone.web.page.FreenetRequest
16 import net.pterodactylus.sone.web.page.FreenetTemplatePage.RedirectException
17 import net.pterodactylus.util.template.Template
18 import net.pterodactylus.util.template.TemplateContext
19 import net.pterodactylus.util.web.Method
20 import net.pterodactylus.util.web.Method.GET
21 import org.junit.Assert.fail
22 import org.junit.Before
23 import org.mockito.ArgumentMatchers.anyBoolean
24 import org.mockito.ArgumentMatchers.anyInt
25 import org.mockito.ArgumentMatchers.anyLong
26 import org.mockito.ArgumentMatchers.anyString
27 import org.mockito.ArgumentMatchers.eq
28 import java.nio.charset.Charset
29 import kotlin.text.Charsets.UTF_8
32 * Base class for web page tests.
34 abstract class WebPageTest2(pageSupplier: (Template, WebInterface) -> SoneTemplatePage) {
36 protected val currentSone = mock<Sone>()
37 private val template = mock<Template>()
38 private val webInterface = deepMock<WebInterface>()
39 protected val core = webInterface.core!!
40 private val eventBus = mock<EventBus>()
41 private val preferences = Preferences(eventBus)
42 private val l10n = webInterface.l10n!!
44 protected val page by lazy { pageSupplier(template, webInterface) }
45 private val httpRequest = mock<HTTPRequest>()
46 protected val freenetRequest = mock<FreenetRequest>()
47 protected val templateContext = TemplateContext()
49 private val toadletContext = deepMock<ToadletContext>()
50 private val getRequestParameters = mutableMapOf<String, MutableList<String>>()
51 private val postRequestParameters = mutableMapOf<String, ByteArray>()
52 private val allSones = mutableMapOf<String, Sone>()
53 private val allPosts = mutableMapOf<String, Post>()
54 private val translations = mutableMapOf<String, String>()
58 whenever(core.preferences).thenReturn(preferences)
59 whenever(core.sones).then { allSones.values }
60 whenever(core.getSone(anyString())).then { allSones[it[0]].asOptional() }
61 whenever(core.getPost(anyString())).then { allPosts[it[0]].asOptional() }
65 fun setupWebInterface() {
66 whenever(webInterface.getCurrentSoneCreatingSession(eq(toadletContext))).thenReturn(currentSone)
67 whenever(webInterface.getCurrentSone(eq(toadletContext), anyBoolean())).thenReturn(currentSone)
68 whenever(webInterface.getCurrentSoneWithoutCreatingSession(eq(toadletContext))).thenReturn(currentSone)
72 fun setupHttpRequest() {
73 whenever(httpRequest.method).thenReturn("GET")
74 whenever(httpRequest.hasParameters()).then { getRequestParameters.isNotEmpty() }
75 whenever(httpRequest.parameterNames).then { getRequestParameters.keys }
76 whenever(httpRequest.isParameterSet(anyString())).then { it[0] in getRequestParameters }
77 whenever(httpRequest.getParam(anyString())).then { getRequestParameters[it[0]]?.firstOrNull() ?: "" }
78 whenever(httpRequest.getParam(anyString(), anyString())).then { getRequestParameters[it[0]]?.firstOrNull() ?: it[1] }
79 whenever(httpRequest.getIntParam(anyString())).then { getRequestParameters[it[0]]?.first()?.toIntOrNull() ?: 0 }
80 whenever(httpRequest.getIntParam(anyString(), anyInt())).then { getRequestParameters[it[0]]?.first()?.toIntOrNull() ?: it[1] }
81 whenever(httpRequest.getLongParam(anyString(), anyLong())).then { getRequestParameters[it[0]]?.first()?.toLongOrNull() ?: it[1] }
82 whenever(httpRequest.getMultipleParam(anyString())).then { getRequestParameters[it[0]]?.toTypedArray() ?: emptyArray<String>() }
83 whenever(httpRequest.getMultipleIntParam(anyString())).then { getRequestParameters[it[0]]?.map { it.toIntOrNull() ?: 0 } ?: emptyArray<Int>() }
84 whenever(httpRequest.getPartAsStringFailsafe(anyString(), anyInt())).then { postRequestParameters[it[0]]?.decode() }
87 private fun ByteArray.decode(charset: Charset = UTF_8) = String(this, charset)
90 fun setupFreenetRequest() {
91 whenever(freenetRequest.method).thenReturn(GET)
92 whenever(freenetRequest.httpRequest).thenReturn(httpRequest)
93 whenever(freenetRequest.toadletContext).thenReturn(toadletContext)
97 fun setupTranslations() {
98 whenever(l10n.getString(anyString())).then { translations[it[0]] ?: it[0] }
101 fun setMethod(method: Method) {
102 whenever(httpRequest.method).thenReturn(method.name)
103 whenever(freenetRequest.method).thenReturn(method)
106 fun addHttpRequestParameter(name: String, value: String) {
107 getRequestParameters[name] = getRequestParameters.getOrElse(name) { mutableListOf<String>() }.apply { add(value) }
110 fun addHttpRequestPart(name: String, value: String) {
111 postRequestParameters[name] = value.toByteArray(UTF_8)
114 fun addSone(id: String, sone: Sone) {
118 fun addPost(id: String, post: Post) {
122 fun addTranslation(key: String, value: String) {
123 translations[key] = value
126 fun verifyNoRedirect(assertions: () -> Unit) {
127 var caughtException: Exception? = null
129 page.handleRequest(freenetRequest, templateContext)
130 } catch (e: Exception) {
133 caughtException?.run { throw this } ?: assertions()
136 fun verifyRedirect(target: String, assertions: () -> Unit) {
138 page.handleRequest(freenetRequest, templateContext)
140 } catch (re: RedirectException) {
141 if (re.target != target) {
145 } catch (e: Exception) {