ref:http://www.cnblogs.com/linglingyuese/articles/linglingyuese-two.html
我们在做性能测试时,有时需要自己编写测试脚本,很多测试工具都支持自定义编写测试脚本,比如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~\n答案是【"+resultNum+"】");//响应数据,对应结果树,其他response code等可以自己点出来 86 results.setSuccessful(true);//告诉系统返回正确还是错误 87 } else if (num > rsNum){ 88 results.setDataEncoding("UTF-8"); 89 results.setResponseData("好像大了点~~~~(>_<)~~~~ \n您输入的是【"+inNum+"】"); 90 results.setSuccessful(false); 91 }else { 92 results.setDataEncoding("UTF-8"); 93 results.setResponseData("好像小了点~~~~(>_<)~~~~ \n您输入的是【"+inNum+"】"); 94 results.setSuccessful(false); 95 } 96 97 }else{ 98 results.setDataEncoding("UTF-8"); 99 results.setResponseData("请输入数字:~~~~(>_<)~~~~ \n您输入的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测试代码示例,用于抛砖引玉,希望大家一起交流,共同 进步.