X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2FLockSonePageTest.kt;h=7136e043436106d13ad353ee44797b7256b2e5c8;hp=44cd81c0e52c7e1f2ae3f83ec1b6719ea4d52ed9;hb=9acbc5bdec4ccb752e0856a501568b0bb6161579;hpb=8883cd2aa6d2294e10259f7a5338e6c16b26c8b2

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/LockSonePageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/LockSonePageTest.kt
index 44cd81c..7136e04 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/LockSonePageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/LockSonePageTest.kt
@@ -14,13 +14,12 @@ class LockSonePageTest : WebPageTest() {
 
 	private val page = LockSonePage(template, webInterface)
 
+	override fun getPage() = page
+
 	@Test
 	fun `locking an invalid local sone redirects to return page`() {
 		addHttpRequestParameter("returnPage", "return.html")
-		expectedException.expect(WebTestUtils.redirectsTo("return.html"))
-		try {
-			page.handleRequest(freenetRequest, templateContext)
-		} finally {
+		verifyRedirect("return.html") {
 			verify(core, never()).lockSone(any<Sone>())
 		}
 	}
@@ -31,10 +30,7 @@ class LockSonePageTest : WebPageTest() {
 		val sone = mock<Sone>()
 		addLocalSone("sone-id", sone)
 		addHttpRequestParameter("returnPage", "return.html")
-		expectedException.expect(WebTestUtils.redirectsTo("return.html"))
-		try {
-			page.handleRequest(freenetRequest, templateContext)
-		} finally {
+		verifyRedirect("return.html") {
 			verify(core).lockSone(sone)
 		}
 	}