1 files changed, 76 insertions, 0 deletions
diff --git a/contrib/testng/src/mill/testng/TestNGRunner.java b/contrib/testng/src/mill/testng/TestNGRunner.java
new file mode 100644
index 00000000..0ad05f76
--- /dev/null
+++ b/contrib/testng/src/mill/testng/TestNGRunner.java
@@ -0,0 +1,76 @@
+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; }
+}