package org.testng.reporters.jq;
import static org.testng.reporters.jq.BasePanel.C;
import static org.testng.reporters.jq.BasePanel.D;
import org.testng.IReporter;
import org.testng.ISuite;
import org.testng.internal.Utils;
import org.testng.reporters.Files;
import org.testng.reporters.XMLStringBuffer;
import org.testng.xml.XmlSuite;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.Arrays;
import java.util.List;
public class Main implements IReporter {
private static final String[] RESOURCES = new String[] {
"jquery-1.7.1.min.js", "testng-reports.css", "testng-reports.js",
"passed.png", "failed.png", "skipped.png", "navigator-bullet.png",
"bullet_point.png", "collapseall.gif"
};
private Model m_model;
private String m_outputDirectory;
@Override
public void generateReport(List<XmlSuite> xmlSuites, List<ISuite> suites,
String outputDirectory) {
m_model = new Model(suites);
m_outputDirectory = outputDirectory;
XMLStringBuffer xsb = new XMLStringBuffer(" ");
// Generate the top banner
new BannerPanel(m_model).generate(xsb);
// All the panels selectable from the navigator
List<INavigatorPanel> panels = Arrays.<INavigatorPanel>asList(
new TestNgXmlPanel(m_model),
new TestPanel(m_model),
new GroupPanel(m_model),
new TimesPanel(m_model),
new ReporterPanel(m_model),
new IgnoredMethodsPanel(m_model),
new ChronologicalPanel(m_model));
// Generate the navigator on the left hand side
new NavigatorPanel(m_model, panels).generate(xsb);
xsb.push(D, C, "wrapper");
xsb.push(D, "class", "main-panel-root");
//
// Generate the main suite panel
//
new SuitePanel(m_model).generate(xsb);
// Generate all the navigator panels
for (INavigatorPanel panel : panels) {
panel.generate(xsb);
}
xsb.pop(D); // main-panel-root
xsb.pop(D); // wrapper
xsb.addString(" </body>\n");
xsb.addString("</html>\n");
String all;
try {
InputStream header = getClass().getResourceAsStream("/header");
if (header == null) {
throw new RuntimeException("Couldn't find resource header");
} else {
for (String fileName : RESOURCES) {
InputStream is = getClass().getResourceAsStream("/" + fileName);
if (is == null) {
throw new AssertionError("Couldn't find resource: " + fileName);
}
Files.copyFile(is, new File(m_outputDirectory, fileName));
}
all = Files.readFile(header);
Utils.writeUtf8File(m_outputDirectory, "index.html", xsb, all);
}
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
}
}