我正在构建一个使用Spring Boot框架，后端使用Postgres数据库，并使用Flyway进行数据库迁移的应用程序。我一直在遇到问题，无法生成既能在Postgres中也能在内嵌的单元测试数据库中产生所需结果的迁移（即使启用了Postgres兼容模式）。因此，我正在考虑在单元测试中使用嵌入式Postgres。
我发现一个有希望的嵌入式postgres实现，但不确定如何设置它仅在Spring Boot的单元测试框架中运行（用于测试Spring Data存储库）。如何使用上述工具或替代版本的嵌入式Postgres进行设置呢？
我是embedded-database-spring-test库的作者，这个库被@MartinVolejnik提到。我认为这个库应该符合您的所有需求（PostgreSQL + Spring Boot + Flyway + 集成测试）。很抱歉您遇到了一些麻烦，因此我创建了一个简单演示应用程序，展示了如何使用这个库与Spring Boot框架。下面我总结了一些您需要做的基本步骤。
Maven配置
添加以下Maven依赖：
<dependency>
<groupId>io.zonky.test</groupId>
<artifactId>embedded-database-spring-test</artifactId>
<version>2.3.0</version>
<scope>test</scope>
</dependency>
Flyway配置
将以下属性添加到您的应用程序配置中：
# Sets the schemas managed by Flyway -> change the xxx value to the name of your schema
# flyway.schemas=xxx // for spring boot 1.x.x
spring.flyway.schemas=xxx // for spring boot 2.x.x
org.flywaydb.test.junit.FlywayTestExecutionListener。因为该库有自己的测试执行监听器，可以优化数据库初始化，如果应用了FlywayTestExecutionListener，则此优化将无效。
示例
演示嵌入式数据库使用的测试类示例：@RunWith(SpringRunner.class)
@DataJpaTest
@AutoConfigureEmbeddedDatabase
public class SpringDataJpaAnnotationTest {
@Autowired
private PersonRepository personRepository;
@Test
public void testEmbeddedDatabase() {
Optional<Person> personOptional = personRepository.findById(1L);
assertThat(personOptional).hasValueSatisfying(person -> {
assertThat(person.getId()).isNotNull();
assertThat(person.getFirstName()).isEqualTo("Dave");
assertThat(person.getLastName()).isEqualTo("Syer");
});
}
}
无法从PostgreSQL JDBC Driver 42.1.1为postgres在jdbc:postgresql://localhost:54436/postgres创建非池化数据源：org.postgresql.util.PSQLException：连接到localhost:54436被拒绝。请检查主机名和端口是否正确，并且postmaster接受TCP/IP连接。 致命错误：数据库系统正在启动 但它仍然可以工作。 - SingleShototj-pg-embedded 组件的最新稳定版本仍在使用 9.4.1208 版本的 jdbc 驱动程序。第一个使用 42.1.x postgres 驱动程序的版本是 otj-pg-embedded:0.11.1，但它依赖于 Spring Boot 的里程碑版本，并且不是 Maven 中央库的一部分。 - Tomáš Vaněk另一个相当简洁的解决该问题的方法是使用 TestContainers 库。唯一需要注意的是它需要 Docker。
集成测试：
@RunWith(SpringRunner.class)
@SpringBootTest
@ContextConfiguration(initializers = {ApplicationTestsIT.Initializer.class})
public class ApplicationTestsIT {
private static int POSTGRES_PORT = 5432;
@Autowired
private FooRepository fooRepository;
@ClassRule
public static PostgreSQLContainer postgres = new PostgreSQLContainer<>("postgres")
.withDatabaseName("foo")
.withUsername("it_user")
.withPassword("it_pass")
.withInitScript("sql/init_postgres.sql");
static class Initializer implements ApplicationContextInitializer<ConfigurableApplicationContext> {
public void initialize(ConfigurableApplicationContext configurableApplicationContext) {
TestPropertyValues.of(
"spring.data.postgres.host=" + postgres.getContainerIpAddress(),
"spring.data.postgres.port=" + postgres.getMappedPort(POSTGRES_PORT),
"spring.data.postgres.username=" + postgres.getUsername(),
"spring.data.postgres.password=" + postgres.getPassword()
).applyTo(configurableApplicationContext.getEnvironment());
}
}
@Test
public void fooRepositoryTestIT() {
...
}
依赖配置:
pom.xml:
<dependency>
<groupId>org.testcontainers</groupId>
<artifactId>postgresql</artifactId>
<scope>test</scope>
</dependency>
build.gradle：
testCompile "org.testcontainers:postgresql:x.x.x"
@Configuration
@Slf4j
public class EmbeddedPostgresConfiguration {
@Bean(destroyMethod = "stop")
public PostgresProcess postgresProcess() throws IOException {
log.info("Starting embedded Postgres");
String tempDir = System.getProperty("java.io.tmpdir");
String dataDir = tempDir + "/database_for_tests";
String binariesDir = System.getProperty("java.io.tmpdir") + "/postgres_binaries";
PostgresConfig postgresConfig = new PostgresConfig(
Version.V10_3,
new AbstractPostgresConfig.Net("localhost", Network.getFreeServerPort()),
new AbstractPostgresConfig.Storage("database_for_tests", dataDir),
new AbstractPostgresConfig.Timeout(60_000),
new AbstractPostgresConfig.Credentials("bob", "ninja")
);
PostgresStarter<PostgresExecutable, PostgresProcess> runtime =
PostgresStarter.getInstance(EmbeddedPostgres.cachedRuntimeConfig(Paths.get(binariesDir)));
PostgresExecutable exec = runtime.prepare(postgresConfig);
PostgresProcess process = exec.start();
return process;
}
@Bean(destroyMethod = "close")
@DependsOn("postgresProcess")
DataSource dataSource(PostgresProcess postgresProcess) {
PostgresConfig postgresConfig = postgresProcess.getConfig();
val config = new HikariConfig();
config.setUsername(postgresConfig.credentials().username());
config.setPassword(postgresConfig.credentials().password());
config.setJdbcUrl("jdbc:postgresql://localhost:" + postgresConfig.net().port() + "/" + postgresConfig.storage().dbName());
return new HikariDataSource(config);
}
}
Maven:
<dependency>
<groupId>ru.yandex.qatools.embed</groupId>
<artifactId>postgresql-embedded</artifactId>
<version>2.9</version>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.postgresql</groupId>
<artifactId>postgresql</artifactId>
</dependency>
HikariDatasource（Spring Boot的默认数据源）来实现正确的连接池。 binariesDir和dataDir用于避免在重复测试中进行昂贵的提取+initdb操作。@AutoConfigureEmbeddedDatabase注释即可：@RunWith(SpringRunner.class)
@AutoConfigureEmbeddedDatabase
@ContextConfiguration("/path/to/app-config.xml")
public class FlywayMigrationIntegrationTest {
@Test
@FlywayTest(locationsForMigrate = "test/db/migration")
public void testMethod() {
// method body...
}
}
并添加Maven依赖项：
<dependency>
<groupId>io.zonky.test</groupId>
<artifactId>embedded-database-spring-test</artifactId>
<version>1.1.0</version>
<scope>test</scope>
</dependency>
@DataJpaTest 一起使用，您需要使用注释 @AutoConfigureTestDatabase(replace = NONE) 来禁用默认测试数据库：@RunWith(SpringRunner.class)
@AutoConfigureTestDatabase(replace = NONE)
@AutoConfigureEmbeddedDatabase
@DataJpaTest
public class SpringDataJpaTest {
// class body...
}
@Documented
@Inherited
@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@AutoConfigureTestDatabase(replace = NONE)
@AutoConfigureEmbeddedDatabase
@DataJpaTest
public @interface PostgresDataJpaTest {
}
然后在你的测试类上方使用它：
@RunWith(SpringRunner.class)
@PostgresDataJpaTest // custom composite annotation
public class SpringDataJpaTest {
// class body...
}
@DataJpaTest下运行，我就得找别的方法了。 - SingleShot@FlywayTest 注解？你可以将其放置在测试方法或测试类上，并且可以在注解中指定迁移路径。 - Martin Volejnik@DataJpaTest 中使用它，因为我们使用了完整的上下文，但我找到了这个链接：https://github.com/spring-projects/spring-boot/issues/5716 所以你可能还需要添加 @ImportAutoConfiguration(FlywayAutoConfiguration.class)。 - Martin Volejnik
@DataJpaTest这样的单元测试可以在本地计算机上运行而无需安装数据库。 - SingleShot