Jmeter CSV操作

统计行号列号

 import java.io.BufferedReader;
 import java.io.FileReader;
 import java.io.File;

 print("========输出结果表格${csvPath}============  "+"${csvPath}");
 print("========Case 表格${csvPath}============  "+"${csvCase}");

 BufferedReader br=new BufferedReader(new FileReader("${csvCase}"));
 //BufferedReader bufRdr = new BufferedReader(new InputStreamReader(new FileInputStream(file), "UTF8"));
 String tmpStr="";
 int rowNum=0;
 while(tmeStr=br.readLine()!=null){
     rowNum++;
 }
 rowNum=rowNum-1;//去掉CSV文件标题行；
 vars.put("rowNum",String.valueOf(rowNum));
 print("-----------------1CSV文件行数：--------"+rowNum+"-------------------");
 log.info("-----------------1CSV文件行数：--------"+rowNum+"-------------------");

 print("-----------------Case源CSV表格路径：--------"+"${csvCase}"+"-------------------");
 //String getRowNum=vars.get("rowNum");

 //int csvNo= Integer.parseInt(vars.get("rowNum"));

 int totalElements= Integer.parseInt(vars.get("rowNum"));
 print("Total_csvNoCSV总行号---->>  "+totalElements);
 vars.put("totalElements", String.valueOf(totalElements));
 //print(${__javaScript("${totalElements}")});
 //print("======javaScript======"+${__javaScript(totalElements)});

prev.setDataEncoding("utf-8");

caseNo,testInput,intent,domain,expectedResult,PASS_FAIL,actualResult,response,description,tester

 import java.util.Collection;
 import java.util.Iterator;
 import java.util.*;
 import java.io.FileWriter;
 import java.util.Arrays;
 import java.io.Writer;
 import java.util.List;
 import com.alibaba.fastjson.JSONPath;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONArray;
 import com.alibaba.fastjson.JSONObject;
 import Util.WriteStreamAppend;
 import Util.Excel;
 import java.io.BufferedReader;
 import java.io.FileReader;
 import java.io.File;
 import org.apache.log4j.Logger;
 import com.cloudminds.csv2excel;
 import com.cloudminds.csv2excel.CsvToExcel;
 import com.cloudminds.csv2excel.CsvToExcel.convertCsvToXls;
 import com.cloudminds.CsvToExcel.convertCsvToXls;
 import java.io.FileOutputStream;
 import java.io.FileReader;
 import java.io.IOException;
 import org.apache.jmeter.assertions;
 import org.apache.jmeter.samplers.SampleResult;
 import org.apache.jmeter.assertions.AssertionResult;
 import org.json.*;

 import org.apache.commons.lang.math.NumberUtils;
 import org.apache.log4j.Logger;
 import org.apache.poi.ss.usermodel.Row;
 import org.apache.poi.ss.usermodel.Workbook;
 import org.apache.poi.xssf.streaming.SXSSFSheet;
 import org.apache.poi.xssf.streaming.SXSSFWorkbook;
 import com.opencsv.CSVReader;
 import com.cloudminds.csv2excel.AppMain;

 //print("========输出结果表格${csvPath}============  "+"${csvPath}");
 //print("========Case 表格${csvPath}============  "+"${csvCase}");
 //
 //BufferedReader br=new BufferedReader(new FileReader("${csvCase}"));
 //String tmpStr="";
 //int rowNum=0;
 //while(tmeStr=br.readLine()!=null){
 //    rowNum++;
 //}
 //rowNum=rowNum-1;//去掉CSV文件标题行；
 //vars.put("rowNum",String.valueOf(rowNum));
 //print("-----------------CSV文件行数：--------"+rowNum+"-------------------");
 //log.info("-----------------CSV文件行数：--------"+rowNum+"-------------------");

 print("${caseNo}");
 print("${testInput}");
 print("${intent}");
 print("${expectedResult}");
 print("${actualResult}");

 String prevQuery = prev.getSamplerData();
 print("当前的接口请求拼接 输出---->>  "+prevQuery);
 //CookieManager manager = ctx.getCurrentSampler().getCookieManager();
 //print(manager);
 String response=prev.getResponseDataAsString();
 print("当前的接口返回值response 输出---->>  "+response);
 String tr1 =response;
 vars.put("rsp",response);
 print("tr1--->>>  "+tr1);
 print("====================  "+${ResponceData});
 // 转换成JSON串
 JSONObject json = JSONObject.parseObject(${ResponceData});
 System.out.println("1当前json--->>   "+ json);
 JSONObject result = JSON.parseObject(${ResponceData});
 System.out.println("2当前tr1--->>   "+ result);

 print("断言$.json====================  "+"${source}"+"  "+"${text}"+"  "+"${intent_json}"+"  "+"${url}");
 Object source = JSONPath.eval(json, "${source}");
 Object text = JSONPath.eval(json, "${text}");
 Object intent_json = JSONPath.eval(json, "${intent_json}");
 Object url = JSONPath.eval(json, "${url}");

 String sou=JSONObject.toJSONString(source);
 String tex=JSONObject.toJSONString(text);
 String inte=JSONObject.toJSONString(intent_json);
 String ur=JSONObject.toJSONString(url);
 print(sou);
 print(tex);
 print(inte);
 print(ur);
 //String arsult="source字段:  "+sou+"  text字段:  "+tex+"  intent字段:  "+inte+"  url字段:  "+ur;
 String arsult="source= "+sou+";"+" text= "+tex+";"+"  intent= "+inte+";"+" url= "+ur+";"+"当前AgentID= "+"${agentId}";
 print(arsult);
 vars.put("arsult",arsult);

 //默认走,分割
 char SEPARATOR = ',';

 public void writeLine(FileWriter writer, String[] params, char separator)
 {
    boolean firstParam = true;
    StringBuilder stringBuilder = new StringBuilder();
    String param = "";
    for (int i = 0; i < params.length; i++)
    {
       param = params[i];
       log.info(param);
       print("===========写入成功============");
          //如果行中的第一个参数，则不需要分隔符
        if (!firstParam)
        {
            stringBuilder.append(separator);
        }
          //append
        stringBuilder.append(param);
        firstParam = false;
    }
    stringBuilder.append("\n");
    print("换行==================== 。");
    log.info(stringBuilder.toString());
    writer.append(stringBuilder.toString());
    print("写入的数据          "+stringBuilder.toString());

 }

 //get path of csv file (creates new one if its not exists)
 print("========检查csv写入============  "+"${csvPath}");
 String csvFile = "${csvPath}";

 print("开始写入数据=======        ");

 print("1断言内容=======        "+"${assert_content}");
 String ErrorValue ="${assert_content}";
 print("2断言内容=======        "+ErrorValue);

 if(prev.getResponseDataAsString().contains("${assert_content}")){
 //大入口统一成功

         if (!(ur=="")) {
             if (!(ur.contains("mp3"))) {
             String fail="FAIL";
                print(arsult);
                vars.put("passorFail",fail);
                Failure = true;//直接判断失败，表示接口跑失败，在结果树中sample是红色的
                FailureMessage =" 当前断言不包含  " + "${assert_content}" + "." ;
                print ( "  未通过当前断言不包含   " + "${assert_content}");   // this goes to stdout
                log.warn( "未通过当前断言不包含 " + "${assert_content}"); // this goes to the JMeter log file
                 } 

             } 

         if (!(inte=="")) {
             if (!((inte.contains("cm_dance")) || (inte.contains("cloudminds")))) {
             String fail="FAIL";
                print(arsult);
                vars.put("passorFail",fail);
                Failure = true;//直接判断失败，表示接口跑失败，在结果树中sample是红色的
                FailureMessage =" 当前断言不包含  " + "${assert_content}" + "." ;
                print ( "  未通过当前断言不包含   " + "${assert_content}");   // this goes to stdout
                log.warn( "未通过当前断言不包含 " + "${assert_content}"); // this goes to the JMeter log file
                 } 

             } 

          if (!(tex==null)) {
             if (!(tex.length()>500)) {
             String fail="FAIL";
                print(arsult);
                vars.put("passorFail",fail);
                Failure = true;//直接判断失败，表示接口跑失败，在结果树中sample是红色的
                FailureMessage =" 当前断言不包含  " + "${assert_content}" + "." ;
                print ( "  未通过当前断言不包含   " + "${assert_content}");   // this goes to stdout
                log.warn( "未通过当前断言不包含 " + "${assert_content}"); // this goes to the JMeter log file
                 } 

             } 

              System.out.println("大入口统一Pass");
              String ps="PASS";
                vars.put("passorFail",ps);

         }

 else{
      String fail="FAIL";
                print(arsult);
                vars.put("passorFail",fail);
                Failure = true;//直接判断失败，表示接口跑失败，在结果树中sample是红色的
     FailureMessage =" 当前断言不包含  " + "${assert_content}" + "." ;
     print ( "  未通过当前断言不包含   " + "${assert_content}");   // this goes to stdout
     log.warn( "未通过当前断言不包含 " + "${assert_content}"); // this goes to the JMeter log file

 }

 String[] params ={"${caseNo}", "${testInput}", "${intent}","${domain}", "${expectedResult}",vars.get("passorFail"), vars.get("arsult") ,vars.get("rsp"),"${description}","${tester}"};

 FileWriter fileWriter = new FileWriter(csvFile, true);
 writeLine(fileWriter, params, SEPARATOR);

 fileWriter.flush();
 fileWriter.close();
 private static Logger logger = Logger.getLogger(AppMain.class);
 fileLoc = CsvToExcel.convertCsvToXls("${csvPath}", csvFile);
 logger.info("文件位置==>>   " + fileLoc);
         

响应断言判断

//响应内容正则提取部分
String ResponseResult = vars.get("ResponseResult");
//断言
String Assertion = vars.get("Assertion"); 

//判断断言是否通过
if(ResponseResult.equals(Assertion) || ResponseResult.equals("Process Case")){
    Failure = false;
    //兼容其他并存的断言
    FailureMessage = "接口响应:\r\n" + prev.getResponseDataAsString() + "\r\n";
}else{
    Failure = true;
    if (prev.getResponseDataAsString().length() < 255) {
        FailureMessage = "接口响应:\r\n" + prev.getResponseDataAsString() + "\r\n";
    }else{
        FailureMessage = "接口响应片段:\r\n" + ResponseResult + "\r\n";
    }
    FailureMessage += "用例断言:\r\n" + Assertion;  

    String logInfo = "\n";
    logInfo = logInfo + prev.getThreadName() + " " +  prev.getSampleLabel() + "\n";
    logInfo = logInfo + prev.getSamplerData() + "\n";
    logInfo = logInfo + prev.getResponseCode() + "\n\n";
    logInfo = logInfo + prev.getResponseDataAsString() + "\n";
    log.info(logInfo);
}

写入csv

import com.csvreader.CsvWriter;

String NewDataPath="C:\\";
CsvWriter cw = new CsvWriter(NewDataPath+"\\test.csv");
cw.writeRecord(new String[]{"test1","test2","test3"});
cw.writeRecord(new String[]{"a","aa","aaa"});
cw.close();

下载javacsv.jar，然后将其放入apache-jmeter-*\lib目录
import com.csvreader.CsvWriter;