X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;ds=sidebyside;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fmain%2FSonePluginTest.kt;h=b846522a40102c925ef04955e6acc09b589800ae;hb=1ce940852c7e96a83233bcb706d0f87a1c655c54;hp=5795285cd959852c3060fec6c3b5cab57a32596a;hpb=d33e7e73ccdc7a1ed8b32b13bbe12f84aa076e6b;p=Sone.git

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
--- 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