src/test/kotlin/net/pterodactylus/sone/web/pages/GetImagePageTest.kt

   1 package net.pterodactylus.sone.web.pages
   2
   3 import net.pterodactylus.sone.data.TemporaryImage
   4 import net.pterodactylus.sone.test.getInstance
   5 import net.pterodactylus.sone.web.baseInjector
   6 import org.hamcrest.MatcherAssert.assertThat
   7 import org.hamcrest.Matchers.contains
   8 import org.hamcrest.Matchers.equalTo
   9 import org.hamcrest.Matchers.notNullValue
  10 import org.junit.Test
  11 import java.net.*
  12
  13 /**
  14  * Unit test for [GetImagePage].
  15  */
  16 class GetImagePageTest {
  17
  18         private val scaffolding = WebPageTest()
  19         private val page = GetImagePage(scaffolding.webInterface)
  20         private val freenetRequest = scaffolding.freenetRequest
  21         private val response = scaffolding.response
  22         private val responseContent = scaffolding.responseContent
  23
  24         @Test
  25         fun `page returns correct path`() {
  26                 assertThat(page.path, equalTo("getImage.html"))
  27         }
  28
  29         @Test
  30         fun `page is not a prefix page`() {
  31                 assertThat(page.isPrefixPage, equalTo(false))
  32         }
  33
  34         @Test
  35         fun `page is not link-excepted`() {
  36                 assertThat(page.isLinkExcepted(URI("")), equalTo(false))
  37         }
  38
  39         @Test
  40         fun `invalid image returns 404 response`() {
  41                 page.handleRequest(freenetRequest, response)
  42                 assertThat(response.statusCode, equalTo(404))
  43                 assertThat(response.statusText, equalTo("Not found."))
  44                 assertThat(response.contentType, equalTo("text/html; charset=utf-8"))
  45                 assertThat(responseContent.toByteArray(), equalTo(ByteArray(0)))
  46         }
  47
  48         @Test
  49         fun `valid image returns response with correct data`() {
  50                 val image = TemporaryImage("temp-id").apply {
  51                         mimeType = "image/test"
  52                         imageData = ByteArray(5, Int::toByte)
  53                 }
  54                 scaffolding.addHttpRequestParameter("image", "temp-id")
  55                 scaffolding.addTemporaryImage("temp-id", image)
  56                 page.handleRequest(freenetRequest, response)
  57                 assertThat(response.statusCode, equalTo(200))
  58                 assertThat(response.contentType, equalTo("image/test"))
  59                 assertThat(responseContent.toByteArray(), equalTo(ByteArray(5, Int::toByte)))
  60                 println(response.headers.map { it.name to it.iterator().asSequence().toList() })
  61                 assertThat(response.headers.map { it.name to it.iterator().asSequence().toList() }, contains(
  62                                 "Content-Disposition" to listOf("attachment; filename=temp-id.test")
  63                 ))
  64         }
  65
  66         @Test
  67         fun `page can be created by dependency injection`() {
  68             assertThat(baseInjector.getInstance<GetImagePage>(), notNullValue())
  69         }
  70
  71 }