✨ Use @TemplatePath in FreenetTemplatePage

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / page / FreenetTemplatePageTest.kt

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..def7410 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
@@ -8,11 +8,11 @@ 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 templateContextFactory = deepMock<TemplateContextFactory>()
        private val loaders = mock<Loaders>()
        private val template = mock<Template>()
        private val page = FreenetTemplatePage("/test/path", templateContextFactory, loaders, template, "invalid-form-password")
@@ -122,4 +122,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(templateContextFactory, loaders)
+               verify(loaders).loadTemplate("template-path")
+       }
+
+       @TemplatePath("template-path")
+       private class TestPage(templateContextFactory: TemplateContextFactory, loaders: Loaders) : FreenetTemplatePage("/", templateContextFactory, loaders, Template(), "") {
+               override fun getPath() = ""
+               override fun isPrefixPage() = false
+               override fun handleRequest(request: FreenetRequest, response: Response) = response
+       }
+
 }