mill.testng.TestNGRunner Maven / Gradle / Ivy
package mill.testng;
import com.beust.jcommander.JCommander;
import org.testng.CommandLineArgs;
import sbt.testing.*;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
class TestNGTask implements Task {
TaskDef taskDef;
TestNGRunner runner;
CommandLineArgs cliArgs;
public TestNGTask(TaskDef taskDef,
TestNGRunner runner,
CommandLineArgs cliArgs){
this.taskDef = taskDef;
this.runner = runner;
this.cliArgs = cliArgs;
}
@Override
public String[] tags() {
return new String[0];
}
@Override
public Task[] execute(EventHandler eventHandler, Logger[] loggers) {
new TestNGInstance(
loggers,
runner.testClassLoader,
cliArgs,
eventHandler
).run();
return new Task[0];
}
@Override
public TaskDef taskDef() {
return taskDef;
}
}
public class TestNGRunner implements Runner {
ClassLoader testClassLoader;
String[] args;
String[] remoteArgs;
public TestNGRunner(String[] args, String[] remoteArgs, ClassLoader testClassLoader) {
this.testClassLoader = testClassLoader;
this.args = args;
this.remoteArgs = remoteArgs;
}
public Task[] tasks(TaskDef[] taskDefs) {
CommandLineArgs cliArgs = new CommandLineArgs();
new JCommander(cliArgs, args); // args is an output parameter of the constructor!
if(cliArgs.testClass == null){
String[] names = new String[taskDefs.length];
for(int i = 0; i < taskDefs.length; i += 1){
names[i] = taskDefs[i].fullyQualifiedName();
}
cliArgs.testClass = String.join(",", names);
}
if (taskDefs.length == 0) return new Task[]{};
else return new Task[]{new TestNGTask(taskDefs[0], this, cliArgs)};
}
public String done() { return null; }
public String[] remoteArgs() { return remoteArgs; }
public String[] args() { return args; }
}