♻️ Copy session-handling code to FreenetRequest

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

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt
index 93047cb..c9ecafe 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt
@@ -1,20 +1,20 @@
 package net.pterodactylus.sone.web.page
 
-import freenet.client.HighLevelSimpleClient
-import net.pterodactylus.sone.test.mock
-import net.pterodactylus.sone.test.whenever
-import net.pterodactylus.util.web.Page
-import org.hamcrest.MatcherAssert.assertThat
-import org.hamcrest.Matchers.equalTo
-import org.hamcrest.Matchers.nullValue
-import org.junit.Test
+import freenet.client.*
+import freenet.clients.http.*
+import net.pterodactylus.sone.test.*
+import net.pterodactylus.util.web.*
+import org.hamcrest.MatcherAssert.*
+import org.hamcrest.Matchers.*
+import org.junit.*
 
 private val highLevelSimpleClient = mock<HighLevelSimpleClient>()
+private val sessionManager = mock<SessionManager>()
 private const val pathPrefix = "/some/prefix/"
 
 class PageToadletFactoryTest {
 
-       private val pageToadletFactory = PageToadletFactory(highLevelSimpleClient, pathPrefix)
+       private val pageToadletFactory = PageToadletFactory(highLevelSimpleClient, sessionManager, pathPrefix)
 
        @Test
        fun `page toadlet without menu name is created without menu name`() {
@@ -39,4 +39,27 @@ class PageToadletFactoryTest {
                assertThat(pageToadlet.path(), equalTo("/some/prefix/path"))
        }
 
+       @Test
+       fun `menu name is added from annotation when no menu name is given`() {
+               val page = TestPageWithMenuName()
+               val pageToadlet = pageToadletFactory.createPageToadlet(page)
+               assertThat(pageToadlet.menuName, equalTo("testName"))
+       }
+
+       @Test
+       fun `menu name from annotation is ignored when menu name is given`() {
+               val page = TestPageWithMenuName()
+               val pageToadlet = pageToadletFactory.createPageToadlet(page, "foo")
+               assertThat(pageToadlet.menuName, equalTo("foo"))
+       }
+
+}
+
+@MenuName("testName")
+private class TestPageWithMenuName : Page<FreenetRequest> {
+
+       override fun getPath() = ""
+       override fun isPrefixPage() = false
+       override fun handleRequest(request: FreenetRequest, response: Response) = response
+
 }