Spring@SQL注释，是否可以在所有测试之前运行一次？

Question

使用Spring进行集成测试，我可以填充一个运行脚本的测试数据库，如下所示...

@Test
@Sql({"/db/schema.sql", "/db/accountConfig.sql", "/db/functions/fnSomething.sql"})
public void verifySomething() {
   ...
}

我使用的是JUnit4、Spring Boot、Java15、TestContainers。

我试过的事情...

• 我曾尝试在我的测试类扩展的类上使用@beforeClass，但它似乎在我的测试之后运行。
• TestContainers确实有一个init脚本函数，但它只接受一个文件，这并不理想。
• 我也尝试过scriptutils.executesqlscript，但是出于某种原因，测试容器不喜欢这样。

下面是我的示例代码，它可以与@sql一起工作，但不能与scriptutils.executesQLScript一起工作。

@ContextConfiguration(initializers = AbstractIntegrationTest.Initializer.class)
public abstract class AbstractIntegrationTest {

    @ClassRule 
    public static MSSQLServerContainer mssqlserver = new MSSQLServerContainer();

  
    public static class Initializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {

        @Override
        public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
            ConfigurableEnvironment environment = configurableApplicationContext.getEnvironment();
            Properties props = new Properties();
            props.put("spring.datasource.driver-class-name", mssqlserver.getDriverClassName());
            props.put("spring.datasource.url", mssqlserver.getJdbcUrl());
            props.put("spring.datasource.username", mssqlserver.getUsername());
            props.put("spring.datasource.password", mssqlserver.getPassword());

            environment
                .getPropertySources()
                .addFirst(new PropertiesPropertySource("myTestDBProps", props));      

            configurableApplicationContext.setEnvironment(environment);
    }
}

我的测试类只是扩展AbstractIntegrationTest。但是使用@sql可以为每个测试用例运行脚本。有人对初始化SQL脚本的更好方法有什么建议吗？尝试了flyway，但它不允许从脚本创建DB。

Answer 1

我建议查看一下在Spring上下文初始化时执行一次SQL脚本的数据库初始化器bean。基本上，根据您使用JDBC还是R2DBC，有两种解决方案。由于要初始化多个脚本，因此应该使用CompositeDatabasePopulator。还要记住导入正确的类，因为它们具有相同的名称，但来自不同的包，这同样取决于jdbc/r2dbc。

要从resources文件夹加载资源，可以使用以下任一选项：

• resource resource=new ClassPathResource（“sql/schema.sql”））
• @value（“classpath：sql/schema.sql”）资源资源

这个解决方案相当灵活，因为您可以使用@testconfiguration为测试上下文定义初始化器bean（请记住，使用这个注释有点棘手，所以我请您参阅这篇文章：Spring的@testconfiguration的怪癖对我帮助很大）。

该解决方案应该适用于任何具有ConnectionFactory可用的解决方案，包括测试容器。

• 使用org.springframework.jdbc.datasource.init.CompositeDatabasePopulator
• 使用org.springframework.jdbc.datasource.init.resourceDatabasePopulator

通过调用populate进行的初始化必须在@PostConstuct中进行，因为Spring Boot不会自动检测到populator。我建议将以下代码段包装在配置类中，并将其包含到测试范围中。

@Autowired
private DataSource dataSource;

@PostConstruct
public void initData() throws SQLException {

    var populator = new CompositeDatabasePopulator();
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/schema.sql")));
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/catalog.sql")));
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/data.sql")));

    populator.populate(dataSource.getConnection());
}

• 使用org.springframework.r2dbc.connection.init.CompositeDatabasePopulator
• 使用org.springframework.r2dbc.connection.init.resourceDatabasePopulator

您可以使用ConnectionFactoryInitializer，它在加载Spring上下文时初始化添加到该初始化器中的所有填充器一次。

@Bean
public ConnectionFactoryInitializer initializer(ConnectionFactory connectionFactory) {

    var populator = new CompositeDatabasePopulator();
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/schema.sql")));
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/catalog.sql")));
    populator.addPopulators(new ResourceDatabasePopulator(new ClassPathResource("sql/data.sql")));


    var initializer = new ConnectionFactoryInitializer();
    initializer.setConnectionFactory(connectionFactory);
    initializer.setDatabasePopulator(populator);

    return initializer;
}