package test; import org.testng.TestNG; import org.testng.xml.XmlClass; import org.testng.xml.XmlSuite; import org.testng.xml.XmlTest; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; import java.util.Collections; public class TestHelper { public static XmlSuite createSuite(String cls, String suiteName) { return createSuite(cls, suiteName, "TmpTest"); } private static XmlSuite createSuite(String cls, String suiteName, String testName) { XmlSuite result = new XmlSuite(); result.setName(suiteName); XmlTest test = new XmlTest(result); test.setName(testName); test.setXmlClasses(Collections.singletonList(new XmlClass(cls))); return result; } public static TestNG createTestNG() throws IOException { return createTestNG(createRandomDirectory()); } public static TestNG createTestNG(XmlSuite suite) throws IOException { return createTestNG(suite, createRandomDirectory()); } public static TestNG createTestNG(XmlSuite suite, Path outputDir) throws IOException { TestNG result = createTestNG(outputDir); result.setXmlSuites(Collections.singletonList(suite)); return result; } private static TestNG createTestNG(Path outputDir) throws IOException { TestNG result = new TestNG(); result.setOutputDirectory(outputDir.toAbsolutePath().toString()); result.setVerbose(-1); return result; } public static Path createRandomDirectory() throws IOException { return Files.createTempDirectory("testng-tmp-"); } }