Selenium+Java：如何根据Web表格同行单元格文本勾选对应复选框

问题背景 #

我是Selenium新手，正从手动测试转向自动化测试。尝试遍历Web表格，根据同行其他单元格的值勾选对应复选框。

表格结构 #

<div class="s-table-container"><table class="s-table">
<thead>
<tr>
<th></th>
<th>Name</th>
<th>Relation</th>
<th>D.O.B</th>
</tr>
</thead>
<tbody>
<tr>
<td>CB1</td>
<td>John</td>
<td>Staff</td>
<td>01-Jan-1995</td>
</tr>
<tr>
<td>CB2</td>
<td>Mary</td>
<td>Spouse</td>
<td>01-Jan-2005</td>
</tr>
<tr>
<td>CB3</td>
<td>Philip</td>
<td>Child</td>
<td>01-Jan-2000</td>
</tr>
<tr>
<td>CB4</td>
<td>Kaleb</td>
<td>Child</td>
<td>01-Jan-2024</td>
</tr>
</tbody>
</table></div>

注：CB代表复选框（Checkbox）

参数与需求 #

从Excel传入String类型的nos和relation参数：

• relation为逗号分隔的字符串（如Spouse,Child），通过String[] paxsel = relation.split(",");拆分
• 需求规则：
  1. 若传入的relation为Employee，匹配表格Relation列文本为Staff时勾选CB1；
  2. 若为Spouse，匹配Relation列文本则勾选对应行复选框；
  3. 若为Child，匹配Relation列且年龄超过731天时勾选；
  4. 若为Infant，匹配表格中Relation为Child且年龄≤731天时勾选；
• 勾选数量达到nos指定数值时停止循环。

问题代码 #

运行时要么抛出索引越界异常，要么无法勾选复选框，代码如下：

public void selectMembers(String nos, String relation) {

        String[] paxsel = relation.split(",");

        try {
            List<WebElement> rows = tblPassengerList.findElements(By.tagName("tr"));

            int relationindex = 2;
            int dobindex = 3;
            WebElement paxRelation;
            WebElement dob;


            for (int i = 0; i < rows.size(); i++) {
                List<WebElement> cells = rows.get(i).findElements(By.tagName("td"));
                for(int j=0;j<Integer.valueOf(nos);)
                if (!cells.isEmpty()) {
                    dob = cells.get(dobindex);
                    paxRelation = cells.get(relationindex);
                    String fdob = dob.getText();

                    DateTimeFormatter formatter = DateTimeFormatter.ofPattern("d-MMM-yyyy");
                    LocalDate fdob1 = LocalDate.parse(fdob, formatter);
                    LocalDate currDate = LocalDate.now();

                    System.out.println(paxsel[j]);
                    if (paxsel[j].equalsIgnoreCase("Infant") && paxRelation.getText().equalsIgnoreCase("Child") && calculateAge(fdob1, currDate) <= 731) {
                        WebElement checkbox = cells.get(0).findElement(By.tagName("input"));
                        checkbox.click();
                        j=j+1;

                    } else if (paxsel[j].equalsIgnoreCase(paxRelation.getText()) && paxRelation.getText().equalsIgnoreCase("Child") && calculateAge(fdob1, currDate) > 731) {
                        WebElement checkbox = cells.get(0).findElement(By.tagName("input"));
                        checkbox.click();
                        j=j+1;


                    } else if (paxsel[j].equalsIgnoreCase(paxRelation.getText())) {
                        WebElement checkbox = cells.get(0).findElement(By.tagName("input"));
                        checkbox.click();
                        j=j+1;
                    }

                }
                break;
            }


    } catch(
    NumberFormatException e)

    {
        throw new RuntimeException(e);
    }

}

    public static long calculateAge(LocalDate fdob, LocalDate currentDate) {
        long days = ChronoUnit.DAYS.between(fdob, currentDate);
        return days;
    }

问题分析 #

1. 索引越界原因：

   • 内层for循环未处理j超过paxsel数组长度的情况，当nos值大于paxsel元素数量时，访问paxsel[j]会抛出数组索引越界异常。
   • 外层循环仅执行一次就break，导致只遍历表头或第一行，无法处理所有行数据。
   • 直接访问cells.get(relationindex)和cells.get(dobindex)未判断单元格数量是否足够，表头行（<thead>里的<tr>）只有<th>没有<td>，会触发索引越界。

2. 无法勾选的原因：

   • 内层循环逻辑错误，每次仅处理当前行，且j的递增依赖当前行匹配成功，若当前行不匹配，j会一直停留在当前值，陷入死循环。
   • 未处理Employee对应Staff的规则，原代码中无相关判断。
   • 日期格式化未指定区域，MMM是月份缩写，若系统区域与表格日期区域不一致，会导致解析失败。

修复后的代码 #

import org.openqa.selenium.NoSuchElementException;
import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.time.format.DateTimeParseException;
import java.time.temporal.ChronoUnit;
import java.util.List;
import java.util.Locale;

public void selectMembers(String nos, String relation) {
    String[] paxsel = relation.split(",");
    int targetCount = Integer.parseInt(nos);
    int selectedCount = 0;

    try {
        // 直接遍历tbody中的行，跳过表头
        List<WebElement> bodyRows = tblPassengerList.findElement(By.tagName("tbody")).findElements(By.tagName("tr"));
        int relationIndex = 2;
        int dobIndex = 3;

        // 指定英文区域解析日期，避免月份缩写解析失败
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("d-MMM-yyyy", Locale.ENGLISH);
        LocalDate currDate = LocalDate.now();

        // 遍历每个需要选择的关系类型
        for (String targetRelation : paxsel) {
            if (selectedCount >= targetCount) break;

            // 遍历表格行找匹配项
            for (WebElement row : bodyRows) {
                if (selectedCount >= targetCount) break;

                List<WebElement> cells = row.findElements(By.tagName("td"));
                // 确保单元格数量足够，避免索引越界
                if (cells.size() <= dobIndex) continue;

                String actualRelation = cells.get(relationIndex).getText().trim();
                String dobText = cells.get(dobIndex).getText().trim();
                LocalDate fdob;

                try {
                    fdob = LocalDate.parse(dobText, formatter);
                } catch (DateTimeParseException e) {
                    System.out.println("日期解析失败：" + dobText);
                    continue;
                }

                long ageDays = calculateAge(fdob, currDate);
                boolean shouldSelect = false;

                // 匹配规则判断
                switch (targetRelation.toLowerCase()) {
                    case "employee":
                        shouldSelect = actualRelation.equalsIgnoreCase("staff");
                        break;
                    case "spouse":
                        shouldSelect = actualRelation.equalsIgnoreCase("spouse");
                        break;
                    case "child":
                        shouldSelect = actualRelation.equalsIgnoreCase("child") && ageDays > 731;
                        break;
                    case "infant":
                        shouldSelect = actualRelation.equalsIgnoreCase("child") && ageDays <= 731;
                        break;
                    default:
                        System.out.println("未知的关系类型：" + targetRelation);
                        break;
                }

                // 勾选复选框，先判断是否已选中避免重复操作
                if (shouldSelect) {
                    WebElement checkbox = cells.get(0).findElement(By.tagName("input"));
                    if (!checkbox.isSelected()) {
                        checkbox.click();
                    }
                    selectedCount++;
                }
            }
        }
    } catch (NumberFormatException e) {
        throw new RuntimeException("nos参数不是有效的数字：" + nos, e);
    } catch (NoSuchElementException e) {
        throw new RuntimeException("表格元素未找到", e);
    }
}

public static long calculateAge(LocalDate fdob, LocalDate currentDate) {
    return ChronoUnit.DAYS.between(fdob, currentDate);
}

修复说明 #

1. 避免索引越界：

   • 直接遍历<tbody>内的行，跳过表头行，避免处理无<td>的表头。
   • 增加单元格数量判断，确保relationIndex和dobIndex在有效范围内。
   • 用selectedCount跟踪已勾选数量，达到targetCount时立即停止所有循环，同时避免j超出数组长度。

2. 修复勾选逻辑：

   • 外层遍历需要的关系类型，内层遍历表格行，逐个匹配，确保每个关系类型都能找到对应行。
   • 补充Employee对应Staff的规则判断。
   • 日期解析指定Locale.ENGLISH，避免区域差异导致的解析失败。
   • 勾选前判断复选框是否已选中，避免重复点击。

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