src/test/kotlin/net/pterodactylus/sone/web/UploadImagePageTest.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.data.TemporaryImage
   7 import net.pterodactylus.sone.test.mock
   8 import net.pterodactylus.sone.test.mockBuilder
   9 import net.pterodactylus.sone.test.whenever
  10 import net.pterodactylus.util.web.Method.POST
  11 import org.hamcrest.MatcherAssert.assertThat
  12 import org.hamcrest.Matchers.equalTo
  13 import org.junit.Test
  14 import org.mockito.Mockito.any
  15 import org.mockito.Mockito.eq
  16 import org.mockito.Mockito.never
  17 import org.mockito.Mockito.verify
  18
  19 /**
  20  * Unit test for [UploadImagePage].
  21  */
  22 class UploadImagePageTest : WebPageTest() {
  23
  24         private val parentAlbum = mock<Album>().apply {
  25                 whenever(id).thenReturn("parent-id")
  26                 whenever(sone).thenReturn(currentSone)
  27         }
  28
  29         override fun getPage() = UploadImagePage(template, webInterface)
  30
  31         @Test
  32         fun `get request does not redirect or upload anything`() {
  33                 page.handleRequest(freenetRequest, templateContext)
  34                 verify(core, never()).createTemporaryImage(any(), any())
  35                 verify(core, never()).createImage(any(), any(), any())
  36         }
  37
  38         @Test
  39         fun `post request without parent results in no permission error page`() {
  40                 request("", POST)
  41                 verifyRedirect("noPermission.html")
  42         }
  43
  44         @Test
  45         fun `post request with parent that is not the current sone results in no permission error page`() {
  46                 request("", POST)
  47                 addHttpRequestParameter("parent", "parent-id")
  48                 whenever(parentAlbum.sone).thenReturn(mock<Sone>())
  49                 addAlbum("parent-id", parentAlbum)
  50                 verifyRedirect("noPermission.html")
  51         }
  52
  53         @Test
  54         fun `post request with empty name redirects to error page`() {
  55                 request("", POST)
  56                 addAlbum("parent-id", parentAlbum)
  57                 addHttpRequestParameter("parent", "parent-id")
  58                 addHttpRequestParameter("title", " ")
  59                 verifyRedirect("emptyImageTitle.html")
  60         }
  61
  62         @Test
  63         fun `uploading an invalid image results in no redirect and message set in template context`() {
  64                 request("", POST)
  65                 addAlbum("parent-id", parentAlbum)
  66                 addHttpRequestParameter("parent", "parent-id")
  67                 addHttpRequestParameter("title", "title")
  68                 addUploadedFile("image", "image.png", "image/png", "no-image.png")
  69                 page.handleRequest(freenetRequest, templateContext)
  70                 verify(core, never()).createTemporaryImage(any(), any())
  71                 assertThat(templateContext["messages"] as String?, equalTo<String>("Page.UploadImage.Error.InvalidImage"))
  72         }
  73
  74         @Test
  75         fun `uploading a valid image uploads image and redirects to album browser`() {
  76                 request("", POST)
  77                 addAlbum("parent-id", parentAlbum)
  78                 addHttpRequestParameter("parent", "parent-id")
  79                 addHttpRequestParameter("title", "Title")
  80                 addHttpRequestParameter("description", "Description")
  81                 addUploadedFile("image", "image.png", "image/png", "image.png")
  82                 val temporaryImage = TemporaryImage("temp-image")
  83                 val imageModifier = mockBuilder<Image.Modifier>()
  84                 val image = mock<Image>().apply {
  85                         whenever(modify()).thenReturn(imageModifier)
  86                 }
  87                 whenever(core.createTemporaryImage(eq("image/png"), any())).thenReturn(temporaryImage)
  88                 whenever(core.createImage(currentSone, parentAlbum, temporaryImage)).thenReturn(image)
  89                 verifyRedirect("imageBrowser.html?album=parent-id") {
  90                         verify(image).modify()
  91                         verify(imageModifier).setWidth(2)
  92                         verify(imageModifier).setHeight(1)
  93                         verify(imageModifier).setTitle("Title")
  94                         verify(imageModifier).setDescription("Description")
  95                         verify(imageModifier).update()
  96                 }
  97         }
  98
  99 }