src/test/kotlin/net/pterodactylus/sone/web/pages/RescuePageTest.kt

   1 package net.pterodactylus.sone.web.pages
   2
   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.sone.web.pages.RescuePage
   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
  11 import org.junit.Test
  12 import org.mockito.ArgumentMatchers.anyLong
  13 import org.mockito.Mockito.never
  14 import org.mockito.Mockito.verify
  15
  16 /**
  17  * Unit test for [RescuePage].
  18  */
  19 class RescuePageTest : WebPageTest() {
  20
  21         private val page = RescuePage(template, webInterface)
  22
  23         private val soneRescuer = mock<SoneRescuer>()
  24
  25         override fun getPage() = page
  26
  27         @Before
  28         fun setupSoneRescuer() {
  29                 whenever(core.getSoneRescuer(currentSone)).thenReturn(soneRescuer)
  30         }
  31
  32         @Test
  33         fun `get request sets rescuer in template context`() {
  34                 page.handleRequest(freenetRequest, templateContext)
  35                 assertThat(templateContext["soneRescuer"], equalTo<Any>(soneRescuer))
  36         }
  37
  38         @Test
  39         fun `post request redirects to rescue page`() {
  40                 setMethod(POST)
  41                 verifyRedirect("rescue.html")
  42         }
  43
  44         @Test
  45         fun `post request with fetch and invalid edition starts next fetch`() {
  46                 setMethod(POST)
  47                 addHttpRequestPart("fetch", "true")
  48                 verifyRedirect("rescue.html") {
  49                         verify(soneRescuer, never()).setEdition(anyLong())
  50                         verify(soneRescuer).startNextFetch()
  51                 }
  52         }
  53
  54         @Test
  55         fun `post request with fetch and valid edition sets edition and starts next fetch`() {
  56                 setMethod(POST)
  57                 addHttpRequestPart("fetch", "true")
  58                 addHttpRequestPart("edition", "123")
  59                 verifyRedirect("rescue.html") {
  60                         verify(soneRescuer).setEdition(123L)
  61                         verify(soneRescuer).startNextFetch()
  62                 }
  63         }
  64
  65 }