如何验证Update/Insert语句实际影响行数与预期值一致？

问题背景 #

现有自动化测试架构中，执行Update/Insert语句时仅在受影响行数为0时判定测试失败，需要扩展为实际受影响行数与预期值不符时（如实际1行、预期4行）同样判定失败，覆盖数据库仅匹配部分目标行的场景。

原代码片段 #

//conn and query already defined and given
Statement stmt = conn.createStatement();
if (!stmt.execute(query)) {
    int updatedRows = stmt.getUpdateCount();
    if(updatedRows == 0) {
        System.out.println("Update or insert query didn't affect any rows!");
    }
}

修改实现 #

核心思路是引入预期受影响行数，将实际行数与预期值做全量对比，而非仅判断是否为0。以下是优化后的代码：

// conn、query 已提前定义，expectedAffectedRows 为测试用例中的预期受影响行数
Statement stmt = conn.createStatement();
if (!stmt.execute(query)) {
    int actualAffectedRows = stmt.getUpdateCount();
    // 示例值，实际需从测试用例配置/数据中动态读取
    int expectedAffectedRows = 4; 

    // 对比实际与预期行数，不符则判定测试失败
    if (actualAffectedRows != expectedAffectedRows) {
        // 自动化测试中抛出断言异常，直接标记测试失败
        throw new AssertionError(String.format(
            "受影响行数不符：预期%d行，实际%d行",
            expectedAffectedRows,
            actualAffectedRows
        ));
        // 若只需日志输出，可替换为：
        // System.out.println(String.format("受影响行数不符：预期%d行，实际%d行", expectedAffectedRows, actualAffectedRows));
    }
}

关键说明 #

• expectedAffectedRows：需从测试用例的预期配置中动态获取（比如测试数据文件、注解参数等），避免硬编码。
• 断言处理：自动化测试场景下，抛出AssertionError（或对应测试框架的断言方法，如JUnit的Assert.assertEquals）会直接标记测试用例失败，符合自动化测试的失败判定逻辑。
• 场景覆盖：既保留了原有的受影响行数为0的失败判定，也覆盖了实际行数与预期不匹配的所有场景（如实际1行、预期4行）。

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