2 * Sone - MarkPostKnownDuringFirstStartHandlerTest.kt - Copyright © 2019 David ‘Bombe’ Roden
4 * This program is free software: you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License as published by
6 * the Free Software Foundation, either version 3 of the License, or
7 * (at your option) any later version.
9 * This program is distributed in the hope that it will be useful,
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 * GNU General Public License for more details.
14 * You should have received a copy of the GNU General Public License
15 * along with this program. If not, see <http://www.gnu.org/licenses/>.
18 package net.pterodactylus.sone.web.notification
20 import com.google.common.eventbus.*
21 import net.pterodactylus.sone.core.event.*
22 import net.pterodactylus.sone.data.*
23 import net.pterodactylus.util.notify.*
24 import org.hamcrest.MatcherAssert.*
25 import org.hamcrest.Matchers.*
27 import java.util.function.Consumer
31 * Unit test for [MarkPostKnownDuringFirstStartHandler].
33 @Suppress("UnstableApiUsage")
34 class MarkPostKnownDuringFirstStartHandlerTest {
36 private val eventBus = EventBus()
37 private val notificationManager = NotificationManager()
38 private val markedPosts = mutableListOf<Post>()
39 private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager, Consumer { markedPosts += it })
42 eventBus.register(handler)
46 fun `post is not marked as known if not during first start`() {
47 eventBus.post(NewPostFoundEvent(post))
48 assertThat(markedPosts, emptyIterable())
52 fun `new post is marked as known during first start`() {
53 notificationManager.addNotification(object : AbstractNotification("first-start-notification") {
54 override fun render(writer: Writer?) = Unit
56 eventBus.post(NewPostFoundEvent(post))
57 assertThat(markedPosts, contains(post))
62 private val post: Post = Post.EmptyPost("post")