🔥 Clean up imports

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / notification / MarkPostKnownDuringFirstStartHandlerTest.kt

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..c2f8e41 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
@@ -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))
        }