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对象去设置结果是成功还是失败