Spring Boot中IntegrationFlow与ServiceActivator的JUnit单元测试咨询

一、先做代码小改造（测试友好） #

原代码中CONST_PATH是硬编码常量，建议改成配置属性，方便测试时替换目录：

@Configuration
@EnableIntegration
public class ConsumeFiles {

    // 替换硬编码常量为配置注入
    @Bean
    public MessageSource<File> fileReadingFromSrc(@Value("${file.directory}") String directoryPath) {
        FileReadingMessageSource source = new FileReadingMessageSource();
        source.setDirectory(new File(directoryPath));
        source.setAutoCreateDirectory(true);
        return source;
    }

    // 其余代码保持不变...
}

同时在项目application.properties中添加配置：

file.directory=/your/actual/directory
period=5000
poller.max.messages.per.poll=1

二、测试依赖准备 #

确保pom.xml（Maven）或build.gradle（Gradle）中包含测试所需依赖：

Maven 示例 #

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-test</artifactId>
        <scope>test</scope>
    </dependency>
    <dependency>
        <groupId>org.springframework.integration</groupId>
        <artifactId>spring-integration-test</artifactId>
        <scope>test</scope>
    </dependency>
</dependencies>

三、两种测试方案 #

方案1：完整集成测试（验证轮询+流程全链路） #

模拟真实场景：文件放入目录后，等待轮询触发流程，验证目标方法是否被正确调用。

@SpringBootTest
@TestPropertySource(properties = {
    "period=100", // 缩短轮询间隔，加快测试
    "poller.max.messages.per.poll=1",
    "file.directory=${java.io.tmpdir}/test-file-dir"
})
class ConsumeFilesIntegrationTest {

    @SpyBean // 用于监控目标方法的调用情况
    private ConsumeFiles consumeFiles;

    @TempDir // JUnit自动创建/清理的临时目录
    private Path testDirectory;

    @Test
    void testFileTriggerFlow() throws IOException, InterruptedException {
        // 1. 在临时目录创建测试文件
        File testFile = testDirectory.resolve("test-sample.txt").toFile();
        Files.writeString(testFile.toPath(), "test content");

        // 2. 等待轮询触发（轮询间隔100ms，等待500ms足够覆盖）
        Thread.sleep(500);

        // 3. 验证目标方法被调用，且参数正确
        ArgumentCaptor<Message<File>> messageCaptor = ArgumentCaptor.forClass(Message.class);
        Mockito.verify(consumeFiles, Mockito.times(1)).controleSignature(messageCaptor.capture());

        Message<File> capturedMsg = messageCaptor.getValue();
        assertEquals(testFile.getName(), capturedMsg.getPayload().getName());
        assertEquals("test content", Files.readString(capturedMsg.getPayload().toPath()));
    }
}

方案2：快速单元测试（跳过轮询，直接触发流程） #

不需要等待轮询，直接从消息源获取文件消息并发送到通道，验证业务逻辑：

@SpringBootTest
@TestPropertySource(properties = {
    "file.directory=${java.io.tmpdir}/quick-test-dir"
})
class ConsumeFilesUnitTest {

    @Autowired
    private MessageSource<File> fileReadingFromSrc;

    @Autowired
    @Qualifier("channel-1")
    private MessageChannel channel1;

    @SpyBean
    private ConsumeFiles consumeFiles;

    @TempDir
    private Path testDirectory;

    @Test
    void testFileProcessingLogic() throws IOException {
        // 1. 创建测试文件
        File testFile = testDirectory.resolve("quick-test.txt").toFile();
        Files.writeString(testFile.toPath(), "quick content");

        // 2. 从消息源获取文件消息
        Message<File> fileMsg = fileReadingFromSrc.receive();
        assertNotNull("未获取到文件消息", fileMsg);

        // 3. 发送消息到通道，触发业务方法
        channel1.send(fileMsg);

        // 4. 验证方法调用和参数
        ArgumentCaptor<Message<File>> messageCaptor = ArgumentCaptor.forClass(Message.class);
        Mockito.verify(consumeFiles, Mockito.times(1)).controleSignature(messageCaptor.capture());

        assertEquals(testFile.getName(), messageCaptor.getValue().getPayload().getName());
    }
}

四、关键注意事项 #

• @TempDir：JUnit 5提供的注解，自动创建临时目录，测试结束后自动清理，避免测试环境污染。
• @SpyBean：Spring测试提供的注解，用于监控真实Bean的方法调用，既保留原Bean的功能，又能验证调用次数和参数。
• 配置覆盖：通过@TestPropertySource在测试中覆盖生产配置，比如缩短轮询间隔、替换测试目录，不影响生产环境配置。

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