JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试-CSDN博客

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试

有一些测试需要做一些预处理程序才能做http请求，在JMeter下可以通过使用参数+BeanShell 预处理程序加工后再发起请求即可。例oauth服务需要通过username,password,client_id,client_secret,grant_type进行请求，这里的密码涉及安全性不能用明文进行传输,需要做预处理

应用提供如下信息：

用户名：username=user

密码：password=888888

盐值：salt=2gjokme8k5p0u3n8ca1n

客户端ID:client_id=test

客户端秘钥:client_secret=16air0bu4l0yjnfkkxfc

授权模式：grant_type=password

发送密码的规则为password=sha(md5(8888882gjokme8k5p0u3n8ca1n))=a43dd2bfc28bbaf137ab5a45d634f44eed06adcf

这里需要先将(密码+盐值)先进行md5，再进行sha才是要发送的最终参数，这里如果只测试一个用户时间，先通过其它程序取得再传入也是很方便，但如果需要做多个用户的测试就非常麻烦了，需要不停的去加工后才能在jmeter里更换，调试起来不方便，如果用参数+BeanShell 预处理程序就可以很方便处理了

下面来实际验证

验证请求

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_jmeter

成功返回

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_变量_02

下面用参数+BeanShell 预处理程序来实现

定义参数

如果换个用户，只需修改这里即可

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_压力测试_03

HTTP请求里值用${xxxx}来定义变量参数，来接收定义的参数或BeanShell 预处理程序生成的变量，这里${passwordPar}是BeanShell 预处理程序生成的

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_jmeter_04

BeanShell预处理程序

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_压力测试_05

处理程序

import sun.misc.BASE64Encoder;

import java.io.UnsupportedEncodingException;

import java.security.MessageDigest;

import java.security.NoSuchAlgorithmException; 

import java.util.Arrays;

import java.util.Map;

import java.util.Set;

import com.alibaba.fastjson.JSONArray;

import com.alibaba.fastjson.JSONObject;

    public static String md5AndSha(String inputText)

    {

        return sha(md5(inputText));

    }

    public static String md5(String inputText) {

        return encrypt(inputText, "md5");

    }

    public static String sha(String inputText) {

        return encrypt(inputText, "sha-1");

    }

    public static String encrypt(String inputText, String algorithmName) {

        if (inputText == null || "".equals(inputText.trim())) {

            thrownew IllegalArgumentException("请输入要加密的内容");

        }

        if (algorithmName == null || "".equals(algorithmName.trim())) {

            algorithmName = "md5";

        }

        String encryptText = null;

        try {

            MessageDigest m = MessageDigest.getInstance(algorithmName);

            m.update(inputText.getBytes("UTF8"));

            byte[] s = m.digest();

            return hex(s);

        } catch (NoSuchAlgorithmException e) {

            e.printStackTrace();

        } catch (UnsupportedEncodingException e) {

            e.printStackTrace();

        }return encryptText;

    }

public static String hex(byte[] arr) {

 StringBuffer sb = new StringBuffer();

for (int i = 0; i < arr.length; ++i) {

  sb.append(Integer.toHexString((arr[i] & 0xFF) | 0x100).substring(1, 3));

 }

return sb.toString();

}

public static String MD5(String data) throws Exception {

  MessageDigest md = MessageDigest.getInstance("MD5");

byte[] array = md.digest(data.getBytes("UTF-8"));

    StringBuilder sb = new StringBuilder();

    for (byte item : array) {

      sb.append(Integer.toHexString((item & 0xFF) | 0x100).substring(1, 3));

    }

    return sb.toString().toUpperCase();

}

public   String generateSignature( Map  data, String key, String signType) throws Exception {

       // Set<String> keySet = data.keySet();

       Set  keySet = data.keySet();

        String[] keyArray = keySet.toArray(new String[keySet.size()]);

        Arrays.sort(keyArray);

        StringBuilder sb = new StringBuilder();

        for (String k : keyArray) {

            if (k.equals("sign")) {

                continue;

            }

            if (data.get(k).toString().trim().length() > 0) //为空，则不参与签名

                sb.append(k).append("=").append(data.get(k).toString().trim()).append("&");

        }

        sb.append("key=").append(key);

        System.out.println(sb);

        if ("MD5".equals(signType)) {

            return MD5(sb.toString()).toUpperCase();

        }else {

            thrownew Exception(String.format("Invalid sign_type: %s", signType));

        }

}



public void getPasswordPar(){    

  String clientId=vars.get("client_id");

  String clientSecret=vars.get("client_secret");

  String username=vars.get("username");

  String password=vars.get("password");

  String salt=vars.get("salt"); 

  String grant_type=vars.get("grant_type"); 

  String passwordPar=md5AndSha(password+salt);   

  vars.put("passwordPar",passwordPar); 

  log.info("加密passwordPar begin");         

  log.info(passwordPar);

  log.info("加密passwordPar end");        

}


log.info("BeanShell 预处理程序");

/*

 * 获取加密参数passwordPar

 */

getPasswordPar();

当请求取得结果后，也可以用BeanShell 后置处理程序将请求返回做解析加工，取出想要的数据，然后存入变量供下一个请求来处理

BeanShell 后置处理程序

JMeter进阶使用变量及BeanShell 预处理程序实现复杂调试_jmeter_06

import com.alibaba.fastjson.JSONArray;

import com.alibaba.fastjson.JSONObject;


log.info("BeanShell 后置处理程序");


//取得请求返回结果

log.info("请求返回结果 begin");

String response = prev.getResponseDataAsString();

log.info(response); 

log.info("请求返回结果 end"); 


JSONObject json=  JSONObject.parseObject(response);

String myToken=json.getString("access_token");

//将access_token存入变量中

vars.put("myToken",myToken);

log.info("取得access_token为："+myToken);