X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fnotification%2FMarkPostKnownDuringFirstStartHandlerTest.kt;h=5b8e41ab1f0a9f052bd5bf75c98da33c5104e44a;hb=438378deab1514f0f608d975ef65f5b7aea44ccb;hp=286648ae25162c6297eab9243aa987c26535d98d;hpb=69bbbebbf548146b03713b398a63247e5f716c85;p=Sone.git

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
index 286648a..5b8e41a 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
@@ -1,5 +1,5 @@
 /**
- * Sone - MarkPostKnownDuringFirstStartHandlerTest.kt - Copyright © 2019 David ‘Bombe’ Roden
+ * Sone - MarkPostKnownDuringFirstStartHandlerTest.kt - Copyright © 2019–2020 David ‘Bombe’ Roden
  *
  * This program is free software: you can redistribute it and/or modify
  * it under the terms of the GNU General Public License as published by
@@ -23,7 +23,7 @@ import net.pterodactylus.sone.data.*
 import net.pterodactylus.util.notify.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
-import java.io.*
+import java.util.function.*
 import kotlin.test.*
 
 /**
@@ -35,7 +35,7 @@ class MarkPostKnownDuringFirstStartHandlerTest {
 	private val eventBus = EventBus()
 	private val notificationManager = NotificationManager()
 	private val markedPosts = mutableListOf<Post>()
-	private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager) { markedPosts += it }
+	private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager, Consumer { markedPosts += it })
 
 	init {
 		eventBus.register(handler)
@@ -49,9 +49,7 @@ class MarkPostKnownDuringFirstStartHandlerTest {
 
 	@Test
 	fun `new post is marked as known during first start`() {
-		notificationManager.addNotification(object : AbstractNotification("first-start-notification") {
-			override fun render(writer: Writer?) = Unit
-		})
+		notificationManager.firstStart()
 		eventBus.post(NewPostFoundEvent(post))
 		assertThat(markedPosts, contains(post))
 	}