Selenium+java - 使用csv文件做数据驱动 - Java

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

前言

早期我们使用TestNG 来做数据驱动进行测试，测试数据是写在测试用例脚本中。这会使得测试脚本的维护工作量很大。因此我们可以将测试的数据和脚本分开。

而我们经常使用会使用csv文件来做为导出数据时的存储文件，因此，可以通过读取 CSV文件存储数据，然后将数据传递给测试脚本进行测试。那么接下来我们将进行讲解使用csv文件做数据驱动。

使用csv做数据驱动测试

使用csv做数据驱动

创建csv数据文件

具体代码示例如下：

import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;import java.io.BufferedReader;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;public class TestCSVData { /**
* 读取csv中的文件
* @param filePath
* @return
*/
public List<Map<String, String>> getData(String filePath){
List<Map<String, String>> list=new ArrayList<Map<String, String>>();;
FileInputStream fileInputStream = null;
try {
fileInputStream = new FileInputStream(filePath);
InputStreamReader inputStreamReader = new InputStreamReader(fileInputStream);
BufferedReader br = new BufferedReader(inputStreamReader);
for (String line = br.readLine(); line != null; line = br.readLine()) {
Map<String, String> map = new HashMap<>();
String key=line.split(",")[0];
String value=line.split(",")[1];
map.put("userName",key);
map.put("passWord",value);
list.add(map);
}
br.close();
} catch (Exception e) {
e.printStackTrace();
}
return list; } @DataProvider
public Object[][] testCSVData() {
List<Map<String, String>> result = getData("d:\\data.csv");
Object[][] files = new Object[result.size()][];
for (int i = 0; i < result.size(); i++) {
files[i] = new Object[]{result.get(i)};
}
return files;
} @Test(dataProvider = "testCSVData")
public void testCSVData(Map<String,String> param) {
System.out.println(param.get("userName")+"\t"+param.get("passWord"));
}
}