X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fpages%2FUnbookmarkPageTest.kt;h=a9f784e543818788a6ebcf04175d7eeaa20b2f17;hp=1123ec8cf86b0bbdda22ac24d352a7ce10269531;hb=2dd40fba7031cffb35a5156435547a5d964535c9;hpb=4d98a47a88176d904960594f1aaaac9e7db15f9a

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..a9f784e 100644
--- 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,23 @@
 package net.pterodactylus.sone.web.pages
 
 import net.pterodactylus.sone.data.Post
+import net.pterodactylus.sone.test.capture
 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.contains
 import org.hamcrest.Matchers.equalTo
 import org.junit.Test
 import org.mockito.Mockito.any
 import org.mockito.Mockito.never
+import org.mockito.Mockito.times
 import org.mockito.Mockito.verify
 
 /**
  * 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`() {
@@ -50,10 +49,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))
 		}
 	}