com.lazerycode.jmeter.json.TestConfig Maven / Gradle / Ivy

Go to download
package com.lazerycode.jmeter.json;

import java.io.File;
import java.io.FileReader;
import java.io.FileWriter;
import java.io.InputStream;
import java.nio.charset.Charset;
import java.util.List;

import org.apache.commons.io.IOUtils;
import org.apache.maven.plugin.MojoExecutionException;

import com.jayway.jsonpath.JsonPath;

/**
 * Allows user to specify the files he wants to check.
 *
 */
public class TestConfig {
	private String jsonData;

	public TestConfig(InputStream jsonFile) throws MojoExecutionException {
		try {
			jsonData = IOUtils.toString(jsonFile, Charset.forName("UTF-8"));
		} catch (Exception ex) {
			throw new MojoExecutionException(ex.getMessage(), ex);
		}
	}

	public TestConfig(File jsonFile) throws MojoExecutionException {
		try (FileReader jsonFileReader = new FileReader(jsonFile)) {
			jsonData = IOUtils.toString(jsonFileReader);
		} catch (Exception ex) {
			throw new MojoExecutionException(ex.getMessage(), ex);
		}
	}

	public String getFullConfig() {
		return jsonData;
	}

	public void writeResultFilesConfigTo(String configLocation) throws MojoExecutionException {
		try (FileWriter file = new FileWriter(configLocation)) {
			file.write(jsonData);
		} catch (Exception ex) {
			throw new MojoExecutionException(ex.getMessage(), ex);
		}
	}

	public void setResultsFileLocations(List resultFileLocations) {
		jsonData = JsonPath.parse(jsonData).set("$.resultFilesLocations", resultFileLocations).jsonString();
	}

	public List getResultsFileLocations() {
		return JsonPath.read(jsonData, "$.resultFilesLocations");
	}

	public void setResultsOutputIsCSVFormat(boolean isCSVFormat) {
		jsonData = JsonPath.parse(jsonData).set("$.resultsOutputIsCSVFormat", isCSVFormat).jsonString();
	}

	public boolean getResultsOutputIsCSVFormat() {
		return JsonPath.read(jsonData, "$.resultsOutputIsCSVFormat");
	}

    public void setGenerateReports(boolean generateReports) {
        jsonData = JsonPath.parse(jsonData).set("$.generateReports", generateReports).jsonString();
    }
    
    public boolean getGenerateReports() {
        return JsonPath.read(jsonData, "$.generateReports");
    }
    /* (non-Javadoc)
     * @see java.lang.Object#hashCode()
     */
    @Override
    public int hashCode() {
        final int prime = 31;
        int result = 1;
        result = prime * result + ((jsonData == null) ? 0 : jsonData.hashCode());
        return result;
    }

    /* (non-Javadoc)
     * @see java.lang.Object#equals(java.lang.Object)
     */
    @Override
    public boolean equals(Object obj) {
        if (this == obj) {
            return true;
        }
        if (obj == null) {
            return false;
        }
        if (!(obj instanceof TestConfig)) {
            return false;
        }
        TestConfig other = (TestConfig) obj;
        if (jsonData == null) {
            if (other.jsonData != null) {
                return false;
            }
        } else if (!jsonData.equals(other.jsonData)) {
            return false;
        }
        return true;
    }
}