X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FFreenetTemplatePageTest.kt;h=2474d08968d3aa381225dd19142973e241f81a29;hp=d992b305d373ecd4a91cdeaac143dca139ee33f8;hb=5201720801a4053f8f324b71fb3e9c36f6cd911e;hpb=09e10dd6e554ce4af6b35e0f56251c6c26bc018e

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
index d992b30..2474d08 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
@@ -2,20 +2,20 @@ package net.pterodactylus.sone.web.page
 
 import net.pterodactylus.sone.main.*
 import net.pterodactylus.sone.test.*
-import net.pterodactylus.util.template.*
 import net.pterodactylus.util.web.*
 import net.pterodactylus.util.web.Method.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
 import org.junit.*
-import org.mockito.ArgumentMatchers.*
+import org.mockito.Mockito.*
+import java.io.*
+import java.nio.charset.StandardCharsets.*
 
 class FreenetTemplatePageTest {
 
-	private val templateContextFactory = mock<TemplateContextFactory>()
+	private val templateRenderer = deepMock<TemplateRenderer>()
 	private val loaders = mock<Loaders>()
-	private val template = mock<Template>()
-	private val page = FreenetTemplatePage("/test/path", templateContextFactory, loaders, template, "invalid-form-password")
+	private val page = TestPage(templateRenderer, loaders)
 
 	@Test
 	fun `path is exposed correctly`() {
@@ -69,7 +69,7 @@ class FreenetTemplatePageTest {
 
 	@Test
 	fun `isEnabled() returns false if full access only is true`() {
-		val page = object : FreenetTemplatePage("/test/path", templateContextFactory, loaders, template, "invalid-form-password") {
+		val page = object : TestPage(templateRenderer, loaders) {
 			override val isFullAccessOnly = true
 		}
 		assertThat(page.isEnabled(mock()), equalTo(false))
@@ -77,7 +77,7 @@ class FreenetTemplatePageTest {
 
 	@Test
 	fun `page with redirect target throws redirect exception on handleRequest`() {
-		val page = object : FreenetTemplatePage("/test/path", templateContextFactory, loaders, template, "invalid-form-password") {
+		val page = object : TestPage(templateRenderer, loaders) {
 			override fun getRedirectTarget(request: FreenetRequest) = "foo"
 		}
 		val request = mock<FreenetRequest>()
@@ -89,7 +89,7 @@ class FreenetTemplatePageTest {
 
 	@Test
 	fun `page with full access only returns unauthorized on handleRequest with non-full access request`() {
-		val page = object : FreenetTemplatePage("/test/path", templateContextFactory, loaders, template, "invalid-form-password") {
+		val page = object : TestPage(templateRenderer, loaders) {
 			override val isFullAccessOnly = true
 		}
 		val request = deepMock<FreenetRequest>()
@@ -122,4 +122,28 @@ class FreenetTemplatePageTest {
 		assertThat(pageResponse.headers, contains(hasHeader("location", "invalid-form-password")))
 	}
 
+	@Test
+	@Dirty
+	fun `freenet template page creates page with correct title`() {
+		val page = object : TestPage(templateRenderer, loaders) {
+			override fun getPageTitle(request: FreenetRequest) = "page title"
+		}
+		val request = deepMock<FreenetRequest>()
+		val pageMakerInteractionFactory = deepMock<PageMakerInteractionFactory>()
+		whenever(pageMakerInteractionFactory.createPageMaker(request.toadletContext, "page title").renderPage()).thenReturn("<page>")
+		setField(page, "pageMakerInteractionFactory", pageMakerInteractionFactory)
+		val response = page.handleRequest(request, Response(ByteArrayOutputStream()))
+		assertThat(response.statusCode, equalTo(200))
+		assertThat((response.content as ByteArrayOutputStream).toString(UTF_8.name()), equalTo("<page>"))
+	}
+
+	@Test
+	fun `template from annotation is loaded`() {
+		verify(loaders).loadTemplate("template-path")
+	}
+
+	@TemplatePath("template-path")
+	@ToadletPath("/test/path")
+	private open class TestPage(templateRenderer: TemplateRenderer, loaders: Loaders) : FreenetTemplatePage(templateRenderer, loaders, "invalid-form-password")
+
 }