se.bjurr.violations.lib.parsers.XUnitParser Maven / Gradle / Ivy

Go to download
package se.bjurr.violations.lib.parsers;

import static java.nio.charset.StandardCharsets.UTF_8;
import static se.bjurr.violations.lib.model.SEVERITY.ERROR;
import static se.bjurr.violations.lib.model.Violation.violationBuilder;
import static se.bjurr.violations.lib.reports.Parser.XUNIT;
import static se.bjurr.violations.lib.util.ViolationParserUtils.findAttribute;
import static se.bjurr.violations.lib.util.ViolationParserUtils.findIntegerAttribute;

import java.io.ByteArrayInputStream;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import javax.xml.stream.XMLStreamConstants;
import javax.xml.stream.XMLStreamReader;
import se.bjurr.violations.lib.ViolationsLogger;
import se.bjurr.violations.lib.model.Violation;
import se.bjurr.violations.lib.util.ViolationParserUtils;

public class XUnitParser implements ViolationsParser {

  @Override
  public Set parseReportOutput(
      final String string, final ViolationsLogger violationsLogger) throws Exception {
    final Set violations = new TreeSet<>();
    try (InputStream input = new ByteArrayInputStream(string.getBytes(UTF_8))) {
      final XMLStreamReader xmlr = ViolationParserUtils.createXmlReader(input);
      final Map specifics = new HashMap<>();
      String testSuiteName = null;
      String file = null;
      String message = null;
      String type = null;
      String prevLocalName = "";
      String testcaseName = "";
      while (xmlr.hasNext()) {
        final int eventType = xmlr.next();
        if (eventType == XMLStreamConstants.START_ELEMENT) {
          if (xmlr.getLocalName().equalsIgnoreCase("testsuite")) {
            prevLocalName = xmlr.getLocalName();
            final Integer failures = findIntegerAttribute(xmlr, "failures").orElse(-1);
            final Integer errors = findIntegerAttribute(xmlr, "errors").orElse(-1);
            final Integer tests = findIntegerAttribute(xmlr, "tests").orElse(-1);
            testSuiteName = findAttribute(xmlr, "name").orElse("");
            file = findAttribute(xmlr, "file").orElse("");
            specifics.put("tests", tests.toString());
            specifics.put("errors", errors.toString());
            specifics.put("failures", failures.toString());
          }
        }

        if (eventType == XMLStreamConstants.START_ELEMENT) {
          if (xmlr.getLocalName().equalsIgnoreCase("testcase")) {
            prevLocalName = xmlr.getLocalName();
            testcaseName = findAttribute(xmlr, "name").orElse("");
            specifics.put("testcaseName", testcaseName);
          }
        }
        if (eventType == XMLStreamConstants.START_ELEMENT) {
          if (xmlr.getLocalName().equalsIgnoreCase("failure")) {
            prevLocalName = xmlr.getLocalName();
            message = findAttribute(xmlr, "message").orElse("");
            type = findAttribute(xmlr, "type").orElse("");
          }
        }

        if (prevLocalName.equalsIgnoreCase("failure")
            && eventType == XMLStreamConstants.CHARACTERS) {
          prevLocalName = "";
          final String content = xmlr.getText();
          String violationFile = testSuiteName;
          if (!file.isEmpty()) {
            violationFile = file;
          }
          violations.add( //
              violationBuilder() //
                  .setFile(violationFile) //
                  .setMessage(message + "\n\n" + content) //
                  .setParser(XUNIT) //
                  .setCategory(type) //
                  .setSeverity(ERROR) //
                  .setSpecifics(specifics) //
                  .setStartLine(0) //
                  .setRule(testcaseName) //
                  .build() //
              );
        }
      }
    }
    return violations;
  }
}