TestNG @Test Annotation and DataProviderClass Example

Oct. 02, 13 · Interview

25.6K Views

In the previous post, we have seen an example where dataProvider attribute has been used3 to test methods with different sets of input data for the same test method.

TestNG provides another attribute dataProviderClass in conjunction with dataProvider to fetch the input data for the test methods from an external class.

The actual class that holds input data is set to the dataProviderClass attribute and datProvider by itself holds the method name where the input data is actually fetched.

Here is a quick example to show how to use dataProviderClass and dataProvide attribute

Code

Service Class

package com.skilledmonster.example;
/**
* Simple calculator service to demonstrate TestNG Framework
*
* @author Jagadeesh Motamarri
* @version 1.0
*/
public interface CalculatorService {
int sum(int a, int b);
int multiply(int a, int b);
int div(int a, int b);
int sub(int a, int b);
}

Service Implementation Class

package com.skilledmonster.example;
/**
* Simple calculator service implementation to demonstrate TestNG Framework
*
* @author Jagadeesh Motamarri
* @version 1.0
*/
public class SimpleCalculator implements CalculatorService {
public int sum(int a, int b) {
return a + b;
}
public int multiply(int a, int b) {
return a * b;
}
public int div(int a, int b) {
return a / b;
}
public int sub(int a, int b) {
return a - b;
}
}

Data Provider Class

package com.skilledmonster.common;
import org.testng.annotations.DataProvider;
/**
* Data Provider class for TestNG test cases
*
* @author Jagadeesh Motamarri
* @version 1.0
*/
public class TestNGDataProvider {
/**
* Data Provider for testing sum of 2 numbers
*
* @return
*/
@DataProvider
public static Object[][] testSumInput() {
return new Object[][] { { 5, 5 }, { 10, 10 }, { 20, 20 } };
}
/**
* Data Provider for testing multiplication of 2 numbers
*
* @return
*/
@DataProvider
public static Object[][] testMultipleInput() {
return new Object[][] { { 5, 5 }, { 10, 10 }, { 20, 20 } };
}
}

Finally, test class that uses dataProviderClass attribute to feed the input data for the test methods

package com.skilledmonster.example;
import org.testng.Assert;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;
import com.skilledmonster.common.TestNGDataProvider;
/**
* Example to demonstrate use of dataProviderClass and dataProvide attributes of TestNG framework
*
* @author Jagadeesh Motamarri
* @version 1.0
*/
public class TestNGAnnotationTestDataProviderExample {
public CalculatorService service;
@BeforeClass
public void init() {
System.out.println("@BeforeClass: The annotated method will be run before the first test method in the current class is invoked.");
System.out.println("init service");
service = new SimpleCalculator();
}
@Test(dataProviderClass = TestNGDataProvider.class, dataProvider = "testSumInput")
public void testSum(int a, int b) {
System.out.println("@Test : testSum()");
int result = service.sum(a, b);
Assert.assertEquals(result, a + b);
}
@Test(dataProviderClass = TestNGDataProvider.class, dataProvider = "testMultipleInput")
public void testMultiple(int a, int b) {
System.out.println("@Test : testMultiple()");
int result = service.multiply(a, b);
Assert.assertEquals(result, a * b);
}
}

Output

As shown in the above console output, each of the testSum() and testMutiple() methods are invoked with different sets of input data using an external class with dataProviderClass attribute.

Advantage

More flexibility and re-usability of commonly used data across several test classes.

Download

Download TestNG DataProvider Example

Testing TestNG Data (computing) Annotation

Published at DZone with permission of Jagadeesh Motamarri. See the original article here.

Opinions expressed by DZone contributors are their own.

TestNG @Test Annotation and DataProviderClass Example

Code

Output

Download

Partner Resources

Comments

Likes