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);

     }

 }