1 package test.tmp; 2 3 import org.testng.Assert; 4 import org.testng.annotations.DataProvider; 5 import org.testng.annotations.Test; 6 7 import java.io.File; 8 import java.io.FileInputStream; 9 import java.io.FileNotFoundException; 10 import java.io.IOException; 11 import java.util.ArrayList; 12 import java.util.Enumeration; 13 import java.util.List; 14 import java.util.Properties; 15 16 17 public class DataDrivenTest { 18 19 @DataProvider(name = "provider") createData()20 public Object[][] createData() throws FileNotFoundException, IOException { 21 Properties p = new Properties(); 22 List<Object> vResult = new ArrayList<>(); 23 p.load(new FileInputStream(new File("c:/t/data.properties"))); 24 for (Enumeration e = p.keys(); e.hasMoreElements(); ) { 25 vResult.add(e.nextElement()); 26 } 27 28 Object[][] result = new Object[vResult.size()][1]; 29 for (int i = 0; i < result.length; i++) { 30 result[i] = new Object[] { vResult.get(i) }; 31 } 32 33 return result; 34 } 35 36 @Test(dataProvider = "provider") foo(int n)37 public void foo(int n) { 38 Assert.assertTrue(n > 0); 39 } 40 41 } 42