Testng注解之dataprovider(参数化)

数据驱动是最常见的自动化测试框架模式之一,TestNG亦支持数据驱动,常用的有@Parameters和@DataProvider两种注解。这次主要是记录dataprovider的用法。
使用DataProvider提供数据有两种形式:

第一种:测试代码和测试数据放在同一个类中;

定义一个函数,函数名随意取,使用@DataProvider注解,注解中用name给这个provider起名字,该函数返回一个二维数组。

package BaseTestCast;

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

public class Testdemo {
   @DataProvider(name = "data")
    public Object[][] createData1() {
        return new Object[][] {
                { "zhangsan", 36},
                { "lisi",20},
                { "wanger",20} };
    }
//定义了属性name时,提取数据时可以使用name,没有定义name时,提取数据时直接使用方法名称。
    @Test(dataProvider = "data") 
    public void testProvider(String n1, Integer n2) {//参数必须和DataProvider返回的一致
        System.out.println(n1 + " " + n2);
    }
}

运行结果:

可以看到该测试方法运行了3次,每一次都把一个DataProvider提供的数组作为自己的参数输入然后执行了测试。

第二种:把所有的数据提供都单独写在一个类里面,当测试数据比较多时,这种方法利于维护。

这种方式需通过dataProviderClass属性来指定@DataProvider类位置,并且@DataProvider注解的方法保持静态(static)。示例:

//数据类:
public class dataProviderDemo {
    @DataProvider(name = "data")
    public static Object[][] createData1() {
        return new Object[][] {
                { "zhangsan", 36 },
                { "lisi",20},
                { "wanger",20}
        };
    }
}
//test类:
public class Testdemo {
    @Test(dataProvider = "data",dataProviderClass = dataProviderDemo.class)
    public void testProvider(String n1, Integer n2) {
        System.out.println(n1 +","+ n2);
    }
}

运行test后,结果:

zhangsan,36
lisi,20
wanger,20

除了上述的return Object[][],@DataProvider注解的方法也允许使用迭代器,区别是迭代器允许你延迟创建自己的测试数据,TestNG会调用迭代器,之后测试方法会一个接一个的调用由迭代器返回的值。在你需要传递很多参数组给测试方法的时候,无须提前创建一堆值。

package BaseTestCast;

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

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

public class Testdemo {
    @DataProvider
    public Iterator providerString() {
        List item = new ArrayList<>();
        for (int i = 0; i < 5; i++) {
            item.add(String.valueOf(i));
        }

        List users = new ArrayList<>();
        for (Object u : item) {
            users.add(new Object[]{u});
        } 
        return users.iterator();
    }
    @Test(dataProvider = "providerString")
    public void testmethod(String param) {
        System.out.println("method received:" + param);
    }
}
 
 

运行结果:
  • 1、使用二维数组作为参数输出,每一个单元作为参数输出
  • 2、使用迭代类型作为参数输出,每一个迭代作为参数输出

带Method参数的DataProvider

如果希望DataProvider对不同的Test方法使用不同的dataprovider,那么在DataProvider中可以使用Method参数。

package BaseTestCast;

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

import java.lang.reflect.Method;

public class Testdemo {

    @Test(dataProvider = "data")
    public void getFirst(String name, int age) {
        System.out.println("print第一组"+name+";"+age);
    }

    @Test(dataProvider = "data")
    public void getSecond(String name, int age) {
        System.out.println("print第二组 " + name +";"+age);
    }
    @Test(dataProvider = "data")
    public void getthird(String name, int age) {
        System.out.println("print第三组 " + name +";"+age);
    }

    @DataProvider(name = "data")
    public Object[][] provider(Method method) {
        Object[][] objects;
        if (method.getName().equals("getFirst")) {              //如果调用该DataProvider的函数是getFirst,那么就返回这个数组
            objects = new Object[][] { { "test11", 11 }, { "test12", 12 } };
        } else if (method.getName().equals("getSecond")) {//如果调用该DataProvider的函数是getSecond,那么就返回这个数组
            objects = new Object[][] { { "test21", 21 }, { "test22", 22 } };
        } else {                                                                    //如果调用该DataProvider的函数不是getFirst也不是getSecond,那么就返回这个数组
            objects = new Object[][] { {"test31",31}, {"test32",32} };
        }

        return objects;
    }

}

运行结果:

这里根据调用DataProvider的函数名,返回了不同的参数。
由于没有指定运行顺序,所以Test会按照字母顺序执行,先执行getFirst,然后getSecond..

参考文档:
1、https://my.oschina.net/771328074/blog/832423
2、https://www.cnblogs.com/lixiaowei395659729/p/8717992.html
3、https://www.cnblogs.com/beifucangqiong/p/11246988.html

你可能感兴趣的:(Testng注解之dataprovider(参数化))