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

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