Spring Boot仓库层测试：无需H2/真实库，仅用Flyway脚本实现

针对你的需求，以下是三种可行的实现方案，按实用性优先级排序：

方案1：配置H2兼容PostgreSQL语法（最轻量） #

通过开启H2的PostgreSQL兼容模式，让你的PostgreSQL语法Flyway脚本可以在内存库中正常执行，同时测试JpaRepository方法。

配置步骤 #

1. 在src/test/resources下创建application-test.yml：

spring:
  datasource:
    url: jdbc:h2:mem:testdb;MODE=PostgreSQL;DATABASE_TO_LOWER=TRUE;DEFAULT_NULL_ORDERING=HIGH
    driver-class-name: org.h2.Driver
    username: sa
    password:
  jpa:
    database-platform: org.hibernate.dialect.PostgreSQLDialect
    hibernate:
      ddl-auto: none # 禁用JPA自动建表，完全依赖Flyway
  flyway:
    enabled: true
    locations: classpath:db/migration # 指向你的Flyway脚本目录

2. 编写仓库测试类：

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

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertNotNull;

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

    @Autowired
    private UserRepository userRepository;

    @Test
    void testSaveAndRetrieveUser() {
        // 准备测试数据
        User testUser = new User();
        testUser.setUsername("harshang");
        testUser.setEmail("harshang@example.com");
        
        // 执行保存
        User savedUser = userRepository.save(testUser);
        
        // 验证结果
        assertNotNull(savedUser.getId());
        User retrievedUser = userRepository.findByUsername("harshang").orElseThrow();
        assertEquals("harshang@example.com", retrievedUser.getEmail());
    }
}

兼容问题处理 #

如果遇到H2不支持的PostgreSQL特有语法（如特定函数），可以：

• 在Flyway脚本中通过/* H2 */和/* PostgreSQL */添加条件注释
• 为H2自定义模拟函数（例如在测试启动时执行CREATE ALIAS语句）

方案2：使用Testcontainers启动临时PostgreSQL容器（最真实） #

通过Testcontainers自动启动一个临时的PostgreSQL容器，完全兼容原生PG语法，无需本地安装数据库，测试完成后自动销毁容器。

依赖配置（Maven） #

<dependencies>
    <dependency>
        <groupId>org.testcontainers</groupId>
        <artifactId>postgresql</artifactId>
        <version>1.18.3</version>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.testcontainers</groupId>
        <artifactId>junit-jupiter</artifactId>
        <version>1.18.3</version>
        <scope>test</scope>
    </dependency>
</dependencies>

测试类实现 #

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.orm.jpa.DataJpaTest;
import org.springframework.test.context.DynamicPropertyRegistry;
import org.springframework.test.context.DynamicPropertySource;
import org.testcontainers.containers.PostgreSQLContainer;
import org.testcontainers.junit.jupiter.Container;
import org.testcontainers.junit.jupiter.Testcontainers;

import static org.junit.jupiter.api.Assertions.assertEquals;

@Testcontainers
@DataJpaTest
public class UserRepositoryTest {

    // 启动临时PostgreSQL容器，使用指定版本镜像
    @Container
    static PostgreSQLContainer<?> postgresContainer = new PostgreSQLContainer<>("postgres:15-alpine")
            .withDatabaseName("test_db")
            .withUsername("test_user")
            .withPassword("test_pass");

    // 动态配置数据源指向容器
    @DynamicPropertySource
    static void setDataSourceProperties(DynamicPropertyRegistry registry) {
        registry.add("spring.datasource.url", postgresContainer::getJdbcUrl);
        registry.add("spring.datasource.username", postgresContainer::getUsername);
        registry.add("spring.datasource.password", postgresContainer::getPassword);
        registry.add("spring.flyway.enabled", () -> true);
    }

    @Autowired
    private UserRepository userRepository;

    @Test
    void testFindByEmail() {
        User user = new User();
        user.setUsername("test_user");
        user.setEmail("test@example.com");
        userRepository.save(user);

        User result = userRepository.findByEmail("test@example.com").orElseThrow();
        assertEquals("test_user", result.getUsername());
    }
}

方案3：Mock仓库层（仅适用于业务逻辑测试） #

如果仅需测试依赖仓库的服务层，无需验证JPA查询或Flyway脚本的正确性，可以用Mockito模拟仓库实现：

import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.util.Optional;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
public class UserServiceTest {

    @Mock
    private UserRepository userRepository;

    @InjectMocks
    private UserService userService;

    @Test
    void testGetUserByUsername() {
        // 模拟仓库返回值
        User mockUser = new User();
        mockUser.setUsername("harshang");
        mockUser.setEmail("harshang@example.com");
        when(userRepository.findByUsername("harshang")).thenReturn(Optional.of(mockUser));

        // 测试服务方法
        User user = userService.getUserByUsername("harshang");
        assertEquals("harshang@example.com", user.getEmail());
    }
}

注意：此方案无法验证Flyway脚本的正确性，也无法测试JPA查询的逻辑是否符合预期，仅适合隔离测试服务层逻辑。

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