包含 SQL 查询的方法的 JUnit 测试

Question

我有一个旧的 Java 项目（没有使用框架/构建工具），它有一个充满 SQL 方法和相应 Bean 类的类。 SQL 方法主要使用 SELECT、INSERT 和 UPDATE 查询，如下所示：

public static void sqlUpdateAge(Connection dbConnection, int age, int id) {

    PreparedStatement s = null;
    ResultSet r = null;

    String sql = "UPDATE person SET age = ? WHERE id = ?";

    try {
        s = dbConnection.prepareStatement(sql);
        s.setInt(1, age);
        s.setInt(2, id);
        s.addBatch();
        s.executeBatch();

    } catch (Exception e) {
        e.printStackTrace();
    } finally {
        try {
            if (r != null)
                r.close();
            if (s != null)
                s.close();
        } catch (SQLException e) {
            e.printStackTrace();
        }
    }
}

当涉及到 SQL 查询时，单元测试的最佳实践是什么？

我能想到的最简单的方法是使用我的开发数据库； 只需在测试类中调用 sqlUpdateAge()，查询数据库中的结果集并 assertTrue 设置的年龄在结果集中。 但是，这会用不必要的数据填满开发数据库，​​我想避免这种情况。

是创建所谓的内存数据库的解决方案还是以某种方式回滚我所做的更改？

如果我需要一个内存数据库：

1. 我将在哪里以及如何创建它？ 直接到测试类，或者可能到配置文件？
2. 我如何将它传递给 updateAge() 方法？

Answer 1

有了 Spring Boot 项目，事情会很容易。 在你的情况下，你有很多策略：

1. 配置H2数据库。 您可以通过使用@BeforeEach注释在setUp方法中创建架构和插入数据来初始化您的数据库。
2. 您可以使用像DbUnit这样的专用框架。
3. 您还必须在单元测试的setUp方法中初始化dbConnection 。

Answer 2

我建议看看你是否可以从自动构建开始。 那是通过引入构建工具（例如 maven 或 gradle） - 或者如果不可能 - scipting 构建。 无论如何，您的目标应该是在代码更改时轻松触发构建和测试。

如果您不能在保证所有单元测试都已运行的情况下对每个更改生成一致的构建，那么首先编写单元测试就真的没有价值。 那是因为否则，您的测试最终将因代码修改而失败，除非您的所有测试都自动运行，否则您不会注意到。

一旦你有了它，你可能会有一些关于如何运行单元或集成测试的提示。

由于您无法从许多应用程序框架提供的测试支持中受益，因此您基本上可以自行决定如何配置数据库测试设置。 在那种情况下，我不认为内存数据库真的是最好的选择，因为：

• 这是一种与您通常使用的数据库技术不同的数据库技术，并且正如代码所示，您没有使用 ORM 来处理不同的 SQL 方言。 在这种情况下，您可能会发现自己处于无法准确测试代码的位置，因为 SQL 方言差异。
• 您需要自己完成内存数据库的所有设置 - 这当然是可能的，但它仍然是您需要维护的一段代码，并且也可能失败。

我能想到的两种选择是：

• 每次运行测试时，使用 Docker 启动您的实际数据库技术。 （这也是您必须为自己编写脚本的内容，但它很可能是您需要执行的一个非常简单且简短的命令）
• 在您使用的测试环境上运行一个测试数据库。 每次运行测试之前，请确保将数据库重置为原始状态。 （最简单的方法是删除现有架构并恢复到原始架构）。 在这种情况下，您需要确保不会针对同一个测试数据库并行运行多个构建。

这些建议仅适用于您有 shell 经验和/或得到操作人员支持的情况。 如果没有，设置 H2 可能更容易、更直接。