TestNG: More than one @DataProvider for one @Test

问题

I'm using TestNG for Eclipse.

Is it possible to give two data providers step by step to the same test-function?

I could put both providers in one, but that is not what I want.

I need (not like in this example) to generate independently data.

@DataProvider(name = "dataSet1")

    public Object[][] createDataX() {
             return new Object[][] { { 1, 1 }, { 2, 2 } };
    }

@DataProvider(name = "dataSet2")

    public Object[][] createDataY() {
             return new Object[][] { { 0, 0 }, { 3, 3 } };
    }

I want to give both providers to the same test. Is this possible?

 @Test(dataProvider = "dataSet1") // ??? and "dataSet2" ???
    public void testThisFunction(int val1, int val2) {
            boolean solution = oracle(val1,val2);
            assert (solution);
   }

回答1:

No, but nothing stops you from merging these two data providers into one and specifying that one as your data provider:

public Object[][] dp1() {
  return new Object[][] {
      new Object[] { "a", "b" },
      new Object[] { "c", "d" },
  };
}

public Object[][] dp2() {
  return new Object[][] {
      new Object[] { "e", "f" },
      new Object[] { "g", "h" },
  };
}

@DataProvider
public Object[][] dp() {
  List<Object[]> result = Lists.newArrayList();
  result.addAll(Arrays.asList(dp1()));
  result.addAll(Arrays.asList(dp2()));
  return result.toArray(new Object[result.size()][]);
}

@Test(dataProvider = "dp")
public void f(String a, String b) {
  System.out.println("f " + a + " " + b);
}

回答2:

Please refer to this answer:

TestNG using multiple DataProviders with single Test method

It is much cleaner and will work for more complex things.

Maybe someone will need it too, I rewrote this method public static T[] concatAll(T[] first, T[]... rest) in a different way:

public static Object[] concat(Object[] first, Object[] second) {
    Object[] result = ArrayUtils.addAll(first, second);
    return result;
}

回答3:

    @DataProvider
    public Object[][] combinedDataProvider() {
        return Stream.of(dp1(), dp2())
                .flatMap(Arrays::stream)
                .toArray(Object[][]::new);
    }