lv.ctco.cukesrest.loadrunner.junit.LoadRunnerFeature Maven / Gradle / Ivy
package lv.ctco.cukesrest.loadrunner.junit;
import com.jayway.restassured.*;
import com.jayway.restassured.filter.Filter;
import cucumber.runtime.Runtime;
import cucumber.runtime.junit.*;
import cucumber.runtime.model.*;
import lv.ctco.cukesrest.internal.*;
import lv.ctco.cukesrest.loadrunner.*;
import org.junit.runner.notification.*;
import org.junit.runners.model.*;
import java.io.*;
import java.util.*;
import java.util.logging.*;
class LoadRunnerFeature extends FeatureRunner {
public static final String LOADRUNNER_OUTPUT_DIR = "target/loadrunner_output";
private final Logger logger = Logger.getLogger(LoadRunnerFeature.class.getName());
private final String featureName;
private final LoadRunnerFilter filter;
public LoadRunnerFeature(CucumberFeature cucumberFeature, Runtime runtime, JUnitReporter jUnitReporter,
LoadRunnerFilter filter) throws InitializationError {
super(cucumberFeature, runtime, jUnitReporter);
this.filter = filter;
this.featureName = cucumberFeature.getGherkinFeature().getName();
}
@Override
public void run(RunNotifier notifier) {
RestAssured.filters(filter);
filter.createLoadRunnerAction();
super.run(notifier);
try {
new File(LOADRUNNER_OUTPUT_DIR).mkdirs();
String fileName = featureName + ".vugen";
File file = new File(LOADRUNNER_OUTPUT_DIR + File.separator + fileName);
OutputStream out = new FileOutputStream(file);
filter.dump(out);
out.close();
logger.info(file.getAbsolutePath());
ArrayList filtersCopy = new ArrayList(RestAssured.filters());
filtersCopy.remove(filter);
RestAssured.replaceFiltersWith(filtersCopy);
} catch (Exception e) {
throw new CukesInternalException(e);
}
}
}