X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FPageToadletFactoryTest.kt;h=a5160db0b4bb9c742131170ec9379e9409946b71;hp=93047cbdfc7b397940d6a64b996d94e32df252e4;hb=2e367a493f70dcd50da7de6af78126ea3c946582;hpb=e8bfde9bfac05a6fb4ad145b5aa523bff7fbbd5a

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..a5160db 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/PageToadletFactoryTest.kt
@@ -4,6 +4,7 @@ import freenet.client.HighLevelSimpleClient
 import net.pterodactylus.sone.test.mock
 import net.pterodactylus.sone.test.whenever
 import net.pterodactylus.util.web.Page
+import net.pterodactylus.util.web.Response
 import org.hamcrest.MatcherAssert.assertThat
 import org.hamcrest.Matchers.equalTo
 import org.hamcrest.Matchers.nullValue
@@ -39,4 +40,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
+
 }