c2f8e415e573867be8b83484fd9caf610c5b9b28
[Sone.git] / src / test / kotlin / net / pterodactylus / sone / web / notification / MarkPostKnownDuringFirstStartHandlerTest.kt
1 /**
2  * Sone - MarkPostKnownDuringFirstStartHandlerTest.kt - Copyright © 2019 David ‘Bombe’ Roden
3  *
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.
8  *
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.
13  *
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/>.
16  */
17
18 package net.pterodactylus.sone.web.notification
19
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.*
26 import java.util.function.*
27 import kotlin.test.*
28
29 /**
30  * Unit test for [MarkPostKnownDuringFirstStartHandler].
31  */
32 @Suppress("UnstableApiUsage")
33 class MarkPostKnownDuringFirstStartHandlerTest {
34
35         private val eventBus = EventBus()
36         private val notificationManager = NotificationManager()
37         private val markedPosts = mutableListOf<Post>()
38         private val handler = MarkPostKnownDuringFirstStartHandler(notificationManager, Consumer { markedPosts += it })
39
40         init {
41                 eventBus.register(handler)
42         }
43
44         @Test
45         fun `post is not marked as known if not during first start`() {
46                 eventBus.post(NewPostFoundEvent(post))
47                 assertThat(markedPosts, emptyIterable())
48         }
49
50         @Test
51         fun `new post is marked as known during first start`() {
52                 notificationManager.firstStart()
53                 eventBus.post(NewPostFoundEvent(post))
54                 assertThat(markedPosts, contains(post))
55         }
56
57 }
58
59 private val post: Post = Post.EmptyPost("post")