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=2880d24e3ece2d7f98c8847966c3691b5cf272c4;hp=bfc2581d6f6d70df101ee7bc6e9bbb78ef853930;hb=110a933c2724aba6a604fc5eed6372ff1e1e6144;hpb=28c3ef5bb54ac66cbbc7b610b21ee4d300180dd6

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 bfc2581..2880d24 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
@@ -1,16 +1,22 @@
 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.*
 
 class FreenetTemplatePageTest {
 
-	private val page = FreenetTemplatePage("/test/path", null, null, "invalid-form-password")
+	private val templateContextFactory = deepMock<TemplateContextFactory>()
+	private val templateRenderer = deepMock<TemplateRenderer>()
+	private val loaders = mock<Loaders>()
+	private val template = mock<Template>()
+	private val page = FreenetTemplatePage("/test/path", templateRenderer, loaders, "invalid-form-password")
 
 	@Test
 	fun `path is exposed correctly`() {
@@ -18,8 +24,8 @@ class FreenetTemplatePageTest {
 	}
 
 	@Test
-	fun `getPageTitle() default implementation returns null`() {
-		assertThat(page.getPageTitle(mock()), nullValue())
+	fun `getPageTitle() default implementation returns empty string`() {
+		assertThat(page.getPageTitle(mock()), equalTo(""))
 	}
 
 	@Test
@@ -64,16 +70,16 @@ class FreenetTemplatePageTest {
 
 	@Test
 	fun `isEnabled() returns false if full access only is true`() {
-		val page = object : FreenetTemplatePage("/test/path", null, null, null) {
-			override fun isFullAccessOnly() = true
+		val page = object : FreenetTemplatePage("/test/path", templateRenderer, loaders, "invalid-form-password") {
+			override val isFullAccessOnly = true
 		}
 		assertThat(page.isEnabled(mock()), equalTo(false))
 	}
 
 	@Test
 	fun `page with redirect target throws redirect exception on handleRequest`() {
-		val page = object : FreenetTemplatePage("/test/path", null, null, null) {
-			override fun getRedirectTarget(request: FreenetRequest?) = "foo"
+		val page = object : FreenetTemplatePage("/test/path", templateRenderer, loaders, "invalid-form-password") {
+			override fun getRedirectTarget(request: FreenetRequest) = "foo"
 		}
 		val request = mock<FreenetRequest>()
 		val response = mock<Response>()
@@ -84,8 +90,8 @@ class FreenetTemplatePageTest {
 
 	@Test
 	fun `page with full access only returns unauthorized on handleRequest with non-full access request`() {
-		val page = object : FreenetTemplatePage("/test/path", null, null, null) {
-			override fun isFullAccessOnly() = true
+		val page = object : FreenetTemplatePage("/test/path", templateRenderer, loaders, "invalid-form-password") {
+			override val isFullAccessOnly = true
 		}
 		val request = deepMock<FreenetRequest>()
 		val response = Response(null)
@@ -117,4 +123,19 @@ class FreenetTemplatePageTest {
 		assertThat(pageResponse.headers, contains(hasHeader("location", "invalid-form-password")))
 	}
 
+	@Test
+	fun `template from annotation is loaded`() {
+		val template = deepMock<Template>()
+		whenever(loaders.loadTemplate("template-path")).thenReturn(template)
+		TestPage(templateRenderer, loaders)
+		verify(loaders).loadTemplate("template-path")
+	}
+
+	@TemplatePath("template-path")
+	private class TestPage(templateRenderer: TemplateRenderer, loaders: Loaders) : FreenetTemplatePage("/", templateRenderer, loaders, "") {
+		override fun getPath() = ""
+		override fun isPrefixPage() = false
+		override fun handleRequest(request: FreenetRequest, response: Response) = response
+	}
+
 }