使用TestNG DataProvider跳过某些测试迭代

Question

我正在使用TestNG数据提供程序读取数据池。xls文件，在名为ReadData的类中包含1017个测试用例和214列。

然后，我将214个String参数传递到名为Enterprise Data的单独类中的@Test注释中。

@Test(dataProvider="autodata", dataProviderClass = ReadAutoData.class)
public void autoentry(String Quote, String Garage_Zip, etc...) {

我在@Test中创建了一个for循环，只执行ceratin迭代的操作（比如1-10），它总共只输入10个测试用例。我的问题是，在运行结束时，它仍然显示“Total tests run:1017”，而不是仅显示10个测试，因为for循环实际上做了任何事情。

// Start for loop for entering auto policies
for (int a = start; a <= end; a++)
{
    if (quote == a)
    {       
        System.out.println("Tier = " + Tier);                       
    } 
}

我知道为什么它显示作为整个数据池运行的测试总数，因为从技术上讲，它传入并运行了数据池中的所有内容，我只是不知道如何只显示实际运行的测试数量。

基本上，我正在寻找一种将@Test注释本身放入for循环的方法，可能是？。。。

编辑：

以下是我当前在ReadAutoData类中读取数据池的代码：

@DataProvider(name = "autodata")
public static Object[][] autodata() {
Object[][] arrayObject = getExcelData("C:/dev/AutoDP.xls","NON-COMBO-DP");
return arrayObject;
}

/**
 * @param File Name
 * @param Sheet Name
 * @return
 */
public static String[][] getExcelData(String fileName, String sheetName) {
    String[][] arrayExcelData = null;
    try {
        FileInputStream fs = new FileInputStream(fileName);
        Workbook wb = Workbook.getWorkbook(fs);
        Sheet sh = wb.getSheet(sheetName);

        int totalNoOfCols = sh.getColumns();
        int totalNoOfRows = sh.getRows();

        arrayExcelData = new String[totalNoOfRows-1][totalNoOfCols];

        for (int i= 1 ; i < totalNoOfRows; i++) {

            for (int j=0; j < totalNoOfCols; j++) {
                arrayExcelData[i-1][j] = sh.getCell(j, i).getContents();
            }

        }
    } catch (FileNotFoundException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
        e.printStackTrace();
    } catch (BiffException e) {
        e.printStackTrace();
    }
    return arrayExcelData;
}

编辑2：

// Start for loop for entering auto policies
    for (int a = start; a <= end; a++)
    {
        if (quote < start) throw new SkipException("Testing skip.");
        if (quote > end) throw new SkipException("Testing skip.");

        if (quote == a)
        {

            System.out.println("Starting Iteration = " + Quote);

然而，它仍然显示1017个测试正在运行。

Answer 1

通过在调用之前实现方法（IINVokedMethodMethodListener）和实现方法（IINVokedMethodMethod method，ITestResult tr）来创建监听器类，您可以获取测试结果对象并获取参数，并决定对该数据集执行什么操作。

Object [] params = tr.getParameters();
// put some logic on params and decide
if ( params is something ) {
tr.setStatus(ITestResult.SKIP);
}

我假设您想运行还是不运行取决于参数。

Answer 2

您始终可以抛出TestNG的SkipException，将跳过的执行标记为跳过而不是通过<代码>跳过异常（code>SkipException）是一种特殊异常，不会被识别为故障。

除此之外，您还可以修改ReadAutoData。autodata可返回较少的数据集（行），例如，仅返回文件中数千行中的前10行。

Answer 3

实现这一点的唯一方法是了解TestNG框架是如何执行的。

这意味着：

1. SkipException不是解决方案，因为它将始终统计所有测试用例，即使它们被跳过
2. 测试数是由DataProvider方法返回的行数

然后，解决方案是从DataProvider注释的方法返回正确数量的测试用例：

public class ReadAutoData {
    private static int indexFrom;
    private static int indexTo;

    @DataProvider(name = "autodata")
    public static Object[][] autodata() {
        // you should probably cache this into static variable
        Object[][] arrayObject = getExcelData("C:/dev/AutoDP.xls","NON-COMBO-DP"); 

        return java.util.Arrays.copyOfRange(arrayObject, indexFrom, indexTo);
    }

    public Class<?> autoDataWithinRange(int from, int to) {
        indexFrom = from;
        indexTo   = to;

        return ReadAutoData.class;
    }
}

@Test(dataProvider="autodata", dataProviderClass = ReadAutoData.autoDataWithinRange(0, 10))
public void autoentry(String Quote, String Garage_Zip, etc...) {

如果您不同时运行测试，这将起作用。如果是，您仍然可以使用ThreadLocals进行小修改来使用它。