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