Selenium动态WebTable如何随机选中2个复选框后退出循环

现有代码的核心问题有三个：

• 没有设置选中计数逻辑，无法判断什么时候选够2个需要终止
• 随机取索引没有去重逻辑，可能会多次点击同一个复选框导致最终选中数量不足2个
• 固定滚动100像素的逻辑不稳定，可能出现复选框没滚动到可视区域导致点击失效的问题

推荐实现方案（无重复选中，性能最优） #

import java.util.HashSet;
import java.util.Random;
import java.util.Set;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.By;

// 先获取表格所有行，绑定行和对应复选框，避免索引错位
List<WebElement> tableRows = driver.findElements(By.xpath("//tbody[@role='rowgroup']//tr"));
Random random = new Random();
Set<Integer> selectedRowIndexes = new HashSet<>();
JavascriptExecutor jse = (JavascriptExecutor)driver;

// 生成2个不重复的行索引
while (selectedRowIndexes.size() < 2) {
    selectedRowIndexes.add(random.nextInt(tableRows.size()));
}

// 遍历选中对应行的复选框，选完直接结束
for (Integer rowIndex : selectedRowIndexes) {
    WebElement targetCheckbox = tableRows.get(rowIndex).findElement(By.xpath(".//label[contains(@class,'mat-checkbox-layout')]"));
    // 滚动到当前复选框位置，保证点击有效
    jse.executeScript("arguments[0].scrollIntoView(true);", targetCheckbox);
    targetCheckbox.click();
}

兼容原有写法的修改方案 #

如果要保留原有单独获取复选框列表的逻辑，可按如下修改：

import java.util.HashSet;
import java.util.Random;
import java.util.Set;
import org.openqa.selenium.JavascriptExecutor;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.By;

List<WebElement> checkboxOptions = driver.findElements(By.xpath("//label[contains(@class,'mat-checkbox-layout')]"));
Random random = new Random();
int selectedCount = 0;
Set<Integer> usedIndexes = new HashSet<>();
JavascriptExecutor jse = (JavascriptExecutor)driver;

while (selectedCount < 2) {
    int index = random.nextInt(checkboxOptions.size());
    // 跳过已经选中过的索引，避免重复点击同一个复选框
    if (usedIndexes.contains(index)) {
        continue;
    }
    jse.executeScript("arguments[0].scrollIntoView(true);", checkboxOptions.get(index));
    checkboxOptions.get(index).click();
    usedIndexes.add(index);
    selectedCount++;
}

注意事项 #

1. 原有代码中chekcboxoptions拼写错误，注意修正为正确的变量名，避免找不到元素报错
2. 推荐使用行和复选框绑定的写法，即使表格结构变动、复选框和行的对应顺序变化，也不会出现选错的问题

内容的提问来源于stack exchange，提问作者Sandy