CODEWITHSUNDEEP
javatestingautomated-teststestngdataprovider
Malte Onken
Malte Onken

Reputation: 769

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

Upvotes: 28

Views: 48727

Answers (5)

Dhanashree K.
Dhanashree K.

Reputation: 39

This is the working fix:

@DataProvider(name = "requestValidator")
public Object[][] getRequestValidationMap() {
    ResponseEntity clientDetailsResponse = ResponseUtil.successResponse(HttpStatus.OK,
            clientDetailsEntityDao);
    ResponseEntity failureResponse = ResponseUtil.handleErrorResponse(HttpStatus.INTERNAL_SERVER_ERROR,
            ERRCONSTANT.CUSTOM_CODE.getErrorDescription(), ERRCONSTANT.CUSTOM_CODE.getCategory(),
            ERRCONSTANT.CUSTOM_CODE.getValue());
    return new Object[][] { { "client", clientDetailsResponse, 200, HttpServletResponse.SC_OK },
            { "error", failureResponse, 500, HttpServletResponse.SC_INTERNAL_SERVER_ERROR } };
}

In the utility method

@Test(dataProvider = "requestValidator")
public void testMultipleArgs(String key, ResponseEntity requestValidatorResponse, int statusCodeExpected, int actualStatusCode) throws Exception {
        
}

Upvotes: 0

Wee
Wee

Reputation: 131

@DataProvider
public Object[][] combinedDataProvider() {
    // Using stream to combine the two separate data providers.
    return Stream.of(dp1(), dp2())
                 .flatMap(Arrays::stream)
                 .toArray(Object[][]::new);
}

Upvotes: 3

Sashko
Sashko

Reputation: 191

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

Upvotes: 1

Dinesh Ahuja
Dinesh Ahuja

Reputation: 1

Yes,

You can write @Test(dataProvider="name_of_first_dataprovider,name_of_second_dataprovider")

Upvotes: -10

Cedric Beust
Cedric Beust

Reputation: 15608

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

Upvotes: 37

Related Questions