最近准备给jmeter增加一个web页面,不想通过上传jmx文件的方式,而是直接根据所填的信息生成jmx文件,下面展示一个最简单的http请求,其他的组件可以参考下面的组装。
import org.apache.jmeter.config.Arguments;
import org.apache.jmeter.control.LoopController;
import org.apache.jmeter.engine.JMeterEngine;
import org.apache.jmeter.engine.JMeterEngineException;
import org.apache.jmeter.engine.StandardJMeterEngine;
import org.apache.jmeter.protocol.http.control.Header;
import org.apache.jmeter.protocol.http.control.HeaderManager;
import org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy;
import org.apache.jmeter.protocol.http.util.HTTPArgument;
import org.apache.jmeter.reporters.ResultCollector;
import org.apache.jmeter.samplers.SampleSaveConfiguration;
import org.apache.jmeter.save.SaveService;
import org.apache.jmeter.testelement.TestElement;
import org.apache.jmeter.testelement.TestPlan;
import org.apache.jmeter.testelement.property.*;
import org.apache.jmeter.util.JMeterUtils;
import org.apache.jorphan.collections.HashTree;
import org.apache.jorphan.collections.ListedHashTree;
import org.testng.annotations.Test;
import org.apache.jmeter.threads.ThreadGroup;
import java.io.File;
import java.io.IOException;
import java.util.*;
public class JmeterTest {
static{
{
JMeterUtils.setJMeterHome(new File("").getAbsolutePath());
JMeterUtils.loadJMeterProperties(new File("src/main/resources/jmeter/jmeter.properties").getAbsolutePath());
JMeterUtils.setProperty("saveservice_properties", "src/main/resources/jmeter/saveservice.properties");
JMeterUtils.setProperty("user_properties", "src/main/resources/jmeter/user.properties");
JMeterUtils.setProperty("upgrade_properties", "src/main/resources/jmeter/upgrade.properties");
JMeterUtils.setProperty("system_properties", "src/main/resources/jmeter/system.properties");
// JMeterUtils.setProperty("proxy.cert.directory", new File("").getAbsolutePath());
JMeterUtils.setLocale(Locale.CHINESE);
JMeterUtils.initLocale();
}
}
@Test
public void test() throws JMeterEngineException, IOException {
File file = new File("D:\\jmeter_jmx\\HTTP Request.jmx");
HashTree hashTree = SaveService.loadTree(file);
JMeterEngine engine = new StandardJMeterEngine();
engine.configure(hashTree);
((StandardJMeterEngine) engine).run();
}
@Test
public void test04() throws IOException {
String url = "172.16.2.143";
String port = "8080";
String api = "/submit/job";
String request = "{}";
//TestPlan
TestPlan testPlan = new TestPlan("Test Plan");
testPlan.setFunctionalMode(false);
testPlan.setSerialized(false);
testPlan.setTearDownOnShutdown(true);
testPlan.setProperty(TestElement.TEST_CLASS, TestPlan.class.getName());
testPlan.setProperty(TestElement.GUI_CLASS,"TestPlanGui");
testPlan.setProperty(new BooleanProperty(TestElement.ENABLED, true));
testPlan.setProperty(new StringProperty("TestPlan.comments", ""));
testPlan.setProperty(new StringProperty("TestPlan.user_define_classpath", ""));
Arguments arguments = new Arguments();
testPlan.setProperty(new TestElementProperty("TestPlan.user_defined_variables",arguments));
ThreadGroup threadGroup = new ThreadGroup();
threadGroup.setNumThreads(1);
threadGroup.setRampUp(1);
threadGroup.setDelay(0);
threadGroup.setDuration(0);
threadGroup.setProperty(new StringProperty(ThreadGroup.ON_SAMPLE_ERROR, "continue"));
threadGroup.setScheduler(false);
threadGroup.setName("Group1");
threadGroup.setProperty(TestElement.TEST_CLASS, ThreadGroup.class.getName());
threadGroup.setProperty(TestElement.GUI_CLASS,"ThreadGroup");
threadGroup.setProperty(new BooleanProperty(TestElement.ENABLED, true));
LoopController loopController = new LoopController();
loopController.setLoops(0);
loopController.setProperty(new BooleanProperty("LoopController.continue_forever",false));
loopController.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.control.gui.LoopControlPanel"));
loopController.setProperty(new StringProperty("TestElement.test_class","org.apache.jmeter.control.LoopController"));
loopController.setProperty(new StringProperty("TestElement.name","Loop Controller"));
loopController.setProperty(new StringProperty("TestElement.enabled","true"));
loopController.setProperty(new StringProperty("LoopController.loops","1"));
threadGroup.setProperty(new TestElementProperty("ThreadGroup.main_controller",loopController));
HTTPSamplerProxy httpSamplerProxy = new HTTPSamplerProxy();
Arguments HTTPsamplerArguments = new Arguments();
HTTPArgument httpArgument = new HTTPArgument();
httpArgument.setProperty(new BooleanProperty("HTTPArgument.always_encode",false));
httpArgument.setProperty(new StringProperty("Argument.value", request));
httpArgument.setProperty(new StringProperty("Argument.metadata","="));
ArrayList<TestElementProperty> list1 = new ArrayList<>();
list1.add(new TestElementProperty("",httpArgument));
HTTPsamplerArguments.setProperty(new CollectionProperty("Arguments.arguments",list1));
httpSamplerProxy.setProperty(new TestElementProperty("HTTPsampler.Arguments",HTTPsamplerArguments));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.domain", url));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.port", port));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.protocol", "http"));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.path", api));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.method", "POST"));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.contentEncoding", ""));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.follow_redirects", true));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.postBodyRaw", true));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.auto_redirects", false));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.use_keepalive", true));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.DO_MULTIPART_POST", false));
httpSamplerProxy.setProperty(new StringProperty("TestElement.gui_class", "org.apache.jmeter.protocol.http.control.gui.HttpTestSampleGui"));
httpSamplerProxy.setProperty(new StringProperty("TestElement.test_class", "org.apache.jmeter.protocol.http.sampler.HTTPSamplerProxy"));
httpSamplerProxy.setProperty(new StringProperty("TestElement.name", "HTTP Request"));
httpSamplerProxy.setProperty(new StringProperty("TestElement.enabled", "true"));
httpSamplerProxy.setProperty(new BooleanProperty("HTTPSampler.postBodyRaw", true));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.embedded_url_re", ""));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.connect_timeout", ""));
httpSamplerProxy.setProperty(new StringProperty("HTTPSampler.response_timeout", ""));
ResultCollector resultCollector = new ResultCollector();
HeaderManager headerManager = new HeaderManager();
SampleSaveConfiguration sampleSaveConfiguration = new SampleSaveConfiguration();
sampleSaveConfiguration.setTime(true);
sampleSaveConfiguration.setLatency(true);
sampleSaveConfiguration.setTimestamp(true);
sampleSaveConfiguration.setSuccess(true);
sampleSaveConfiguration.setLabel(true);
sampleSaveConfiguration.setCode(true);
sampleSaveConfiguration.setMessage(true);
sampleSaveConfiguration.setThreadName(true);
sampleSaveConfiguration.setDataType(true);
sampleSaveConfiguration.setEncoding(false);
sampleSaveConfiguration.setAssertions(true);
sampleSaveConfiguration.setSubresults(true);
sampleSaveConfiguration.setResponseData(false);
sampleSaveConfiguration.setSamplerData(false);
sampleSaveConfiguration.setAsXml(false);
sampleSaveConfiguration.setFieldNames(true);
sampleSaveConfiguration.setResponseHeaders(false);
sampleSaveConfiguration.setRequestHeaders(false);
//sampleSaveConfiguration.setAssertionResultsFailureMessage(true); responseDataOnError
sampleSaveConfiguration.setAssertionResultsFailureMessage(true);
//sampleSaveConfiguration.setsserAtionsResultsToSave(0); assertionsResultsToSave
sampleSaveConfiguration.setBytes(true);
sampleSaveConfiguration.setSentBytes(true);
sampleSaveConfiguration.setUrl(true);
sampleSaveConfiguration.setThreadCounts(true);
sampleSaveConfiguration.setIdleTime(true);
sampleSaveConfiguration.setConnectTime(true);
resultCollector.setProperty(new BooleanProperty("ResultCollector.error_logging",false));
resultCollector.setProperty(new ObjectProperty("saveConfig",new SampleSaveConfiguration()));
resultCollector.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.visualizers.ViewResultsFullVisualizer"));
resultCollector.setProperty(new StringProperty("TestElement.name","View Results Tree"));
resultCollector.setProperty(new StringProperty("TestElement.enabled","true"));
resultCollector.setProperty(new StringProperty("filename",""));
ArrayList<TestElementProperty> list2 = new ArrayList<>();
Header header = new Header();
header.setProperty(new StringProperty("Header.name","Content-Type"));
header.setProperty(new StringProperty("Header.value","application/json"));
TestElementProperty HeaderElement = new TestElementProperty("",header);
list2.add(HeaderElement);
headerManager.setProperty(new CollectionProperty("HeaderManager.headers",list2));
headerManager.setProperty(new StringProperty("TestElement.test_class","org.apache.jmeter.protocol.http.control.HeaderManager"));
headerManager.setProperty(new StringProperty("TestElement.name","HTTP Header Manager"));
headerManager.setProperty(new StringProperty("TestElement.enabled","true"));
headerManager.setProperty(new StringProperty("TestElement.gui_class","org.apache.jmeter.protocol.http.gui.HeaderPanel"));
ListedHashTree hashTreeResultCollectorAndHeaderManager = new ListedHashTree();
hashTreeResultCollectorAndHeaderManager.add(resultCollector);
hashTreeResultCollectorAndHeaderManager.add(headerManager);
ListedHashTree hashTreeHTTPSamplerProxy = new ListedHashTree();
hashTreeHTTPSamplerProxy.add(httpSamplerProxy,hashTreeResultCollectorAndHeaderManager);
ListedHashTree hashTreeThreadGroup = new ListedHashTree();
hashTreeThreadGroup.add(threadGroup,hashTreeHTTPSamplerProxy);
ListedHashTree hashTreeTestPlan = new ListedHashTree();
hashTreeTestPlan.add(testPlan,hashTreeThreadGroup);
//可以将生成的hashTree保存成一个jmx文件 与Jmeter自己生成的做对比
//SaveService.saveTree(hashTreeTestPlan, new FileOutputStream("D:\\jmeter_jmx\\test.jmx"));
JMeterEngine engine = new StandardJMeterEngine();
//engine.configure(SaveService.loadTree(new File("D:\\jmeter_jmx\\test.jmx")));
engine.configure(hashTreeTestPlan);
((StandardJMeterEngine) engine).run();
}
}
上方代码中Test04生成的jmx文件如下所示
<?xml version="1.0" encoding="UTF-8"?>
<jmeterTestPlan version="1.2" properties="5.0" jmeter="5.2.1">
<hashTree>
<TestPlan guiclass="TestPlanGui" testclass="TestPlan" testname="Test Plan" enabled="true">
<boolProp name="TestPlan.functional_mode">false</boolProp>
<boolProp name="TestPlan.serialize_threadgroups">false</boolProp>
<boolProp name="TestPlan.tearDown_on_shutdown">true</boolProp>
<stringProp name="TestPlan.comments"></stringProp>
<stringProp name="TestPlan.user_define_classpath"></stringProp>
<elementProp name="TestPlan.user_defined_variables" elementType="Arguments">
<collectionProp name="Arguments.arguments"/>
</elementProp>
</TestPlan>
<hashTree>
<ThreadGroup guiclass="ThreadGroup" testclass="ThreadGroup" testname="Group1" enabled="true">
<intProp name="ThreadGroup.num_threads">1</intProp>
<intProp name="ThreadGroup.ramp_time">1</intProp>
<longProp name="ThreadGroup.delay">0</longProp>
<longProp name="ThreadGroup.duration">0</longProp>
<stringProp name="ThreadGroup.on_sample_error">continue</stringProp>
<boolProp name="ThreadGroup.scheduler">false</boolProp>
<elementProp name="ThreadGroup.main_controller" elementType="LoopController" guiclass="LoopControlPanel" testclass="LoopController" testname="Loop Controller" enabled="true">
<boolProp name="LoopController.continue_forever">false</boolProp>
<stringProp name="LoopController.loops">1</stringProp>
</elementProp>
</ThreadGroup>
<hashTree>
<HTTPSamplerProxy guiclass="HttpTestSampleGui" testclass="HTTPSamplerProxy" testname="HTTP Request" enabled="true">
<elementProp name="HTTPsampler.Arguments" elementType="Arguments">
<collectionProp name="Arguments.arguments">
<elementProp name="" elementType="HTTPArgument">
<boolProp name="HTTPArgument.always_encode">false</boolProp>
<stringProp name="Argument.value">{}</stringProp>
<stringProp name="Argument.metadata">=</stringProp>
</elementProp>
</collectionProp>
</elementProp>
<stringProp name="HTTPSampler.domain">172.16.2.143</stringProp>
<stringProp name="HTTPSampler.port">8080</stringProp>
<stringProp name="HTTPSampler.protocol">http</stringProp>
<stringProp name="HTTPSampler.path">/submit/job</stringProp>
<stringProp name="HTTPSampler.method">POST</stringProp>
<stringProp name="HTTPSampler.contentEncoding"></stringProp>
<boolProp name="HTTPSampler.follow_redirects">true</boolProp>
<boolProp name="HTTPSampler.postBodyRaw">true</boolProp>
<boolProp name="HTTPSampler.auto_redirects">false</boolProp>
<boolProp name="HTTPSampler.use_keepalive">true</boolProp>
<boolProp name="HTTPSampler.DO_MULTIPART_POST">false</boolProp>
<stringProp name="HTTPSampler.embedded_url_re"></stringProp>
<stringProp name="HTTPSampler.connect_timeout"></stringProp>
<stringProp name="HTTPSampler.response_timeout"></stringProp>
</HTTPSamplerProxy>
<hashTree>
<ResultCollector guiclass="ViewResultsFullVisualizer" testname="View Results Tree" enabled="true">
<boolProp name="ResultCollector.error_logging">false</boolProp>
<objProp>
<name>saveConfig</name>
<value class="SampleSaveConfiguration">
<time>true</time>
<latency>true</latency>
<timestamp>true</timestamp>
<success>true</success>
<label>true</label>
<code>true</code>
<message>true</message>
<threadName>true</threadName>
<dataType>true</dataType>
<encoding>false</encoding>
<assertions>true</assertions>
<subresults>true</subresults>
<responseData>false</responseData>
<samplerData>false</samplerData>
<xml>false</xml>
<fieldNames>true</fieldNames>
<responseHeaders>false</responseHeaders>
<requestHeaders>false</requestHeaders>
<responseDataOnError>false</responseDataOnError>
<saveAssertionResultsFailureMessage>true</saveAssertionResultsFailureMessage>
<assertionsResultsToSave>0</assertionsResultsToSave>
<bytes>true</bytes>
<sentBytes>true</sentBytes>
<url>true</url>
<threadCounts>true</threadCounts>
<idleTime>true</idleTime>
<connectTime>true</connectTime>
</value>
</objProp>
<stringProp name="filename"></stringProp>
</ResultCollector>
<hashTree/>
<HeaderManager guiclass="HeaderPanel" testclass="HeaderManager" testname="HTTP Header Manager" enabled="true">
<collectionProp name="HeaderManager.headers">
<elementProp name="" elementType="Header">
<stringProp name="Header.name">Content-Type</stringProp>
<stringProp name="Header.value">application/json</stringProp>
</elementProp>
</collectionProp>
</HeaderManager>
<hashTree/>
</hashTree>
</hashTree>
</hashTree>
</hashTree>
</jmeterTestPlan>
如果需要添加其他元素的,可以参考jmeter生成jmx的结构,在此基础上自行拼装