package org.testng.xml;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.UUID;
import org.testng.TestNG;
import org.testng.TestNGException;
import org.testng.collections.Lists;
import org.testng.collections.Maps;
import org.testng.xml.dom.ParentSetter;
import static org.testng.xml.XmlSuite.ParallelMode.skipDeprecatedValues;
public class XmlTest implements Cloneable {
public static final int DEFAULT_TIMEOUT_MS = Integer.MAX_VALUE;
private XmlSuite m_suite;
private String m_name;
private Integer m_verbose = XmlSuite.DEFAULT_VERBOSE;
private Boolean m_isJUnit = XmlSuite.DEFAULT_JUNIT;
private int m_threadCount= -1;
private List<XmlClass> m_xmlClasses = Lists.newArrayList();
private Map<String, String> m_parameters = Maps.newHashMap();
private XmlSuite.ParallelMode m_parallel;
private List<XmlMethodSelector> m_methodSelectors = Lists.newArrayList();
private List<XmlPackage> m_xmlPackages = Lists.newArrayList();
private String m_timeOut;
private Boolean m_skipFailedInvocationCounts = XmlSuite.DEFAULT_SKIP_FAILED_INVOCATION_COUNTS;
private Map<String, List<Integer>> m_failedInvocationNumbers = null;
private Boolean m_preserveOrder = XmlSuite.DEFAULT_PRESERVE_ORDER;
private int m_index;
private Boolean m_groupByInstances;
private Boolean m_allowReturnValues = null;
private Map<String, String> m_xmlDependencyGroups = Maps.newHashMap();
public XmlTest(XmlSuite suite, int index) {
init(suite, index);
}
public XmlTest(XmlSuite suite) {
init(suite, 0);
}
private void init(XmlSuite suite, int index) {
m_suite = suite;
m_suite.getTests().add(this);
m_index = index;
m_name = TestNG.DEFAULT_COMMAND_LINE_TEST_NAME
+ " " + UUID.randomUUID().toString();
}
public XmlTest() {
}
public void setXmlPackages(List<XmlPackage> packages) {
m_xmlPackages = Lists.newArrayList(packages);
}
public List<XmlPackage> getXmlPackages() {
return m_xmlPackages;
}
public List<XmlPackage> getPackages() {
return getXmlPackages();
}
public void setPackages(List<XmlPackage> p) {
setXmlPackages(p);
}
public List<XmlMethodSelector> getMethodSelectors() {
return m_methodSelectors;
}
public void setMethodSelectors(List<XmlMethodSelector> methodSelectors) {
m_methodSelectors = Lists.newArrayList(methodSelectors);
}
public XmlSuite getSuite() {
return m_suite;
}
public List<String> getIncludedGroups() {
List<String> result = Lists.newArrayList();
if (m_xmlGroups != null && m_xmlGroups.getRun() != null) {
result.addAll(m_xmlGroups.getRun().getIncludes());
}
result.addAll(m_suite.getIncludedGroups());
return Collections.unmodifiableList(result);
}
@Deprecated
public void setClassNames(List<XmlClass> classes) {
m_xmlClasses = classes;
}
public List<XmlClass> getXmlClasses() {
return m_xmlClasses;
}
public List<XmlClass> getClasses() {
return getXmlClasses();
}
public void setClasses(List<XmlClass> c) {
setXmlClasses(c);
}
public void setXmlClasses(List<XmlClass> classes) {
m_xmlClasses = classes;
}
public String getName() {
return m_name;
}
public void setName(String name) {
m_name = name;
}
public void setVerbose(int v) {
m_verbose = v;
}
public int getThreadCount() {
return m_threadCount > 0 ? m_threadCount : getSuite().getThreadCount();
}
public void setThreadCount(int threadCount) {
m_threadCount = threadCount;
}
public void setIncludedGroups(List<String> g) {
if (m_xmlGroups == null) {
m_xmlGroups = new XmlGroups();
}
if (m_xmlGroups.getRun() == null) {
m_xmlGroups.setRun(new XmlRun());
}
List<String> includes = m_xmlGroups.getRun().getIncludes();
includes.clear();
includes.addAll(g);
}
public void setExcludedGroups(List<String> g) {
if (m_xmlGroups == null) {
m_xmlGroups = new XmlGroups();
}
if (m_xmlGroups.getRun() == null) {
m_xmlGroups.setRun(new XmlRun());
}
List<String> excludes = m_xmlGroups.getRun().getExcludes();
excludes.clear();
excludes.addAll(g);
}
public List<String> getExcludedGroups() {
List<String> result = new ArrayList<>();
if (m_xmlGroups != null && m_xmlGroups.getRun() != null) {
result.addAll(m_xmlGroups.getRun().getExcludes());
}
result.addAll(m_suite.getExcludedGroups());
return Collections.unmodifiableList(result);
}
public void addIncludedGroup(String g) {
if (m_xmlGroups == null) {
m_xmlGroups = new XmlGroups();
m_xmlGroups.setRun(new XmlRun());
}
m_xmlGroups.getRun().getIncludes().add(g);
}
public void addExcludedGroup(String g) {
if (m_xmlGroups == null) {
m_xmlGroups = new XmlGroups();
}
if (m_xmlGroups.getRun() == null) {
m_xmlGroups.setRun(new XmlRun());
}
m_xmlGroups.getRun().getExcludes().add(g);
}
public int getVerbose() {
Integer result = m_verbose;
if (null == result || XmlSuite.DEFAULT_VERBOSE.equals(m_verbose)) {
result = m_suite.getVerbose();
}
if (null != result) {
return result;
} else {
return 1;
}
}
public boolean getGroupByInstances() {
Boolean result = m_groupByInstances;
if (result == null || XmlSuite.DEFAULT_GROUP_BY_INSTANCES.equals(m_groupByInstances)) {
result = m_suite.getGroupByInstances();
}
if (result != null) {
return result;
} else {
return XmlSuite.DEFAULT_GROUP_BY_INSTANCES;
}
}
public void setGroupByInstances(boolean f) {
m_groupByInstances = f;
}
public boolean isJUnit() {
Boolean result = m_isJUnit;
if (null == result || XmlSuite.DEFAULT_JUNIT.equals(result)) {
result = m_suite.isJUnit();
}
return result;
}
public void setJUnit(boolean isJUnit) {
m_isJUnit = isJUnit;
}
public void setJunit(boolean isJUnit) {
setJUnit(isJUnit);
}
public void setSkipFailedInvocationCounts(boolean skip) {
m_skipFailedInvocationCounts = skip;
}
public boolean skipFailedInvocationCounts() {
Boolean result = m_skipFailedInvocationCounts;
if (null == result) {
result = m_suite.skipFailedInvocationCounts();
}
return result;
}
public void addMetaGroup(String name, List<String> metaGroup) {
if (m_xmlGroups == null) {
m_xmlGroups = new XmlGroups();
}
XmlDefine define = new XmlDefine();
define.setName(name);
define.getIncludes().addAll(metaGroup);
m_xmlGroups.getDefines().add(define);
}
public void addMetaGroup(String name, String... metaGroup) {
addMetaGroup(name, Arrays.asList(metaGroup));
}
public void setMetaGroups(Map<String, List<String>> metaGroups) {
for (Map.Entry<String, List<String>> entry : metaGroups.entrySet()) {
addMetaGroup(entry.getKey(), entry.getValue());
}
}
public Map<String, List<String>> getMetaGroups() {
if (m_xmlGroups == null) {
return Collections.emptyMap();
}
Map<String, List<String>> result = Maps.newHashMap();
List<XmlDefine> defines = m_xmlGroups.getDefines();
for (XmlDefine xd : defines) {
result.put(xd.getName(), xd.getIncludes());
}
return result;
}
public void setParameters(Map<String, String> parameters) {
m_parameters = parameters;
}
public void addParameter(String key, String value) {
m_parameters.put(key, value);
}
public String getParameter(String name) {
String result = m_parameters.get(name);
if (null == result) {
result = m_suite.getParameter(name);
}
return result;
}
public Map<String, String> getAllParameters() {
Map<String, String> result = Maps.newHashMap();
result.putAll(getSuite().getParameters());
result.putAll(m_parameters);
return result;
}
public Map<String, String> getLocalParameters() {
return m_parameters;
}
@Deprecated
public Map<String, String> getParameters() {
return getAllParameters();
}
@Deprecated
public Map<String, String> getTestParameters() {
return getLocalParameters();
}
public void setParallel(XmlSuite.ParallelMode parallel) {
m_parallel = skipDeprecatedValues(parallel);
}
public XmlSuite.ParallelMode getParallel() {
XmlSuite.ParallelMode result;
if (null != m_parallel || XmlSuite.DEFAULT_PARALLEL.equals(m_parallel)) {
result = m_parallel;
}
else {
result = m_suite.getParallel();
}
return result;
}
public String getTimeOut() {
String result = m_suite.getTimeOut();
if (null != m_timeOut) {
result = m_timeOut;
}
return result;
}
public long getTimeOut(long def) {
long result = def;
if (getTimeOut() != null) {
result = Long.parseLong(getTimeOut());
}
return result;
}
public void setTimeOut(long timeOut) {
m_timeOut = Long.toString(timeOut);
}
private void setTimeOut(String timeOut) {
m_timeOut = timeOut;
}
public void setExpression(String expression) {
setBeanShellExpression(expression);
}
public void setBeanShellExpression(String expression) {
List<XmlMethodSelector> selectors = getMethodSelectors();
if (selectors.size() > 0) {
selectors.get(0).setExpression(expression);
} else if (expression != null) {
XmlMethodSelector xms = new XmlMethodSelector();
xms.setExpression(expression);
xms.setLanguage("BeanShell");
getMethodSelectors().add(xms);
}
}
public String getExpression() {
List<XmlMethodSelector> selectors = getMethodSelectors();
if (selectors.size() > 0) {
return selectors.get(0).getExpression();
} else {
return null;
}
}
public String toXml(String indent) {
return XmlWeaver.asXml(this, indent);
}
@Override
public Object clone() {
XmlTest result = new XmlTest(getSuite());
result.setName(getName());
result.setIncludedGroups(getIncludedGroups());
result.setExcludedGroups(getExcludedGroups());
result.setJUnit(isJUnit());
result.setParallel(getParallel());
result.setVerbose(getVerbose());
result.setParameters(getLocalParameters());
result.setXmlPackages(getXmlPackages());
result.setTimeOut(getTimeOut());
Map<String, List<String>> metagroups = getMetaGroups();
for (Map.Entry<String, List<String>> group: metagroups.entrySet()) {
result.addMetaGroup(group.getKey(), group.getValue());
}
return result;
}
public List<Integer> getInvocationNumbers(String method) {
if (m_failedInvocationNumbers == null) {
m_failedInvocationNumbers = Maps.newHashMap();
for (XmlClass c : getXmlClasses()) {
for (XmlInclude xi : c.getIncludedMethods()) {
List<Integer> invocationNumbers = xi.getInvocationNumbers();
if (invocationNumbers.size() > 0) {
String methodName = c.getName() + "." + xi.getName();
m_failedInvocationNumbers.put(methodName, invocationNumbers);
}
}
}
}
List<Integer> result = m_failedInvocationNumbers.get(method);
if (result == null) {
return Lists.newArrayList();
} else {
return result;
}
}
@Deprecated
public void setPreserveOrder(String preserveOrder) {
setPreserveOrder(Boolean.valueOf(preserveOrder));
}
public void setPreserveOrder(Boolean preserveOrder) {
m_preserveOrder = preserveOrder;
}
public Boolean getPreserveOrder() {
if (m_preserveOrder == null) {
return m_suite.getPreserveOrder();
}
return m_preserveOrder;
}
public void setSuite(XmlSuite result) {
m_suite = result;
}
public Boolean getAllowReturnValues() {
if (m_allowReturnValues != null) return m_allowReturnValues;
else return getSuite().getAllowReturnValues();
}
public void setAllowReturnValues(Boolean allowReturnValues) {
m_allowReturnValues = allowReturnValues;
}
public int getIndex() {
return m_index;
}
@Override
public int hashCode() {
final int prime = 31;
int result = 1;
result = prime * result
+ ((m_xmlGroups == null || m_xmlGroups.getRun() == null) ? 0 : m_xmlGroups.getRun().getExcludes().hashCode());
result = prime
* result
+ ((m_failedInvocationNumbers == null) ? 0 : m_failedInvocationNumbers
.hashCode());
result = prime * result
+ ((m_xmlGroups == null || m_xmlGroups.getRun() == null) ? 0 : m_xmlGroups.getRun().getIncludes().hashCode());
result = prime * result + ((m_isJUnit == null) ? 0 : m_isJUnit.hashCode());
result = prime * result
+ ((m_xmlGroups == null) ? 0 : m_xmlGroups.getDefines().hashCode());
result = prime * result
+ ((m_methodSelectors == null) ? 0 : m_methodSelectors.hashCode());
result = prime * result + ((m_name == null) ? 0 : m_name.hashCode());
result = prime * result
+ ((m_parallel == null) ? 0 : m_parallel.hashCode());
result = prime * result
+ ((m_parameters == null) ? 0 : m_parameters.hashCode());
result = prime * result
+ ((m_preserveOrder == null) ? 0 : m_preserveOrder.hashCode());
result = prime
* result
+ ((m_skipFailedInvocationCounts == null) ? 0
: m_skipFailedInvocationCounts.hashCode());
result = prime * result + m_threadCount;
result = prime * result + ((m_timeOut == null) ? 0 : m_timeOut.hashCode());
result = prime * result + ((m_verbose == null) ? 0 : m_verbose.hashCode());
result = prime * result
+ ((m_xmlClasses == null) ? 0 : m_xmlClasses.hashCode());
result = prime * result
+ ((m_xmlPackages == null) ? 0 : m_xmlPackages.hashCode());
return result;
}
@Override
public boolean equals(Object obj) {
if (this == obj) {
return true;
}
if (obj == null)
return XmlSuite.f();
if (getClass() != obj.getClass())
return XmlSuite.f();
XmlTest other = (XmlTest) obj;
if (m_xmlGroups == null) {
if (other.m_xmlGroups != null)
return XmlSuite.f();
} else {
if (other.m_xmlGroups == null) {
return false;
}
if ((m_xmlGroups.getRun() == null && other.m_xmlGroups != null)
|| m_xmlGroups.getRun() != null && other.m_xmlGroups == null) {
return false;
}
if (!m_xmlGroups.getRun().getExcludes().equals(other.m_xmlGroups.getRun().getExcludes())) {
return XmlSuite.f();
}
if (!m_xmlGroups.getRun().getIncludes().equals(other.m_xmlGroups.getRun().getIncludes())) {
return XmlSuite.f();
}
if (!m_xmlGroups.getDefines().equals(other.m_xmlGroups.getDefines())) {
return false;
}
}
if (m_failedInvocationNumbers == null) {
if (other.m_failedInvocationNumbers != null)
return XmlSuite.f();
} else if (!m_failedInvocationNumbers
.equals(other.m_failedInvocationNumbers))
return XmlSuite.f();
if (m_isJUnit == null) {
if (other.m_isJUnit != null && ! other.m_isJUnit.equals(XmlSuite.DEFAULT_JUNIT))
return XmlSuite.f();
} else if (!m_isJUnit.equals(other.m_isJUnit))
return XmlSuite.f();
if (m_methodSelectors == null) {
if (other.m_methodSelectors != null)
return XmlSuite.f();
} else if (!m_methodSelectors.equals(other.m_methodSelectors))
return XmlSuite.f();
if (m_name == null) {
if (other.m_name != null)
return XmlSuite.f();
} else if (!m_name.equals(other.m_name))
return XmlSuite.f();
if (m_parallel == null) {
if (other.m_parallel != null)
return XmlSuite.f();
} else if (!m_parallel.equals(other.m_parallel))
return XmlSuite.f();
if (m_parameters == null) {
if (other.m_parameters != null)
return XmlSuite.f();
} else if (!m_parameters.equals(other.m_parameters))
return XmlSuite.f();
if (m_preserveOrder == null) {
if (other.m_preserveOrder != null)
return XmlSuite.f();
} else if (!m_preserveOrder.equals(other.m_preserveOrder))
return XmlSuite.f();
if (m_skipFailedInvocationCounts == null) {
if (other.m_skipFailedInvocationCounts != null)
return XmlSuite.f();
} else if (!m_skipFailedInvocationCounts
.equals(other.m_skipFailedInvocationCounts))
return XmlSuite.f();
if (m_threadCount != other.m_threadCount)
return XmlSuite.f();
if (m_timeOut == null) {
if (other.m_timeOut != null)
return XmlSuite.f();
} else if (!m_timeOut.equals(other.m_timeOut))
return XmlSuite.f();
if (m_verbose == null) {
if (other.m_verbose != null)
return XmlSuite.f();
} else if (!m_verbose.equals(other.m_verbose))
return XmlSuite.f();
if (m_xmlClasses == null) {
if (other.m_xmlClasses != null)
return XmlSuite.f();
} else if (!m_xmlClasses.equals(other.m_xmlClasses))
return XmlSuite.f();
if (m_xmlPackages == null) {
if (other.m_xmlPackages != null)
return XmlSuite.f();
} else if (!m_xmlPackages.equals(other.m_xmlPackages))
return XmlSuite.f();
return true;
}
public void addXmlDependencyGroup(String group, String dependsOn) {
if (! m_xmlDependencyGroups.containsKey(group)) {
m_xmlDependencyGroups.put(group, dependsOn);
} else {
throw new TestNGException("Duplicate group dependency found for group \"" + group + "\""
+ ", use a space-separated list of groups in the \"depends-on\" attribute");
}
}
public Map<String, String> getXmlDependencyGroups() {
if (m_xmlGroups != null) {
Map<String, String> result = Maps.newHashMap();
List<XmlDependencies> deps = m_xmlGroups.getDependencies();
for (XmlDependencies d : deps) {
result.putAll(d.getDependencies());
}
return result;
} else {
return m_xmlDependencyGroups;
}
}
@ParentSetter
public void setXmlSuite(XmlSuite suite) {
m_suite = suite;
}
private XmlGroups m_xmlGroups;
public void setGroups(XmlGroups xmlGroups) {
m_xmlGroups = xmlGroups;
}
public XmlGroups getXmlGroups() {
return m_xmlGroups;
}
public boolean nameMatchesAny(List<String> names) {
return names.contains(getName());
}
}