如何为Spring Boot类库编写集成测试？

嘿，我太懂你现在的处境了——自己搭了Spring Boot服务和依赖的类库，功能跑顺了但类库缺集成测试，而且类库因为用的是java-library插件，没加Spring Boot的官方插件，想测从数据接收、处理到存库的完整流程对吧？别担心，我来一步步教你搞定这件事：

第一步：调整类库的build.gradle配置 #

因为你的类库没加org.springframework.boot插件，得手动引入Spring Boot测试相关的依赖，还好你已经用了io.spring.dependency-management插件，它会帮我们管理版本号，不用自己硬写。在dependencies块里加上这些：

testImplementation 'org.springframework.boot:spring-boot-starter-test'
testImplementation 'com.h2database:h2' // 用内存数据库做测试，不用搭真实库，方便又快

第二步：创建测试专用的Spring配置类 #

因为类库不是完整的Spring Boot应用，没有主启动类，所以我们要手动写一个配置类来加载类库的组件和测试需要的配置。在src/test/java下建一个配置类，比如LibraryTestConfig：

@Configuration
@ComponentScan(basePackages = "com.your.library.package") // 替换成你类库的实际包路径
@EnableJpaRepositories(basePackages = "com.your.library.repository") // 如果用了JPA仓库就加这个
@EntityScan(basePackages = "com.your.library.entity") // 扫描你的实体类
@PropertySource("classpath:application-test.properties") // 加载测试用的配置文件
public class LibraryTestConfig {
    // 如果需要手动配置数据源或者其他Bean，也可以在这写，不过H2的话Spring Boot会自动帮忙配置
}

第三步：编写集成测试类 #

现在可以写真正的集成测试了，我们用@SpringBootTest指定刚才的配置类，再加上@Transactional保证测试后数据库回滚，避免脏数据影响下一次测试：

@SpringBootTest(classes = LibraryTestConfig.class)
@Transactional
class LibraryIntegrationTest {

    @Autowired
    private DataProcessingService dataProcessingService; // 替换成你类库的核心服务类
    @Autowired
    private DataRepository dataRepository; // 替换成你类库的仓库类

    @Test
    void testFullDataFlow() {
        // 模拟从外部服务传来的输入数据
        RawInputData input = new RawInputData("测试数据", "2024-05-20");

        // 调用类库的处理存储方法
        dataProcessingService.processAndSave(input);

        // 验证数据是否正确存储到数据库
        Optional<ProcessedData> savedData = dataRepository.findByInputKey("测试数据");
        assertThat(savedData).isPresent();
        assertThat(savedData.get().getProcessedContent()).isEqualTo("处理后的测试数据"); // 替换成你预期的处理结果
    }
}

第四步：添加测试配置文件 #

在src/test/resources下创建application-test.properties，配置H2内存数据库和其他测试需要的参数：

# H2内存数据库配置
spring.datasource.url=jdbc:h2:mem:testdb
spring.datasource.driverClassName=org.h2.Driver
spring.datasource.username=sa
spring.datasource.password=

# JPA配置，自动创建表，测试结束后删除
spring.jpa.hibernate.ddl-auto=create-drop

# 可选：开启H2控制台，方便调试时看数据库内容
spring.h2.console.enabled=true

一些额外的小提示 #

• 单元测试和集成测试要分开：单元测试用Mockito Mock掉依赖，只测单个方法的逻辑；集成测试就像上面这样，测完整的流程，从入口到数据库。
• 如果你的类库用到了其他Spring组件（比如Redis、MQ），只要在build.gradle里加上对应的测试依赖（比如spring-boot-starter-data-redis-test），再在测试配置里加上相关配置就行。
• 运行测试的话，直接用Gradle的test任务就行，和普通Java项目一样。

备注：内容来源于stack exchange，提问作者StefanoF02