Jmeter使用自定义编写代码

原文地址：http://blog.csdn.net/li_ok/article/details/1487685

我们在做性能测试时,有时需要自己编写测试脚本,很多测试工具都支持自定义编写测试脚本,比如LoadRunner就有很多自定义脚本的协议,比如"C Vuser","Java Vuser"等协议.同样,Jmeter也支持自定义编写的测试代码,不过与LoadRunner不同的是,Jmeter没有自带编译器,需要借助第三方编译器才能实现.下面举一个简单的Java自定义测试代码例子,使用Java编译器编写测试代码(Java编译器可以用Eclipse,JBulider等),实现功能为:判断输入的数字是否等于你指定的数，如果等于，则返回成功，如果小于，则提示该输入数字太小，如果大于，则提示该数字太大，如果不为数字，提示输入数字。然后在放到Jmeter中模拟10个用户测试,同时运行这段代码,具体实现如下:

一、开始编写前的准备

1、打开Java编译器，新建一个项目TestNumber，然后新建一个包test。

2、从Jmeter的安装目录lib/ext中拷贝两个文件ApacheJMeter_core.jar和ApacheJMeter_java.jar到TestNumber项目中，然后引入这两个JAR包。

3、在test包中新建一个类，类名为TestNum，该类继承AbstractJavaSamplerClient类，AbstractJavaSamplerClient存在于ApacheJMeter_java.jar这个JAR包中，引用即可调用。

4、TestNum类在继承AbstractJavaSamplerClient类的时候，需要实现四个方法，分别是

setupTest()：初始化方法，用于初始化性能测试时的每个线程；

getDefaultParameters()：主要用于设置传入的参数；

runTest()：为性能测试时的线程运行体；

teardownTest()：测试结束方法，用于结束性能测试中的每个线程。

 

二、具体的代码实现

1 package test;
  2 
  3 import org.apache.jmeter.config.Arguments;
  4 import org.apache.jmeter.protocol.java.sampler.AbstractJavaSamplerClient;
  5 import org.apache.jmeter.protocol.java.sampler.JavaSamplerContext;
  6 import org.apache.jmeter.samplers.SampleResult;
  7 
  8 public class TestNum extends AbstractJavaSamplerClient{
  9     
 10     private SampleResult results;
 11     
 12     /**
 13      * 输入的数字
 14      */
 15     private String inNum;
 16     
 17     /**
 18      * 需要匹配的数字
 19      */
 20     private String resultNum;
 21 
 22     /**
 23      * 初始化方法，初始化性能测试时的每个线程
 24      * 实际运行时每个线程仅执行一次，在测试方法运行前执行，类似于LoadRunner中的init方法
 25      */
 26     public void setupTest(JavaSamplerContext jsc) {
 27         results = new SampleResult();
 28         
 29         inNum = jsc.getParameter("inNum", "");
 30         resultNum = jsc.getParameter("resultNum", "");
 31         
 32         if (inNum != null && inNum.length() > 0){
 33             results.setSamplerData(inNum);
 34         }
 35         
 36         if (resultNum != null && resultNum.length() > 0){
 37             results.setSamplerData(resultNum);
 38         }
 39     }
 40     
 41     /**
 42      * 设置传入参数
 43      * 可以设置多个，已设置的参数会显示到Jmeter参数列表中
 44      */
 45     public Arguments getDefaultParameters() {
 46         Arguments params = new Arguments();
 47         params.addArgument("inNum","");
 48         params.addArgument("resultNum", "66");
 49         return params;
 50     }
 51 
 52     /**
 53      * 性能测试时的线程运行体
 54      * 测试执行的循环体，根据线程数和循环次数的不同可执行多次，类似于Loadrunner中的Action方法
 55      */
 56     public SampleResult runTest(JavaSamplerContext arg0) {
 57         boolean flag = false;
 58         //定义一个事务，表示这是事务的起始点，类似于Loadrunner中的lr.start_transaction
 59         results.sampleStart();
 60         
 61         for (int i = inNum.length();--i >= 0;){
 62             if (!Character.isDigit(inNum.charAt(i))){
 63                 flag = false;
 64             }else{
 65                 flag = true;
 66             }
 67         }
 68         
 69         for (int j = resultNum.length();--j >= 0;){
 70             if (!Character.isDigit(resultNum.charAt(j))){
 71                 flag = false;
 72             }else{
 73                 flag = true;
 74             }
 75         }
 76         //定义一个事务，表示这是事务的结束点，类似于Loadrunner中的lr.end_transaction
 77         results.sampleEnd();
 78         
 79         if (flag){
 80             Integer num = Integer.parseInt(inNum);
 81             Integer rsNum = Integer.parseInt(resultNum);
 82             
 83             if (num == rsNum){
 84                 results.setDataEncoding("UTF-8");//因为响应的数据有中文，所以最好先设置编码
 85                 results.setResponseData("恭喜你，答对了O(∩_∩)O~
答案是【"+resultNum+"】");//响应数据，对应结果树，其他response code等可以自己点出来
 86                 results.setSuccessful(true);//告诉系统返回正确还是错误
 87             } else if (num > rsNum){
 88                 results.setDataEncoding("UTF-8");
 89                 results.setResponseData("好像大了点~~~~(>_<)~~~~ 
您输入的是【"+inNum+"】");
 90                 results.setSuccessful(false);
 91             }else {
 92                 results.setDataEncoding("UTF-8");
 93                 results.setResponseData("好像小了点~~~~(>_<)~~~~ 
您输入的是【"+inNum+"】");
 94                 results.setSuccessful(false);
 95             }
 96             
 97         }else{
 98             results.setDataEncoding("UTF-8");
 99             results.setResponseData("请输入数字：~~~~(>_<)~~~~ 
您输入的inNum是【"+inNum+"】，resultNum是【"+resultNum+"】");
100             results.setSuccessful(false);
101         }
102         
103         return results;
104     }
105 
106     /**
107      * 测试结束方法，结束测试中的每个线程
108      * 实际运行时，每个线程仅执行一次，在测试方法运行结束后执行，类似于Loadrunner中的End方法
109      */
110     public void teardownTest(JavaSamplerContext arg0) {
111     }
112 
113 }

三、Jmeter运行分析

1、将上述代码打包成jar包，生成的包名称为TestNumber.jar，将jar包拷贝到Jmeter的安装目录lib/ext下面。

2、运行Jmeter，添加线程组及java请求，显示如下：

其中，inNum为我们输入的值，因为需要用到多个用户，避免填写的麻烦，我们用随机数函数来随机抽取数字验证，resultNum为匹配的结果。

3、添加监听器，这里我们添加查看结果树和聚合报告就好。

4、结果显示如下图：

查看结果树：

我们可以看到，10个请求中，有一个请求回答正确，响应数据对应了我们的代码，表示执行成功。

聚合报告：

可以看到如上信息，请求用户为10个，因为我们的代码基本上没有任何含义，执行速度很快，所以其他值基本为0.

通过上面的例子我们可以发现,使用Jmeter自定义Java测试代码,配合Jmeter自带的函数,就可以实现出LoadRunner中"Java Vuser"协议的绝大多数功能,而且是没有用户数限制和完全免费的(嘿嘿).上面的例子非常简单,而且没有任何实际意义,只是一个简单的Jmeter测试代码示例,用于抛砖引玉,希望大家一起交流,共同 进步. 

 资料下载地址：http://files.cnblogs.com/linglingyuese/TestNumber.zip