使用Selenium和Java对比CSV数据与Web表格：解决数据解析难题

// 假设已初始化 driver 和 mytable1
WebElement mytable1 = driver.findElement(By.cssSelector("#dStocks1"));
List<WebElement> rows_table1 = mytable1.findElements(By.tagName("tr"));
int rows_count1 = rows_table1.size();

String path = (filePath); // filePath 为 CSV 文件路径
BufferedReader br = null;
String line;
String splitBy = ",(?=([^\"]|\"[^\"]*\")*$)"; // 尝试处理带引号的逗号
br = new BufferedReader(new FileReader(path));
br.readLine(); // 跳过CSV头
while ((line = br.readLine()) != null) {
    // 内部循环逻辑可能导致问题
    for (int col = 1; col < line.length();) { // line.length() 不等于列数
        for (int row = 1; row < rows_count1;) {
            String[] cells = line.split(splitBy);
            // ... 后续对比逻辑 ...
            String col1 = cells[col].substring(1, cells[col].length() - 1); // 索引可能越界
            // ...
            br.readLine(); // 再次读取行，跳过数据
            col++;
        }
        row++;
    }
}

import java.io.File;
import java.io.FileNotFoundException;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

public class CSVReaderUtility {

    private static final String COMMA_DELIMITER = ",";

    /**
     * 从CSV行中提取字段值。
     * 使用Scanner按逗号分隔，可以更好地处理带引号的字段。
     *
     * @param line CSV文件中的一行字符串
     * @return 包含该行所有字段值的List<String>
     */
    private static List<String> getRecordFromLine(String line) {
        List<String> values = new ArrayList<>();
        try (Scanner rowScanner = new Scanner(line)) {
            // 设置分隔符
            rowScanner.useDelimiter(COMMA_DELIMITER);
            while (rowScanner.hasNext()) {
                // 移除可能的首尾空格或引号（如果需要，可在此处添加更复杂的去引号逻辑）
                String value = rowScanner.next().trim();
                // 简单的去引号处理，如果字段被双引号包围
                if (value.startsWith("\"") && value.endsWith("\"") && value.length() > 1) {
                    value = value.substring(1, value.length() - 1);
                }
                values.add(value);
            }
        }
        return values;
    }

    /**
     * 读取整个CSV文件并将其内容存储为List<List<String>>。
     *
     * @param filePath CSV文件的路径
     * @return 包含所有CSV行和字段的列表
     * @throws FileNotFoundException 如果文件不存在
     */
    public static List<List<String>> readCsvFile(String filePath) throws FileNotFoundException {
        List<List<String>> records = new ArrayList<>();
        try (Scanner scanner = new Scanner(new File(filePath))) {
            // 跳过CSV文件的标题行（如果存在）
            if (scanner.hasNextLine()) {
                scanner.nextLine(); // 读取并丢弃标题行
            }
            while (scanner.hasNextLine()) {
                records.add(getRecordFromLine(scanner.nextLine()));
            }
        }
        return records;
    }

    public static void main(String[] args) {
        try {
            // 示例用法：假设CSV文件在项目的 src/test/resources 目录下
            String csvFilePath = "src/test/resources/test.csv";
            List<List<String>> csvData = readCsvFile(csvFilePath);
            System.out.println("CSV Data:");
            csvData.forEach(System.out::println);
        } catch (FileNotFoundException e) {
            System.err.println("CSV file not found: " + e.getMessage());
        }
    }
}

import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.testng.Assert; // 或者 org.junit.Assert;
import org.testng.annotations.AfterClass;
import org.testng.annotations.BeforeClass;
import org.testng.annotations.Test;

import java.io.FileNotFoundException;
import java.util.List;
import java.util.concurrent.TimeUnit;

public class WebTableCsvComparison {

    private WebDriver driver;
    private List<List<String>> csvData;
    private final String CSV_FILE_PATH = "src/test/resources/test.csv"; // 确保路径正确
    private final String WEB_PAGE_URL = "http://your-web-application-url.com"; // 替换为你的网页URL

    @BeforeClass
    public void setup() throws FileNotFoundException {
        // 设置WebDriver路径
        System.setProperty("webdriver.chrome.driver", "path/to/chromedriver"); // 替换为你的chromedriver路径
        driver = new ChromeDriver();
        driver.manage().window().maximize();
        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);

        // 加载CSV数据
        csvData = CSVReaderUtility.readCsvFile(CSV_FILE_PATH);
        if (csvData.isEmpty()) {
            System.out.println("警告: CSV文件为空或只包含标题行。");
        } else {
            System.out.println("成功加载CSV数据，共 " + csvData.size() + " 行。");
        }

        // 导航到网页
        driver.get(WEB_PAGE_URL);
    }

    @Test
    public void compareWebTableWithCsvData() {
        // 定位Web表格元素
        WebElement webTable = driver.findElement(By.cssSelector("#dStocks1")); // 替换为你的表格CSS选择器
        List<WebElement> webTableRows = webTable.findElements(By.tagName("tr"));

        // 假设CSV数据不包含标题行，且与Web表格的行数匹配
        // 如果CSV包含标题行，并且在readCsvFile中已经跳过，那么这里的csvData就是实际数据行
        // 如果Web表格也有标题行，需要跳过
        int webTableDataRowsStartIndex = 1; // 假设Web表格第一行是标题，从第二行开始是数据
        int csvDataRowsStartIndex = 0; // 假设csvData已经是纯数据，从第一行开始是数据

        // 确保Web表格数据行数与CSV数据行数一致
        Assert.assertTrue(webTableRows.size() - webTableDataRowsStartIndex <= csvData.size(),
                "Web表格数据行数 (" + (webTableRows.size() - webTableDataRowsStartIndex) + ") 多于CSV数据行数 (" + csvData.size() + ")");
        Assert.assertTrue(webTableRows.size() - webTableDataRowsStartIndex >= csvData.size(),
                "Web表格数据行数 (" + (webTableRows.size() - webTableDataRowsStartIndex) + ") 少于CSV数据行数 (" + csvData.size() + ")");


        for (int i = webTableDataRowsStartIndex; i < webTableRows.size(); i++) {
            WebElement webTableRow = webTableRows.get(i);
            List<WebElement> webTableCells = webTableRow.findElements(By.tagName("td"));

            // 获取对应的CSV数据行
            List<String> currentCsvRow = csvData.get(i - webTableDataRowsStartIndex + csvDataRowsStartIndex);

            // 确保Web表格列数与CSV数据列数一致
            Assert.assertTrue(webTableCells.size() <= currentCsvRow.size(),
                    "Web表格第 " + (i + 1) + " 行的列数 (" + webTableCells.size() + ") 多于CSV数据列数 (" + currentCsvRow.size() + ")");
            Assert.assertTrue(webTableCells.size() >= currentCsvRow.size(),
                    "Web表格第 " + (i + 1) + " 行的列数 (" + webTableCells.size() + ") 少于CSV数据列数 (" + currentCsvRow.size() + ")");


            for (int j = 0; j < webTableCells.size(); j++) {
                String webTableCellText = webTableCells.get(j).getText().trim();
                String csvCellText = currentCsvRow.get(j).trim();

                // 执行对比
                System.out.println(String.format("对比行 %d, 列 %d: Web='%s', CSV='%s'",
                        i, j, webTableCellText, csvCellText));
                Assert.assertEquals(webTableCellText, csvCellText,
                        String.format("数据不匹配在行 %d, 列 %d. 预期(CSV): '%s', 实际(Web): '%s'",
                                i, j, csvCellText, webTableCellText));
            }
        }
        System.out.println("所有Web表格数据与CSV数据对比成功！");
    }

    @AfterClass
    public void tearDown() {
        if (driver != null) {
            driver.quit();
        }
    }
}