🎨 Add helper for simulating first start

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

diff --git a/src/test/kotlin/net/pterodactylus/sone/web/notification/NewRemotePostHandlerTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/notification/NewRemotePostHandlerTest.kt
index a86bbf1..a0d7bc6 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/web/notification/NewRemotePostHandlerTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/NewRemotePostHandlerTest.kt
@@ -20,12 +20,13 @@ package net.pterodactylus.sone.web.notification
 import com.google.common.eventbus.*
 import net.pterodactylus.sone.core.event.*
 import net.pterodactylus.sone.data.*
+import net.pterodactylus.sone.data.Post.*
+import net.pterodactylus.sone.data.impl.*
 import net.pterodactylus.sone.notify.*
 import net.pterodactylus.util.notify.*
 import net.pterodactylus.util.template.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
-import java.io.*
 import kotlin.test.*
 
 /**
@@ -44,26 +45,46 @@ class NewRemotePostHandlerTest {
        }
 
        @Test
-       fun `handler adds post to new-post notification`() {
-               eventBus.post(NewPostFoundEvent(post))
-               assertThat(notification.elements, contains(post))
+       fun `handler adds remote post to new-post notification`() {
+               eventBus.post(NewPostFoundEvent(remotePost))
+               assertThat(notification.elements, contains(remotePost))
        }
 
        @Test
-       fun `handler adds notification to notification manager`() {
-               eventBus.post(NewPostFoundEvent(post))
+       fun `handler does not add local post to new-post notification`() {
+               eventBus.post(NewPostFoundEvent(localPost))
+               assertThat(notification.elements, emptyIterable())
+       }
+
+       @Test
+       fun `handler adds notification for remote post to notification manager`() {
+               eventBus.post(NewPostFoundEvent(remotePost))
                assertThat(notificationManager.notifications, contains<Notification>(notification))
        }
 
        @Test
+       fun `handler does not add notification for local post to notification manager`() {
+               eventBus.post(NewPostFoundEvent(localPost))
+               assertThat(notificationManager.notifications, emptyIterable())
+       }
+
+       @Test
        fun `handler does not add notification to notification manager during first start`() {
-               notificationManager.addNotification(object : AbstractNotification("first-start-notification") {
-                       override fun render(writer: Writer?) = Unit
-               })
-               eventBus.post(NewPostFoundEvent(post))
+               notificationManager.firstStart()
+               eventBus.post(NewPostFoundEvent(remotePost))
                assertThat(notificationManager.notifications, not(hasItem(notification)))
        }
 
 }
 
-private val post: Post = Post.EmptyPost("post")
+private val remoteSone: Sone = IdOnlySone("remote-sone")
+private val remotePost: Post = object : EmptyPost("remote-post") {
+       override fun getSone() = remoteSone
+}
+
+private val localSone: Sone = object : IdOnlySone("local-sone") {
+       override fun isLocal() = true
+}
+private val localPost: Post = object : EmptyPost("local-post") {
+       override fun getSone() = localSone
+}