Java接口自动化测试(四) — 使用Excel做数据驱动

一、说明

1. 数据驱动测试的核心：是测试数据与测试脚本分离，实现测试脚本的参数化，例如：在使用工具测试时，常常会使用到参数化设置；
2. 使用数据驱动测试方便后期维护，提高脚本的可重用性；
3. 做数据驱动的方式有多种例如：Excel、CSV、MySQL等

二、使用Excel实现数据驱动

1、在pom文件添加POI依赖

       <!--操作Excel数据-->
        <!-- https://mvnrepository.com/artifact/org.apache.poi/poi -->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.17</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.17</version>
        </dependency>
        <!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml-schemas -->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml-schemas</artifactId>
            <version>3.17</version>
        </dependency>

2、新建一个Class命名为ExcelUtils

package com.test.excel;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.xssf.usermodel.XSSFCell;
import org.apache.poi.xssf.usermodel.XSSFRow;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import java.io.FileInputStream;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author lh
 * @date 2020/7/3
 * @description
 */
public class ExcelUtils {
    /**
     * 读取Excel数据
     * @return List
     * @throws Exception
     */
    public static List<Map<String ,Object>> readExcle() throws Exception{
        //用流的方式读取Excel文件
        FileInputStream fis = new FileInputStream("C:\Users\Think\Desktop\test.xlsx");
        //获取Excel工作簿
        XSSFWorkbook xf = new XSSFWorkbook(fis);
        //获取第一个sheet
        XSSFSheet sheet = xf.getSheetAt(0);
        //获取第一行
        int firstRow = sheet.getFirstRowNum();
        //获取最后一行
        int lastRow = sheet.getLastRowNum();
      //  System.out.println(lastRow);
        //用于map设置key值，自定义
        String columns[] = {"username", "password"};
        List< Map<String ,Object>> list = new ArrayList<>();

        for (int i = 0;i<=lastRow;i++){
            XSSFRow row = sheet.getRow(i);
            if (row!=null){
                //获取第一行，第一列
                int firstcell =  row.getFirstCellNum();
                //获取最后一行，最后一列
                int lastcell = row.getLastCellNum();
                Map<String ,Object> map = new HashMap<>();
                for (int j = 0;j<columns.length;j++){
                    XSSFCell cell = row.getCell(j);
                    if (cell == null){
                        continue;
                    }
                    //封装成map
                    map.put(columns[j], getCellFormatValue(cell));
                }
                //System.out.println(map);
                //将map放入List
                list.add(map);
            }
            fis.close();
            xf.close();
        }
        return list;
    }
    //格式化数值类型
    public static Object getCellFormatValue(Cell cell){
        Object cellValue = null;
        /**
         * getCellTypeEnum()方法是枚举类型，用于判断单元格值类型，有以下五种格式：
         *     _NONE(-1),
         *     NUMERIC(0),
         *     STRING(1),
         *     FORMULA(2),
         *     BLANK(3),
         *     BOOLEAN(4),
         *     ERROR(5);
         */
        switch(cell.getCellTypeEnum()){
            case STRING:
                cellValue = cell.getStringCellValue();
                break;
            case NUMERIC:
                /**如果是数字类型转换成数字类型，但是初始化数字加的.0,因此可以转换成int类型去掉.0
                 * cell.getNumericCellValue()
                 */
                cellValue = new Double(cell.getNumericCellValue()).intValue();
                break;
            case BOOLEAN:
                break;
            case FORMULA:
                break;
        }
        return cellValue;
    }
    //测试
    public  static void main(String[] arg) throws Exception{
        ExcelUtils excelTest = new ExcelUtils();
        excelTest.readExcle();
    }
}

三、在测试用例中使用

1. 在testng中使用需要新创建一个方法getDataMethod()，返回Object[][]类型
2. 使用注解@DataProvider，为测试方法提供数据
3. 在测试方法上使用@Test(dataProvider = "getDataMethod")接收数据
4. 使用数据驱动有一个好处，测试方法可以写一个，根据参数频繁调用一个方法（和上一篇对比https://www.cnblogs.com/liho/p/13225137.html）

package com.test.httpclient;

import com.alibaba.fastjson.JSONObject;
import com.test.excel.ExcelUtils;
import org.testng.Assert;
import org.testng.annotations.AfterTest;
import org.testng.annotations.BeforeTest;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import java.util.List;
import java.util.Map;

/**
 * @author lh
 * @date 2020/6/8
 * @description
 */
@Test
public class LoginCase {
    HttpClientPost httpClientPost = new HttpClientPost();
    @DataProvider
    public Object[][] getDataMethod() throws Exception{
　　　　　//读取Excel中的数据
        List<Map<String,Object>> result = ExcelUtils.readExcle();
        Object[][] params = new Object[result.size()][];
        for (int i = 0; i < result.size(); i++) {
            params[i] = new Object[]{result.get(i)};
        }
        return params;
    }
    @BeforeTest
    public void setup(){
        System.out.println("用例执行前执行");
    }
　　 //一个测试方法
    @Test(dataProvider = "getDataMethod")
    public void test1(Map<String, Object> param) throws Exception {
        JSONObject result = httpClientPost.LoginPost(param);
        Assert.assertEquals(result.get("code"),200);
    }
    @AfterTest
    public void teardown(){
        System.out.println("用例执行完后执行");
    }
}

四、查看运行结果