Junit Spring 避免加载两次应用程序上下文数据源

Question

我有这个配置类：

@ComponentScan(
        basePackages = { 
                "mypackage.controller",
                "mypackage.service",
                "mypackage.repository" 
        }
)
@TestPropertySource(locations="classpath:configuration.properties")
@Import({
    H2Configuration.class
})
public class TestConfiguration {
}

@Configuration
public class H2Configuration {

    @Bean
    public DataSource dataSource() throws SQLException {
        EmbeddedDatabaseBuilder builder = new EmbeddedDatabaseBuilder();
        EmbeddedDatabase db = builder
                .setType(EmbeddedDatabaseType.H2)
                .addScript("h2/create.sql")
                .addScript("h2/insert.sql")
                .build();
        db.getConnection().setAutoCommit(false);
        return db;
    }

}

我有这两个类测试：

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(loader=AnnotationConfigContextLoader.class, classes = { TestConfiguration.class })
public class FirstRepositoryTest {

    @Autowired
    MyFirstRepositoryImpl repository;

    @Before
    public void initTest() {
    }

    @Test(expected = NullPointerException.class)
    public void testNullRecords() {
        repository.foo(null, null);
    }
}


@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(loader=AnnotationConfigContextLoader.class, classes = { TestConfiguration.class })
public class SecondRepositoryTest {

    @Autowired
    MySecondRepositoryImpl repository;

    @Before
    public void initTest() {
    }

    @Test(expected = NullPointerException.class)
    public void testSomethingNullRecords() {
        repository.something(null, null);
    }
}

如果我为每个类运行一次 junit 测试，一切顺利。

在全新安装阶段测试失败，因为应用程序上下文被初始化了两次。

例如，它尝试创建两次 h2 表并执行两次 insert.sql 脚本。

我必须做些什么来初始化 h2 数据库和应用程序上下文一次？

谢谢

Answer 1

我认为您可以开始查看有关 Integration Testing 的 Spring 文档。

将事务测试用于集成测试 (@Transactional) 也是一种很好的做法，它会在每个测试结束时回滚：请参阅 Transaction Management。

为避免为每个测试类重新创建ApplicationContext 的成本，缓存可按此处的说明使用：Context Caching。

对于嵌入式数据库的集成测试，您还可以找到文档：Testing Data Access Logic with an Embedded Database。 上一个链接中的注释，与您的用例匹配：

但是，如果您希望创建一个共享的嵌入式数据库 在测试套件中，考虑使用 Spring TestContext Framework 并在 Spring 中将嵌入式数据库配置为 bean ApplicationContext 如创建嵌入式数据库中所述 使用 Spring XML 并以编程方式创建嵌入式数据库。

希望你能找到一些有用的参考资料。

Answer 2

我从Embedded Database Support 的 Spring Boot 文档中找到的另一个好技巧：

他们说：

如果您在测试中使用此功能，您可能会注意到 整个测试套件都重用相同的数据库 您使用的应用程序上下文的数量。如果你想确定 每个上下文都有一个单独的嵌入式数据库，您应该设置 spring.datasource.generate-unique-name 为 true。

因此，要使每个 EmbeddedDatabase 独一无二，您可以尝试使用以下方法创建它们：

EmbeddedDatabase db = new EmbeddedDatabaseBuilder()
                      .generateUniqueName(true)
                      ...
                      .build();

Answer 3

在单元测试中，您必须保证每个测试都是可重复的，并且与上下文无关。因此，仅加载一次上下文并不是一个好主意。最好在执行后重置。为此，您可以在测试类中使用 @DirtiesContext(classMode = ClassMode.AFTER_CLASS)

因此，当启动下一个 junit 类时，您将强制您的上下文重新启动

Answer 4

因此，失败的原因是当您在清理/安装过程中运行测试时，数据库 (H2) 驻留在内存中。创建/插入脚本在第一次测试运行后已经执行。在此之后的任何后续测试执行都将导致重新执行相同的脚本并且会发生错误。

使用DROP TABLE IF EXISTS <table name>; 更新您的创建脚本。这将确保表被删除然后重新创建。

注意：我不确定您为什么明确指定 AnnotationConfigContextLoader。我认为，没有它，跑步者SpringJUnit4ClassRunner 将缓存未更改的上下文。我不知道具体是不是这种情况。