2 * Sone - StartupHandlerTest.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.utils.*
23 import net.pterodactylus.util.notify.*
24 import net.pterodactylus.util.template.*
25 import org.hamcrest.MatcherAssert.*
26 import org.hamcrest.Matchers.*
30 * Unit test for [StartupHandler].
32 class StartupHandlerTest {
34 private val eventBus = EventBus()
35 private val notificationManager = NotificationManager()
36 private val notification = TemplateNotification("", Template())
37 private val executor = TestScheduledThreadPoolExecutor()
40 eventBus.register(StartupHandler(notificationManager, notification, executor))
44 fun shutdownExecutor() = executor.shutdown()
47 fun `handler adds notification to manager on startup`() {
48 eventBus.post(Startup())
49 assertThat(notificationManager.notifications, contains<Notification>(notification))
53 fun `handler registers command on startup`() {
54 eventBus.post(Startup())
55 assertThat(executor.scheduleds, hasSize(1))
59 fun `registered command removes notification from manager`() {
60 eventBus.post(Startup())
61 executor.scheduleds.single().command()
62 assertThat(notificationManager.notifications, emptyIterable())