@DataProvider ITestContext 参数

 1 package roger.testng;
 2 
 3 import java.util.Random;
 4 
 5 import org.testng.ITestContext;
 6 import org.testng.annotations.DataProvider;
 7 import org.testng.annotations.Test;
 8 
 9 /*
10  *  数据提供者在方法签名中声明了一个 ITestContext 类型的参数
11  *  testng 会将当前的测试上下文设置给它
12  *  
13  */
14 public class TestDataProviderITestContext {
15     @DataProvider
16     public Object[][] randomIntegers(ITestContext context) {
17         String[] groups= context.getIncludedGroups();
18         int size = 2;
19         for (String group : groups) {
20             System.out.println("--------------" + group);
21             if (group.equals("function-test")) {
22                 size = 10;
23                 break;
24             }
25         }
26         
27         Object[][] result = new Object[size][];
28         Random r = new Random();
29         for (int i = 0; i < size; i++) {
30             result[i] = new Object[] {new Integer(r.nextInt())};
31         }
32         
33         return result;
34     }
35     
36     // 如果在 unite-test 组中执行, 将返回2个随机整数构成数组;
37     // 如果在 function-test 组中执行, 将返回 10 个随机整数构成数组
38     @Test(dataProvider = "randomIntegers", groups = {"unit-test", "function-test"})
39     public void random(Integer n) {
40         System.out.println(n);
41     }
42     
43 }