@DataProvider Method 参数传递

package roger.testng;

import java.lang.reflect.Method;

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

/*
 *  Method 参数 Demo
 *  如果数据提供者的第一个参数是 java.lang.reflect.Method,
 *  testng 传递这个将调用的测试方法
 *  如：调用的方法名为 two, 数据提供者就返回2
 */
public class TestDataProviderMethod {
    // Method 表示当前测试的方法
    @DataProvider
    public Object[][] provideNumbers(Method method) {
        Object[][] result = null;
        // 根据方法名称判断
        if (method.getName().equals("one")) {
            result = new Object[][] { new Object[] { 1 } };
        } else if (method.getName().equals("two")) {
            result = new Object[][] { new Object[] { 2 } };
        } else {
            result = new Object[][] { new Object[] { 3 } };
        }
        return result;
    }

    @Test(dataProvider = "provideNumbers")
    public void one(int param) {
        System.out.println("one received: " + param);
    }

    @Test(dataProvider = "provideNumbers")
    public void two(int param) {
        System.out.println("Two received: " + param);
    }

    @Test(dataProvider = "provideNumbers")
    public void three(int param) {
        System.out.println("Three received: " + param);
    }

}