JUnit新手求助：如何Stub方法内的本地WebClient变量？

针对你的场景，核心是通过控制依赖注入的webClientList和预先设置静态映射mapClientAccounts，让方法内获取到你Stub好的WebClient实例。下面是基于JUnit 5 + Mockito的具体实现方案：

步骤1：核心测试逻辑实现 #

import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.springframework.web.reactive.function.client.WebClient;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.HashMap;
import java.util.Map;

class YourServiceTest { // 替换为你的实际服务类名

    @InjectMocks
    private YourService targetService; // 注入被测试的服务类

    @Mock
    private WebClient mockWebClient; // 模拟WebClient实例

    private Map<String, WebClient> mockWebClientList;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);

        // 初始化模拟的webClientList，关联自定义key和Mock WebClient
        mockWebClientList = new HashMap<>();
        String clientKey = "test-client";
        mockWebClientList.put(clientKey, mockWebClient);

        // 将模拟的map赋值给目标类的webClientList字段
        targetService.webClientList = mockWebClientList;

        // 设置静态映射：让指定userName对应上面的clientKey
        YourService.mapClientAccounts.put("test-user", clientKey);
    }

    @AfterEach
    void tearDown() {
        // 清理静态变量，避免测试用例间状态污染
        YourService.mapClientAccounts.clear();
    }

    @Test
    void callRemoteListenerService_shouldUseStubbedWebClient() throws FileNotFoundException {
        // 准备测试参数
        String uuid = "test-uuid";
        String userName = "test-user";
        boolean mode = true;
        File testFile = new File("test-file.txt");

        // 执行测试方法
        targetService.callRemoteListenerService(uuid, userName, mode, testFile);

        // 可添加验证逻辑，比如验证WebClient的特定方法是否被调用
        // 示例：Mockito.verify(mockWebClient).post();
    }
}

关键细节说明 #

• 处理webClientList：由于它是@Autowired的依赖，我们直接创建包含Mock WebClient的Map，赋值给目标类的对应字段，替代Spring自动注入的实例。
• 处理静态mapClientAccounts：测试前通过静态字段设置userName到clientKey的映射，确保方法内mapClientAccounts.get(userName)能拿到正确的key，从而取出Mock的WebClient。
• 静态变量清理：在@AfterEach中清空静态Map，防止不同测试用例之间的状态干扰。

Spring Boot测试简化写法 #

如果你的项目基于Spring Boot，也可以用@MockBean直接替换容器中的依赖，写法更简洁：

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mockito;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.web.reactive.function.client.WebClient;

import java.io.File;
import java.io.FileNotFoundException;
import java.util.Map;

@SpringBootTest
class YourServiceTest {

    @Autowired
    private YourService targetService;

    @MockBean
    private WebClient mockWebClient;

    @MockBean(name = "oauthproviders")
    private Map<String, WebClient> webClientList;

    @BeforeEach
    void setUp() {
        String clientKey = "test-client";
        Mockito.when(webClientList.get(clientKey)).thenReturn(mockWebClient);
        YourService.mapClientAccounts.put("test-user", clientKey);
    }

    @Test
    void callRemoteListenerService_test() throws FileNotFoundException {
        // 测试逻辑同前
    }
}

内容的提问来源于stack exchange，提问作者S.R.D