TestNG中用Parameters或DataProvider为测试方法传入参数

转载于网络

 

一、设置参数

测试方法是可以带有参数的。每个测试方法都可以带有任意数量的参数，并且可以通过使用TestNG的@Parameters向方法传递正确的参数。

设置方式有两种方法：使用 testng.xml 或者 Data Providers。

（一）使用 testng.xml 设置参数

1.如果只使用相对简单的参数，可以在 testng.xml文件中指定：

@Parameters({ "first-name" })

@Test

public void testSingleString(String firstName) {

System.out.println("Invoked testString " + firstName);

assert "Cedric".equals(firstName);

}

在这段代码中，我们让 firstName 参数能够接到XML文件中叫做 first-name参数的值。这个XML参数被定义在 testng.xml：

<suite name="My suite">

<parameter name="first-name"  value="Cedric"/>

<test name="Simple example">

 

类似的，它也可以用在 @Before/After 和 @Factory 注解上：

@Parameters({ "datasource", "jdbcDriver" })

@BeforeMethod

public void beforeTest(String ds, String driver) {

m_dataSource = ...;                    // 查询数据源的值

m_jdbcDriver = driver;

}

这次有两个Java参数 ds 和 driver 会分别接收到来自属性datasource 和 jdbc-driver所指定的值。

 

2.参数也可以通过 Optional 注释来声明：

@Parameters("db")

@Test

public void testNonExistentParameter(@Optional("mysql") String db) { ... }

如果在你的testng.xml文件中没有找到"db"，你的测试方法就会使用 @Optional 中的值："mysql"。

 

3.@Parameters 可以被放置到如下位置：

1.在任何已经被 @Test, @Before/After 或 @Factory 注解过的地方。

2.在测试类中至多被放到一个构造函数签。这样，TestNG才能在需要的时候使用 testng.xml中特定的参数来实例化这个类。这个特性可以被用作初始化某些类中的值，以便稍后会被类中其他的方法所使用。

 

注意：

XML中的参数会按照Java参数在注解中出现的顺序被映射过去，并且如果数量不匹配，TestNG会报错。

参数是有作用范围的。在testng.xml中，你即可以在<suite> 标签下声明，也可以在<test>下声明。如果两个参数都有相同的名字，那么，定义在<test>中的有优先权。这在你需要覆盖某些测试中特定参数的值时，会非常方便。

 

（二）使用DataProviders提供参数

在 testng.xml 中指定参数可能会有如下的不足：

1.如果你压根不用 testng.xml.

2.你需要传递复杂的参数，或者从Java中创建参数（复杂对象，对象从属性文件或者数据库中读取的etc...）

这样的话，你就可以使用Data Provider来给需要的测试提供参数。所谓数据提供者，就是一个能返回对象数组的数组的方法，并且这个方法被@DataProvider注解标注：

DataProvider的定义如下：

 

@DataProvider(name = "range-provider")

public Object[][] rangeData() {

int lower = 5;

int upper = 10;

return new Object[][] {

{ lower-1, lower, upper, false },

{ lower, lower, upper, true },

{ lower+1, lower, upper, true },

{ upper, lower, upper, true},

{ upper+1, lower, upper, false },

};

}

 

调用DataProvider的方式如下：

@Test(dataProvider = "range-provider")

public void testIsBetween(int n, int lower,int upper, boolean expected)

{

println("Received " + n + " " + lower + "-"+ upper + " expected: " + expected);

Assert.assertEquals(expected, isBetween(n, lower, upper));

}

 

被@Test标注的方法通过dataProvider属性指明其数据提供商。这个名字必须与@DataProvider(name="...")中的名字相一致。

 

DataProvider返回的是一个Object的二维数组，二维数组中的每个一维数组都会传递给调用函数，作为参数使用。运行的时候，会发现，@Test标识的test method被执行的次数和object[][]包含的一维数组的个数是一致的，而@Test标识的函数的参数个数，也和object内一维数组内的元素数是一致的。

 

运行后的输出结果如下：

Received 4 5-10 expected: false

Received 5 5-10 expected: true

Received 6 5-10 expected: true

Received 10 5-10 expected: true

Received 11 5-10 expected: false

 

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

Parameter Suite

Total tests run: 5, Failures: 0, Skips: 0

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

 

（三）DataProviders扩展

默认的情况下，数据提供者会查找当前的测试类或者测试类的基类。如果你希望它能够被其他的类所使用，那么就要将其指定为static，并且通过 dataProviderClass属性指定要使用的类：

public static classStaticProvider {

@DataProvider(name = "create")

public static Object[][] createData() {

  return new Object[][] {

    new Object[] { new Integer(42) }

  }

}

}

 

public class MyTest {

@Test(dataProvider = "create", dataProviderClass =StaticProvider.class)

public void test(Integer n) {

  // ...

}

}

 

Data Provider方法可以返回如下两种类型中的一种：

1.含有多个对象的数组(Object[][])，其中第一个下标指明了测试方法要调用的次数，第二个下标则完全与测试方法中的参数类型和个数相匹配。上面的例子已经说明。

2.另外一个是迭代器Iterator<Object[]>。二者的区别是迭代器允许你延迟创建自己的测试数据。TestNG会调用迭代器，之后测试方法会一个接一个的调用由迭代器返回的值。在你需要传递很多参数组给测试组的时候，这样你无须提前创建一堆值。

 

下面是使用JDK5 的例子 (注意 JDK 1.4 的例子不适用泛型)：

public Iterator createData() {

return new MyIterator(DATA);

)

@DataProvider(name = "test1")

public Iterator

 

如果你声明的 @DataProvider 使用 java.lang.reflect.Method作为第一个参数，TestNG会把当前的测试方法当成参数传给第一个参数。这一点在你的多个测试方法使用相同的@DataProvider的时候，并且你想要依据具体的测试方法返回不同的值时，特别有用。

例如，下面的代码它内部的 @DataProvider 中的测试方法的名字：

@DataProvider(name = "dp")

public Object[][] createData(Method m) {

System.out.println(m.getName()); 

return new Object[][] { new Object[] { "Cedric" }};

}

 

@Test(dataProvider = "dp")

public void test1(String s) {

}

@Test(dataProvider = "dp")

public void test2(String s) {

}

所以会显示：

test1

test2

 

Data provider可以通过属性 parallel实现并行运行：

@DataProvider(parallel = true)

// ...

使用XML文件运行的data provider享有相同的线程池，默认的大小是10.你可以通过修改该在<suite> 标签中的值来更改：

<suite name="Suite1" data-provider-thread-count="20" >

...

如果你需要让指定的几个data provider运行在不同的线程中，那么就必须通过不同的xml文件来运行。

 

二、依赖方法

有些时候，需要按照特定顺序调用测试方法。

1.确保在进行更多的方法测试之前，有一定数量的测试方法已经成功完成。

2.在初始化测试的时候，同时希望这个初始化方法也是一个测试方法（ @Before/After不会出现在最后生成的报告中）。

 

为此，你可以使用 @Test 中的 dependsOnMethods 或 dependsOnGroups 属性。

这两种依赖：

1.Hard dependencies（硬依赖）。所有的被依赖方法必须成功运行。只要有一个出问题，测试就不会被调用，并且在报告中被标记为SKIP。

2.Soft dependencies（软依赖）。 即便是有些依赖方法失败了，也一样运行。如果你只是需要保证你的测试方法按照顺序执行，而不关心他们的依赖方法是否成功。那么这种机制就非常有用。可以通过添加"alwaysRun=true" 到 @Test 来实现软依赖。

 

硬依赖的例子：

@Test

public void serverStartedOk() {}

@Test(dependsOnMethods = { "serverStartedOk" })

public void method1() {}

 

此例中，method1() 依赖于方法 serverStartedOk()，从而保证 serverStartedOk()总是先运行。

也可以让若干方法依赖于组：

@Test(groups = { "init" })

public void serverStartedOk() {}

@Test(groups = { "init" })

public void initEnvironment() {}

 

@Test(dependsOnGroups = { "init.* })

public void method1() {}

本例中,method1()依赖于匹配正则表达式"init.*"的组，由此保证了serverStartedOk()和 initEnvironment() 总是先于 method1()被调用。

 

注意：正如前面所说的那样，在相同组中的调用可是在夸测试中不保证顺序的。

 

如果你使用硬依赖，并且被依赖方法失败(alwaysRun=false，即默认是硬依赖)，依赖方法则不是被标记为FAIL而是SKIP。被跳过的方法会被在最后的报告中标记出来（HTML既不用红色也不是绿色所表示），主要是被跳过的方法不是必然失败，所以被标出来做以区别。

 

无论dependsOnGroups 还是 dependsOnMethods 都可以接受正则表达式作为参数。对于 dependsOnMethods，如果被依赖的方法有多个重载，那么所有的重载方法都会被调用。如果你只希望使用这些重载中的一个，那么就应该使用 dependsOnGroups。

 

三、类级注解

通常 @Test也可以用来标注类，而不仅仅是方法：

@Test

public class Test1 {

public void test1() {

}

 

public void test2() {

}

}

处于类级的 @Test会使得类中所有的public方法成为测试方法，而不管他们是否已经被标注。当然，你仍然可以用 @Test 注解重复标注测试方法，特别是要为其添加一些特别的属性时。

 

例如：

@Test

public class Test1 {

public void test1() {

}

 

@Test(groups = "g1")

public void test2() {

}

}

上例中 test1() 和 test2()都被处理，不过在此之上 test2() 现在还属于组 "g1"。