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 `page returns correct path`() {
  39                 assertThat(page.path, equalTo("deleteImage.html"))
  40         }
  41
  42         @Test
  43         fun `page requires login`() {
  44                 assertThat(page.requiresLogin(), equalTo(true))
  45         }
  46
  47         @Test
  48         fun `get request with invalid image redirects to invalid page`() {
  49                 request("", GET)
  50                 verifyRedirect("invalid.html")
  51         }
  52
  53         @Test
  54         fun `get request with image from non-local sone redirects to no permissions page`() {
  55                 request("", GET)
  56                 whenever(sone.isLocal).thenReturn(false)
  57                 addImage("image-id", image)
  58                 addHttpRequestParameter("image", "image-id")
  59                 verifyRedirect("noPermission.html")
  60         }
  61
  62         @Test
  63         fun `get request with image from local sone sets image in template context`() {
  64                 request("", GET)
  65                 addImage("image-id", image)
  66                 addHttpRequestParameter("image", "image-id")
  67                 page.processTemplate(freenetRequest, templateContext)
  68                 assertThat(templateContext["image"], equalTo<Any>(image))
  69         }
  70
  71         @Test
  72         fun `post request with abort delete flag set redirects to image browser`() {
  73                 request("", POST)
  74                 addImage("image-id", image)
  75                 addHttpRequestParameter("image", "image-id")
  76                 addHttpRequestParameter("abortDelete", "true")
  77                 verifyRedirect("imageBrowser.html?image=image-id")
  78         }
  79
  80         @Test
  81         fun `post request deletes image and redirects to image browser`() {
  82                 request("", POST)
  83                 addImage("image-id", image)
  84                 addHttpRequestParameter("image", "image-id")
  85                 verifyRedirect("imageBrowser.html?album=album-id") {
  86                         verify(webInterface.core).deleteImage(image)
  87                 }
  88         }
  89
  90 }