@DataProvider ITestContext 参数

 package roger.testng;

 import java.util.Random;

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

 /*
  *  数据提供者在方法签名中声明了一个 ITestContext 类型的参数
  *  testng 会将当前的测试上下文设置给它
  *
  */
 public class TestDataProviderITestContext {
     @DataProvider
     public Object[][] randomIntegers(ITestContext context) {
         String[] groups= context.getIncludedGroups();
         int size = 2;
         for (String group : groups) {
             System.out.println("--------------" + group);
             if (group.equals("function-test")) {
                 size = 10;
                 break;
             }
         }

         Object[][] result = new Object[size][];
         Random r = new Random();
         for (int i = 0; i < size; i++) {
             result[i] = new Object[] {new Integer(r.nextInt())};
         }

         return result;
     }

     // 如果在 unite-test 组中执行, 将返回2个随机整数构成数组;
     // 如果在 function-test 组中执行, 将返回 10 个随机整数构成数组
     @Test(dataProvider = "randomIntegers", groups = {"unit-test", "function-test"})
     public void random(Integer n) {
         System.out.println(n);
     }

 }

通过 testng.xml 指定运行 unite-test 组还是 function-test 组。