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

<dependencies>
    <!-- Selenium Java Client -->
    <dependency>
        <groupId>org.seleniumhq.selenium</groupId>
        <artifactId>selenium-java</artifactId>
        <version>4.1.2</version> <!-- 请使用最新稳定版本 -->
    </dependency>
    <!-- JUnit 5 -->
    <dependency>
        <groupId>org.junit.jupiter</groupId>
        <artifactId>junit-jupiter-api</artifactId>
        <version>5.8.2</version> <!-- 请使用最新稳定版本 -->
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.junit.jupiter</groupId>
        <artifactId>junit-jupiter-engine</artifactId>
        <version>5.8.2</version> <!-- 请使用最新稳定版本 -->
        <scope>test</scope>
    </dependency>
</dependencies>

package com.example.selenium;

import org.junit.jupiter.api.AfterAll;
import org.junit.jupiter.api.BeforeAll;
import org.junit.jupiter.api.Test;
import org.openqa.selenium.By;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
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;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertTrue;

public class TableSortVerificationTest {

    private static WebDriver driver;
    private static WebDriverWait wait;
    private static final String BASE_URL = "https://sakshingp.github.io/assignment/";

    @BeforeAll
    public static void setup() {
        // 设置 ChromeDriver 路径
        // System.setProperty("webdriver.chrome.driver", "path/to/your/chromedriver.exe");
        // 如果 ChromeDriver 已经在 PATH 中，则无需设置

        ChromeOptions options = new ChromeOptions();
        // options.addArguments("--headless"); // 无头模式运行
        options.addArguments("--start-maximized"); // 最大化窗口

        driver = new ChromeDriver(options);
        wait = new WebDriverWait(driver, Duration.ofSeconds(10)); // 显式等待，最长10秒
        driver.get(BASE_URL + "login.html");

        // 执行登录操作
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("username"))).sendKeys("Lakshay");
        driver.findElement(By.id("password")).sendKeys("Wingify");
        driver.findElement(By.id("log-in")).click();

        // 等待登录成功并跳转到 home 页面
        wait.until(ExpectedConditions.urlContains("home.html"));
        System.out.println("登录成功，当前URL: " + driver.getCurrentUrl());
    }

    @AfterAll
    public static void teardown() {
        if (driver != null) {
            driver.quit();
        }
    }

    // ... 后续测试方法将在此处添加
}

// 承接上文的 TableSortVerificationTest 类
public class TableSortVerificationTest {
    // ... setup 和 teardown 方法 ...

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

        // 1. 定位并点击“Amount”列头以触发排序
        WebElement amountHeader = wait.until(ExpectedConditions.elementToBeClickable(By.id("amount")));
        amountHeader.click();
        System.out.println("已点击 'Amount' 列头进行排序。");

        // 2. 提取排序后的金额数据
        // 这里的XPath定位到表格中所有金额的<span>元素
        List<WebElement> amountElements = driver.findElements(By.xpath("//table[@id='transactionsTable']/tbody/tr/td/span[@class='text-danger' or @class='text-success']"));

        // 用于存储从网页中提取并清洗后的数值
        List<Double> actualSortedAmounts = new ArrayList<>();

        for (WebElement element : amountElements) {
            String text = element.getText();
            // 3. 数据清洗与类型转换
            // 移除 "USD", "," 和空格，然后转换为 Double
            String cleanedText = text.replaceAll("USD", "")
                                     .replaceAll(",", "")
                                     .trim();
            try {
                // 某些值可能以 "-" 开头表示负数，例如 "- 244.00 USD"
                // 需要进一步处理，确保 Double.parseDouble 可以正确解析
                double amount = Double.parseDouble(cleanedText);
                actualSortedAmounts.add(amount);
            } catch (NumberFormatException e) {
                System.err.println("无法解析金额: " + cleanedText + "，原始文本: " + text);
                // 根据实际情况选择抛出异常或跳过
            }
        }
        System.out.println("从网页提取并清洗后的金额列表 (实际顺序): " + actualSortedAmounts);

        // 4. 程序内部排序
        // 创建一个副本进行内部排序，不改变原始提取顺序
        List<Double> expectedSortedAmounts = new ArrayList<>(actualSortedAmounts);
        Collections.sort(expectedSortedAmounts); // 默认升序排序
        System.out.println("程序内部排序后的金额列表 (预期升序): " + expectedSortedAmounts);

        // 5. 验证排序结果
        // 比较实际排序结果与预期排序结果
        assertEquals(expectedSortedAmounts, actualSortedAmounts, "金额列未能正确按升序排序！");
        System.out.println("金额列已成功按升序排序验证通过！");

        // 如果需要验证降序，可以再次点击 'Amount' 列头，然后对 expectedSortedAmounts 进行 Collections.reverse()
        // amountHeader.click(); // 再次点击切换到降序
        // Collections.reverse(expectedSortedAmounts);
        // actualSortedAmounts.clear(); // 重新提取降序数据
        // ... 重新提取和验证降序逻辑 ...
    }
}