X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fweb%2Fnotification%2FMarkPostKnownDuringFirstStartHandlerTest.kt;h=c2f8e415e573867be8b83484fd9caf610c5b9b28;hp=746eb476d0748a0a557238b161091baf5845bcb1;hb=58386d1ecdcbafd43c29692d4c165054eea9e0ba;hpb=3b926c638ec74eaf41e54463de6e85629f4e1be8

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 746eb47..c2f8e41 100644
--- a/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/MarkPostKnownDuringFirstStartHandlerTest.kt
@@ -23,8 +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.concurrent.atomic.*
+import java.util.function.*
 import kotlin.test.*
 
 /**
@@ -35,8 +34,8 @@ class MarkPostKnownDuringFirstStartHandlerTest {
 
 	private val eventBus = EventBus()
 	private val notificationManager = NotificationManager()
-	private val markedPostAsKnown = AtomicBoolean()
-	private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager) { if (it == post) markedPostAsKnown.set(true) }
+	private val markedPosts = mutableListOf<Post>()
+	private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager, Consumer { markedPosts += it })
 
 	init {
 		eventBus.register(handler)
@@ -45,16 +44,14 @@ class MarkPostKnownDuringFirstStartHandlerTest {
 	@Test
 	fun `post is not marked as known if not during first start`() {
 		eventBus.post(NewPostFoundEvent(post))
-		assertThat(markedPostAsKnown.get(), equalTo(false))
+		assertThat(markedPosts, emptyIterable())
 	}
 
 	@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(markedPostAsKnown.get(), equalTo(true))
+		assertThat(markedPosts, contains(post))
 	}
 
 }