Beanshell断言

Beanshell断言示例一：通过从数据库查询结果集，和从接口返回中解析json格式的返回内容，比较两者来验证sample请求是否成功

1>先通过JDBC PreProcessor把测试计划中所有接口请求涉及到的表的列名和对应的数据类型从information_schema库的COLUMNS表中查询出来存到一个变量里面

SELECT DISTINCT column_name, data_type

FROM COLUMNS
 WHERE table_name IN ("t_bd_block",
                      "t_bd_build",
                      "t_bd_build_detail",
                      "t_bd_build_education",
                      "t_bd_build_hospital",
                      "t_bd_build_project",
                      "t_bd_build_schdist",
                      "t_bd_build_tag",
                      "t_bd_build_traffic",
                      "t_bd_floor",
                      "t_bd_house",
                      "t_bd_house_type",
                      "t_bd_picture",
                      "t_bd_project",
                      "t_bd_seo",
                      "t_bd_tag",
                      "t_bd_unit",
                      "t_bd_use",
                      "t_bd_use_sub",
                      "t_city",
                      "t_county",
                      "t_developer_company",
                      "t_idcenter",
                      "t_nation",
                      "t_poi",
                      "t_poi_education",
                      "t_poi_education_pic",
                      "t_poi_enrollment_plan",
                      "t_poi_graduation",
                      "t_poi_hospital",
                      "t_poi_station_pic",
                      "t_poi_sub",
                      "t_poi_subway_stations",
                      "t_poi_traffic_bus_station",
                      "t_poi_traffic_line",
                      "t_poi_traffic_line_station",
                      "t_poi_traffic_station",
                      "t_property_company",
                      "t_province",
                      "t_region",
                      "t_static_type");

2>把第1步中查出来的getColumnType解析出来，以键值对的形式存到变量vars供后面使用，因为json格式的数据有些类型的数据要加双引号，有些不需要，这时候就需要知道字段类型来判断下

ArrayList getColumnType1=vars.getObject("getColumnType");

ArrayList column_name = new ArrayList();

ArrayList data_type = new ArrayList();

for(HashMap row : getColumnType1){

Iterator it = row.entrySet().iterator();

while(it.hasNext()){

Map.Entry pair = (Map.Entry)it.next();

log.info(pair.getKey()+"="+pair.getValue());

if(pair.getKey().toString().equals("column_name")){

column_name.add(pair.getValue().toString());

}else if(pair.getKey().toString().equals("data_type")){

data_type.add(pair.getValue().toString());

}

}

}

for(int i=0;i<column_name.size();i++){

vars.put(column_name.get(i).toUpperCase(),data_type.get(i));

}

3>从数据库中把要验证的结果查询出来存到变量里面，变量存的是结果集

4> 对结果进行断言

ArrayList sqlResult1=vars.getObject("sqlResult");

String json = "";

if(sqlResult1.size()==0){

json= "[]";  //如果jdbc request查询结果为空，返回[]

}else{

for(HashMap row : sqlResult1){

String r="";

Iterator it = row.entrySet().iterator();

ArrayList keyAndValue = new ArrayList();

while(it.hasNext()){

Map.Entry pair = (Map.Entry)it.next();

log.info("pair:"+pair);

log.info(pair.getKey()+"="+pair.getValue());

if(pair.getValue()==null){

continue;

}else{

String[] subKey = pair.getKey().toString().split("_");

String key = subKey[0];

for(int k=1; k<subKey.length; k++){

key += Character.toUpperCase(subKey[k].charAt(0))+ subKey[k].substring(1);

}

if("varchar".equals(vars.get(pair.getKey().toString().toUpperCase())) || "char".equals(vars.get(pair.getKey().toString().toUpperCase())) || "text".equals(vars.get(pair.getKey().toString().toUpperCase()))){

keyAndValue.add("\""+key+"\":\""+pair.getValue()+"\",");

}else{

keyAndValue.add("\""+key+"\":"+pair.getValue()+",");

}

}

} //while循环结束

Collections.sort(keyAndValue);

for(int i=0;i<keyAndValue.size();i++){

r+=keyAndValue.get(i);

}

r="{"+r.substring(0,r.length()-1)+"},";

json+=r;

}

if(sqlResult1.size() > 1){

json="{\"code\":\"0001\",\"data\":{\"A-G\":["+json.substring(0,json.length()-1)+"]},\"message\":\"ok\"}";

}else{

json="{\"code\":\"0001\",\"data\":{"+json.substring(0,json.length()-1)+"},\"message\":\"ok\"}";

}

}

log.info(json);

String response_data = prev.getResponseDataAsString();

log.info(response_data);

import org.json.*;

log.info("进去了json对象");

JSONObject sql_data_obj = new JSONObject(json);

JSONObject response_data_obj = new JSONObject(response_data);

log.info("进去了json对象");

log.info("拿到了response的json对象"+response_data_obj);

log.info("拿到了sql的json对象"+sql_data_obj);

String responses_str = response_data_obj.get("data").get("A-G").toString();

String sqls_str = sql_data_obj.get("data").get("A-G").toString();

log.info("进去了responses的A-G"+responses_str);

JSONArray responses_array = new JSONArray(responses_str);

JSONArray sqls_array = new JSONArray(sqls_str);

//String[] result = new String[responses_array.length()];

boolean flag=true;

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

JSONObject response_obj = new JSONObject(responses_array.get(i).toString());

JSONObject sql_obj = new JSONObject(sqls_array.get(i).toString());

//    String name = app_obj.get("cityId").toString();

Iterator it = response_obj.keys();

while(it.hasNext()){

//获取map的key

String key = (String) it.next();

System.out.println("key: "+key);

//得到value的值

Object response_value = response_obj.get(key);

Object sql_value=sql_obj.get(key);

if(!(sql_value.toString().equals(response_value.toString()))){

flag=false;

break;

}

//递归遍历

//                jsonObj.put(key, traveseJson(value));

}

//    result[i] = name;

}

//vars.put("result", Arrays.toString(result));

if(flag)//请求的response中有包含自定义字符，则断言结果为false

{

Failure=false;

String message="${__time(yyyy-MM-dd HH:mm:ss,)} 结果比对成功";

FailureMessage=message;

log.info(FailureMessage);

}else

{

Failure=true;

String message="${__time(yyyy-MM-dd HH:mm:ss,)} 结果比对失败";

String response="本次请求response数据："+prev.getResponseDataAsString()+"----------";

String need_assert="需要断言的数据："+ json;

FailureMessage=message+response+need_assert;

}

Beanshell断言示例二：从SampleResult对象中获取请求的响应结果码，如果不是200，那就直接让断言失败，如果成功，才去获取json格式的返回内容中的某个值，然后判断后再通过Fqilure对象去设置结果是成功还是失败