Spring Boot单元测试中嵌入式Postgres的配置咨询

我来帮你搞定这个配置——用嵌入式Postgres完全匹配生产环境的数据库，再也不用纠结Flyway脚本在不同数据库里的兼容问题了！下面是一步步的配置指南：

1. 先添加依赖 #

推荐用Zonky的embedded-postgres-spring-boot-starter，它是专门为Spring Boot整合的嵌入式Postgres库，省掉很多手动配置的麻烦。

如果用Maven，在pom.xml里加：

<dependency>
    <groupId>io.zonky.test</groupId>
    <artifactId>embedded-postgres-spring-boot-starter</artifactId>
    <scope>test</scope>
</dependency>

Gradle的话，在build.gradle里加：

testImplementation 'io.zonky.test:embedded-postgres-spring-boot-starter'

2. 配置测试环境属性 #

在src/test/resources下新建application-test.properties（或者用YAML格式的application-test.yml），配置数据库和Flyway的相关参数：

Properties格式： #

# 激活测试环境
spring.profiles.active=test

# 嵌入式Postgres的自定义配置（可选，默认会自动生成数据库名、用户名密码）
zonky.test.postgres.database.name=my_test_db
zonky.test.postgres.username=test_user
zonky.test.postgres.password=test_pass

# 让Flyway用和生产一样的迁移脚本
spring.flyway.locations=classpath:db/migration
spring.flyway.enabled=true

YAML格式： #

spring:
  profiles:
    active: test
  flyway:
    locations: classpath:db/migration
    enabled: true

zonky:
  test:
    postgres:
      database:
        name: my_test_db
      username: test_user
      password: test_pass

要是你想指定和生产一致的Postgres版本，直接加一行zonky.test.postgres.version=15（替换成你用的版本号）就行，保证测试和生产的数据库版本完全对齐。

3. 编写测试类 #

不管是全栈的@SpringBootTest还是数据层的@DataJpaTest，只要加上@ActiveProfiles("test")，就能自动加载嵌入式Postgres的配置：

示例1：验证Flyway迁移的测试 #

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.ActiveProfiles;

@SpringBootTest
@ActiveProfiles("test")
public class MigrationValidationTest {

    @Test
    void flywayMigrationsShouldRunSuccessfully() {
        // 只要Spring Boot能正常启动，就说明Flyway迁移执行成功了
        // 你也可以在这里加逻辑验证表结构、初始数据是否符合预期
    }
}

示例2：数据访问层测试 #

import org.junit.jupiter.api.Test;
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import org.springframework.test.context.ActiveProfiles;

@DataJpaTest
@ActiveProfiles("test")
public class UserRepositoryTest {

    // 注入你的Repository
    // @Autowired
    // private UserRepository userRepository;

    @Test
    void shouldSaveAndRetrieveUser() {
        // 在这里编写数据操作的测试逻辑，完全基于Postgres环境
    }
}

4. 几个关键提醒 #

• 脚本不用再兼容H2：现在测试用的是真实的Postgres（嵌入式），直接用生产环境的Flyway脚本就行，彻底解决之前的兼容问题。
• 第一次启动慢：第一次运行测试时，库会自动下载对应版本的Postgres二进制包，可能需要几分钟，之后再跑就快了。
• 自动清理资源：测试结束后，嵌入式Postgres会自动关闭，临时文件也会被清理，不用手动处理。

这样配置完，你的单元测试就和生产环境用完全一样的数据库环境，Flyway脚本的效果也能和生产保持一致啦！

内容的提问来源于stack exchange，提问作者SingleShot