org.continuity.commons.jmeter.TestPlanWriter Maven / Gradle / Ivy

Go to download
package org.continuity.commons.jmeter;

import java.io.IOException;
import java.nio.file.Path;
import java.util.Locale;
import java.util.Map;

import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.ListedHashTree;

import net.sf.markov4jmeter.testplangenerator.JMeterEngineGateway;
import net.sf.markov4jmeter.testplangenerator.TestPlanGenerator;
import net.sf.markov4jmeter.testplangenerator.util.CSVHandler;

/**
 * @author Henning Schulz
 *
 */
public class TestPlanWriter {

	/**
	 * Default contructor not initializing JMeter.
	 */
	public TestPlanWriter() {
	}

	/**
	 * Creates and initializes JMeter.
	 *
	 * @param jmeterHome
	 *            The path to the root JMeter configuration folder.
	 */
	public TestPlanWriter(String jmeterHome) {
		init(jmeterHome);
	}

	/**
	 * Initializes JMeter.
	 *
	 * @param jmeterHome
	 *            The path to the root JMeter configuration folder.
	 */
	public void init(String jmeterHome) {
		JMeterEngineGateway.getInstance().initJMeter(jmeterHome, "bin/jmeter.properties", Locale.ENGLISH);
		JMeterUtils.initLogging();
	}

	private final GeneratorAdapter generatorAdapter = new GeneratorAdapter();

	private final CSVHandler csvHandler = new CSVHandler(CSVHandler.LINEBREAK_TYPE_UNIX);

	public Path write(ListedHashTree testPlanTree, Map behavior, Path outputDir) {
		Path testplanPath = outputDir.resolve("testplan.jmx");
		boolean succeeded = generatorAdapter.writeOutput(testPlanTree, testplanPath.toString());

		for (Map.Entry entry : behavior.entrySet()) {
			Path csvPath = outputDir.resolve(entry.getKey());

			try {
				csvHandler.writeValues(csvPath.toString(), entry.getValue());
			} catch (SecurityException | NullPointerException | IOException e) {
				e.printStackTrace();
				succeeded = false;
				break;
			}
		}

		if (!succeeded) {
			return null;
		}

		return testplanPath;
	}

	private static class GeneratorAdapter extends TestPlanGenerator {

		@Override
		public boolean writeOutput(ListedHashTree testPlanTree, String outputFilename) {
			return super.writeOutput(testPlanTree, outputFilename);
		}

	}

}