接口自动化：HttpClient + TestNG + Java(五) - 接口测试数据驱动

在上一篇中，我们实现了post方法请求的封装和测试，这一篇我们做测试数据的分离。

 

5.1 测试数据处理思路

5.1.1 测试数据分离

这里我想到，我们之前的测试demo里，全都是在testNG类里去写入测试请求的相关数据。

如果测试到达一定规模，这样去维护测试数据会当然是不行的。

比较好的办法应该是将测试数据分离出来，通过外部文件来管理，然后读取文件驱动测试。

这个外部文件用Excel来做会比较合适，设计如下：

考虑用表单来驱动请求的方法，暂时写上了Get,Post,Put和Delete方法。当然其实我还没有去实现put和delete方法，后续再添加。

表单的内容分别记录：

• 测试用例编号
• 测试名称
• 协议
• 接口地址
• 检查点
• 状态码
• key1...keyN传入参数

 之前用到的url这样的不会轻易变动的数据存放到另一个property文件中去。

 

5.1.2 配置数据分离

接下来将测试中所用到的配置独立出去：

在src/main/resource下新建config.properties文件。

Host = https://api.apishop.net/
testData = \\src\\test\\resources\\APIcase.xls

 主要用于存放用到的外部文件位置。

 

5.2 数据读取和处理

5.2.1 测试基类读取配置

在项目中src/test/java下新建包com.test.api，创建类TestApi.java，写入以下内容：

package com.test.api;

import java.io.FileInputStream;
import java.util.Properties;

public class TestAPI {
    public Properties prop;
    public String excelPath;
    public String host; 
    
    //构造函数
    public testAPI() {
        try {
　　　　　　　//数据流的形式读取配置文件
            prop = new Properties();
            FileInputStream fis = new FileInputStream(System.getProperty("user.dir")+
"/src/main/resources/config.properties");
            prop.load(fis);
        } catch (Exception e) {
            e.printStackTrace();
        }
        
        host = prop.getProperty("Host");
        excelPath=prop.getProperty("testData");
    }
}

把这个类做为我们测试基类，后续在创建测试时只要继承这个类就可以了。

 

5.2.2 POI读取外部Excel

在项目中src/main/java下的test.com.utils包中，新建ExcelProcess类，写入以下代码：

package com.test.utils;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;

import org.apache.poi.hssf.usermodel.HSSFCell;
import org.apache.poi.hssf.usermodel.HSSFSheet;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
import org.apache.poi.ss.usermodel.CellType;

public class ExcelProcess {public static Object[][] proessExcel(String filePath,int sheetId) throws IOException{
             
             //数据流读入excel
            File file = new File(System.getProperty("user.dir")+filePath);
            FileInputStream fis = new FileInputStream(file);
            HSSFWorkbook wb = new HSSFWorkbook(fis);
            
            //读取特定表单并计算行列数
            HSSFSheet sheet = wb.getSheetAt(sheetId);
            int numberOfRow = sheet.getPhysicalNumberOfRows();
            int numberOfCell = sheet.getRow(0).getLastCellNum();

            //将表单数据处理存入dtt对象
            Object[][] dttData = new Object[numberOfRow][numberOfCell];
            for(int i=0;i<numberOfRow;i++){
                if(null==sheet.getRow(i)||"".equals(sheet.getRow(i))){
                    continue;
                }
                for(int j=0;j<numberOfCell;j++) {
                    if(null==sheet.getRow(i).getCell(j)||"".equals(sheet.getRow(i).getCell(j))){
                        continue;
                    }
                    HSSFCell cell = sheet.getRow(i).getCell(j);
                    cell.setCellType(CellType.STRING);
                    dttData[i][j] = cell.getStringCellValue();
                }
            }     
            return dttData;
        }
}

 

5.2.3 jsonPath处理反馈信息

在本篇第一部分的设计中，我将每个测试的验证点写在了外部excel文件里，基本的想法是，从反馈信息内拿出相应的键值，与excel内的验证点相比较。

所以这里我需要改造之前的JSONParser，去实现特定键值对的获取。这里我用jsonPath去实现：

package com.test.utils;

import com.alibaba.fastjson.JSONObject;
import com.jayway.jsonpath.JsonPath;
import com.jayway.jsonpath.ReadContext;

import net.minidev.json.JSONArray;

public class JSONParser {
    
    public boolean isResponseCorrect(JSONObject jo,String checkpoint,String passValue){
　　　　 //用jsonpath处理json，获取result中特定键值
        ReadContext context = JsonPath.parse(jo);
        JSONArray result = context.read("$.result.."+checkpoint);
        String resultString = result.get(0).toString();
        if(resultString.equals(passValue)){
            return true;
        }else{
            return false;
        }
    }
}

jsonpath本文不再赘述，感兴趣的可以去搜索一下相关教程。本段代码也只是简单的用".."去实现对result元素下的所有元素进行遍历和匹配。

 

5.3 实现测试类

5.3.1 TestNG类实现测试

重写TestPost测试类，改写后代码如下：

package com.test.api;

import org.testng.annotations.Test;

import com.alibaba.fastjson.JSONObject;
import com.test.client.RestfulClient;
import com.test.utils.ExcelProcess;
import com.test.utils.JSONParser;

import java.io.IOException;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import org.apache.http.NameValuePair;
import org.apache.http.client.ClientProtocolException;
import org.apache.http.message.BasicNameValuePair;
import org.testng.Assert;
import org.testng.annotations.BeforeClass;

public class TestPost extends TestAPI{
    RestfulClient client;
    JSONObject responseBody;
    int responseCode;
    String url;
    String postBody;
    Object[][] excelData;
    HashMap<String, String> hashHead;
  
  @BeforeClass
  public void setup() throws ClientProtocolException, IOException {
      //读取用例excel
      excelData = ExcelProcess.proessExcel(excelPath, 1);
      
      //实例化client
      client = new RestfulClient();      
      
      //设置好请求头部
      hashHead = new HashMap<String, String>();
      hashHead.put("Content-Type", "application/x-www-form-urlencoded");   
  }
  
  @Test
  public void testPostRequest() throws ClientProtocolException, IOException {
    //从第二行开始遍历表单，跳过表头
    for(int i=1;i<excelData.length;i++){
        //从特定位置读取测试数据
        String address = excelData[i][3].toString();
        url = host+address;
        String checkPoint = excelData[i][4].toString();
        String checkValue = excelData[i][5].toString();
        //用NameValuePair存储所有请求参数
        List<NameValuePair> keys = new ArrayList<NameValuePair>();
        for(int j=7;j<excelData[i].length-2;j=j+2){
            //因为每种请求的参数个数不确定，在这里进行非空判断
            if(excelData[i][j]==null){
                break;
            }
            NameValuePair pair = new BasicNameValuePair(excelData[i][j].toString(),excelData[i][j+1].toString());
            keys.add(pair);             
        }
      
      //发出请求
      client.sendPost(url, keys, hashHead);

      responseBody = client.getBodyInJSON();
      responseCode = client.getCodeInNumber();

      JSONParser jParser = new JSONParser();
      boolean result = jParser.isResponseCorrect(responseBody, checkPoint, checkValue);
      
      //断言判断结果
      Assert.assertTrue(result);
      Assert.assertEquals(responseCode, 200);
    }
  }
}

测试类继承前文的TestAPI类，setup方法读取excel表单，设置好请求头部，testPostRequest发送请求并分析结果。

 

5.3.2 测试结果

用以下外部测试数据驱动测试：

可以看到，测试用例中包含了两个不同的接口测试，测试均通过。

 

注：

在TestNG测试类里，在读取测试数据时，我采取的是读取固定位置的参数，显然这样的处理在代码健壮性上是比较差的，可以进一步改进。

另外测试数据中的protocol我只是进行了设计，还没有真正处理。