X-Git-Url: https://git.pterodactylus.net/?p=Sone.git;a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fmain%2FSonePluginTest.kt;h=d1e2cecb5309ec0a23d2ecd93736a4bdc4dc2676;hp=9388cb06598c445b4e9baf8db03eebf4468e74d9;hb=4af887c75667793c197b564f16b7e95b1f63cbc7;hpb=18ada47554280f444bb829dafc4d40926312066e

diff --git a/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt b/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
index 9388cb0..d1e2cec 100644
--- a/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
@@ -1,28 +1,34 @@
 package net.pterodactylus.sone.main
 
+import com.google.common.eventbus.*
 import com.google.inject.*
 import freenet.client.async.*
 import freenet.l10n.BaseL10n.LANGUAGE.*
 import freenet.node.*
 import freenet.pluginmanager.*
 import net.pterodactylus.sone.core.*
+import net.pterodactylus.sone.core.event.*
 import net.pterodactylus.sone.fcp.*
 import net.pterodactylus.sone.freenet.wot.*
 import net.pterodactylus.sone.test.*
 import net.pterodactylus.sone.web.*
+import net.pterodactylus.sone.web.notification.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
+import org.junit.experimental.categories.*
 import org.mockito.Mockito.*
+import java.io.*
+import java.util.concurrent.atomic.*
 import kotlin.test.*
 
 /**
  * Unit test for [SonePlugin].
  */
 @Dirty
+@Category(NotParallel::class)
 class SonePluginTest {
 
-	private var injector = mockInjector()
-	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>()
@@ -64,11 +70,18 @@ class SonePluginTest {
 		assertThat(injector.getInstance<WebOfTrustConnector>(), notNullValue())
 	}
 
-	private fun runSonePluginWithRealInjector(): Injector {
+	@Test
+	fun `notification handler can be created`() {
+		val injector: Injector = runSonePluginWithRealInjector()
+		assertThat(injector.getInstance<NotificationHandler>(), notNullValue())
+	}
+
+	private fun runSonePluginWithRealInjector(injectorConsumer: (Injector) -> Unit = {}): Injector {
 		lateinit var injector: Injector
-		val sonePlugin = SonePlugin {
+		sonePlugin = SonePlugin {
 			Guice.createInjector(*it).also {
 				injector = it
+				injectorConsumer(it)
 			}
 		}
 		sonePlugin.setLanguage(ENGLISH)
@@ -83,21 +96,154 @@ class SonePluginTest {
 		verify(core).start()
 	}
 
-}
+	@Test
+	fun `notification handler is being requested`() {
+		sonePlugin.runPlugin(pluginRespirator)
+		assertThat(getInjected(NotificationHandler::class.java), notNullValue())
+	}
 
-private fun mockInjector() = mock<Injector>().apply {
-	val injected = mutableMapOf<Pair<TypeLiteral<*>, Annotation?>, Any>()
-	fun mockValue(clazz: Class<*>) = false.takeIf { clazz.name == java.lang.Boolean::class.java.name } ?: mock(clazz)
-	whenever(getInstance(any<Key<*>>())).then {
-		injected.getOrPut((it.getArgument(0) as Key<*>).let { it.typeLiteral to it.annotation }) {
-			it.getArgument<Key<*>>(0).typeLiteral.type.typeName.toClass().let(::mockValue)
+	@Test
+	fun `ticker shutdown is being requested`() {
+		sonePlugin.runPlugin(pluginRespirator)
+		assertThat(getInjected(TickerShutdown::class.java), notNullValue())
+	}
+
+	private class FirstStartListener(private val firstStartReceived: AtomicBoolean) {
+		@Subscribe
+		fun firstStart(@Suppress("UNUSED_PARAMETER") firstStart: FirstStart) {
+			firstStartReceived.set(true)
+		}
+	}
+
+	@Test
+	fun `first-start event is sent to event bus when first start is true`() {
+		File("sone.properties").delete()
+		val firstStartReceived = AtomicBoolean()
+		runSonePluginWithRealInjector {
+			val eventBus = it.getInstance(EventBus::class.java)
+			eventBus.register(FirstStartListener(firstStartReceived))
 		}
+		assertThat(firstStartReceived.get(), equalTo(true))
 	}
-	whenever(getInstance(any<Class<*>>())).then {
-		injected.getOrPut(TypeLiteral.get(it.getArgument(0) as Class<*>) to null) {
-			it.getArgument<Class<*>>(0).let(::mockValue)
+
+	@Test
+	fun `first-start event is not sent to event bus when first start is false`() {
+		File("sone.properties").deleteAfter {
+			writeText("# empty")
+			val firstStartReceived = AtomicBoolean()
+			runSonePluginWithRealInjector {
+				val eventBus = it.getInstance(EventBus::class.java)
+				eventBus.register(FirstStartListener(firstStartReceived))
+			}
+			assertThat(firstStartReceived.get(), equalTo(false))
+		}
+	}
+
+	private class ConfigNotReadListener(private val configNotReadReceiver: AtomicBoolean) {
+		@Subscribe
+		fun configNotRead(@Suppress("UNUSED_PARAMETER") configNotRead: ConfigNotRead) {
+			configNotReadReceiver.set(true)
+		}
+	}
+
+	@Test
+	fun `config-not-read event is sent to event bus when new config is true`() {
+		File("sone.properties").deleteAfter {
+			writeText("Invalid")
+			val configNotReadReceived = AtomicBoolean()
+			runSonePluginWithRealInjector {
+				val eventBus = it.getInstance(EventBus::class.java)
+				eventBus.register(ConfigNotReadListener(configNotReadReceived))
+			}
+			assertThat(configNotReadReceived.get(), equalTo(true))
+		}
+	}
+
+	@Test
+	fun `config-not-read event is not sent to event bus when first start is true`() {
+		File("sone.properties").delete()
+		val configNotReadReceived = AtomicBoolean()
+		runSonePluginWithRealInjector {
+			val eventBus = it.getInstance(EventBus::class.java)
+			eventBus.register(ConfigNotReadListener(configNotReadReceived))
 		}
+		assertThat(configNotReadReceived.get(), equalTo(false))
 	}
+
+	@Test
+	fun `config-not-read event is not sent to event bus when new config is false`() {
+		File("sone.properties").deleteAfter {
+			writeText("# comment")
+			val configNotReadReceived = AtomicBoolean()
+			runSonePluginWithRealInjector {
+				val eventBus = it.getInstance(EventBus::class.java)
+				eventBus.register(ConfigNotReadListener(configNotReadReceived))
+			}
+			assertThat(configNotReadReceived.get(), equalTo(false))
+		}
+	}
+
+	private class StartupListener(private val startupReceived: () -> Unit) {
+		@Subscribe
+		fun startup(@Suppress("UNUSED_PARAMETER") 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(@Suppress("UNUSED_PARAMETER") 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
+
+	private val injected =
+			mutableMapOf<Pair<TypeLiteral<*>, Annotation?>, Any>()
+
+	private val injector = mock<Injector>().apply {
+		fun mockValue(clazz: Class<*>) = false.takeIf { clazz.name == java.lang.Boolean::class.java.name } ?: mock(clazz)
+		whenever(getInstance(any<Key<*>>())).then {
+			injected.getOrPut((it.getArgument(0) as Key<*>).let { it.typeLiteral to it.annotation }) {
+				it.getArgument<Key<*>>(0).typeLiteral.type.typeName.toClass().let(::mockValue)
+			}
+		}
+		whenever(getInstance(any<Class<*>>())).then {
+			injected.getOrPut(TypeLiteral.get(it.getArgument(0) as Class<*>) to null) {
+				it.getArgument<Class<*>>(0).let(::mockValue)
+			}
+		}
+	}
+
 }
 
 private fun String.toClass(): Class<*> = SonePlugin::class.java.classLoader.loadClass(this)
+
+private fun File.deleteAfter(action: File.() -> Unit) = try {
+	action(this)
+} finally {
+	this.delete()
+}