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