Minor, extract ProgramWithDependencyOnCompiler for tests

Also use ProcessBuilder instead of GeneralCommandLine for simplicity.
2021-04-29 19:35:52 +02:00
parent d397efb2bd
commit f63dac26e6
2 changed files with 68 additions and 42 deletions
@@ -0,0 +1,62 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.integration
+
+import junit.framework.TestCase
+import org.jetbrains.kotlin.cli.AbstractCliTest
+import org.jetbrains.kotlin.cli.common.ExitCode
+import org.jetbrains.kotlin.cli.jvm.K2JVMCompiler
+import org.jetbrains.kotlin.utils.PathUtil
+import java.io.File
+
+class ProgramWithDependencyOnCompiler(
+    private val tmpdir: File,
+    private val programText: String,
+) {
+    private lateinit var program: File
+
+    fun compile() {
+        val programSource = File(tmpdir, "program.kt")
+        programSource.writeText(programText)
+
+        program = File(tmpdir, "program")
+        val (output, exitCode) = AbstractCliTest.executeCompilerGrabOutput(
+            K2JVMCompiler(),
+            listOf(
+                programSource.path,
+                "-d", program.absolutePath,
+                "-cp", PathUtil.kotlinPathsForDistDirectory.compilerPath.absolutePath,
+            ),
+        )
+        TestCase.assertEquals("Compilation failed:\n$output", ExitCode.OK, exitCode)
+    }
+
+    fun run(workingDirectory: File, vararg arguments: String): String = runJava(
+        workingDirectory,
+        "-cp",
+        listOf(
+            program.absolutePath,
+            PathUtil.kotlinPathsForDistDirectory.compilerPath.absolutePath,
+        ).joinToString(File.pathSeparator),
+        "ProgramKt",
+        *arguments,
+    )
+
+    private fun runJava(workingDirectory: File, vararg arguments: String): String {
+        val pb = ProcessBuilder()
+            .directory(workingDirectory)
+            .command(KotlinIntegrationTestBase.getJavaRuntime().absolutePath, *arguments)
+        val process = pb.start()
+        val stdout = String(process.inputStream.readBytes())
+        val stderr = String(process.errorStream.readBytes())
+        process.waitFor()
+
+        TestCase.assertEquals("Exit code should be 0.", 0, process.exitValue())
+        TestCase.assertEquals("Stderr should be empty.", "", stderr)
+
+        return stdout.trimEnd()
+    }
+}