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

   1 package net.pterodactylus.sone.web.pages
   2
   3 import net.pterodactylus.sone.data.Sone
   4 import net.pterodactylus.sone.test.mock
   5 import net.pterodactylus.sone.web.pages.UntrustPage
   6 import net.pterodactylus.util.web.Method.POST
   7 import org.junit.Test
   8 import org.mockito.ArgumentMatchers.any
   9 import org.mockito.ArgumentMatchers.eq
  10 import org.mockito.Mockito.never
  11 import org.mockito.Mockito.verify
  12
  13 /**
  14  * Unit test for [UntrustPage].
  15  */
  16 class UntrustPageTest : WebPageTest() {
  17
  18         private val page = UntrustPage(template, webInterface)
  19
  20         override fun getPage() = page
  21
  22         @Test
  23         fun `get request does not redirect`() {
  24                 page.handleRequest(freenetRequest, templateContext)
  25                 verify(core, never()).untrustSone(eq(currentSone), any())
  26         }
  27
  28         @Test
  29         fun `post request without sone parameter does not untrust but redirects`() {
  30                 setMethod(POST)
  31                 addHttpRequestPart("returnPage", "return.html")
  32                 verifyRedirect("return.html") {
  33                         verify(core, never()).untrustSone(eq(currentSone), any())
  34                 }
  35         }
  36
  37         @Test
  38         fun `post request with invalid sone parameter does not untrust but redirects`() {
  39                 setMethod(POST)
  40                 addHttpRequestPart("returnPage", "return.html")
  41                 addHttpRequestPart("sone", "no-sone")
  42                 verifyRedirect("return.html") {
  43                         verify(core, never()).untrustSone(eq(currentSone), any())
  44                 }
  45         }
  46
  47         @Test
  48         fun `post request with valid sone parameter untrusts and redirects`() {
  49                 setMethod(POST)
  50                 addHttpRequestPart("returnPage", "return.html")
  51                 addHttpRequestPart("sone", "sone-id")
  52                 val sone = mock<Sone>()
  53                 addSone("sone-id", sone)
  54                 verifyRedirect("return.html") {
  55                         verify(core).untrustSone(currentSone, sone)
  56                 }
  57         }
  58
  59 }