1 package net.pterodactylus.sone.web.pages
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.POST
7 import org.hamcrest.MatcherAssert.assertThat
8 import org.hamcrest.Matchers.equalTo
9 import org.junit.Before
11 import org.mockito.ArgumentMatchers.anyLong
12 import org.mockito.Mockito.never
13 import org.mockito.Mockito.verify
16 * Unit test for [RescuePage].
18 class RescuePageTest : WebPageTest() {
20 private val page = RescuePage(template, webInterface)
22 private val soneRescuer = mock<SoneRescuer>()
24 override fun getPage() = page
27 fun setupSoneRescuer() {
28 whenever(core.getSoneRescuer(currentSone)).thenReturn(soneRescuer)
32 fun `get request sets rescuer in template context`() {
33 page.handleRequest(freenetRequest, templateContext)
34 assertThat(templateContext["soneRescuer"], equalTo<Any>(soneRescuer))
38 fun `post request redirects to rescue page`() {
40 verifyRedirect("rescue.html")
44 fun `post request with fetch and invalid edition starts next fetch`() {
46 addHttpRequestPart("fetch", "true")
47 verifyRedirect("rescue.html") {
48 verify(soneRescuer, never()).setEdition(anyLong())
49 verify(soneRescuer).startNextFetch()
54 fun `post request with fetch and valid edition sets edition and starts next fetch`() {
56 addHttpRequestPart("fetch", "true")
57 addHttpRequestPart("edition", "123")
58 verifyRedirect("rescue.html") {
59 verify(soneRescuer).setEdition(123L)
60 verify(soneRescuer).startNextFetch()