package net.pterodactylus.sone.web.pages

import net.pterodactylus.sone.data.TemporaryImage
import org.hamcrest.MatcherAssert.assertThat
import org.hamcrest.Matchers.contains
import org.hamcrest.Matchers.equalTo
import org.junit.Test

/**
 * Unit test for [GetImagePage].
 */
class GetImagePageTest {

	private val scaffolding = WebPageTest()
	private val page = GetImagePage(scaffolding.webInterface)
	private val freenetRequest = scaffolding.freenetRequest
	private val response = scaffolding.response
	private val responseContent = scaffolding.responseContent

	@Test
	fun `page returns correct path`() {
		assertThat(page.path, equalTo("getImage.html"))
	}

	@Test
	fun `page is not a prefix page`() {
		assertThat(page.isPrefixPage, equalTo(false))
	}

	@Test
	fun `page is not link-excepted`() {
		assertThat(page.isLinkExcepted(null), equalTo(false))
	}

	@Test
	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(responseContent.toByteArray(), equalTo(ByteArray(0)))
	}

	@Test
	fun `valid image returns response with correct data`() {
		val image = TemporaryImage("temp-id").apply {
			mimeType = "image/test"
			imageData = ByteArray(5, Int::toByte)
		}
		scaffolding.addHttpRequestParameter("image", "temp-id")
		scaffolding.addTemporaryImage("temp-id", image)
		page.handleRequest(freenetRequest, response)
		assertThat(response.statusCode, equalTo(200))
		assertThat(response.contentType, equalTo("image/test"))
		assertThat(responseContent.toByteArray(), 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")
		))
	}

}