Selenium Java：验证网页表格数据排序功能

<dependencies>
    <!-- Selenium Java Client -->
    <dependency>
        <groupId>org.seleniumhq.selenium</groupId>
        <artifactId>selenium-java</artifactId>
        <version>4.1.2</version> <!-- 请使用最新稳定版本 -->
    </dependency>
    <!-- JUnit 4 -->
    <dependency>
        <groupId>junit</groupId>
        <artifactId>junit</artifactId>
        <version>4.13.2</version>
        <scope>test</scope>
    </dependency>
    <!-- WebDriverManager (可选，用于自动管理浏览器驱动) -->
    <dependency>
        <groupId>io.github.bonigarcia</groupId>
        <artifactId>webdrivermanager</artifactId>
        <version>5.0.3</version> <!-- 请使用最新稳定版本 -->
        <scope>test</scope>
    </dependency>
</dependencies>

package com.example.seleniumtests;

import io.github.bonigarcia.wdm.WebDriverManager;
import org.junit.AfterClass;
import org.junit.Assert;
import org.junit.BeforeClass;
import org.junit.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import java.time.Duration;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.stream.Collectors;

public class TableSortingVerification {

    private static WebDriver driver;
    private static WebDriverWait wait;

    // 辅助方法：提取并清洗金额数据
    private List<Double> extractAndCleanAmountData() {
        // 定位所有金额列的span元素
        List<WebElement> amountElements = driver.findElements(By.xpath("//table[@id='transactionsTable']/tbody/tr/td/span"));
        List<Double> cleanedAmounts = new ArrayList<>();

        for (WebElement element : amountElements) {
            String text = element.getText();
            // 清除货币符号 "USD", 逗号 ",", 空格 " "
            String cleanedText = text.replaceAll("USD", "").replaceAll(",", "").trim();
            try {
                // 转换为 Double 类型
                cleanedAmounts.add(Double.parseDouble(cleanedText));
            } catch (NumberFormatException e) {
                System.err.println("无法解析金额: " + text + " - " + e.getMessage());
                // 根据需求处理错误，例如跳过或抛出异常
            }
        }
        return cleanedAmounts;
    }

    @BeforeClass
    public static void setup() {
        // 自动管理 ChromeDriver (推荐使用 WebDriverManager)
        WebDriverManager.chromedriver().setup();
        driver = new ChromeDriver();
        driver.manage().window().maximize();
        wait = new WebDriverWait(driver, Duration.ofSeconds(10)); // 显式等待

        // 导航到登录页面并登录
        String loginUrl = "https://sakshingp.github.io/assignment/login.html";
        driver.get(loginUrl);

        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("username"))).sendKeys("Lakshay");
        driver.findElement(By.id("password")).sendKeys("Wingify");
        driver.findElement(By.id("log-in")).click();

        // 等待登录成功并跳转到主页
        wait.until(ExpectedConditions.urlContains("home.html"));
        System.out.println("成功登录并进入主页.");
    }

    @Test
    public void testAmountColumnSorting() {
        System.out.println("开始验证金额列排序...");

        // 1. 提取原始（未排序）金额数据
        List<Double> initialAmounts = extractAndCleanAmountData();
        System.out.println("原始金额数据: " + initialAmounts);

        // 2. 准备预期升序数据
        List<Double> expectedAscendingAmounts = new ArrayList<>(initialAmounts);
        Collections.sort(expectedAscendingAmounts); // 默认升序
        System.out.println("预期升序数据: " + expectedAscendingAmounts);

        // 3. 触发 UI 升序排序操作
        WebElement amountHeader = driver.findElement(By.id("amount"));
        amountHeader.click(); // 第一次点击通常是升序

        // 4. 提取实际升序排序后的数据
        // 等待表格内容更新，这里可以根据实际页面加载情况调整等待策略
        // 简单等待一段时间，或者等待某个元素状态变化
        wait.until(ExpectedConditions.stalenessOf(amountHeader)); // 假设点击后页面有刷新或部分更新
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.xpath("//table[@id='transactionsTable']/tbody/tr/td/span"))); // 确保元素可见

        List<Double> actualAscendingAmounts = extractAndCleanAmountData();
        System.out.println("实际升序数据: " + actualAscendingAmounts);

        // 5. 验证升序排序
        Assert.assertEquals("金额列升序排序验证失败！", expectedAscendingAmounts, actualAscendingAmounts);
        System.out.println("金额列升序排序验证成功！");

        // --- 验证降序排序（可选） ---
        System.out.println("\n开始验证金额列降序排序...");

        // 1. 准备预期降序数据
        List<Double> expectedDescendingAmounts = new ArrayList<>(expectedAscendingAmounts);
        Collections.reverse(expectedDescendingAmounts); // 反转升序列表得到降序
        System.out.println("预期降序数据: " + expectedDescendingAmounts);

        // 2. 触发 UI 降序排序操作 (再次点击通常是降序)
        amountHeader.click();

        // 3. 提取实际降序排序后的数据
        wait.until(ExpectedConditions.stalenessOf(amountHeader)); // 假设点击后页面有刷新或部分更新
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.xpath("//table[@id='transactionsTable']/tbody/tr/td/span"))); // 确保元素可见

        List<Double> actualDescendingAmounts = extractAndCleanAmountData();
        System.out.println("实际降序数据: " + actualDescendingAmounts);

        // 4. 验证降序排序
        Assert.assertEquals("金额列降序排序验证失败！", expectedDescendingAmounts, actualDescendingAmounts);
        System.out.println("金额列降序排序验证成功！");
    }

    @AfterClass
    public static void tearDown() {
        if (driver != null) {
            driver.quit();
            System.out.println("浏览器已关闭.");
        }
    }
}