1 package net.pterodactylus.sone.web
3 import com.google.inject.*
4 import freenet.clients.http.*
5 import net.pterodactylus.sone.main.*
6 import net.pterodactylus.sone.test.*
7 import net.pterodactylus.sone.web.page.*
8 import net.pterodactylus.util.web.*
10 import org.junit.rules.*
11 import org.mockito.Mockito.*
13 class PageToadletRegistryTest {
15 private val pageMaker = mock<PageMaker>()
16 private val toadletContainer = mock<ToadletContainer>()
17 private val pageToadletFactory = mock<PageToadletFactory>()
18 private val sonePlugin = mock<SonePlugin>()
20 private val injector = Guice.createInjector(
21 PageMaker::class.isProvidedBy(pageMaker),
22 ToadletContainer::class.isProvidedBy(toadletContainer),
23 PageToadletFactory::class.isProvidedBy(pageToadletFactory),
24 SonePlugin::class.isProvidedBy(sonePlugin)
29 val expectedException: ExpectedException = ExpectedException.none()
30 private val pageToadletRegistry = injector.getInstance<PageToadletRegistry>()
33 fun `registry adds navigation category to page maker`() {
34 pageToadletRegistry.registerToadlets()
35 verify(pageMaker).addNavigationCategory("/Sone/index.html", "Navigation.Menu.Sone.Name", "Navigation.Menu.Sone.Tooltip", sonePlugin)
38 private val page = TestPage()
41 fun `adding a page without menuname will add it correctly`() {
42 val toadletWithoutMenuname = createPageToadlet()
43 whenever(pageToadletFactory.createPageToadlet(page)).thenReturn(toadletWithoutMenuname)
44 pageToadletRegistry.addPage(page)
45 pageToadletRegistry.registerToadlets()
46 verify(toadletContainer).register(toadletWithoutMenuname, null, "/Sone/", true, false)
50 fun `adding a page with menuname will add it correctly`() {
51 val toadletWithMenuname = createPageToadlet("Test")
52 whenever(pageToadletFactory.createPageToadlet(page)).thenReturn(toadletWithMenuname)
53 pageToadletRegistry.addPage(page)
54 pageToadletRegistry.registerToadlets()
55 verify(toadletContainer).register(toadletWithMenuname, "Navigation.Menu.Sone.Name", "/Sone/", true, "Navigation.Menu.Sone.Item.Test.Name", "Navigation.Menu.Sone.Item.Test.Tooltip", false, toadletWithMenuname)
59 fun `adding a page after registering will throw an exception`() {
60 val toadletWithMenuname = createPageToadlet("Test")
61 whenever(pageToadletFactory.createPageToadlet(page)).thenReturn(toadletWithMenuname)
62 pageToadletRegistry.registerToadlets()
63 expectedException.expect(IllegalStateException::class.java)
64 pageToadletRegistry.addPage(page)
68 fun `unregistering toadlets will remove category link`() {
69 pageToadletRegistry.unregisterToadlets()
70 verify(pageMaker).removeNavigationCategory("Navigation.Menu.Sone.Name")
74 fun `unregistering toadlets will unregister them from the container`() {
75 val toadletWithMenuname = createPageToadlet("Test")
76 whenever(pageToadletFactory.createPageToadlet(page)).thenReturn(toadletWithMenuname)
77 pageToadletRegistry.addPage(page)
78 pageToadletRegistry.registerToadlets()
79 pageToadletRegistry.unregisterToadlets()
80 verify(toadletContainer).unregister(toadletWithMenuname)
83 private fun createPageToadlet(menuName: String? = null) =
84 mock<PageToadlet>().apply {
85 whenever(this.path()).thenReturn("/Sone/")
86 whenever(this.menuName).thenReturn(menuName)
89 private class TestPage : Page<FreenetRequest> {
90 override fun getPath() = ""
91 override fun isPrefixPage() = false
92 override fun handleRequest(freenetRequest: FreenetRequest, response: Response) = response