X-Git-Url: https://git.pterodactylus.net/?a=blobdiff_plain;f=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fmain%2FSonePluginTest.kt;fp=src%2Ftest%2Fkotlin%2Fnet%2Fpterodactylus%2Fsone%2Fmain%2FSonePluginTest.kt;h=677823c6e26352f5077350f9a13f4402392d7b7f;hb=8281d81a4412af542753a4fc56b2c1a8c52000d4;hp=23c679bf0eabfb2a50df2630743429ef768b4911;hpb=6fd3d46cae8b06225c862a8b5d0a6aef35328e21;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 23c679b..677823c 100644
--- a/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
+++ b/src/test/kotlin/net/pterodactylus/sone/main/SonePluginTest.kt
@@ -1,11 +1,13 @@
 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.*
@@ -14,6 +16,8 @@ import net.pterodactylus.sone.web.notification.*
 import org.hamcrest.MatcherAssert.*
 import org.hamcrest.Matchers.*
 import org.mockito.Mockito.*
+import java.io.*
+import java.util.concurrent.atomic.*
 import kotlin.test.*
 
 /**
@@ -70,11 +74,12 @@ class SonePluginTest {
 		assertThat(injector.getInstance<NotificationHandler>(), notNullValue())
 	}
 
-	private fun runSonePluginWithRealInjector(): Injector {
+	private fun runSonePluginWithRealInjector(injectorConsumer: (Injector) -> Unit = {}): Injector {
 		lateinit var injector: Injector
 		val sonePlugin = SonePlugin {
 			Guice.createInjector(*it).also {
 				injector = it
+				injectorConsumer(it)
 			}
 		}
 		sonePlugin.setLanguage(ENGLISH)
@@ -95,6 +100,42 @@ class SonePluginTest {
 		assertThat(getInjected(NotificationHandler::class.java), notNullValue())
 	}
 
+	@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(object : Any() {
+				@Subscribe
+				fun firstStart(firstStart: FirstStart) {
+					firstStartReceived.set(true)
+				}
+			})
+		}
+		sonePlugin.runPlugin(pluginRespirator)
+		assertThat(firstStartReceived.get(), equalTo(true))
+	}
+
+	@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(object : Any() {
+					@Subscribe
+					fun firstStart(firstStart: FirstStart) {
+						firstStartReceived.set(true)
+					}
+				})
+			}
+			sonePlugin.runPlugin(pluginRespirator)
+			assertThat(firstStartReceived.get(), equalTo(false))
+		}
+	}
+
 	private fun <T> getInjected(clazz: Class<T>, annotation: Annotation? = null): T? =
 			injected[TypeLiteral.get(clazz) to annotation] as? T
 
@@ -118,3 +159,9 @@ class SonePluginTest {
 }
 
 private fun String.toClass(): Class<*> = SonePlugin::class.java.classLoader.loadClass(this)
+
+private fun File.deleteAfter(action: File.() -> Unit) = try {
+	action(this)
+} finally {
+	this.delete()
+}