mirror of https://github.com/sbt/sbt.git
72 lines
2.1 KiB
Scala
72 lines
2.1 KiB
Scala
|
package example.test
|
||
|
|
|
||
|
|
import java.io.File
|
||
|
|
import java.io.PrintWriter
|
||
|
|
import java.nio.file.Files
|
||
|
|
import sbt.io.IO
|
||
|
|
import verify.BasicTestSuite
|
||
|
|
|
||
|
|
trait ShellScriptUtil extends BasicTestSuite {
|
||
|
|
val isWindows: Boolean =
|
||
|
|
sys.props("os.name").toLowerCase(java.util.Locale.ENGLISH).contains("windows")
|
||
|
|
|
||
|
|
private val javaBinDir = new File("launcher-package/integration-test/bin").getAbsolutePath
|
||
|
|
|
||
|
|
private def retry[A1](f: () => A1, maxAttempt: Int = 10): A1 =
|
||
|
|
try {
|
||
|
|
f()
|
||
|
|
} catch {
|
||
|
|
case _ if maxAttempt <= 1 =>
|
||
|
|
Thread.sleep(100)
|
||
|
|
retry(f, maxAttempt - 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
val sbtScript =
|
||
|
|
if (isWindows) new File("launcher-package/target/universal/stage/bin/sbt.bat")
|
||
|
|
else new File("launcher-package/target/universal/stage/bin/sbt")
|
||
|
|
|
||
|
|
/**
|
||
|
|
* testOutput is a helper function to create a test for shell script.
|
||
|
|
*/
|
||
|
|
inline def testOutput(
|
||
|
|
name: String,
|
||
|
|
javaOpts: String = "",
|
||
|
|
sbtOpts: String = "",
|
||
|
|
sbtOptsFileContents: String = "",
|
||
|
|
javaToolOptions: String = ""
|
||
|
|
)(args: String*)(f: List[String] => Any) =
|
||
|
|
test(name) {
|
||
|
|
val workingDirectory = Files.createTempDirectory("sbt-launcher-package-test").toFile
|
||
|
|
retry(() => IO.copyDirectory(new File("launcher-package/citest"), workingDirectory))
|
||
|
|
|
||
|
|
try
|
||
|
|
val sbtOptsFile = new File(workingDirectory, ".sbtopts")
|
||
|
|
sbtOptsFile.createNewFile()
|
||
|
|
val writer = new PrintWriter(sbtOptsFile)
|
||
|
|
try {
|
||
|
|
writer.write(sbtOptsFileContents)
|
||
|
|
} finally {
|
||
|
|
writer.close()
|
||
|
|
}
|
||
|
|
val path = sys.env.getOrElse("PATH", sys.env("Path"))
|
||
|
|
val out = scala.sys.process
|
||
|
|
.Process(
|
||
|
|
Seq(sbtScript.getAbsolutePath) ++ args,
|
||
|
|
workingDirectory,
|
||
|
|
"JAVA_OPTS" -> javaOpts,
|
||
|
|
"SBT_OPTS" -> sbtOpts,
|
||
|
|
"JAVA_TOOL_OPTIONS" -> javaToolOptions,
|
||
|
|
if (isWindows)
|
||
|
|
"JAVACMD" -> new File(javaBinDir, "java").getAbsolutePath()
|
||
|
|
else
|
||
|
|
"PATH" -> (javaBinDir + File.pathSeparator + path)
|
||
|
|
)
|
||
|
|
.!!
|
||
|
|
.linesIterator
|
||
|
|
.toList
|
||
|
|
f(out)
|
||
|
|
()
|
||
|
|
finally IO.delete(workingDirectory)
|
||
|
|
}
|
||
|
|
}
|