TestNG: More than one @DataProvider for one @Test

Malte Onken picture Malte Onken · May 31, 2012 · Viewed 32.4k times · Source

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);
   }

Answer

Cedric Beust picture Cedric Beust · May 31, 2012

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);
}