在TestNG DataProvider中迭代CSV文件

Question

我对如何最好地实现一个简单的DataProvider有点困惑，以前没有这样做过。

我有一个非常简单的逗号分隔的.csv文件：

978KAL,625JBH,876SSH,452GSH

我只需要读入并迭代记录，对每个记录运行相同的测试，直到完成为止。

String csvFile = "src/test/resources/registrationsData.csv";
    BufferedReader br = null;
    String line = "";
    String cvsSplitBy = ",";


    @DataProvider(name="getRegistrations")
    private Object[] getCSVTestData() {
        Object [] registrationsObject;
        try {

            br = new BufferedReader(new FileReader(csvFile));
            while ((line = br.readLine()) != null) {

                // use comma as separator
                String [] registrations = line.split(cvsSplitBy);

                System.out.println( registrations[0] + "," + registrations[1]);

            }

        } catch//File not found & IOException handling here
        registrationsObject = new Object[][]{registrations};
        return registrationsObject;
    }



    @Test(dataProvider = "getRegistrations")
    public void getRegistrations(String registration){

        Object[] objRegArray =  getCSVTestData();

        for(int i=0; objRegArray.length>i; i++){
        //run tests for every record in the array (csv file)
    }
 }

我不清楚如何（和/或最好的方式）从objRegArray对象检索每个记录。

我猜这是一个基本的集合问题；谁能给我指正确的路吗？

Answer 1

请使用我下面的解释检查此代码：

package click.webelement.testng;

import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import java.io.File;
import java.io.FileNotFoundException;
import java.util.Iterator;
import java.util.Scanner;

public class OneLineCSV {

    final static String CSV_FILE = "/path_to_file/oneline.csv";
    final static String DELIMETER = ",";

    @DataProvider(name = "test")
    public Iterator<Object[]> testDP(){
        try {
            Scanner scanner = new Scanner(new File(CSV_FILE)).useDelimiter(DELIMETER);
            return new Iterator<Object[]>() {
                @Override
                public boolean hasNext() {
                    return scanner.hasNext();
                }
                @Override
                public Object[] next() {
                    return new Object[]{scanner.next()};
                }
            };
        } catch (FileNotFoundException e) {
            e.printStackTrace();
            return null;
        }
    }

    @Test(dataProvider = "test")
    public void testOneLineCSV(String value){
        System.out.println(value);
    }

}

因此，我将使用scanner类，这样它就可以方便地将字符串解析为令牌。

我还将在数据提供程序中使用返回迭代器 的功能，因为扫描器就是这样设计的。您只需用新的迭代器包装它，该迭代器将Scanner.Next()返回的String转换为new Object[]{scanner.Next}。