✨ Send shutdown event on termination

[Sone.git] / src / test / kotlin / net / pterodactylus / sone / main / SonePluginTest.kt

diff --git a/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt b/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
index 5795285..b846522 100644 (file)
--- a/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
@@ -26,7 +26,7 @@ import kotlin.test.*
 @Dirty
 class SonePluginTest {
 
-       private val sonePlugin by lazy { SonePlugin { injector } }
+       private var sonePlugin = SonePlugin { injector }
        private val pluginRespirator = deepMock<PluginRespirator>()
        private val node = deepMock<Node>()
        private val clientCore = deepMock<NodeClientCore>()
@@ -76,7 +76,7 @@ class SonePluginTest {
 
        private fun runSonePluginWithRealInjector(injectorConsumer: (Injector) -> Unit = {}): Injector {
                lateinit var injector: Injector
-               val sonePlugin = SonePlugin {
+               sonePlugin = SonePlugin {
                        Guice.createInjector(*it).also {
                                injector = it
                                injectorConsumer(it)
@@ -115,7 +115,6 @@ class SonePluginTest {
                        val eventBus = it.getInstance(EventBus::class.java)
                        eventBus.register(FirstStartListener(firstStartReceived))
                }
-               sonePlugin.runPlugin(pluginRespirator)
                assertThat(firstStartReceived.get(), equalTo(true))
        }
 
@@ -128,7 +127,6 @@ class SonePluginTest {
                                val eventBus = it.getInstance(EventBus::class.java)
                                eventBus.register(FirstStartListener(firstStartReceived))
                        }
-                       sonePlugin.runPlugin(pluginRespirator)
                        assertThat(firstStartReceived.get(), equalTo(false))
                }
        }
@@ -149,7 +147,6 @@ class SonePluginTest {
                                val eventBus = it.getInstance(EventBus::class.java)
                                eventBus.register(ConfigNotReadListener(configNotReadReceived))
                        }
-                       sonePlugin.runPlugin(pluginRespirator)
                        assertThat(configNotReadReceived.get(), equalTo(true))
                }
        }
@@ -162,7 +159,6 @@ class SonePluginTest {
                        val eventBus = it.getInstance(EventBus::class.java)
                        eventBus.register(ConfigNotReadListener(configNotReadReceived))
                }
-               sonePlugin.runPlugin(pluginRespirator)
                assertThat(configNotReadReceived.get(), equalTo(false))
        }
 
@@ -175,11 +171,45 @@ class SonePluginTest {
                                val eventBus = it.getInstance(EventBus::class.java)
                                eventBus.register(ConfigNotReadListener(configNotReadReceived))
                        }
-                       sonePlugin.runPlugin(pluginRespirator)
                        assertThat(configNotReadReceived.get(), equalTo(false))
                }
        }
 
+       private class StartupListener(private val startupReceived: () -> Unit) {
+               @Subscribe
+               fun startup(startup: Startup) {
+                       startupReceived()
+               }
+       }
+
+       @Test
+       fun `startup event is sent to event bus`() {
+               val startupReceived = AtomicBoolean()
+               runSonePluginWithRealInjector {
+                       val eventBus = it.getInstance(EventBus::class.java)
+                       eventBus.register(StartupListener { startupReceived.set(true) })
+               }
+               assertThat(startupReceived.get(), equalTo(true))
+       }
+
+       private class ShutdownListener(private val shutdownReceived: () -> Unit) {
+               @Subscribe
+               fun shutdown(shutdown: Shutdown) {
+                       shutdownReceived()
+               }
+       }
+
+       @Test
+       fun `shutdown event is sent to event bus on terminate`() {
+               val shutdownReceived = AtomicBoolean()
+               runSonePluginWithRealInjector {
+                       val eventBus = it.getInstance(EventBus::class.java)
+                       eventBus.register(ShutdownListener { shutdownReceived.set(true) })
+               }
+               sonePlugin.terminate()
+               assertThat(shutdownReceived.get(), equalTo(true))
+       }
+
        private fun <T> getInjected(clazz: Class<T>, annotation: Annotation? = null): T? =
                        injected[TypeLiteral.get(clazz) to annotation] as? T