Spring Boot测试中Mockito处理同一方法多HTTP调用的问题求助

嘿，我完全懂你现在的困扰——同一个业务方法里嵌套了两次HTTP调用，想用Mockito分别mock它们的返回结果对吧？这在单元测试里太常见了，给你分享几个实用的解决办法：

1. 按调用顺序直接指定返回值 #

如果两次HTTP调用的顺序是固定的，最简单的方式就是用thenReturn()链式调用，指定第一次和第二次调用的返回结果：

// 假设你依赖的HTTP客户端是RestTemplate，先把它Mock出来
@Mock
private RestTemplate restTemplate;

// 构造mock返回对象
AccountResponse mockAccountResp = new AccountResponse();
mockAccountResp.setId("123");

BalanceResponse mockBalanceResp = new BalanceResponse();
mockBalanceResp.setAmount(BigDecimal.valueOf(1000));

// 链式指定两次调用的返回值：第一次返回账户信息，第二次返回余额信息
when(restTemplate.getForObject(anyString(), eq(AccountResponse.class)))
    .thenReturn(mockAccountResp)
    .thenReturn(mockBalanceResp);

// 执行你的测试方法
accountService.methodWithTwoHttpCalls();

这样Mockito会严格按照调用顺序返回结果，第一次调用该HTTP方法时返回mockAccountResp，第二次返回mockBalanceResp。

2. 通过请求参数区分两次调用 #

如果两次HTTP调用的参数（比如请求URL、请求体）不一样，用ArgumentCaptor捕获请求参数，再根据参数返回对应结果会更精准：

@Mock
private RestTemplate restTemplate;
@Captor
private ArgumentCaptor<String> urlCaptor; // 捕获请求URL的参数

// 定义mock返回对象
AccountResponse accountResp = new AccountResponse();
BalanceResponse balanceResp = new BalanceResponse();

// 使用thenAnswer根据捕获的参数动态返回结果
when(restTemplate.getForObject(urlCaptor.capture(), eq(Object.class)))
    .thenAnswer(invocation -> {
        String capturedUrl = urlCaptor.getValue();
        // 根据URL判断是哪一次调用
        if (capturedUrl.contains("/api/accounts")) {
            return accountResp;
        } else if (capturedUrl.contains("/api/balances")) {
            return balanceResp;
        }
        throw new IllegalArgumentException("Unknown request URL");
    });

// 执行测试
accountService.methodWithTwoHttpCalls();

// 还可以验证调用次数和参数是否符合预期
verify(restTemplate, times(2)).getForObject(anyString(), any());
List<String> allCapturedUrls = urlCaptor.getAllValues();
assertEquals("预期的账户接口URL", allCapturedUrls.get(0));
assertEquals("预期的余额接口URL", allCapturedUrls.get(1));

这种方法不仅能区分两次调用，还能验证请求参数的正确性，适合需要严格校验请求的场景。

3. 根据调用次数动态返回结果 #

如果两次调用的参数几乎一样，但你需要按调用次数区分，也可以在thenAnswer里获取调用次数来判断：

@Mock
private RestTemplate restTemplate;

when(restTemplate.postForObject(anyString(), any(), eq(GenericResponse.class)))
    .thenAnswer(invocation -> {
        // 获取当前Mock方法的调用次数
        int callCount = Mockito.mockingDetails(restTemplate).getInvocations().size();
        return callCount == 1 ? firstResponse : secondResponse;
    });

注意：这种方式要确保当前测试方法中该Mock对象只会被调用这两次，否则计数会受其他调用影响。

额外注意事项 #

• 确保你的测试类上添加了@ExtendWith(MockitoExtension.class)（JUnit 5）或者@RunWith(MockitoJUnitRunner.class)（JUnit 4），这样Mockito才能正确初始化@Mock和@InjectMocks对象；
• 如果你的HTTP调用是通过FeignClient，Mock方式和上面一致，只需要把RestTemplate换成你的FeignClient接口即可。

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