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=d68060903290f13612940e777b4492cc98489151;hp=d992b305d373ecd4a91cdeaac143dca139ee33f8;hb=8f6aa25c5a6bfc76430800027533bfe5e2ee336c;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..d680609 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,18 @@ 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 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 +67,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 +75,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 +87,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 +120,13 @@ class FreenetTemplatePageTest {
 		assertThat(pageResponse.headers, contains(hasHeader("location", "invalid-form-password")))
 	}
 
+	@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")
+
 }