使用@MethodSource和返回ArrayList的方法进行参数化的JUnit测试

时间:2019-06-28 12:30:09

标签: java junit5 parametrized-testing

我想为具有三个输入参数的方法实现参数化的JUnit测试。我希望测试能够通过这三个输入参数的所有可能组合的笛卡尔乘积进行测试。 为此,我有一个生成笛卡尔乘积并将其存储在Arraylist中的方法。 如何访问测试方法的单个值?我已经读过有关返回参数流的信息,但是我想为参数生成值,而不是显式地编写它们。

@ParameterizedTest
@MethodSource("generateCartesianProduct")
public void myTest(int x, int y, int z) {        
    Assertions.assertTrue(methodToTest(
           x, y, z));
}

private static whatToReturnHere?? generateCartesianProduct() {
    int[] x = {1, 2, 3};
    int[] y = {4, 5, 6};
    int[] z = {7, 8, 9};
    ArrayList<Integer> list = new ArrayList<>();
    ArrayList<ArrayList> result = new ArrayList<>();

    for (int i = 0; i < x.length; i++) {
        for (int j = 0; j < y.length; j++) {
            for (int k = 0; k < z.length; k++) {
                list = new ArrayList<>();
                list.add(x[i]);
                list.add(y[j]);
                list.add(z[k]);
                result.add(list);
            }
        }
    }
    return result;
}

1 个答案:

答案 0 :(得分:0)

尝试以下操作并返回org.junit.jupiter.params.provider.Arguments(如@MethodSource的javadoc所建议。

@ParameterizedTest
@MethodSource("generateCartesianProduct")
public void myTest(final int x, final int y, final int z) {
    System.out.println(x + " " + y + " " + z);
}

private static List<Arguments> generateCartesianProduct() {
    final int[] x = { 1, 2, 3 };
    final int[] y = { 4, 5, 6 };
    final int[] z = { 7, 8, 9 };
    final List<Integer> list = new ArrayList<>();
    final List<Arguments> result = new ArrayList<>();

    for (final int element : x) {
        for (final int element2 : y) {
            for (final int element3 : z) {
                final Object[] values = { element, element2, element3 };
                result.add(() -> values);
            }
        }
    }
    return result;
}

编辑:众所周知,有一个番石榴函数可以生成笛卡尔积:Lists.cartesianProduct