1

这是我的代码。我从两个 CSV 文件（一个用于用户，一个用于餐厅）中读取数据，并且我在 Selenium Webdriver 中提供了两个不同的测试类和两个不同的数组，以便对测试进行参数化。

有什么办法我不在这里重复这么多代码吗？

final String FILE_PATH = "src/test/resources/250.csv";
final String FILE_PATH2 = "src/test/resources/places.csv";

    //read CSV file and supply data for test purposes
    CSVReader reader = new CSVReader(new FileReader(FILE_PATH));
    ArrayList<ArrayList<String>> array = new ArrayList<ArrayList<String>>();
    String[] nextLine;
    while ((nextLine = reader.readNext()) != null) {
        ArrayList<String> list = new ArrayList<String>();
        for (int i = 0; i < 5; i++) { // 5 is the number of columns
            list.add(nextLine[i]);
        }
        array.add(list);

    }
    reader.close();

    CSVReader reader2 = new CSVReader(new FileReader(FILE_PATH2));
    ArrayList<ArrayList<String>> array2 = new ArrayList<ArrayList<String>>();
    String[] nextLine2;
    while ((nextLine2 = reader2.readNext()) != null) {
        ArrayList<String> list2 = new ArrayList<String>();
        for (int i = 0; i < 5; i++) { // 5 is the number of columns
            list2.add(nextLine2[i]);
        }
        array2.add(list2);

    }
    reader2.close();
4

4 回答 4

3

声明一个...函数怎么样

ArrayList<ArrayList<String>> fromFileToArray(String filename)
于 2013-08-11T11:36:29.037 回答
1 
  List<List<String>> getList(String filePath){
    CSVReader reader = new CSVReader(new FileReader(filePath));
    ArrayList<ArrayList<String>> array = new ArrayList<ArrayList<String>>();
    String[] nextLine;
    while ((nextLine = reader.readNext()) != null) {
        ArrayList<String> list = new ArrayList<String>();
        for (int i = 0; i < 5; i++) { // 5 is the number of columns
            list.add(nextLine2[i]);
        }
        array.add(list);

    }
    reader.close();

    return array;
  }
  • 获取列表（文件路径）
  • 获取列表（FILE_PATH2）
于 2013-08-11T11:39:35.027 回答
1

您创建一个方法如下

ArrayList<ArrayList<String>> read (FileReader reader)
{
   CSVReader reader = new CSVReader(reader);
   ArrayList<ArrayList<String>> array = new ArrayList<ArrayList<String>>();
   String[] nextLine;
   while ((nextLine = reader.readNext()) != null) {
    ArrayList<String> list = new ArrayList<String>();
    for (int i = 0; i < 5; i++) { // 5 is the number of columns
        list.add(nextLine[i]);
    }
    array.add(list);

}
reader.close();
return array;
}

然后你可以调用下面的方法两次

ArrayList<ArrayList<String>> array1 = read(new FileReader(FILE_PATH));
ArrayList<ArrayList<String>> array2 = read(new FileReader(FILE_PATH2));
于 2013-08-11T11:40:28.367 回答
1 
public ArrayList<ArrayList<String>> getCSVContent(String filepath) throws Exception {
    CSVReader reader = new CSVReader(new FileReader(FILE_PATH));
    ArrayList<ArrayList<String>> array = new ArrayList<ArrayList<String>>();
    String[] nextLine;
    while ((nextLine = reader.readNext()) != null) {
        ArrayList<String> list = new ArrayList<String>();
        for (int i = 0; i < 5; i++) { // 5 is the number of columns
            list.add(nextLine[i]);
        }
        array.add(list);
    }
    reader.close();
    return array;
}
于 2013-08-11T11:41:34.067 回答