如何在testNG中将动态参数传递给dataProvider方法

Question

我试图通过动态输入要从中提取数据的工作表的名称，以这种方式重用现有的数据提供程序，将Excel文件中的数据加载到我的测试框架中。

示例：我正在从Login凭据表加载数据，该表代表成功登录测试的数据。第二个测试是失败的登录测试，它从InvalidLogin凭据表加载数据。第三个测试从第三个表UserInformation等中提取数据。我所遵循的udemy课程并没有真正涵盖这个主题，我觉得整个事情的实际实现也不是最好的，但我只是跟着它，因为这是我第一次处理Selenium和Excel。

主要问题似乎是将此工作表名称参数发送到作为数据提供者的getData方法。出于某种原因，我认为解决方案可能非常简单，只是我无法理解它。

我尝试了XML参数属性，但它并没有解决我的问题。我想根据测试动态发送此参数。

我已经尝试在我的测试方法和数据提供者方法上使用@Parameters注释，但这也不起作用。将其放在测试方法上的问题是因为我假设数据提供者注释也试图从Excel文件中检索String sheetName，而我的测试甚至无法运行。

在数据提供程序方法本身上使用它并不是一种选择，因为我必须生成尽可能多的数据提供程序方法和发送的参数（这会破坏重用现有数据提供程序的整个要点）。

从excel文件中提取数据的方法

public Object[][] testData(String excelPath, String sheetName) {

        Object[][] data = null;

        try {
            ExcelUtil excelUtil = new ExcelUtil(excelPath, sheetName);

            XSSFWorkbook workbook = new XSSFWorkbook(excelPath);
            XSSFSheet sheet = workbook.getSheet(sheetName);
            int rowCount = excelUtil.getRowCount();
            int colCount = excelUtil.getColCount();

            data = new Object[rowCount - 1][colCount];

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

                Row row = sheet.getRow(i);
                for (int j = 0; j < colCount; j++) {

                    Cell cell = row.getCell(j);
                    switch (cell.getCellType()) {

                        case STRING:
                            data[i - 1][j] = excelUtil.getCellDataString(i, j);
                            break;


                        case NUMERIC:
                            data[i - 1][j] = String.valueOf(excelUtil.getCellDataNumeric(i, j));
                            break;
                    }
                }
            }
        } catch (Exception ex) {
            LOGGER.error("Error while loading data from Excel file: " + ex.getMessage());
        }

        return data;
    }

dataProvider方法本身

 @DataProvider(name = "TeamoLoginData")
    public Object[][] getData(){
        String excelPath = "src/main/java/apps/teamo/testdata/Book 1.xlsx";
        return testData(excelPath, "LoginCredentials");
    }

试验方法：

@Test(dataProvider = "TeamoLoginData", dataProviderClass = ExcelDataProvider.class)
    public void loginSuccessful(Method method, String username, String password) {

        ExtentTestManager.startTest(methodName, "Successful login");
        TeamoLoginPage teamoLoginPage = new TeamoLoginPage(getDriver());
        TeamoHomePage teamoHomePage = new TeamoHomePage(getDriver());

            getDriver().get(PropertyManager.getInstance().getBaseURL());
            teamoLoginPage.enterUsername(username);
            teamoLoginPage.enterPassword(password);

            teamoLoginPage.signIn();

            Assert.assertTrue(teamoHomePage.changeLogIsDisplayed());
    }

是否有方法将带有工作表名称的字符串作为参数传递到某个位置，以便测试获得相关信息？

Answer 1

选项A：

我建议您看看这里所描述的ITestContext。

提供以下两种测试方法：第一种是设置属性（=工作表名称）的测试方法，第二种是测试方法（取决于第一种方法），由数据提供程序驱动，该数据提供程序使用我们指定的工作表中的数据。

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

public class DemoTestClass {

    @Test
    public void testTabnameSetting() {
        //Simulating toggling between multiple flows.
        String tabName = "LoginCredentials";
        Reporter.getCurrentTestResult().getTestContext().setAttribute("sheetName", tabName);
    }

    @Test(dependsOnMethods = "testTabnameSetting", dataProvider = "getData")
    public void loginSuccess(String username, String pass) {
        //test method logic goes here. .....


    }

    @DataProvider
    public Object[][] getData(ITestContext context) {
        String sheetNameToParse = context.getAttribute("sheetName").toString();

        String excelPath = "src/main/java/apps/teamo/testdata/Book 1.xlsx";
        return testData(excelPath, sheetNameToParse);
    }
}

============

选项B：您可以为不同的目的定义不同的测试（在套件中），如

public class DemoTestClass {
@Test(dataProvider = "getData")
public void   testSheetA(String username, String pass){
....}

@Test(dataProvider = "getData")
public void
    testSheetB(String username, String pass){...}

@Test(dataProvider = "getData")
public void
    testSheetC(String username, String pass){....}

}

另一个选项是重新组织将要分析测试名称的数据提供程序，从而提供适合测试的工作表名称：

@DataProvider
public  Object[][] getData(ITestNGMethod testContext)
{
    String excelPath = "src/main/java/apps/teamo/testdata/Book 1.xlsx";
    if(testContext.getMethodName().equals("testSheetA"))
    {
        return testData(excelPath, "sheetA");
    }
    else if(testContext.getMethodName().equals("testSheetB"))
    {

        return testData(excelPath, "sheetB");
    }
    .....

}

在本文中，我解释了如何从头开始设置selenium java maven testNG项目。

希望这对你有帮助。问候，