Jmeter使用自定義編寫代碼

轉自：https://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()：測試結束方法，用於結束性能測試中的每個線程。

 

二、具體的代碼實現

package test;

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;

public class TestNum extends AbstractJavaSamplerClient{

    private SampleResult results;

    /**
     * 輸入的數字
     */
    private String inNum;

    /**
     * 需要匹配的數字
     */
    private String resultNum;

    /**
     * 初始化方法，初始化性能測試時的每個線程
     * 實際運行時每個線程僅執行一次，在測試方法運行前執行，類似於LoadRunner中的init方法
     */
    public void setupTest(JavaSamplerContext jsc) {
        results = new SampleResult();

        inNum = jsc.getParameter("inNum", "");
        resultNum = jsc.getParameter("resultNum", "");

        if (inNum != null && inNum.length() > 0){
            results.setSamplerData(inNum);
        }

        if (resultNum != null && resultNum.length() > 0){
            results.setSamplerData(resultNum);
        }
    }

    /**
     * 設置傳入參數
     * 可以設置多個，已設置的參數會顯示到Jmeter參數列表中
     */
    public Arguments getDefaultParameters() {
        Arguments params = new Arguments();
        params.addArgument("inNum","");
        params.addArgument("resultNum", "66");
        return params;
    }

    /**
     * 性能測試時的線程運行體
     * 測試執行的循環體，根據線程數和循環次數的不同可執行多次，類似於Loadrunner中的Action方法
     */
    public SampleResult runTest(JavaSamplerContext arg0) {
        boolean flag = false;
        //定義一個事務，表示這是事務的起始點，類似於Loadrunner中的lr.start_transaction
        results.sampleStart();

        for (int i = inNum.length();--i >= 0;){
            if (!Character.isDigit(inNum.charAt(i))){
                flag = false;
            }else{
                flag = true;
            }
        }

        for (int j = resultNum.length();--j >= 0;){
            if (!Character.isDigit(resultNum.charAt(j))){
                flag = false;
            }else{
                flag = true;
            }
        }
        //定義一個事務，表示這是事務的結束點，類似於Loadrunner中的lr.end_transaction
        results.sampleEnd();

        if (flag){
            Integer num = Integer.parseInt(inNum);
            Integer rsNum = Integer.parseInt(resultNum);

            if (num == rsNum){
                results.setDataEncoding("UTF-8");//因為響應的數據有中文，所以最好先設置編碼
                results.setResponseData("恭喜你，答對了O(∩_∩)O~\n答案是【"+resultNum+"】");//響應數據，對應結果樹，其他response code等可以自己點出來
                results.setSuccessful(true);//告訴系統返回正確還是錯誤
            } else if (num > rsNum){
                results.setDataEncoding("UTF-8");
                results.setResponseData("好像大了點~~~~(>_<)~~~~ \n您輸入的是【"+inNum+"】");
                results.setSuccessful(false);
            }else {
                results.setDataEncoding("UTF-8");
                results.setResponseData("好像小了點~~~~(>_<)~~~~ \n您輸入的是【"+inNum+"】");
                results.setSuccessful(false);
            }

        }else{
            results.setDataEncoding("UTF-8");
            results.setResponseData("請輸入數字：~~~~(>_<)~~~~ \n您輸入的inNum是【"+inNum+"】，resultNum是【"+resultNum+"】");
            results.setSuccessful(false);
        }

        return results;
    }

    /**
     * 測試結束方法，結束測試中的每個線程
     * 實際運行時，每個線程僅執行一次，在測試方法運行結束后執行，類似於Loadrunner中的End方法
     */
    public void teardownTest(JavaSamplerContext arg0) {
    }

}

三、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