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=a9c1c1770651720ec3b271464a4b490ea4a1334f;hpb=7eac5c60bdb241d7b98e548cf974d41e22af1b5c

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)
 		}
 	}