From: David ‘Bombe’ Roden Date: Wed, 11 Dec 2019 18:01:05 +0000 (+0100) Subject: 🚧 Add handler for startup event X-Git-Tag: v81^2~5^2~42 X-Git-Url: https://git.pterodactylus.net/?a=commitdiff_plain;h=742d893ed6941719f4612887197429ea903ab62f;p=Sone.git 🚧 Add handler for startup event --- diff --git a/src/main/kotlin/net/pterodactylus/sone/web/notification/StartupHandler.kt b/src/main/kotlin/net/pterodactylus/sone/web/notification/StartupHandler.kt new file mode 100644 index 0000000..392aa5c --- /dev/null +++ b/src/main/kotlin/net/pterodactylus/sone/web/notification/StartupHandler.kt @@ -0,0 +1,37 @@ +/** + * Sone - StartupHandler.kt - Copyright © 2019 David ‘Bombe’ Roden + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package net.pterodactylus.sone.web.notification + +import com.google.common.eventbus.* +import net.pterodactylus.sone.core.event.* +import net.pterodactylus.util.notify.* +import java.util.concurrent.* +import java.util.concurrent.TimeUnit.* + +/** + * Handler for the [Startup] event notification. + */ +class StartupHandler(private val notificationManager: NotificationManager, private val notification: TemplateNotification, val ticker: ScheduledExecutorService) { + + @Subscribe + fun startup(startup: Startup) { + notificationManager.addNotification(notification) + ticker.schedule({ notificationManager.removeNotification(notification) }, 2, MINUTES) + } + +} diff --git a/src/test/kotlin/net/pterodactylus/sone/web/notification/StartupHandlerTest.kt b/src/test/kotlin/net/pterodactylus/sone/web/notification/StartupHandlerTest.kt new file mode 100644 index 0000000..38e7f1b --- /dev/null +++ b/src/test/kotlin/net/pterodactylus/sone/web/notification/StartupHandlerTest.kt @@ -0,0 +1,65 @@ +/** + * Sone - StartupHandlerTest.kt - Copyright © 2019 David ‘Bombe’ Roden + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program. If not, see . + */ + +package net.pterodactylus.sone.web.notification + +import com.google.common.eventbus.* +import net.pterodactylus.sone.core.event.* +import net.pterodactylus.sone.utils.* +import net.pterodactylus.util.notify.* +import net.pterodactylus.util.template.* +import org.hamcrest.MatcherAssert.* +import org.hamcrest.Matchers.* +import kotlin.test.* + +/** + * Unit test for [StartupHandler]. + */ +class StartupHandlerTest { + + private val eventBus = EventBus() + private val notificationManager = NotificationManager() + private val notification = TemplateNotification("", Template()) + private val executor = TestScheduledThreadPoolExecutor() + + init { + eventBus.register(StartupHandler(notificationManager, notification, executor)) + } + + @AfterTest + fun shutdownExecutor() = executor.shutdown() + + @Test + fun `handler adds notification to manager on startup`() { + eventBus.post(Startup()) + assertThat(notificationManager.notifications, contains(notification)) + } + + @Test + fun `handler registers command on startup`() { + eventBus.post(Startup()) + assertThat(executor.scheduleds, hasSize(1)) + } + + @Test + fun `registered command removes notification from manager`() { + eventBus.post(Startup()) + executor.scheduleds.single().command() + assertThat(notificationManager.notifications, emptyIterable()) + } + +}