如何模拟带有Service Activator注解的Bean？附测试代码场景

针对你这个场景，我整理了两种测试思路：一种是单元测试，专注验证MessageHandler的业务逻辑；另一种是集成测试，验证Spring上下文是否正确注册这个Bean并完成通道绑定。

一、单元测试：验证MessageHandler的处理逻辑 #

这种方式不需要启动Spring上下文，直接测试handler()方法返回的MessageHandler实例的功能，简单高效：

import org.junit.jupiter.api.Test;
import org.springframework.messaging.Message;
import org.springframework.messaging.MessagingException;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

class MyClassTest {

    // 直接实例化MyClass，因为handler是public方法
    private MyClass myClass = new MyClass();

    @Test
    void testHandleMessageLogic() throws MessagingException {
        // 1. 获取MessageHandler实例
        MessageHandler handler = myClass.handler();

        // 2. 模拟Message对象
        Message<?> mockMessage = mock(Message.class);
        String testPayload = "test SFTP content";
        when(mockMessage.getPayload()).thenReturn(testPayload);

        // 3. 捕获System.out输出（因为你的处理器是打印payload）
        ByteArrayOutputStream outContent = new ByteArrayOutputStream();
        System.setOut(new PrintStream(outContent));

        // 4. 调用处理器的handleMessage方法
        handler.handleMessage(mockMessage);

        // 5. 验证输出是否符合预期
        assertEquals(testPayload + System.lineSeparator(), outContent.toString());

        // 6. 恢复System.out的默认输出
        System.setOut(System.out);
    }
}

如果你的MyClass里有依赖（比如logger是注入的），可以结合Mockito的@Mock和@InjectMocks来模拟依赖：

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 org.slf4j.Logger;
import static org.mockito.Mockito.verify;

@ExtendWith(MockitoExtension.class)
class MyClassWithDependenciesTest {

    @Mock
    private Logger logger; // 模拟MyClass中的Logger依赖

    @InjectMocks
    private MyClass myClass;

    @Test
    void testHandlerCreationAndLoggerCall() {
        // 调用handler方法
        MessageHandler handler = myClass.handler();
        
        // 验证logger的info方法被正确调用
        verify(logger).info("I am here");
        
        // 后续可以继续测试handleMessage逻辑...
    }
}

二、集成测试：验证Spring上下文的Bean注册与通道绑定 #

如果你需要验证@ServiceActivator是否正确绑定到sftpChannel，以及Spring是否正确管理这个Bean，可以用Spring Boot的集成测试：

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.messaging.Message;
import org.springframework.messaging.support.GenericMessage;
import java.io.ByteArrayOutputStream;
import java.io.PrintStream;
import static org.junit.jupiter.api.Assertions.assertEquals;

@SpringBootTest(classes = MyClass.class)
class MyClassIntegrationTest {

    @Autowired
    private MessageChannel sftpChannel;

    @Test
    void testServiceActivatorMessageProcessing() throws InterruptedException {
        String testPayload = "test SFTP payload from channel";
        Message<String> testMessage = new GenericMessage<>(testPayload);

        // 捕获System.out输出
        ByteArrayOutputStream outContent = new ByteArrayOutputStream();
        System.setOut(new PrintStream(outContent));

        // 发送消息到sftpChannel，触发ServiceActivator的处理器
        sftpChannel.send(testMessage);

        // 因为消息处理可能是异步的，短暂等待一下（如果是同步处理可以省略）
        Thread.sleep(100);

        // 验证处理器是否正确打印了payload
        assertEquals(testPayload + System.lineSeparator(), outContent.toString());

        // 恢复默认输出
        System.setOut(System.out);
    }
}

注意点 #

• 如果你的logger是静态成员，Mockito无法直接模拟，建议改成依赖注入的方式（比如通过构造方法或@Autowired注入Logger），这样测试更友好。
• 集成测试会启动Spring上下文，速度比单元测试慢，所以优先用单元测试验证逻辑，集成测试验证上下文配置。

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