package org.testng.xml;
import static org.testng.internal.Utils.isStringNotBlank;
import org.testng.collections.Lists;
import org.testng.log4testng.Logger;
import org.testng.reporters.XMLStringBuffer;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.nio.charset.Charset;
import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Properties;
public abstract class LaunchSuite {
private static final Logger LOGGER = Logger.getLogger(LaunchSuite.class);
protected boolean m_temporary;
protected LaunchSuite(boolean isTemp) {
m_temporary = isTemp;
}
public boolean isTemporary() {
return m_temporary;
}
public abstract File save(File directory);
public abstract XMLStringBuffer getSuiteBuffer();
public static class ExistingSuite extends LaunchSuite {
private File m_suitePath;
public ExistingSuite(File path) {
super(false);
m_suitePath = path;
}
@Override
public XMLStringBuffer getSuiteBuffer() {
throw new UnsupportedOperationException("Not implemented yet");
}
@Override
public File save(File directory) {
return m_suitePath;
}
}
private abstract static class CustomizedSuite extends LaunchSuite {
protected String m_projectName;
protected String m_suiteName;
protected Map<String, String> m_parameters;
private XMLStringBuffer m_suiteBuffer;
private CustomizedSuite(final String projectName,
final String className,
final Map<String, String> parameters,
final String annotationType)
{
super(true);
m_projectName = projectName;
m_suiteName = className;
m_parameters = parameters;
}
protected XMLStringBuffer createContentBuffer() {
XMLStringBuffer suiteBuffer = new XMLStringBuffer();
suiteBuffer.setDocType("suite SYSTEM \"" + Parser.TESTNG_DTD_URL + "\"");
Properties attrs = new Properties();
attrs.setProperty("parallel", XmlSuite.ParallelMode.NONE.toString());
attrs.setProperty("name", m_suiteName);
suiteBuffer.push("suite", attrs);
if (m_parameters != null) {
for (Map.Entry<String, String> entry : m_parameters.entrySet()) {
Properties paramAttrs = new Properties();
paramAttrs.setProperty("name", entry.getKey());
paramAttrs.setProperty("value", entry.getValue());
suiteBuffer.push("parameter", paramAttrs);
suiteBuffer.pop("parameter");
}
}
initContentBuffer(suiteBuffer);
suiteBuffer.pop("suite");
return suiteBuffer;
}
@Override
public XMLStringBuffer getSuiteBuffer() {
if (null == m_suiteBuffer) {
m_suiteBuffer = createContentBuffer();
}
return m_suiteBuffer;
}
protected abstract void initContentBuffer(XMLStringBuffer suiteBuffer);
@Override
public File save(File directory) {
final File suiteFile = new File(directory, "temp-testng-customsuite.xml");
saveSuiteContent(suiteFile, getSuiteBuffer());
return suiteFile;
}
protected void saveSuiteContent(final File file, final XMLStringBuffer content) {
try (OutputStreamWriter fw = new OutputStreamWriter(new FileOutputStream(file), Charset.forName("UTF-8"))) {
fw.write(content.getStringBuffer().toString());
} catch (IOException ioe) {
LOGGER.error("IO Exception", ioe);
}
}
}
static class MethodsSuite extends CustomizedSuite {
protected Collection<String> m_methodNames;
protected String m_className;
protected int m_logLevel;
MethodsSuite(final String projectName,
final String className,
final Collection<String> methodNames,
final Map<String, String> parameters,
final String annotationType,
final int logLevel) {
super(projectName, className, parameters, annotationType);
m_className = className;
m_methodNames = methodNames;
m_logLevel = logLevel;
}
@Override
protected void initContentBuffer(XMLStringBuffer suiteBuffer) {
Properties testAttrs = new Properties();
testAttrs.setProperty("name", m_className);
testAttrs.setProperty("verbose", String.valueOf(m_logLevel));
suiteBuffer.push("test", testAttrs);
suiteBuffer.push("classes");
Properties classAttrs = new Properties();
classAttrs.setProperty("name", m_className);
if ((null != m_methodNames) && (m_methodNames.size() > 0)) {
suiteBuffer.push("class", classAttrs);
suiteBuffer.push("methods");
for (Object methodName : m_methodNames) {
Properties methodAttrs = new Properties();
methodAttrs.setProperty("name", (String) methodName);
suiteBuffer.addEmptyElement("include", methodAttrs);
}
suiteBuffer.pop("methods");
suiteBuffer.pop("class");
}
else {
suiteBuffer.addEmptyElement("class", classAttrs);
}
suiteBuffer.pop("classes");
suiteBuffer.pop("test");
}
}
static class ClassesAndMethodsSuite extends CustomizedSuite {
protected Map<String, Collection<String>> m_classes;
protected int m_logLevel;
ClassesAndMethodsSuite(final String projectName,
final Map<String, Collection<String>> classes,
final Map<String, String> parameters,
final String annotationType,
final int logLevel) {
super(projectName, "Custom suite", parameters, annotationType);
m_classes = classes;
m_logLevel = logLevel;
}
@Override
protected void initContentBuffer(XMLStringBuffer suiteBuffer) {
Properties testAttrs = new Properties();
testAttrs.setProperty("name", m_projectName);
testAttrs.setProperty("verbose", String.valueOf(m_logLevel));
suiteBuffer.push("test", testAttrs);
suiteBuffer.push("classes");
for(Map.Entry<String, Collection<String>> entry : m_classes.entrySet()) {
Properties classAttrs = new Properties();
classAttrs.setProperty("name", entry.getKey());
Collection<String> methodNames= sanitize(entry.getValue());
if ((null != methodNames) && (methodNames.size() > 0)) {
suiteBuffer.push("class", classAttrs);
suiteBuffer.push("methods");
for (String methodName : methodNames) {
Properties methodAttrs = new Properties();
methodAttrs.setProperty("name", methodName);
suiteBuffer.addEmptyElement("include", methodAttrs);
}
suiteBuffer.pop("methods");
suiteBuffer.pop("class");
}
else {
suiteBuffer.addEmptyElement("class", classAttrs);
}
}
suiteBuffer.pop("classes");
suiteBuffer.pop("test");
}
private Collection<String> sanitize(Collection<String> source) {
if(null == source) {
return null;
}
List<String> result= Lists.newArrayList();
for(String name: source) {
if(isStringNotBlank(name)) {
result.add(name);
}
}
return result;
}
}
static class ClassListSuite extends CustomizedSuite {
protected Collection<String> m_packageNames;
protected Collection<String> m_classNames;
protected Collection<String> m_groupNames;
protected int m_logLevel;
ClassListSuite(final String projectName,
final Collection<String> packageNames,
final Collection<String> classNames,
final Collection<String> groupNames,
final Map<String, String> parameters,
final String annotationType,
final int logLevel) {
super(projectName, "Custom suite", parameters, annotationType);
m_packageNames = packageNames;
m_classNames = classNames;
m_groupNames = groupNames;
m_logLevel = logLevel;
}
@Override
protected void initContentBuffer(XMLStringBuffer suiteBuffer) {
Properties testAttrs = new Properties();
testAttrs.setProperty("name", m_projectName);
testAttrs.setProperty("verbose", String.valueOf(m_logLevel));
suiteBuffer.push("test", testAttrs);
if (null != m_groupNames) {
suiteBuffer.push("groups");
suiteBuffer.push("run");
for (String groupName : m_groupNames) {
Properties includeAttrs = new Properties();
includeAttrs.setProperty("name", groupName);
suiteBuffer.addEmptyElement("include", includeAttrs);
}
suiteBuffer.pop("run");
suiteBuffer.pop("groups");
}
if ((m_packageNames != null) && (m_packageNames.size() > 0)) {
suiteBuffer.push("packages");
for (String packageName : m_packageNames) {
Properties packageAttrs = new Properties();
packageAttrs.setProperty("name", packageName);
suiteBuffer.addEmptyElement("package", packageAttrs);
}
suiteBuffer.pop("packages");
}
if ((m_classNames != null) && (m_classNames.size() > 0)) {
suiteBuffer.push("classes");
for (String className : m_classNames) {
Properties classAttrs = new Properties();
classAttrs.setProperty("name", className);
suiteBuffer.addEmptyElement("class", classAttrs);
}
suiteBuffer.pop("classes");
}
suiteBuffer.pop("test");
}
}
}