TestNG数据驱动DataProvider(4种方式)

TestNG数据驱动DataProvider(4种方式)

1、DataProvider用法(从二维数组中读取)

1、定义方法，加上@DataProvider注解

• 定义一个方法，方法名随意取，使用@DataProvider注解，注解中给这个provider起名字，该方法返回一个二维数组。例如：@DataProvider(name="testData")

• 该方法返回的二维数组每一行代表一次测试的参数，每一行的元素跟Test中需要的参数一一对应

3、TestCase使用DataProvider

需要使用DataProvider时，该方法@Test加上@DataProvider注解和其名称

例如：@Test(dataProvider="testData")

1.1、测试例子：testProvider.java

package com.example;

import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

/**
 * Description :
 *
 * @author : HMF
 * Date : Created in 11:24 2023/11/16
 * @version :
 */
public class testProvider {
    @DataProvider(name="testData")
    public Object[][] provider(){
        Object [][] provider = new Object [6][2];
        for (int i = 0; i < provider.length; i++) {
            provider[i][0] = "value"+i;
            provider[i][1] = i+100;
        }

        return provider;
    }

    @Test(dataProvider="testData")
    public void getName(String value,int num){
        System.out.println(value+","+num);

    }
    
}

1.2、执行结果

运行过程中，打印 dataprovider提供的二维数组参数，每次测试可用来入参。

value0,100
value1,101
value2,102
value3,103
value4,104
value5,105

===============================================
Default Suite
Total tests run: 6, Failures: 0, Skips: 0
===============================================

Process finished with exit code 0

2、DataProvider用法(从Excel中读取(XSS方式获取直接放入Object))

1、定义方法，加上@DataProvider注解

• 定义一个方法，方法名随意取，使用@DataProvider注解，注解中给这个provider起名字，例如：@DataProvider(name="testData")

• 该方法读取excel，数据放入二维数组

• 该方法返回的二维数组每一行代表一次测试的参数，每一行的元素跟Test中需要的参数一一对应

3、TestCase使用DataProvider

需要使用DataProvider时，该方法@Test加上@DataProvider注解和其名称

例如：@Test(dataProvider="testData"

2.1、excel文件

testdata.xlsx

id	name	age
1	小王	18
2	王凯	19
3	李杰	18
4	陈小小	19
5	胡晓	19
6	刘梅	19

2.2、测试例子testProvider_excel.java

2.3.1、pom.xml引用

<!-- org.apache.poi依赖-->
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi</artifactId>
            <version>3.14</version>
        </dependency>
        <dependency>
            <groupId>org.apache.poi</groupId>
            <artifactId>poi-ooxml</artifactId>
            <version>3.14</version>
        </dependency>

2.3.2、测试例子testProvider_excel.java

package com.example;

import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.xssf.usermodel.XSSFSheet;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

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

/**
 * Description :
 *
 * @author : HMF
 * Date : Created in 10:25 2023/11/23
 * @version :
 */
public class testProvider_excel {
    @Test(dataProvider = "testdata")
    public void testMethod(int id,String name, int age) {
        System.out.println("Id:" + id + ", Name:" + name + ", Age:" + age);
    }
    @DataProvider(name = "testdata")
    public Object[][] readData() throws IOException {
        File file = new File("src/main/resources/testdata.xlsx");
        FileInputStream fis = new FileInputStream(file);
        XSSFWorkbook wb = new XSSFWorkbook(fis);
        XSSFSheet sheet1 = wb.getSheetAt(0);
        int rowcount = sheet1.getPhysicalNumberOfRows();
        int columnCount = sheet1.getRow(0).getLastCellNum();
        Object objects[][] = new Object[rowcount-1][columnCount];
        int rowCounter = 0;
        Iterator<Row> rowIterator = sheet1.iterator();
        boolean firstRow = true;
        while (rowIterator.hasNext()) {
            Row currentRow = rowIterator.next();
            if (firstRow) {