HttpClient +Testng(2)使用testng的DataProvicer来进行数据驱动

在FirstTest.java 文件中，我们可看到请求URL、请求参数，这些都是属于测试数据。这里我们使用testng中自带的@DataProvider注解来做数据驱动，数据源文件可以是EXCEL，XML，甚至可以是TXT文本。这里我们使用EXCEL数据源。

1、大概流程

每个测试运行前，会先通过@DataProvider的方法来读取EXCEL中数据，该注解的方法会返回一个Object[] []。然后带@Test的测试方法，如果希望从这个 DataProvider 的接收数据，则需要标识获取数据源的dataProvider的名称，也就是这个注解的名字。那么对应的DataProvider会把读取的数据传给该test方法

借用该文档中的一个图来更清楚的说明这个流程

2、创建一个Excel表格

在目录main下新建一个目录TestData,并在该目录下新建一个excel表格，直接命名为LoginTest。然后打开该文档，修改sheet页名称为login.之后把上面http请求URL、请求参数都放到excel表格中。

这里必须把表格第一行用来命名所有参数的名称。测试数据是从第二行开始的。

如下图所示：

3、创建一个读取EXCEL表格的类

这里我们直接在test->java下创建一个包，命名为utl

之后在utl中创建一个类ExcelReader.java

里面的内容如下所示：

package utl;import org.apache.poi.hssf.usermodel.HSSFCell;import org.apache.poi.hssf.usermodel.HSSFRow;import org.apache.poi.hssf.usermodel.HSSFSheet;import org.apache.poi.hssf.usermodel.HSSFWorkbook;import org.apache.poi.poifs.filesystem.POIFSFileSystem;import java.io.File;import java.io.FileInputStream;import java.text.DecimalFormat;import java.util.ArrayList;import java.util.List;public class ExcelReader {    /**     * 读取excel文件，excel中不含合并单元格     * @param path     * @param sheetName     * excel表格要注意每行的最后一格不能为空     * 如果要为空，则需要手动随便填写字符，然后再按del键删除才可以     * 目前没找到这个解决办法，后续会继续查找解决办法     * @return     */    public static String[][] getExpectationData(String path, String sheetName) {        try {            File file = new File(path);            FileInputStream fis = new FileInputStream(file);            POIFSFileSystem POIStream = new POIFSFileSystem(fis);            HSSFWorkbook workBook = new HSSFWorkbook(POIStream);            //得到工作表            HSSFSheet sheet1 = workBook.getSheet(sheetName);            //得到总行数            int rowNum = sheet1.getLastRowNum();            List<String[]> results = new ArrayList<String[]>();            for (int i=1;i<=rowNum;i++){                //当前行                HSSFRow row = sheet1.getRow(i);                int colNum = row.getLastCellNum();                String[] data = new String[colNum];                //当前行所有列                for (int j = 0; j < colNum; j++) {                    try {                        data[j] = getCellValue(row.getCell(j));                    }catch (NullPointerException e){ //如果单元格为空的时候，则用这个来处理                        data[j] = "";                    }                }                //把data[]数组的数据存在list<[]>中                results.add(data);            }            fis.close();            String[][] returnArray = new String[results.size()][rowNum];            for (int i = 0; i < returnArray.length; i++) {                returnArray[i] = (String[]) results.get(i);            }            return returnArray;        }catch (Exception e){            return null;        }    }    /**     * 对Excel的各个单元格的格式进行判断并转换     */    public static String getCellValue(HSSFCell cell) {        String cellValue = "";        DecimalFormat df = new DecimalFormat("#");        switch (cell.getCellType()) {            case HSSFCell.CELL_TYPE_STRING:                cellValue =cell.getRichStringCellValue().getString().trim();                break;            case HSSFCell.CELL_TYPE_NUMERIC:                cellValue =df.format(cell.getNumericCellValue()).toString();                break;            case HSSFCell.CELL_TYPE_BOOLEAN:                cellValue = String.valueOf(cell.getBooleanCellValue()).trim();                break;            case HSSFCell.CELL_TYPE_FORMULA:                cellValue =cell.getCellFormula();                break;            default:                cellValue = "";        }        return cellValue;    }}

4、在类FirstTest.java中创建一个含有DataProvider注解的方法

修改类FirstTest.java，在该类中添加一个方法，如下面内容：

@DataProvider(name = "LoginData")public Object[][] login() {    // 测试数据准备    String file = "." + File.separator + "TestData" + File.separator + "LoginTest.xls";    Object[][] records ;    records = ExcelReader.getExpectationData(file, "login");    return records ;}

5、修改类FirstTest.java中的Test方法

Test方法内容如下所示：

@Test(dataProvider = "LoginData")public void loginJDTest(String caseDescription,String loginURL, String uuid,String eid,String fp,String _t, String loginType,String loginname,String nloginpwd, String chkRememberMe, String authcode,String pubKey,String sa_token,String seqSid) {    System.out.println("=====" + caseDescription + "=====");    //创建一个httppost请求    httppost = new HttpPost(loginURL);    //创建Post请求参数    List<NameValuePair> formparams1 = new ArrayList<NameValuePair>();    formparams1.add(new BasicNameValuePair("uuid",uuid));    formparams1.add(new BasicNameValuePair("eid",eid));    formparams1.add(new BasicNameValuePair("fp",fp));    formparams1.add(new BasicNameValuePair("_t",_t));    formparams1.add(new BasicNameValuePair("loginType",loginType));    formparams1.add(new BasicNameValuePair("loginname",loginname));    formparams1.add(new BasicNameValuePair("nloginpwd",nloginpwd));    formparams1.add(new BasicNameValuePair("chkRememberMe",chkRememberMe));    formparams1.add(new BasicNameValuePair("authcode",authcode));    formparams1.add(new BasicNameValuePair("pubKey",pubKey));    formparams1.add(new BasicNameValuePair("sa_token",sa_token));    formparams1.add(new BasicNameValuePair("seqSid",seqSid));    try {        httppost.setEntity(new UrlEncodedFormEntity(formparams1,"UTF-8"));        response = httpClient.execute(httppost);        entity = response.getEntity();        // 在这里可以用Jsoup之类的工具对返回结果进行分析，以判断创建是否成功        postResult = EntityUtils.toString(entity, "UTF-8");        System.out.println("查看登录接口请求返回的结果：" + postResult);    } catch (Exception e) {        e.printStackTrace();    }    httppost.releaseConnection();}

上面可看出，loginJDTest方法带了若干个参数，这里定义的参数要注意：

参数的个数必须和之前创建excel表格中的一样，如excel表格里面有14列参数，则这里也必须对应14个参数。而且定义参数的顺序也必须和excel表格里面参数的顺序保持一样。

6、运行程序

再次运行程序，运行结束后查看日志，一切正常，如下图所示