X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FMarkAsKnownPageTest.kt;h=d9a132ab01868f3af6483a61c615ff25d5fccadd;hp=a132a70196c704b6358fc39ff81f2b6637bf3b71;hb=2dd40fba7031cffb35a5156435547a5d964535c9;hpb=04a7ae5658a170f12eb9118acfb2df1305e571ac

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt
index a132a70..d9a132a 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt
@@ -4,6 +4,7 @@ import net.pterodactylus.sone.data.Post
 import net.pterodactylus.sone.data.PostReply
 import net.pterodactylus.sone.data.Sone
 import net.pterodactylus.sone.test.mock
+import net.pterodactylus.util.web.Method.POST
 import org.hamcrest.MatcherAssert.assertThat
 import org.hamcrest.Matchers.equalTo
 import org.junit.Test
@@ -12,11 +13,7 @@ import org.mockito.Mockito.verify
 /**
  * Unit test for [MarkAsKnownPage].
  */
-class MarkAsKnownPageTest: WebPageTest() {
-
-	private val page = MarkAsKnownPage(template, webInterface)
-
-	override fun getPage() = page
+class MarkAsKnownPageTest: WebPageTest(::MarkAsKnownPage) {
 
 	@Test
 	fun `page returns correct path`() {
@@ -36,6 +33,7 @@ class MarkAsKnownPageTest: WebPageTest() {
 
 	@Test
 	fun `posts can be marked as known`() {
+		setMethod(POST)
 		addHttpRequestPart("returnPage", "return.html")
 		addHttpRequestPart("type", "post")
 		addHttpRequestPart("id", "post1 post2 post3")
@@ -50,6 +48,7 @@ class MarkAsKnownPageTest: WebPageTest() {
 
 	@Test
 	fun `replies can be marked as known`() {
+		setMethod(POST)
 		addHttpRequestPart("returnPage", "return.html")
 		addHttpRequestPart("type", "reply")
 		addHttpRequestPart("id", "reply1 reply2 reply3")
@@ -64,6 +63,7 @@ class MarkAsKnownPageTest: WebPageTest() {
 
 	@Test
 	fun `sones can be marked as known`() {
+		setMethod(POST)
 		addHttpRequestPart("returnPage", "return.html")
 		addHttpRequestPart("type", "sone")
 		addHttpRequestPart("id", "sone1 sone2 sone3")
@@ -78,6 +78,7 @@ class MarkAsKnownPageTest: WebPageTest() {
 
 	@Test
 	fun `different type redirects to invalid page`() {
+		setMethod(POST)
 		addHttpRequestPart("type", "foo")
 		verifyRedirect("invalid.html")
 	}