com.slickqa.resultsimport.junit.JUnitUnMarshaller Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of slick-results-import-tool Show documentation
This tool for importing various test framework reports, such as junit xml report
There is a newer version: 1.0.0-3
Show newest version
package com.slickqa.resultsimport.junit;

import com.slickqa.resultsimport.junit.model.JUnitTestSuites;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringEscapeUtils;
import com.slickqa.resultsimport.junit.model.JUnitTestSuite;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import javax.xml.stream.XMLInputFactory;
import javax.xml.stream.XMLStreamException;
import javax.xml.stream.XMLStreamReader;
import java.io.ByteArrayInputStream;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;

public class JUnitUnMarshaller {
    private static final String systemOutStart = "", systemOutEnd = "";
    private static final String systemErrStart = "", systemErrEnd = "";

    public static JUnitTestSuite unmarshalTestSuite(InputStream stream) throws JAXBException, XMLStreamException, IOException {
        JAXBContext context = JAXBContext.newInstance(JUnitTestSuite.class);

        XMLInputFactory xif = XMLInputFactory.newFactory();
        XMLStreamReader xsr = xif.createXMLStreamReader(createEscapedJUnitInputStream(stream));
        xsr = xif.createFilteredReader(xsr, reader -> {
            if (reader.getEventType() == XMLStreamReader.CHARACTERS) {
                return reader.getText().trim().length() > 0;
            }
            return true;
        });

        Unmarshaller unmarshaller = context.createUnmarshaller();
        return (JUnitTestSuite) unmarshaller.unmarshal(xsr);
    }

    public static JUnitTestSuites unmarshalTestSuites(String fileName) throws JAXBException, XMLStreamException, IOException {
        FileInputStream fileInputStreamJava;
        fileInputStreamJava =  new FileInputStream(Paths.get(fileName).toFile());

        JAXBContext context = JAXBContext.newInstance(JUnitTestSuites.class);

        XMLInputFactory xif = XMLInputFactory.newFactory();
        XMLStreamReader xsr = xif.createXMLStreamReader(createEscapedJUnitInputStream(fileInputStreamJava));
        xsr = xif.createFilteredReader(xsr, reader -> {
            if (reader.getEventType() == XMLStreamReader.CHARACTERS) {
                return reader.getText().trim().length() > 0;
            }
            return true;
        });

        Unmarshaller unmarshaller = context.createUnmarshaller();

        JUnitTestSuites tempSuites = new JUnitTestSuites();
        try {
            tempSuites = (JUnitTestSuites) unmarshaller.unmarshal(xsr);
        } catch (ClassCastException e) {
            fileInputStreamJava =  new FileInputStream(Paths.get(fileName).toFile());
            JUnitTestSuite testSuite = JUnitUnMarshaller.unmarshalTestSuite(fileInputStreamJava);
            List tempList = new ArrayList();
            tempList.add(testSuite);
            tempSuites.setTestSuites(tempList);
        }

        return tempSuites;
    }

    public static InputStream createEscapedJUnitInputStream(InputStream stream) throws IOException {
        try {

            String contents = IOUtils.toString(stream, Charset.defaultCharset());

            int systemOutStartIndex = contents.indexOf(systemOutStart);
            int systemOutEndIndex = contents.indexOf(systemOutEnd);
            if (systemOutStartIndex > 0 && systemOutEndIndex > 0) {
                systemOutStartIndex += systemOutStart.length();
                contents = contents.substring(0, systemOutStartIndex) + StringEscapeUtils.escapeXml10(contents.substring(systemOutStartIndex, systemOutEndIndex)) + contents.substring(systemOutEndIndex, contents.length());
            }

            int systemErrStartIndex = contents.indexOf(systemErrStart);
            int systemErrEndIndex = contents.indexOf(systemErrEnd);
            if (systemErrStartIndex > 0 && systemErrEndIndex > 0) {
                systemErrStartIndex += systemErrStart.length();
                contents = contents.substring(0, systemErrStartIndex) + StringEscapeUtils.escapeXml10(contents.substring(systemErrStartIndex, systemErrEndIndex)) + contents.substring(systemErrEndIndex, contents.length());
            }

            return new ByteArrayInputStream(contents.getBytes());
        } finally {
            stream.close();
        }
    }
}