X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpage%2FFreenetRequestTest.kt;h=3295bddec69cd441063e1a00475af91b070b550a;hp=391991c35e1c77e805a76d9820b38b5763719821;hb=d5cd41949b7e62fdc40ef7390661dcda8c18a139;hpb=cd1f93dc057bdc6e89baeb41db0f749f2287224e

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetRequestTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetRequestTest.kt
index 391991c..3295bdd 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetRequestTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/page/FreenetRequestTest.kt
@@ -1,15 +1,15 @@
 package net.pterodactylus.sone.web.page
 
 import freenet.clients.http.*
-import freenet.l10n.*
+import freenet.clients.http.SessionManager.*
 import freenet.support.api.*
 import net.pterodactylus.sone.test.*
 import net.pterodactylus.util.web.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
 import org.junit.*
+import org.mockito.*
 import org.mockito.Mockito.*
-import org.mockito.Mockito.mock
 import java.net.*
 
 class FreenetRequestTest {
@@ -18,8 +18,8 @@ class FreenetRequestTest {
 	private val method = Method.GET
 	private val httpRequest = mock(HTTPRequest::class.java)
 	private val toadletContext = mock(ToadletContext::class.java)
-	private val l10n = mock<BaseL10n>()
-	private val request = FreenetRequest(uri, method, httpRequest, toadletContext, l10n)
+	private val sessionManager = mock<SessionManager>()
+	private val request = FreenetRequest(uri, method, httpRequest, toadletContext, sessionManager)
 
 	@Test
 	fun `uri is retained correctly`() {
@@ -42,8 +42,29 @@ class FreenetRequestTest {
 	}
 
 	@Test
-	fun `l10n is retained correctly`() {
-		assertThat(request.l10n, equalTo(l10n))
+	fun `null is returned if no session exists`() {
+		assertThat(request.existingSession, nullValue())
+	}
+
+	@Test
+	fun `existing session can be retrieved`() {
+		val session = mock<Session>()
+		whenever(sessionManager.useSession(toadletContext)).thenReturn(session)
+		assertThat(request.existingSession, sameInstance(session))
+	}
+
+	@Test
+	fun `existing session is returned if it exists`() {
+		val session = mock<Session>()
+		whenever(sessionManager.useSession(toadletContext)).thenReturn(session)
+		assertThat(request.session, sameInstance(session))
+	}
+
+	@Test
+	fun `new session is returned if none exists`() {
+		val session = mock<Session>()
+		whenever(sessionManager.createSession(anyString(), ArgumentMatchers.eq(toadletContext))).thenReturn(session)
+		assertThat(request.session, sameInstance(session))
 	}
 
 }