package net.pterodactylus.sone.web.pages import com.google.common.eventbus.* import freenet.clients.http.* import freenet.support.* import freenet.support.api.* import net.pterodactylus.sone.core.* import net.pterodactylus.sone.data.* import net.pterodactylus.sone.freenet.* import net.pterodactylus.sone.freenet.wot.* import net.pterodactylus.sone.main.* import net.pterodactylus.sone.test.deepMock import net.pterodactylus.sone.test.get import net.pterodactylus.sone.test.mock import net.pterodactylus.sone.test.whenever import net.pterodactylus.sone.utils.* import net.pterodactylus.sone.web.* import net.pterodactylus.sone.web.page.* import net.pterodactylus.sone.web.page.FreenetTemplatePage.RedirectException import net.pterodactylus.util.notify.* import net.pterodactylus.util.template.* import net.pterodactylus.util.web.* import net.pterodactylus.util.web.Method.* import org.junit.Assert.* import org.mockito.ArgumentMatchers.* import org.mockito.ArgumentMatchers.eq import java.io.* import java.net.* import java.nio.charset.* import java.util.* import kotlin.text.Charsets.UTF_8 /** * Base class for web page tests. */ open class WebPageTest(pageSupplier: (WebInterface, Loaders, TemplateRenderer) -> SoneTemplatePage = { _, _, _ -> mock() }) { val currentSone = mock() val loaders = mock() val templateRenderer = mock() val webInterface = deepMock() val core = webInterface.core val eventBus = mock() val preferences = Preferences(eventBus) open val page by lazy { pageSupplier(webInterface, loaders, templateRenderer) } val httpRequest = mock() val freenetRequest = mock() init { whenever(freenetRequest.uri).thenReturn(mock()) } val soneRequest by lazy { freenetRequest.toSoneRequest(core, webInterface) } val templateContext = TemplateContext() val toadletContext = deepMock() val responseContent = ByteArrayOutputStream() val response = Response(responseContent) private val requestHeaders = mutableMapOf() private val getRequestParameters = mutableMapOf>() private val postRequestParameters = mutableMapOf() private val uploadedFileNames = mutableMapOf() private val uploadedFileContentTypes = mutableMapOf() private val uploadedFileResources = mutableMapOf() private val ownIdentities = mutableSetOf() private val allSones = mutableMapOf() private val localSones = mutableMapOf() private val allPosts = mutableMapOf() private val allPostReplies = mutableMapOf() private val perPostReplies = mutableMapOf() private val allAlbums = mutableMapOf() private val allImages = mutableMapOf() private val notifications = mutableMapOf() private val translations = mutableMapOf() private val translation = object : Translation { override val currentLocale = Locale.ENGLISH override fun translate(key: String) = translations[key] ?: key } init { setupCore() setupWebInterface() setupHttpRequest() setupFreenetRequest() } private fun setupCore() { whenever(core.preferences).thenReturn(preferences) whenever(core.identityManager.allOwnIdentities).then { ownIdentities } whenever(core.sones).then { allSones.values } whenever(core.getSone(anyString())).then { allSones[it[0]] } whenever(core.localSones).then { localSones.values } whenever(core.getLocalSone(anyString())).then { localSones[it[0]] } whenever(core.getPost(anyString())).then { allPosts[it[0]] } whenever(core.getPostReply(anyString())).then { allPostReplies[it[0]] } whenever(core.getReplies(anyString())).then { perPostReplies[it[0]].asList() } whenever(core.getAlbum(anyString())).then { allAlbums[it[0]] } whenever(core.getImage(anyString())).then { allImages[it[0]] } whenever(core.getImage(anyString(), anyBoolean())).then { allImages[it[0]] } whenever(core.getTemporaryImage(anyString())).thenReturn(null) } private fun setupWebInterface() { whenever(webInterface.getCurrentSone(eq(toadletContext))).thenReturn(currentSone) whenever(webInterface.getNotifications(currentSone)).then { notifications.values } whenever(webInterface.getNotification(anyString())).then { notifications[it[0]].asOptional() } whenever(webInterface.translation).thenReturn(translation) } private fun setupHttpRequest() { whenever(httpRequest.method).thenReturn("GET") whenever(httpRequest.getHeader(anyString())).then { requestHeaders[it.get(0).toLowerCase()] } whenever(httpRequest.hasParameters()).then { getRequestParameters.isNotEmpty() } whenever(httpRequest.parameterNames).then { getRequestParameters.keys } whenever(httpRequest.isParameterSet(anyString())).then { it[0] in getRequestParameters } whenever(httpRequest.getParam(anyString())).then { getRequestParameters[it[0]]?.firstOrNull() ?: "" } whenever(httpRequest.getParam(anyString(), anyString())).then { getRequestParameters[it[0]]?.firstOrNull() ?: it[1] } whenever(httpRequest.getIntParam(anyString())).then { getRequestParameters[it[0]]?.first()?.toIntOrNull() ?: 0 } whenever(httpRequest.getIntParam(anyString(), anyInt())).then { getRequestParameters[it[0]]?.first()?.toIntOrNull() ?: it[1] } whenever(httpRequest.getLongParam(anyString(), anyLong())).then { getRequestParameters[it[0]]?.first()?.toLongOrNull() ?: it[1] } whenever(httpRequest.getMultipleParam(anyString())).then { getRequestParameters[it[0]]?.toTypedArray() ?: emptyArray() } whenever(httpRequest.getMultipleIntParam(anyString())).then { getRequestParameters[it[0]]?.map { it.toIntOrNull() ?: 0 } ?: emptyArray() } whenever(httpRequest.isPartSet(anyString())).then { it[0] in postRequestParameters } whenever(httpRequest.getPartAsStringFailsafe(anyString(), anyInt())).then { postRequestParameters[it[0]]?.decode()?.take(it[1]) ?: "" } whenever(httpRequest.getUploadedFile(anyString())).then { it.get(0).takeIf { it in uploadedFileNames } ?.let { name -> UploadedFile(uploadedFileNames[name]!!, uploadedFileContentTypes[name]!!, uploadedFileResources[name]!!) } } } private class UploadedFile(private val filename: String, private val contentType: String, private val resourceName: String) : HTTPUploadedFile { override fun getFilename() = filename override fun getContentType() = contentType override fun getData() = javaClass.getResourceAsStream(resourceName).readBytes().let(::SimpleReadOnlyArrayBucket) } private fun ByteArray.decode(charset: Charset = UTF_8) = String(this, charset) private fun setupFreenetRequest() { whenever(freenetRequest.method).thenReturn(GET) whenever(freenetRequest.httpRequest).thenReturn(httpRequest) whenever(freenetRequest.toadletContext).thenReturn(toadletContext) } fun setMethod(method: Method) { whenever(httpRequest.method).thenReturn(method.name) whenever(freenetRequest.method).thenReturn(method) } fun request(uri: String) { whenever(httpRequest.path).thenReturn(uri) whenever(freenetRequest.uri).thenReturn(URI(uri)) } fun addHttpRequestHeader(name: String, value: String) { requestHeaders[name.toLowerCase()] = value } fun addHttpRequestParameter(name: String, value: String) { getRequestParameters[name] = getRequestParameters.getOrElse(name) { mutableListOf() }.apply { add(value) } } fun addHttpRequestPart(name: String, value: String) { postRequestParameters[name] = value.toByteArray(UTF_8) } fun unsetCurrentSone() { whenever(webInterface.getCurrentSone(eq(toadletContext))).thenReturn(null) } fun addOwnIdentity(ownIdentity: OwnIdentity) { ownIdentities += ownIdentity } fun addSone(id: String, sone: Sone) { allSones[id] = sone } fun addLocalSone(id: String, localSone: Sone) { localSones[id] = localSone } fun addPost(id: String, post: Post) { allPosts[id] = post } fun addPostReply(id: String, postReply: PostReply) { allPostReplies[id] = postReply postReply.postId?.also { perPostReplies[it] = postReply } } fun addAlbum(id: String, album: Album) { allAlbums[id] = album } fun addImage(id: String, image: Image) { allImages[id] = image } fun addTranslation(key: String, value: String) { translations[key] = value } fun addNotification(id: String, notification: Notification) { notifications[id] = notification } fun addTemporaryImage(id: String, temporaryImage: TemporaryImage) { whenever(core.getTemporaryImage(id)).thenReturn(temporaryImage) } fun addUploadedFile(name: String, filename: String, contentType: String, resource: String) { uploadedFileNames[name] = filename uploadedFileContentTypes[name] = contentType uploadedFileResources[name] = resource } fun verifyNoRedirect(assertions: () -> Unit) { var caughtException: Exception? = null try { page.handleRequest(freenetRequest, templateContext) } catch (e: Exception) { caughtException = e } caughtException?.run { throw this } ?: assertions() } fun verifyRedirect(target: String, assertions: () -> Unit = {}) { try { page.handleRequest(freenetRequest, templateContext) fail() } catch (re: RedirectException) { if (re.target != target) { throw re } assertions() } catch (e: Exception) { throw e } } }