Tests for ITestJvmFlag{Contributor,Interceptor}.

src/test/kotlin/com/beust/kobalt/internal/GenericRunnerTest.kt

@@ -0,0 +1,75 @@
+package com.beust.kobalt.internal
+
+import com.beust.kobalt.Args
+import com.beust.kobalt.TestConfig
+import com.beust.kobalt.TestModule
+import com.beust.kobalt.api.*
+import com.beust.kobalt.maven.dependency.FileDependency
+import com.beust.kobalt.project
+import org.assertj.core.api.Assertions.assertThat
+import org.testng.annotations.BeforeClass
+import org.testng.annotations.Test
+import javax.inject.Inject
+
+
+/**
+ * Test ITestJvmFlagContributor and ITestJvmFlagInterceptor.
+ */
+class DependencyTest @Inject constructor() {
+    private val project : Project get() = project { name = "dummy" }
+    private val classpath = listOf(FileDependency("/tmp/a.jar"))
+    private val context = KobaltContext(Args())
+    private val contributor = object : ITestJvmFlagContributor {
+        override fun testJvmFlagsFor(project: Project, context: KobaltContext,
+                currentFlags: List<String>): List<String> {
+            return listOf("-agent", "foo")
+        }
+    }
+    private val A_JAR = "/tmp/a.jar"
+    private val B_JAR = "/tmp/b.jar"
+
+    private val interceptor = object : ITestJvmFlagInterceptor {
+        override fun testJvmFlagsFor(project: Project, context: KobaltContext,
+                currentFlags: List<String>): List<String> {
+            return currentFlags.map { if (it == A_JAR) B_JAR else it }
+        }
+    }
+
+    @BeforeClass
+    fun beforeClass() {
+        Kobalt.init(TestModule())
+    }
+
+    private fun runTest(pluginInfo: IPluginInfo, expected: List<String>) {
+        val result = TestNgRunner().calculateAllJvmArgs(project, context, TestConfig(project),
+                classpath, pluginInfo)
+        assertThat(result).isEqualTo(expected)
+    }
+
+    @Test
+    fun noContributorsNoInterceptors() {
+        runTest(BasePluginInfo(), listOf("-classpath", A_JAR))
+    }
+
+    @Test
+    fun contributorOnly() {
+        runTest(BasePluginInfo().apply { testJvmFlagContributors.add(contributor) },
+                listOf("-classpath", A_JAR, "-agent", "foo"))
+    }
+
+    @Test
+    fun interceptorOnly() {
+        runTest(BasePluginInfo().apply { testJvmFlagInterceptors.add(interceptor) },
+                listOf("-classpath", B_JAR))
+    }
+
+    @Test
+    fun contributorAndInterceptor() {
+        runTest(BasePluginInfo().apply {
+                testJvmFlagContributors.add(contributor)
+                testJvmFlagInterceptors.add(interceptor)
+            },
+            listOf("-classpath", B_JAR, "-agent", "foo"))
+    }
+}
+