Spring boot 和 Flyway：在集成测试之前清除数据库数据

Question

我正在使用 Spring 引导框架 (v2.0.2) 构建 REST 服务，我在数据库上运行集成测试。 我在谷歌上搜索了很多关于在测试前清理数据库的文章，但不幸的是，我发现它们要么效率低下要么是黑客攻击，而不是 Spring Boot。 你能不能容忍我并提出一个解决这个问题的好方法？

理想情况下，我认为不应该在每次测试之前清除数据库，而是在某些组之前清除数据库，例如套件或每个测试类。 找到的建议之一如下所示：

@Autowired
protected Flyway flyway;

@Before
public void init() {
    flyway.clean();
    flyway.migrate();
}

在每次测试之前重建数据库，显然效率不高。 将此更改为静态上下文并使用@BeforeClass不起作用，因为 Spring 不注入静态字段。

是否有一些很好的方法可以从静态上下文中访问这个 flyway bean，以使该解决方案起作用？

这里的子问题： Flyway 有一个命令 clean，它不仅清除数据，而且删除所有内容，然后 migrate 命令再次执行迁移。 这似乎也是开销。 由于无论如何都会在启动时检查迁移，因此我认为没有必要在每个测试组之前拆除和重建所有内容。 只需清除数据就足够了。 您能否就如何实现这一目标提出一些建议？

总而言之，我正在寻找一种在每组集成测试（例如，每个类）之前删除数据库数据（如果可能，不删除表）的标准方法。 我想每个人在使用 Spring boot 时都会面临这个任务，所以也许框架本身考虑了一些不错的解决方案。

谢谢！

Answer 1

您可以为您的测试创建配置文件。 它将在所有测试之前运行一次。

@Configuration
public class TestConfig {
@Bean
public FlywayMigrationStrategy clean() {
    return flyway -> {
        flyway.clean();
        flyway.migrate();
    };
}
}

Answer 2

这个答案很有用，但它并没有让我一路走好，所以我想我会回来添加一个答案，以防其他人想要解决同样的问题。 上面的 bean 定义非常棒。

弹簧型材有 5 种左右的可能性。 我查看了文档以及人们如何使用它们，但走了另一条路。 Maven 有 6 个范围，但在这种情况下有用的是运行时和测试。

当我深入研究弹簧轮廓以及人们可以在它们之间切换的各种方式时，我的情况似乎有点过于复杂。 我只是希望我的被测数据库被创建、结构化并填充一些数据，以便我可以在我的 jpa spring boot 应用程序中测试存储库。 我不想花 4 个小时来设置配置文件。 并不是说从长远来看这不是一项值得的努力，只是我想让事情有所进展。

当我执行 spring-boot:run 时，我希望迁移非测试数据库，但我不希望那里有任何用于测试的 crud 数据。

因此，在实时应用程序中，我想要一个几乎为空的数据库，并且在测试期间，我希望 flyway 清理数据库，运行版本化迁移并使用测试数据填充它。

上面的答案让我找到了一个解决方案，当我的项目接近生产时，我可能会将其折叠到弹簧配置文件中。

事实证明，spring-boot-test 提供了一个@TestConfiguration注释，您可以将其附加到 src/test/ 层次结构中的任何类。 我创建了一个 FlywayConfiguration 类，其中包含上面提供的 bean 定义：

package com.foo.fooservice;

import org.flywaydb.core.Flyway;
import org.springframework.boot.autoconfigure.flyway.FlywayMigrationStrategy;
import org.springframework.boot.test.context.TestConfiguration;
import org.springframework.context.annotation.Bean;

@TestConfiguration
public class FlywayMigrationConfig {

@Bean
public static FlywayMigrationStrategy cleanMigrateStrategy(){

        return flyway -> {
                flyway.clean();
                flyway.migrate();
                };
        }
}

所以现在，如果我想在测试中使用它，我会在适当的测试类中添加另一个漂亮的注释 - @Includes，@TestConfiguration 注释的伴随 - 这样我就可以像以前一样使用这个配置@BeforeClass 像这样：

@DataJpaTest
@Import(FlywayMigrationConfig.class) 
@AutoConfigureTestDatabase(replace = AutoConfigureTestDatabase.Replace.NONE)
class AccountUserRepoTest {



    @Autowired 
    private AccountUserRepo accountUserRepo;

    @Autowired
    private FlywayMigrationStrategy strategy;

这允许我在每个测试类的基础上注入这种飞行迁移策略。 Spring 不会将您的 bean 自动注入到每个测试类中，您现在只需将 @Includes 注释添加到适当的测试类中即可使用此配置。 您不必在要使用它的每个测试类中定义 bean。 只需使用@Includes(your@TestCongiguration-annoted-class)。

我碰巧使用的是 postgresSQL 而不是 H2，因为我认为如果我正在对存储库实体进行集成测试，我可能会针对我将在生产中使用的内容进行测试。

另外：src/main/resources 将 jdbc 和 flyway 属性设置为开发模式名称和 jdbc url。

src/test/resources/application.properties 将模式名称设置为“test”（您可以随意命名）。

您可能不想要这种方法的一个缺点是粒度 - 为您以这种方式配置的每个测试类清理和重新填充数据库。

我个人喜欢这个，因为对于我正在测试的每个存储库类，我都希望刷新数据。 我也喜欢这样，如果我正在处理特定的测试类，那么在这种粒度级别上进行配置意味着“运行测试”开箱即用。 无需在 IDE 中进行特殊配置即可使其工作。