26

我们想针对一组数据值运行我们的一些测试，验证每个测试的相同条件是否成立。数据当前存储在平面文件或简单的 Excel 电子表格中。

我的第一个想法是创建一个 TestNG DataProvider，它将从文件中加载数据，并用于为每个数据值调用一次测试方法。我的问题是不同的测试需要从不同的文件加载数据，并且似乎没有任何方法可以将参数发送到 DataProvider。 有谁知道这是否可能？

理想情况下，我希望我的代码如下所示（简化示例）：

public class OddTest {
    @DataProvider(name = "excelLoader")
    public Iterator<Object[]> loadExcelData(String fileName) {
        ...
    }

    @Test(dataProvider = "excelLoader" dataProviderParameters = { "data.xls" })
    public void checkIsOddWorks(int num)
        assertTrue(isOdd(num));
    }
}
4

5 回答 5

21

取自TestNG 文档：

如果您将@DataProvider 声明java.lang.reflect.Method为第一个参数，TestNG 将传递此第一个参数的当前测试方法。当多个测试方法使用相同的 @DataProvider 并且您希望它根据它提供数据的测试方法返回不同的值时，这特别有用。

例如，以下代码在其@DataProvider 中打印测试方法的名称：

@DataProvider(name = "dp")
public Object[][] createData(Method m) {
  System.out.println(m.getName());  // print test method name
  return new Object[][] { new Object[] { "Cedric" }};
}

@Test(dataProvider = "dp")
  public void test1(String s) {
}

@Test(dataProvider = "dp")
  public void test2(String s) {
}

因此将显示：

test1
test2

这也可以结合 desolat 提供的解决方案，从上下文中确定数据和相应的方法：

    @DataProvider(name = "dp")
    public Object[][] foodp(ITestContext ctx, Method method) {
        // ...
    }
于 2009-08-12T20:48:14.743 回答
18

您可以使用TestNG 的依赖注入功能访问 DataProvider 中所有已定义的参数。这是一些需要“test_param”参数的示例 DataProvider：

@DataProvider(name = "usesParameter")
public Object[][] provideTestParam(ITestContext context) {
    String testParam = context.getCurrentXmlTest().getParameter("test_param");
    return new Object[][] {{ testParam }};
}

这需要在您中定义“test_param” suite.xml：

<!DOCTYPE suite SYSTEM "http://testng.org/testng-1.0.dtd" >
<suite name="suite">
    <parameter name="test_param" value="foo" />
    <test name="tests">
        <classes>
            ...
        </classes>
    </test>
</suite>

有关 ITestContext 类的详细信息，请参阅TestNG JavaDoc。

于 2011-01-17T15:56:38.850 回答
7

一种更通用的方法是使用groups注释来构建自定义值列表：

@DataProvider(name = "excelLoader")
public Object[][] createData(Method m) {
    ArrayList<Object[]> excelFiles = new ArrayList<Object[]>;
    // iterate over all the groups listed in the annotation
    for (String excelFile : ((Test) m.getAnnotation(Test.class)).groups()) {
        // add each to the list
        excelFiles.add(new Object[] { excelFile });
    }
    // convert the list to an array
    return excelFiles.toArray(new Object[excelFiles.size()]);
}

@Test(dataProvider = "excelLoader", groups = { "data1", "data2" })
public void test1(String excelFile) {
    // we will test "data1.xls" and "data2.xls" in this test
    String testExcelFile = excelFile + ".xls";
}

@Test(dataProvider = "excelLoader", groups = { "data2", "data3" })
public void test2(String excelFile) {
    // we will test "data2.xls" and "data3.xls" in this test
    String testExcelFile = excelFile + ".xls";
}

或者，您也可以创建自己的注释类，该类接受自定义元素，以便您可以做更多类似的事情：

@Retention(java.lang.annotation.RetentionPolicy.RUNTIME)
@Target({METHOD, TYPE, CONSTRUCTOR})
public @interface FilesToTest {
    public String[] value() default {};
}

@DataProvider(name = "excelLoader")
public Object[][] createData(Method m) {
    ArrayList<Object[]> excelFiles = new ArrayList<Object[]>;
    // iterate over all the groups listed in the annotation
    for (String excelFile : ((FilesToTest) m.getAnnotation(FilesToTest.class)).value()) {
        // add each to the list
        excelFiles.add(new Object[] { excelFile });
    }
    // convert the list to an array
    return excelFiles.toArray(new Object[excelFiles.size()]);
}

@Test(dataProvider = "excelLoader")
@FilesToTest({ "data1.xls", "data2.xls" })
public void myTest(String excelFile) {
    // we will test "data1.xls" and "data2.xls" in this test
}
于 2016-01-30T00:31:53.443 回答
1

yshua 的回答有点局限，因为您仍然必须对数据提供程序中的文件路径进行硬编码。这意味着您必须更改源代码，然后重新编译才能重新运行测试。这违背了使用 XML 文件来配置测试运行的目的。

一个更好的，绝对更hacky的解决方案是创建一个在套件之前运行的虚拟@test方法，将您的文件路径作为参数并将此信息保存在包含这些测试方法的类中。

这个解决方案并不完美，但在 TestNG 允许更好的参数传递（也许这已经改变）之前，这可能对您的需求是可行的。

于 2010-03-02T17:11:21.200 回答
0

要添加到我上面的答案，以下是如何使用 EasyTest 框架完成此操作的完整代码：

@RunWith(DataDrivenTestRunner.class)
public class MyTestClass {

@Test
@DataLoader(filePaths={myTestFile.xls}, loaderType=LoaderType.EXCEL)
public void testFirstMethod(@Param()
Map<String, Object> inputData) {
    System.out.print("Executing testFirstMethod:");
    System.out.println("library Id : " + inputData.get("LibraryId"));

}

@Test
@DataLoader(filePaths={mySecondTestFile.xls}, loaderType=LoaderType.EXCEL)
public void testSecondMethod(@Param(name="input")
MyClassObject inputData) {
    System.out.print("Executing testSecondMethod:");
    System.out.println("library Id : " + inputData.get("LibraryId"));

}

等等。如果您想了解更多关于 @DataLoader 注释在 EasyTest 中的工作原理，请查看以下内容：https ://github.com/EaseTech/easytest/wiki/EasyTest-:-Loading-Data-using-Excel

请注意，您可以使用 XML、Excel、CSV 或您自己的自定义加载器来加载数据，并且所有这些都可以一次在同一个测试类中使用，如本示例所示：https ://github.com/EaseTech/easytest/blob /master/src/test/java/org/easetech/easytest/example/TestCombinedLoadingAndWriting.java

我希望它有用。

于 2012-11-08T17:43:53.577 回答