🎨 Clean up imports

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / pages / MarkAsKnownPageTest.kt

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..0bb49e0 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/MarkAsKnownPageTest.kt
@@ -1,22 +1,18 @@
 package net.pterodactylus.sone.web.pages
 
-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 org.hamcrest.MatcherAssert.assertThat
-import org.hamcrest.Matchers.equalTo
-import org.junit.Test
-import org.mockito.Mockito.verify
+import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.test.*
+import net.pterodactylus.sone.web.*
+import net.pterodactylus.util.web.Method.*
+import org.hamcrest.MatcherAssert.*
+import org.hamcrest.Matchers.*
+import org.junit.*
+import org.mockito.Mockito.*
 
 /**
  * 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`() {
@@ -31,15 +27,16 @@ class MarkAsKnownPageTest: WebPageTest() {
        @Test
        fun `page returns correct title`() {
                addTranslation("Page.MarkAsKnown.Title", "mark as known page title")
-               assertThat(page.getPageTitle(freenetRequest), equalTo("mark as known page title"))
+               assertThat(page.getPageTitle(soneRequest), equalTo("mark as known page title"))
        }
 
        @Test
        fun `posts can be marked as known`() {
+               setMethod(POST)
                addHttpRequestPart("returnPage", "return.html")
                addHttpRequestPart("type", "post")
                addHttpRequestPart("id", "post1 post2 post3")
-               val posts = listOf(mock<Post>(), mock<Post>())
+               val posts = listOf(mock<Post>(), mock())
                addPost("post1", posts[0])
                addPost("post3", posts[1])
                verifyRedirect("return.html") {
@@ -50,10 +47,11 @@ 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")
-               val replies = listOf(mock<PostReply>(), mock<PostReply>())
+               val replies = listOf(mock<PostReply>(), mock())
                addPostReply("reply1", replies[0])
                addPostReply("reply3", replies[1])
                verifyRedirect("return.html") {
@@ -64,10 +62,11 @@ 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")
-               val sones = listOf(mock<Sone>(), mock<Sone>())
+               val sones = listOf(mock<Sone>(), mock())
                addSone("sone1", sones[0])
                addSone("sone3", sones[1])
                verifyRedirect("return.html") {
@@ -78,8 +77,14 @@ class MarkAsKnownPageTest: WebPageTest() {
 
        @Test
        fun `different type redirects to invalid page`() {
+               setMethod(POST)
                addHttpRequestPart("type", "foo")
                verifyRedirect("invalid.html")
        }
 
+       @Test
+       fun `page can be created by dependency injection`() {
+           assertThat(baseInjector.getInstance<MarkAsKnownPage>(), notNullValue())
+       }
+
 }