package org.testng;
import org.testng.collections.Lists;
import org.testng.collections.Objects;
import org.testng.internal.SuiteRunnerMap;
import org.testng.internal.Utils;
import org.testng.internal.thread.graph.IWorker;
import org.testng.xml.XmlSuite;
import java.util.Collection;
import java.util.List;
import java.util.Map;
public class SuiteRunnerWorker implements IWorker<ISuite> {
private SuiteRunner m_suiteRunner;
private Integer m_verbose;
private String m_defaultSuiteName;
private SuiteRunnerMap m_suiteRunnerMap;
public SuiteRunnerWorker(ISuite suiteRunner,
SuiteRunnerMap suiteRunnerMap,
int verbose,
String defaultSuiteName)
{
m_suiteRunnerMap = suiteRunnerMap;
m_suiteRunner = (SuiteRunner) suiteRunner;
m_verbose = verbose;
m_defaultSuiteName = defaultSuiteName;
}
private void runSuite(SuiteRunnerMap suiteRunnerMap , XmlSuite xmlSuite)
{
if (m_verbose > 0) {
String allFiles = " " + (xmlSuite.getFileName() != null
? xmlSuite.getFileName() : m_defaultSuiteName) +
'\n';
Utils.log("TestNG", 0, "Running:\n" + allFiles);
}
SuiteRunner suiteRunner = (SuiteRunner) suiteRunnerMap.get(xmlSuite);
suiteRunner.run();
if (xmlSuite.getVerbose() > 0) {
SuiteResultCounts counts = new SuiteResultCounts();
counts.calculateResultCounts(xmlSuite, suiteRunnerMap);
StringBuilder bufLog = new StringBuilder("\n===============================================\n")
.append(xmlSuite.getName());
bufLog.append("\nTotal tests run: ")
.append(counts.m_total).append(", Failures: ").append(counts.m_failed)
.append(", Skips: ").append(counts.m_skipped);
if(counts.m_confFailures > 0 || counts.m_confSkips > 0) {
bufLog.append("\nConfiguration Failures: ").append(counts.m_confFailures)
.append(", Skips: ").append(counts.m_confSkips);
}
bufLog.append("\n===============================================\n");
System.out.println(bufLog.toString());
}
}
@Override
public void run() {
runSuite(m_suiteRunnerMap, m_suiteRunner.getXmlSuite());
}
@Override
public int compareTo(IWorker<ISuite> arg0) {
return 0;
}
@Override
public List<ISuite> getTasks() {
List<ISuite> suiteRunnerList = Lists.newArrayList();
suiteRunnerList.add(m_suiteRunner);
return suiteRunnerList;
}
@Override
public String toString() {
return Objects.toStringHelper(getClass())
.add("name", m_suiteRunner.getName())
.toString();
}
@Override
public long getTimeOut()
{
return m_suiteRunner.getXmlSuite().getTimeOut(Long.MAX_VALUE);
}
@Override
public int getPriority()
{
return 0;
}
}
class SuiteResultCounts {
int m_total = 0;
int m_skipped = 0;
int m_failed = 0;
int m_confFailures = 0;
int m_confSkips = 0;
public void calculateResultCounts(XmlSuite xmlSuite, SuiteRunnerMap suiteRunnerMap)
{
ISuite iSuite = suiteRunnerMap.get(xmlSuite);
if (iSuite != null) {
Map<String, ISuiteResult> results = iSuite.getResults();
if (results != null) {
Collection<ISuiteResult> tempSuiteResult = results.values();
for (ISuiteResult isr : tempSuiteResult) {
ITestContext ctx = isr.getTestContext();
int skipped = ctx.getSkippedTests().size();
int failed = ctx.getFailedTests().size() + ctx.getFailedButWithinSuccessPercentageTests().size();
m_skipped += skipped;
m_failed += failed;
m_confFailures += ctx.getFailedConfigurations().size();
m_confSkips += ctx.getSkippedConfigurations().size();
m_total += ctx.getPassedTests().size() + failed + skipped;
}
for (XmlSuite childSuite : xmlSuite.getChildSuites()) {
calculateResultCounts(childSuite, suiteRunnerMap);
}
}
}
}
}