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