com.hp.octane.integrations.testresults.GherkinXmlWritableTestResult Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of integrations-sdk Show documentation
There is a newer version: 2.24.3.5
package com.hp.octane.integrations.testresults;

import com.hp.octane.integrations.dto.tests.TestRunResult;
import org.apache.commons.lang3.StringUtils;
import org.w3c.dom.Element;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamWriter;
import java.util.HashMap;
import java.util.Map;

public class GherkinXmlWritableTestResult implements XmlWritableTestResult {
    private Map attributes;
    private Element contentElement;
    private final int ERROR_MESSAGE_MAX_SIZE = System.getProperty("octane.sdk.tests.error_message_max_size") != null ? Integer.parseInt(System.getProperty("octane.sdk.tests.error_message_max_size")) : 512*512;

    public GherkinXmlWritableTestResult(String name, Element xmlElement, long duration, TestRunResult status) {
        this.attributes = new HashMap<>();
        this.attributes.put("name", name);
        this.attributes.put("duration", String.valueOf(duration));
        this.attributes.put("status", status.value());
        this.contentElement = xmlElement;
    }

    public Map getAttributes() {
        return attributes;
    }

    public Element getXmlElement() {
        return contentElement;
    }

    public void writeXmlElement(XMLStreamWriter writer) throws XMLStreamException {
        writer.writeStartElement("gherkin_test_run");
        if (attributes != null) {
            for (Map.Entry entry : attributes.entrySet()) {
                writer.writeAttribute(entry.getKey(), entry.getValue());
            }
        }
        writeXmlElement(writer, contentElement);
        writer.writeEndElement();
    }

    private void writeXmlElement(XMLStreamWriter writer, Element rootElement) throws XMLStreamException {
        if (rootElement != null) {
            writer.writeStartElement(rootElement.getTagName());
            for (int a = 0; a < rootElement.getAttributes().getLength(); a++) {
                String attrName = rootElement.getAttributes().item(a).getNodeName();
                writer.writeAttribute(attrName, rootElement.getAttribute(attrName));
            }
            NodeList childNodes = rootElement.getChildNodes();
            for (int c = 0; c < childNodes.getLength(); c++) {
                Node child = childNodes.item(c);
                if (child instanceof Element) {
                    writeXmlElement(writer, (Element) child);
                } else if (child.getNodeType() == Node.CDATA_SECTION_NODE) {
                    if(child.getParentNode() != null && "error_message".equals(child.getParentNode().getNodeName())){
                        String errorMassage = child.getNodeValue();
                        errorMassage = errorMassage.length() > ERROR_MESSAGE_MAX_SIZE ? StringUtils.abbreviate(errorMassage,ERROR_MESSAGE_MAX_SIZE) : errorMassage;
                        writer.writeCharacters(errorMassage);
                    } else {
                        writer.writeCharacters(child.getNodeValue());
                    }
                }
            }
            writer.writeEndElement();
        }
    }
}