Java Selenium：验证网页表格数值列的排序功能

<dependencies>
    <dependency>
        <groupId>org.seleniumhq.selenium</groupId>
        <artifactId>selenium-java</artifactId>
        <version>4.1.2</version> <!-- 使用您需要的版本 -->
    </dependency>
    <dependency>
        <groupId>org.junit.jupiter</groupId>
        <artifactId>junit-jupiter-api</artifactId>
        <version>5.8.1</version> <!-- 使用您需要的版本 -->
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.junit.jupiter</groupId>
        <artifactId>junit-jupiter-engine</artifactId>
        <version>5.8.1</version> <!-- 使用您需要的版本 -->
        <scope>test</scope>
    </dependency>
</dependencies>

import org.junit.jupiter.api.*;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import java.time.Duration;

public class TableSortVerificationTest {

    private static WebDriver driver;
    private static WebDriverWait wait;

    @BeforeAll
    public static void setup() {
        // 设置ChromeDriver路径，请根据您的实际路径修改
        System.setProperty("webdriver.chrome.driver", "path/to/your/chromedriver.exe");
        driver = new ChromeDriver();
        driver.manage().window().maximize(); // 最大化浏览器窗口
        wait = new WebDriverWait(driver, Duration.ofSeconds(10)); // 初始化显式等待
    }

    @BeforeEach
    public void login() {
        driver.get("https://sakshingp.github.io/assignment/login.html");

        // 等待用户名输入框可见并输入用户名
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("username"))).sendKeys("Lakshay");
        // 等待密码输入框可见并输入密码
        wait.until(ExpectedConditions.visibilityOfElementLocated(By.id("password"))).sendKeys("Wingify");

        // 点击登录按钮
        wait.until(ExpectedConditions.elementToBeClickable(By.id("log-in"))).click();

        // 登录后等待页面跳转到home.html
        wait.until(ExpectedConditions.urlContains("home.html"));
    }

    // ... 后续测试方法 ...

    @AfterAll
    public static void tearDown() {
        if (driver != null) {
            driver.quit(); // 关闭浏览器
        }
    }
}

// ... 在 TableSortVerificationTest 类中添加 ...

import org.openqa.selenium.WebElement;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class TableSortVerificationTest {
    // ... setup, login, tearDown methods ...

    @Test
    @DisplayName("验证Amount列的升序排序")
    public void verifyAmountColumnAscendingSort() {
        // 1. 点击“Amount”列头，触发升序排序
        WebElement amountHeader = wait.until(ExpectedConditions.elementToBeClickable(By.id("amount")));
        amountHeader.click();

        // 添加一个短暂的等待，确保页面数据已经重新加载和排序
        // 更好的做法是等待特定元素状态变化，例如等待表格行的文本发生变化或加载指示器消失
        // 这里我们简单等待表格第一行第一个单元格的文本变得“陈旧”，表示页面内容已刷新
        wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.xpath("//*[@id=\"transactionsTable\"]/tbody/tr[1]/td[1]"))));

        // 2. 提取并清洗表格数据
        List<Double> actualAmounts = extractAndCleanAmountData();

        // 3. 创建一个期望的排序列表
        List<Double> expectedSortedAmounts = new ArrayList<>(actualAmounts);
        Collections.sort(expectedSortedAmounts); // 默认升序排序

        // 4. 验证实际数据是否与期望的升序数据一致
        Assertions.assertEquals(expectedSortedAmounts, actualAmounts, "Amount列未按升序排列！");
        System.out.println("Amount列已成功验证为升序排列。");
    }

    @Test
    @DisplayName("验证Amount列的降序排序")
    public void verifyAmountColumnDescendingSort() {
        // 首先点击一次，确保进入升序状态
        WebElement amountHeader = wait.until(ExpectedConditions.elementToBeClickable(By.id("amount")));
        amountHeader.click();
        wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.xpath("//*[@id=\"transactionsTable\"]/tbody/tr[1]/td[1]"))));

        // 再次点击“Amount”列头，触发降序排序
        amountHeader.click();
        wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.xpath("//*[@id=\"transactionsTable\"]/tbody/tr[1]/td[1]"))));

        // 2. 提取并清洗表格数据
        List<Double> actualAmounts = extractAndCleanAmountData();

        // 3. 创建一个期望的排序列表
        List<Double> expectedSortedAmounts = new ArrayList<>(actualAmounts);
        Collections.sort(expectedSortedAmounts, Collections.reverseOrder()); // 降序排序

        // 4. 验证实际数据是否与期望的降序数据一致
        Assertions.assertEquals(expectedSortedAmounts, actualAmounts, "Amount列未按降序排列！");
        System.out.println("Amount列已成功验证为降序排列。");
    }

    /**
     * 辅助方法：提取并清洗“Amount”列的数据
     * @return 包含Double类型金额的列表
     */
    private List<Double> extractAndCleanAmountData() {
        // 定位所有“Amount”列的数据单元格（假设“Amount”是第5列，且数据在span标签内）
        List<WebElement> amountElements = driver.findElements(By.xpath("//*[@id=\"transactionsTable\"]/tbody/tr/td[5]/span"));
        List<Double> amounts = new ArrayList<>();

        for (WebElement element : amountElements) {
            String text = element.getText();
            // 清洗字符串：移除“USD”、逗号、空格，并解析为Double
            // 示例数据可能包含负号，如 "- 250.00 USD"
            String cleanedText = text.replaceAll("USD", "")
                                     .replaceAll(",", "")
                                     .trim(); // 移除首尾空格
            amounts.add(Double.parseDouble(cleanedText));
        }
        return amounts;
    }
}