X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FGetImagePageTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FGetImagePageTest.kt;h=151f1ae527160c56abe522c1a6da11312bbec6a6;hp=fb96a4c0d34d5647c8eff62153ad62d479e36e3d;hb=e9e8f21fdfe533e8946a64d53dabaf48c3c296c3;hpb=05fb821e72072bde52f383bdc5a988da67f66d0c

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/GetImagePageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/GetImagePageTest.kt
index fb96a4c..151f1ae 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/GetImagePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/GetImagePageTest.kt
@@ -1,9 +1,8 @@
 package net.pterodactylus.sone.web.pages
 
 import net.pterodactylus.sone.data.TemporaryImage
-import net.pterodactylus.sone.web.pages.GetImagePage
-import net.pterodactylus.sone.web.pages.WebPageTest
 import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.contains
 import org.hamcrest.Matchers.equalTo
 import org.junit.Test
 
@@ -33,6 +32,8 @@ class GetImagePageTest : WebPageTest() {
 	fun `invalid image returns 404 response`() {
 		page.handleRequest(freenetRequest, response)
 		assertThat(response.statusCode, equalTo(404))
+		assertThat(response.statusText, equalTo("Not found."))
+		assertThat(response.contentType, equalTo("text/html; charset=utf-8"))
 		assertThat(responseBytes, equalTo(ByteArray(0)))
 	}
 
@@ -48,6 +49,10 @@ class GetImagePageTest : WebPageTest() {
 		assertThat(response.statusCode, equalTo(200))
 		assertThat(response.contentType, equalTo("image/test"))
 		assertThat(responseBytes, equalTo(ByteArray(5, Int::toByte)))
+		println(response.headers.map { it.name to it.iterator().asSequence().toList() })
+		assertThat(response.headers.map { it.name to it.iterator().asSequence().toList() }, contains(
+				"Content-Disposition" to listOf("attachment; filename=temp-id.test")
+		))
 	}
 
 }