89 lines
2.8 KiB
Scala
89 lines
2.8 KiB
Scala
package org.broadinstitute.sting.queue
|
|
|
|
import org.broadinstitute.sting.utils.Utils
|
|
import org.testng.Assert
|
|
import org.broadinstitute.sting.commandline.CommandLineProgram
|
|
import org.broadinstitute.sting.BaseTest
|
|
import org.broadinstitute.sting.queue.util.ProcessController
|
|
|
|
class QScriptTest extends BaseTest {
|
|
|
|
protected val stingDir = "./"
|
|
|
|
/**
|
|
* execute the test
|
|
* @param name the name of the test
|
|
* @param args the argument list
|
|
* @param expectedException the expected exception or null if no exception is expected.
|
|
*/
|
|
def executeTest(name: String, args: String, expectedException: Class[_]) = {
|
|
var command = Utils.escapeExpressions(args)
|
|
|
|
// add the logging level to each of the integration test commands
|
|
|
|
command = Utils.appendArray(command, "-l", "WARN", "-startFromScratch", "-tempDir", "integrationtests")
|
|
|
|
// run the executable
|
|
var gotAnException = false
|
|
|
|
val instance = new QCommandLine
|
|
QScriptTest.runningCommandLines += instance
|
|
try {
|
|
println("Executing test %s with Queue arguments: %s".format(name, Utils.join(" ",command)))
|
|
CommandLineProgram.start(instance, command)
|
|
} catch {
|
|
case e =>
|
|
gotAnException = true
|
|
if (expectedException != null) {
|
|
// we expect an exception
|
|
println("Wanted exception %s, saw %s".format(expectedException, e.getClass))
|
|
if (expectedException.isInstance(e)) {
|
|
// it's the type we expected
|
|
println(String.format(" => %s PASSED", name))
|
|
} else {
|
|
e.printStackTrace()
|
|
Assert.fail("Test %s expected exception %s but got %s instead".format(
|
|
name, expectedException, e.getClass))
|
|
}
|
|
} else {
|
|
// we didn't expect an exception but we got one :-(
|
|
throw new RuntimeException(e)
|
|
}
|
|
} finally {
|
|
instance.shutdown()
|
|
QScriptTest.runningCommandLines -= instance
|
|
}
|
|
|
|
// catch failures from the integration test
|
|
if (expectedException != null) {
|
|
if (!gotAnException)
|
|
// we expected an exception but didn't see it
|
|
Assert.fail("Test %s expected exception %s but none was thrown".format(name, expectedException.toString))
|
|
} else {
|
|
if (CommandLineProgram.result != 0)
|
|
throw new RuntimeException("Error running the GATK with arguments: " + args)
|
|
}
|
|
}
|
|
}
|
|
|
|
object QScriptTest {
|
|
private var runningCommandLines = Set.empty[QCommandLine]
|
|
|
|
Runtime.getRuntime.addShutdownHook(new Thread {
|
|
/** Cleanup as the JVM shuts down. */
|
|
override def run = {
|
|
try {
|
|
ProcessController.shutdown()
|
|
} catch {
|
|
case _ => /*ignore */
|
|
}
|
|
runningCommandLines.foreach(commandLine =>
|
|
try {
|
|
commandLine.shutdown()
|
|
} catch {
|
|
case _ => /* ignore */
|
|
})
|
|
}
|
|
})
|
|
}
|