com.blazemeter.taurus.junit.TaurusReporter Maven / Gradle / Ivy

Go to download
package com.blazemeter.taurus.junit;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.logging.Logger;

import org.json.*;

public class TaurusReporter {

    private FileWriter outStream;
    private static final Logger log = Logger.getLogger(TaurusReporter.class.getName());

    public TaurusReporter(String file_name) {
        log.info("LDJSON file: " + file_name);
        try {
            outStream = new FileWriter(new File(file_name));
        } catch (IOException e) {
            throw new RuntimeException("Failed to open file " + file_name, e);
        }
    }

    public void writeSample(Sample sample) {
        try {
            JSONObject obj = new JSONObject();
            obj.put("test_case", sample.getLabel());
            obj.put("test_suite", sample.getSuite());
            obj.put("start_time", sample.getStartTime());
            obj.put("duration", sample.getDuration());
            obj.put("status", sample.getStatus());
            obj.put("error_msg", sample.getErrorMessage());
            obj.put("error_trace", sample.getErrorTrace());
            JSONObject extras = new JSONObject();
            extras.put("full_name", sample.getFullName());
            obj.put("extras", extras);
            outStream.write(obj.toString() + "\n");
            outStream.flush();
        } catch (Exception e) {
            log.severe("Ex: " + Utils.getStackTrace(e));
        }
    }

    public void close() throws Exception {
        log.info("Closing LDJSON reporter stream");
        outStream.close();
    }
}