🎨 Clean up imports

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

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/pages/UnbookmarkPageTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/pages/UnbookmarkPageTest.kt
index 1123ec8..0180187 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/pages/UnbookmarkPageTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/pages/UnbookmarkPageTest.kt
@@ -1,24 +1,18 @@
 package net.pterodactylus.sone.web.pages
 
-import net.pterodactylus.sone.data.Post
-import net.pterodactylus.sone.test.mock
-import net.pterodactylus.sone.test.whenever
-import net.pterodactylus.util.web.Method.POST
-import org.hamcrest.MatcherAssert.assertThat
-import org.hamcrest.Matchers.equalTo
-import org.junit.Test
-import org.mockito.Mockito.any
-import org.mockito.Mockito.never
-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 [UnbookmarkPage].
  */
-class UnbookmarkPageTest: WebPageTest() {
-
-       private val page = UnbookmarkPage(template, webInterface)
-
-       override fun getPage() = page
+class UnbookmarkPageTest: WebPageTest(::UnbookmarkPage) {
 
        @Test
        fun `page returns correct path`() {
@@ -33,7 +27,7 @@ class UnbookmarkPageTest: WebPageTest() {
        @Test
        fun `page returns correct title`() {
                addTranslation("Page.Unbookmark.Title", "unbookmark page title")
-               assertThat(page.getPageTitle(freenetRequest), equalTo("unbookmark page title"))
+               assertThat(page.getPageTitle(soneRequest), equalTo("unbookmark page title"))
        }
 
        @Test
@@ -50,10 +44,9 @@ class UnbookmarkPageTest: WebPageTest() {
                val notLoadedPost2 = mock<Post>()
                whenever(core.bookmarkedPosts).thenReturn(setOf(loadedPost1, loadedPost2, notLoadedPost1, notLoadedPost2))
                verifyRedirect("bookmarks.html") {
-                       verify(core).unbookmarkPost(notLoadedPost1)
-                       verify(core).unbookmarkPost(notLoadedPost2)
-                       verify(core, never()).unbookmarkPost(loadedPost1)
-                       verify(core, never()).unbookmarkPost(loadedPost2)
+                       val postCaptor = capture<Post>()
+                       verify(core, times(2)).unbookmarkPost(postCaptor.capture())
+                       assertThat(postCaptor.allValues, contains(notLoadedPost1, notLoadedPost2))
                }
        }
 
@@ -79,4 +72,9 @@ class UnbookmarkPageTest: WebPageTest() {
                }
        }
 
+       @Test
+       fun `page can be created by dependency injection`() {
+           assertThat(baseInjector.getInstance<UnbookmarkPage>(), notNullValue())
+       }
+
 }