1 package net.pterodactylus.sone.web
3 import net.pterodactylus.sone.core.SoneRescuer
4 import net.pterodactylus.sone.test.mock
5 import net.pterodactylus.sone.test.whenever
6 import net.pterodactylus.util.web.Method.GET
7 import net.pterodactylus.util.web.Method.POST
8 import org.hamcrest.MatcherAssert.assertThat
9 import org.hamcrest.Matchers.equalTo
10 import org.junit.Before
12 import org.mockito.ArgumentMatchers.anyLong
13 import org.mockito.Mockito.never
14 import org.mockito.Mockito.verify
17 * Unit test for [RescuePage].
19 class RescuePageTest : WebPageTest() {
21 private val page = RescuePage(template, webInterface)
23 private val soneRescuer = mock<SoneRescuer>()
25 override fun getPage() = page
28 fun setupSoneRescuer() {
29 whenever(core.getSoneRescuer(currentSone)).thenReturn(soneRescuer)
33 fun `get request sets rescuer in template context`() {
35 page.handleRequest(freenetRequest, templateContext)
36 assertThat(templateContext["soneRescuer"], equalTo<Any>(soneRescuer))
40 fun `post request redirects to rescue page`() {
42 verifyRedirect("rescue.html")
46 fun `post request with fetch and invalid edition starts next fetch`() {
48 addHttpRequestParameter("fetch", "true")
49 verifyRedirect("rescue.html") {
50 verify(soneRescuer, never()).setEdition(anyLong())
51 verify(soneRescuer).startNextFetch()
56 fun `post request with fetch and valid edition sets edition and starts next fetch`() {
58 addHttpRequestParameter("fetch", "true")
59 addHttpRequestParameter("edition", "123")
60 verifyRedirect("rescue.html") {
61 verify(soneRescuer).setEdition(123L)
62 verify(soneRescuer).startNextFetch()