TestNG 使用多个 DataProvider 和单个测试方法 [英] TestNG using multiple DataProviders with single Test method
问题描述
我一直在寻找一种在我的测试方法中使用多个 DataProvider 的方法.我的场景如下:
I have been looking for a way to use multiple DataProviders in my test method. My scenario is as follows:
假设我们有一个 DataProvider 类:
lets say we have a DataProvider class:
@Test
public class ExampleDataProvider {
/**
* Returns the list of shape codes.
*
* @return the collection shape codes.
*/
@DataProvider(name = "ShapeCodes")
public static Object[][] getShapeCodes() {
return new Object[][] { new Object[] { Shape.Square },
new Object[] { Shape.Triangle }
};
}
/**
* Returns the list of color codes.
*
* @return the collection of color codes.
*/
@DataProvider(name = "ColorCodes")
public static Object[][] geColorCodes() {
return new Object[][] { new Object[] { Color.Green },
new Object[] { Color.Red }
};
}
}
现在在我的测试方法中,我想针对所有场景组合运行:
Now in my Test method I want to run for all combinations of scenarios:
- 绿色广场
- 红场
- 绿色三角形
- 红三角
鉴于我无法使用 @Test
批注指定多个 DataProvider,我应该如何在我的代码中实现这一点
How should I achieve this in my code, given that I cant specify multiple DataProviders with @Test
annotation
@Test(dataProvider = "ShapeCodes", dataProviderClass = ExampleDataProvider.class)
public void test(String ShapeCode, String ColorCode) throws IOException {
.............
/* tests for color shape combination */
.............
}
我发现了一个类似的问题和一个@解决方法 但我仍然想知道是否有更好的方法来处理这个问题.
EDIT : I found a similar problem and a @ workaround but I am still wondering if there are better ways to handle this.
推荐答案
由于缺乏更好的方法,我决定坚持解决方案.以下是如何实现上述场景的示例:
For the lack of a better approach, I decided to stick to the workaround. here is an example of how the above scenario could be implemented:
@Test
public class ExampleDataProvider {
/**
* Returns the list of shape codes.
*
* @return the collection shape codes.
*/
@DataProvider(name = "ShapeCodes")
public static Object[][] getShapeCodes() {
return new Object[][] { new Object[] { Shape.Square },
new Object[] { Shape.Triangle }
};
}
/**
* Returns the list of color codes.
*
* @return the collection of color codes.
*/
@DataProvider(name = "ColorCodes")
public static Object[][] geColorCodes() {
return new Object[][] { new Object[] { Color.Green },
new Object[] { Color.Red }
};
}
/**
* Returns the list object codes providing a color shape combination.
*
* @return the collection of object codes.
*/
@DataProvider(name = "objectCodes")
public static Object[][] getObjectCodes(){
return combine(geColorCodes(), getShapeCodes());
}
/**
* Returns the list of combination of color and shape codes.
*
* @return the collection of combined color and shape codes.
*/
public static Object[][] combine(Object[][] a1, Object[][] a2){
List<Object[]> objectCodesList = new LinkedList<Object[]>();
for(Object[] o : a1){
for(Object[] o2 : a2){
objectCodesList.add(concatAll(o, o2));
}
}
return objectCodesList.toArray(new Object[0][0]);
}
@SafeVarargs
public static <T> T[] concatAll(T[] first, T[]... rest) {
//calculate the total length of the final object array after the concat
int totalLength = first.length;
for (T[] array : rest) {
totalLength += array.length;
}
//copy the first array to result array and then copy each array completely to result
T[] result = Arrays.copyOf(first, totalLength);
int offset = first.length;
for (T[] array : rest) {
System.arraycopy(array, 0, result, offset, array.length);
offset += array.length;
}
return result;
}
}
这样我就可以单独使用我的颜色代码和形状代码,并且还允许我使用组合.
This way I get to use my Color codes and Shape codes separately and also provides me to use the combination.
所以,我的测试方法看起来像:
So, my test methods would look like:
@Test(dataProvider = "objectCodes", dataProviderClass = ExampleDataProvider.class)
public void test(String ShapeCode, String ColorCode) throws IOException {
.............
/* tests for color shape combination */
.............
}
@Test(dataProvider = "ShapeCodes", dataProviderClass = ExampleDataProvider.class)
public void test(String ShapeCode) throws IOException {
.............
/* tests for shapes */
.............
}
@Test(dataProvider = "ColorCodes", dataProviderClass = ExampleDataProvider.class)
public void test(String ColorCode) throws IOException {
.............
/* tests for colors */
.............
}
这篇关于TestNG 使用多个 DataProvider 和单个测试方法的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持IT屋!