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=f08d3aed42a5d5704abdbe5541563c80ca5a6d46;hb=8f6aa25c5a6bfc76430800027533bfe5e2ee336c;hpb=4c5a7fd8b4f2d2e223304b7ad4fb9829d011cbe9

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 f08d3ae..d680609 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetTemplatePageTest.kt
@@ -1,78 +1,132 @@
 package net.pterodactylus.sone.web.page
 
+import net.pterodactylus.sone.main.*
 import net.pterodactylus.sone.test.*
+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.Mockito.*
 
 class FreenetTemplatePageTest {
 
+	private val templateRenderer = deepMock<TemplateRenderer>()
+	private val loaders = mock<Loaders>()
+	private val page = TestPage(templateRenderer, loaders)
+
 	@Test
 	fun `path is exposed correctly`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.path, equalTo("/test/path"))
 	}
 
 	@Test
-	fun `getPageTitle() default implementation returns null`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
-		assertThat(page.getPageTitle(mock()), nullValue())
+	fun `getPageTitle() default implementation returns empty string`() {
+		assertThat(page.getPageTitle(mock()), equalTo(""))
 	}
 
 	@Test
 	fun `isPrefixPage() default implementation returns false`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.isPrefixPage, equalTo(false))
 	}
 
 	@Test
 	fun `getStylesheets() default implementation returns empty collection`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.styleSheets, empty())
 	}
 
 	@Test
 	fun `getShortcutIcon() default implementation returns null`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.shortcutIcon, nullValue())
 	}
 
 	@Test
 	fun `getRedirectTarget() default implementation returns null`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.getRedirectTarget(mock()), nullValue())
 	}
 
 	@Test
 	fun `getAdditionalLinkNodes() default implementation returns empty collection`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.getAdditionalLinkNodes(mock()), empty())
 	}
 
 	@Test
 	fun `isFullAccessOnly() default implementation returns false`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.isFullAccessOnly, equalTo(false))
 	}
 
 	@Test
 	fun `isLinkExcepted() default implementation returns false`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.isLinkExcepted(mock()), equalTo(false))
 	}
 
 	@Test
 	fun `isEnabled() returns true if full access only is false`() {
-		val page = FreenetTemplatePage("/test/path", null, null, null)
 		assertThat(page.isEnabled(mock()), equalTo(true))
 	}
 
 	@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 : TestPage(templateRenderer, loaders) {
+			override val isFullAccessOnly = true
 		}
 		assertThat(page.isEnabled(mock()), equalTo(false))
 	}
 
+	@Test
+	fun `page with redirect target throws redirect exception on handleRequest`() {
+		val page = object : TestPage(templateRenderer, loaders) {
+			override fun getRedirectTarget(request: FreenetRequest) = "foo"
+		}
+		val request = mock<FreenetRequest>()
+		val response = mock<Response>()
+		val pageResponse = page.handleRequest(request, response)
+		assertThat(pageResponse.statusCode, anyOf(equalTo(302), equalTo(307)))
+		assertThat(pageResponse.headers, contains(hasHeader("location", "foo")))
+	}
+
+	@Test
+	fun `page with full access only returns unauthorized on handleRequest with non-full access request`() {
+		val page = object : TestPage(templateRenderer, loaders) {
+			override val isFullAccessOnly = true
+		}
+		val request = deepMock<FreenetRequest>()
+		val response = Response(null)
+		val pageResponse = page.handleRequest(request, response)
+		assertThat(pageResponse.statusCode, equalTo(401))
+	}
+
+	@Test
+	fun `page redirects on POST without form password`() {
+		val request = deepMock<FreenetRequest>().apply {
+			whenever(httpRequest.getPartAsStringFailsafe(any(), anyInt())).thenReturn("")
+			whenever(method).thenReturn(POST)
+		}
+		val response = Response(null)
+		val pageResponse = page.handleRequest(request, response)
+		assertThat(pageResponse.statusCode, anyOf(equalTo(302), equalTo(307)))
+		assertThat(pageResponse.headers, contains(hasHeader("location", "invalid-form-password")))
+	}
+
+	@Test
+	fun `page redirects on POST with invalid password`() {
+		val request = deepMock<FreenetRequest>().apply {
+			whenever(httpRequest.getPartAsStringFailsafe(any(), anyInt())).thenReturn("invalid")
+			whenever(method).thenReturn(POST)
+		}
+		val response = Response(null)
+		val pageResponse = page.handleRequest(request, response)
+		assertThat(pageResponse.statusCode, anyOf(equalTo(302), equalTo(307)))
+		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")
+
 }