使用Apache Jmeter编写自定义Java测试代码开展并发测试

我们在做性能测试时,有时需要自己编写测试脚本,很多测试工具都支持自定义编写测试脚本,比如LoadRunner就有很多自定义脚本的协议,比如"C Vuser","Java Vuser"等协议。同样,Jmeter也支持自定义编写的测试代码,不过与LoadRunner不同的是,Jmeter没有自带编译器,需要借助第三方编译器才能实现。

1) 下面举一个简单的Java自定义测试代码例子,使用Java编译器编写测试代码(Java编译器可以用Eclipse),实现功能为:在测试前输入任意一个字符串,然后判断该字符串的长度是否大于5,如果大于则测试结果成功,否则测试结果失败,然后再放到Jmeter中模拟10个用户测试,同时运行这段代码,具体实现如下:

打开Java编译器,新建一个项目"TestLength",然后新建包"com.jmeter".

2) 从Jmeter的安装目录lib/ext中拷贝两个文件"ApacheJMeter_core.jar"和"ApacheJMeter_java.jar"到"Tester"的项目中,然后引入这两个JAR文件.

3) 在"com.jmeter"包中新建,名字叫"TestLength",不过这个类要继承"AbstractJavaSamplerClient"类,如果项目引入步骤二中的两个文件,就可以找到"AbstractJavaSamplerClient"类了.

4) "TestLength"类在继承"AbstractJavaSamplerClient"类的同时也会继承四个方法,分别是"getDefaultParameters","setupTest","runTest"和"teardownTest"方法.

u getDefaultParameters方法主要用于设置传入的参数;

u setupTest方法为初始化方法,用于初始化性能测试时的每个线程.

u runTest方法为性能测试时的线程运行体;

u teardownTest方法为测试结束方法,用于结束性能测试中的每个线程.

5) 具体实现代码如下.

package com.jmeter; import org.apache.log.Logger; import org.apache.jmeter.config.Arguments; import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient; import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext; import org.apache.jmeter.samplers.SampleResult; /** * Jmeter自定义脚本测试类. <br> * @author ices <br> * @version 1.0.0 2013-3-18 上午8:43:13 <br> * @see AbstractJavaSamplerClient * @since JDK 1.6.0 */ public class TestLength extends AbstractJavaSamplerClient { /** * 输出到Jmeter控制台的日志类. * 需要引用Jmeter lib目录下的logkit-2.0.jar. */ private Logger log = getLogger(); /** * 运行结果. */ private SampleResult results; /** * Jmeter控制台输入的参数. */ private String testStr; /** * 初始化方法，实际运行时每个线程仅执行一次，在测试方法运行前执行，<br> * 类似于LoadRunner中的init方法. * {@inheritDoc} * @see org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient#setupTest( * org.apache.jmeter.protocol.java.sampler.JavaSamplerContext) * @author ices 2013-3-18 上午8:44:51 <br> */ public void setupTest(JavaSamplerContext arg0) { log.info("execute setupTest..."); results = new SampleResult(); testStr = arg0.getParameter("testStr", ""); if (testStr != null && testStr.length() > 0) { results.setSamplerData(testStr); } } /** * 设置传入的参数，可以设置多个，已设置的参数会显示到Jmeter的参数列表中. * {@inheritDoc} * @see org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient * #getDefaultParameters() * @author ices 2013-3-18 上午8:45:02 <br> */ public Arguments getDefaultParameters() { log.info("execute getDefaultParameters..."); Arguments params = new Arguments(); /* * 定义一个参数，显示到Jmeter的参数列表中， * 第一个参数为参数默认的显示名称， * 第二个参数为默认值 */ params.addArgument("testStr", ""); return params; } /** * 测试执行的循环体，根据线程数和循环次数的不同可执行多次，类似于LoadRunner中的Action方法 * {@inheritDoc} * @see org.apache.jmeter.protocol.java.sampler.JavaSamplerClient#runTest( * org.apache.jmeter.protocol.java.sampler.JavaSamplerContext) * @author ices 2013-3-18 上午8:45:18 <br> */ public SampleResult runTest(JavaSamplerContext arg0) { log.info("execute runTest..."); // //定义一个事务，表示这是事务的起始点，类似于LoadRunner的lr.start_transaction // results.sampleStart(); // //定义一个事务，表示这是事务的结束点，类似于LoadRunner的lr.end_transaction // results.sampleEnd(); if(testStr.length() < 5){ log.info("fail..."); //用于设置运行结果的成功或失败，如果是"false"则表示结果失败，否则则表示成功 results.setSuccessful(false); } else { log.info("Success..."); results.setSuccessful(true); } return results; } /** * 结束方法，实际运行时每个线程仅执行一次，在测试方法运行结束后执行，<br> * 类似于LoadRunner中的end方法. * {@inheritDoc} * @see org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient#teardownTest( * org.apache.jmeter.protocol.java.sampler.JavaSamplerContext) * @author ices 2013-3-18 上午8:45:53 <br> */ public void teardownTest(JavaSamplerContext arg0) { } }

6) 把上面的例子打包,然后把生成的"TestLength.jar"文件拷贝到Jmeter的安装目录lib/ext下.

7) 打开bin\jmeter.bat运行Jmeter，进入Apache Jmeter控制台。

添加一个线程组。

上图中，“Number of Thread(users)”为并发的线程数，“Loop Count”为循环次数。

然后在该线程组下面添加一个Java请求(在Sampler中)

在Java请求的类名称中选择咱们刚创建的类"com.jmeter.TestLength",在下面参数列表的"testStr"后面输入要测试的字符串，然后添加一个监听器(聚合报告)。

设置一下模拟的用户数就可以测试了.如果测试不成功,Jmeter会在它自己个输出框中抛出这个字符串.

通过上面的例子我们可以发现,使用Jmeter自定义Java测试代码,配合Jmeter自带的函数,就可以实现出LoadRunner中"Java Vuser"协议的绝大多数功能,而且是没有用户数限制和完全免费的。