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

Question

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

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

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

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

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

在数据提供者方法本身上使用它不是一个选项，因为我必须创建与发送的参数一样多的数据提供者方法（这会破坏重用现有 dataProvider 的全部意义）。

从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，因为它一直是described here。

在下面提供两个@Test 方法： 第一个测试方法设置属性（=工作表名称）， 第二个 @Test 方法（取决于第一个）- 由数据提供者驱动，该数据提供者使用我们指定的工作表中的数据。

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");
    }
    .....

}

在本文中，我解释了如何从头开始to set-up selenium java maven testNG project。

希望这对您有所帮助。 亲切的问候，