From c922f4176046a911deec47dc71f70b0052703233 Mon Sep 17 00:00:00 2001
From: =?utf8?q?David=20=E2=80=98Bombe=E2=80=99=20Roden?=
 <bombe@pterodactylus.net>
Date: Mon, 28 Nov 2016 20:37:02 +0100
Subject: [PATCH] Add unit test for logout page

---
 .../net/pterodactylus/sone/web/LogoutPageTest.kt   | 59 ++++++++++++++++++++++
 1 file changed, 59 insertions(+)
 create mode 100644 src/test/kotlin/net/pterodactylus/sone/web/LogoutPageTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/LogoutPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/LogoutPageTest.kt
new file mode 100644
index 0000000..65b75e0
--- /dev/null
+++ b/src/test/kotlin/net/pterodactylus/sone/web/LogoutPageTest.kt
@@ -0,0 +1,59 @@
+package net.pterodactylus.sone.web
+
+import net.pterodactylus.sone.test.whenever
+import net.pterodactylus.sone.web.WebTestUtils.redirectsTo
+import org.hamcrest.MatcherAssert.assertThat
+import org.hamcrest.Matchers.equalTo
+import org.junit.Test
+import org.mockito.Mockito.verify
+
+/**
+ * Unit test for [LogoutPage].
+ */
+class LogoutPageTest : WebPageTest() {
+
+	private val page = LogoutPage(template, webInterface)
+
+	@Test
+	fun `page unsets current sone and redirects to index`() {
+		expectedException.expect(redirectsTo("index.html"))
+		try {
+			page.handleRequest(freenetRequest, templateContext)
+		} finally {
+			verify(webInterface).setCurrentSone(toadletContext, null)
+		}
+	}
+
+	@Test
+	fun `page is not enabled if sone requires full access and request does not have full access`() {
+		core.preferences.isRequireFullAccess = true
+		assertThat(page.isEnabled(toadletContext), equalTo(false))
+	}
+
+	@Test
+	fun `page is disabled if no sone is logged in`() {
+		unsetCurrentSone()
+		assertThat(page.isEnabled(toadletContext), equalTo(false))
+	}
+
+	@Test
+	fun `page is disabled if sone is logged in but there is only one sone`() {
+		whenever(core.localSones).thenReturn(listOf(currentSone))
+		assertThat(page.isEnabled(toadletContext), equalTo(false))
+	}
+
+	@Test
+	fun `page is enabled if sone is logged in and there is more than one sone`() {
+		whenever(core.localSones).thenReturn(listOf(currentSone, currentSone))
+		assertThat(page.isEnabled(toadletContext), equalTo(true))
+	}
+
+	@Test
+	fun `page is enabled if full access is required and present and sone is logged in and there is more than one sone`() {
+		core.preferences.isRequireFullAccess = true
+		whenever(toadletContext.isAllowedFullAccess).thenReturn(true)
+		whenever(core.localSones).thenReturn(listOf(currentSone, currentSone))
+		assertThat(page.isEnabled(toadletContext), equalTo(true))
+	}
+
+}
-- 
2.7.4