io.cucumber.junitxmlformatter.MessagesToJunitXmlWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of junit-xml-formatter Show documentation

Renders Cucumber Messages as JUnit XML

There is a newer version: 0.7.1

package io.cucumber.junitxmlformatter;

import io.cucumber.messages.types.Envelope;

import javax.xml.stream.XMLStreamException;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.nio.charset.StandardCharsets;

import static java.util.Objects.requireNonNull;

/**
 * Writes the message output of a test run as single page html report.
 * 
 * Note: Messages are first collected and only written once the stream is closed.
 *
 * @see Cucumber JUnit XML Formatter - README.md
 */
public class MessagesToJunitXmlWriter implements AutoCloseable {

    private final OutputStreamWriter out;
    private final XmlReportData data = new XmlReportData();
    private boolean streamClosed = false;

    public MessagesToJunitXmlWriter(OutputStream out) {
        this.out = new OutputStreamWriter(
                requireNonNull(out),
                StandardCharsets.UTF_8
        );
    }

    /**
     * Writes a cucumber message to the xml output.
     *
     * @param envelope the message
     * @throws IOException if an IO error occurs
     */
    public void write(Envelope envelope) throws IOException {
        if (streamClosed) {
            throw new IOException("Stream closed");
        }
        data.collect(envelope);
    }

    /**
     * Closes the stream, flushing it first. Once closed further write()
     * invocations will cause an IOException to be thrown. Closing a closed
     * stream has no effect.
     *
     * @throws IOException if an IO error occurs
     */
    @Override
    public void close() throws IOException {
        if (streamClosed) {
            return;
        }

        try {
            new XmlReportWriter(data).writeXmlReport(out);
        } catch (XMLStreamException e) {
            throw new IOException("Error while transforming.", e);
        } finally {
            try {
                out.close();
            } finally {
                streamClosed = true;
            }
        }
    }
}