使用Java和Selenium比较CSV数据与网页表格：健壮解析与高效验证策略

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

public class CsvDataReader {

    private static final String COMMA_DELIMITER = ",";

    /**
     * 从一行CSV文本中解析出字段列表
     * @param line CSV文件中的一行文本
     * @return 包含该行所有字段的列表
     */
    private static List<String> getRecordFromLine(String line) {
        List<String> values = new ArrayList<>();
        // 使用Scanner解析一行，并指定逗号为分隔符
        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文件，并将数据存储为二维列表
     * @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))) {
            if (scanner.hasNextLine()) {
                scanner.nextLine(); // 跳过CSV文件的标题行（如果存在）
            }
            while (scanner.hasNextLine()) {
                records.add(getRecordFromLine(scanner.nextLine()));
            }
        }
        return records;
    }

    public static void main(String[] args) {
        String csvFilePath = "src/test/resources/test.csv"; // 示例CSV文件路径
        try {
            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: " + csvFilePath);
            e.printStackTrace();
        }
    }
}

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; // 假设使用TestNG进行断言

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

public class WebTableCsvComparator {

    public static void main(String[] args) {
        // 1. 初始化WebDriver
        System.setProperty("webdriver.chrome.driver", "path/to/chromedriver"); // 替换为你的chromedriver路径
        WebDriver driver = new ChromeDriver();
        driver.manage().timeouts().implicitlyWait(10, TimeUnit.SECONDS);

        try {
            // 2. 导航到包含表格的网页
            driver.get("http://your-website.com/table-page"); // 替换为你的网页URL

            // 3. 读取CSV数据
            String csvFilePath = "src/test/resources/test.csv"; // 替换为你的CSV文件路径
            List<List<String>> csvData = CsvDataReader.readCsvFile(csvFilePath);
            System.out.println("CSV Data for comparison: " + csvData);

            // 4. 获取网页表格元素
            WebElement webTable = driver.findElement(By.cssSelector("#dStocks1")); // 替换为你的表格CSS选择器
            List<WebElement> webTableRows = webTable.findElements(By.tagName("tr"));

            // 假设CSV数据和网页表格都有标题行，且我们已经跳过了CSV的标题行
            // 如果网页表格也有标题行，并且我们不想比对它，可以从索引1开始
            int startRowIndexForWebTable = 1; // 假设第一行是标题
            int startRowIndexForCsvData = 0; // CsvDataReader已经跳过了标题行，所以从0开始

            // 确保CSV数据和网页表格的行数匹配
            if (csvData.size() != (webTableRows.size() - startRowIndexForWebTable)) {
                System.err.println("Warning: CSV data row count (" + csvData.size() +
                                   ") does not match web table data row count (" +
                                   (webTableRows.size() - startRowIndexForWebTable) + ")");
                // 可以选择抛出异常或继续比对尽可能多的数据
            }

            // 5. 逐行逐列比对数据
            System.out.println("\n--- Starting Data Comparison ---");
            for (int i = 0; i < csvData.size(); i++) {
                List<String> csvRow = csvData.get(i);
                int webTableRowIndex = i + startRowIndexForWebTable;

                if (webTableRowIndex >= webTableRows.size()) {
                    System.err.println("No corresponding web table row for CSV row " + (i + 1));
                    break; // CSV行多于网页表格行
                }

                List<WebElement> webTableCells = webTableRows.get(webTableRowIndex).findElements(By.tagName("td"));

                // 确保列数匹配
                if (csvRow.size() != webTableCells.size()) {
                    System.err.println("Warning: Column count mismatch at row " + (i + 1) +
                                       ". CSV columns: " + csvRow.size() +
                                       ", Web table columns: " + webTableCells.size());
                    // 可以选择跳过此行或继续比对尽可能多的列
                }

                for (int j = 0; j < csvRow.size(); j++) {
                    if (j >= webTableCells.size()) {
                        System.err.println("No corresponding web table cell for CSV cell at row " + (i + 1) + ", column " + (j + 1));
                        break; // CSV列多于网页表格列
                    }

                    String csvCellValue = csvRow.get(j);
                    String webTableCellValue = webTableCells.get(j).getText().trim(); // 获取文本并去除空格

                    System.out.println("Comparing Row " + (i + 1) + ", Column " + (j + 1) + ":");
                    System.out.println("  CSV: '" + csvCellValue + "'");
                    System.out.println("  Web: '" + webTableCellValue + "'");

                    try {
                        Assert.assertEquals(webTableCellValue, csvCellValue,
                                "Mismatch at Row " + (i + 1) + ", Column " + (j + 1));
                        System.out.println("  -> PASSED");
                    } catch (AssertionError e) {
                        System.err.println("  -> FAILED: " + e.getMessage());
                        // 可以在这里记录失败，或者抛出异常中断测试
                    }
                }
            }
            System.out.println("\n--- Data Comparison Finished ---");

        } catch (FileNotFoundException e) {
            System.err.println("Error: CSV file not found. " + e.getMessage());
        } finally {
            // 6. 关闭WebDriver
            driver.quit();
        }
    }
}