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

   1 package net.pterodactylus.sone.web.pages
   2
   3 import net.pterodactylus.sone.data.Album
   4 import net.pterodactylus.sone.data.Image
   5 import net.pterodactylus.sone.data.Sone
   6 import net.pterodactylus.sone.test.mock
   7 import net.pterodactylus.sone.test.whenever
   8 import net.pterodactylus.sone.web.pages.WebPageTest
   9 import net.pterodactylus.sone.web.pages.DeleteImagePage
  10 import net.pterodactylus.util.web.Method.GET
  11 import net.pterodactylus.util.web.Method.POST
  12 import org.hamcrest.MatcherAssert.assertThat
  13 import org.hamcrest.Matchers.equalTo
  14 import org.junit.Before
  15 import org.junit.Test
  16 import org.mockito.Mockito.verify
  17
  18 /**
  19  * Unit test for [DeleteImagePage].
  20  */
  21 class DeleteImagePageTest: WebPageTest() {
  22
  23         private val page = DeleteImagePage(template, webInterface)
  24         private val image = mock<Image>()
  25         private val sone = mock<Sone>()
  26
  27         override fun getPage() = page
  28
  29         @Before
  30         fun setupImage() {
  31                 val album = mock<Album>()
  32                 whenever(album.id).thenReturn("album-id")
  33                 whenever(image.id).thenReturn("image-id")
  34                 whenever(image.sone).thenReturn(sone)
  35                 whenever(image.album).thenReturn(album)
  36                 whenever(sone.isLocal).thenReturn(true)
  37         }
  38
  39         @Test
  40         fun `page returns correct path`() {
  41                 assertThat(page.path, equalTo("deleteImage.html"))
  42         }
  43
  44         @Test
  45         fun `page requires login`() {
  46                 assertThat(page.requiresLogin(), equalTo(true))
  47         }
  48
  49         @Test
  50         fun `get request with invalid image redirects to invalid page`() {
  51                 request("", GET)
  52                 verifyRedirect("invalid.html")
  53         }
  54
  55         @Test
  56         fun `get request with image from non-local sone redirects to no permissions page`() {
  57                 request("", GET)
  58                 whenever(sone.isLocal).thenReturn(false)
  59                 addImage("image-id", image)
  60                 addHttpRequestParameter("image", "image-id")
  61                 verifyRedirect("noPermission.html")
  62         }
  63
  64         @Test
  65         fun `get request with image from local sone sets image in template context`() {
  66                 request("", GET)
  67                 addImage("image-id", image)
  68                 addHttpRequestParameter("image", "image-id")
  69                 page.processTemplate(freenetRequest, templateContext)
  70                 assertThat(templateContext["image"], equalTo<Any>(image))
  71         }
  72
  73         @Test
  74         fun `post request with abort delete flag set redirects to image browser`() {
  75                 request("", POST)
  76                 addImage("image-id", image)
  77                 addHttpRequestParameter("image", "image-id")
  78                 addHttpRequestParameter("abortDelete", "true")
  79                 verifyRedirect("imageBrowser.html?image=image-id")
  80         }
  81
  82         @Test
  83         fun `post request deletes image and redirects to image browser`() {
  84                 request("", POST)
  85                 addImage("image-id", image)
  86                 addHttpRequestParameter("image", "image-id")
  87                 verifyRedirect("imageBrowser.html?album=album-id") {
  88                         verify(webInterface.core).deleteImage(image)
  89                 }
  90         }
  91
  92 }