JUnit5中如何Mock RestTemplate的exchange方法？

你遇到的核心问题是Mock的RestTemplate实例没有被正确注入到被测的抽象类实例中，或者参数匹配未命中目标重载方法，导致实际调用了真实的exchange方法。以下是具体解决步骤：

核心原因 #

1. 抽象类中的restTemplate字段是通过@Autowired注入的，若测试时未将Mock的RestTemplate替换掉该字段，代码仍会使用真实实例。
2. exchange方法存在多个重载版本，泛用的any()可能匹配到错误的重载，导致Mock规则不生效。

解决方案一：纯Mockito测试（无Spring容器依赖） #

步骤说明 #

1. 为抽象类创建一个测试用的子类（因为抽象类无法直接实例化）。
2. 使用@Mock创建RestTemplate的Mock实例，通过@InjectMocks自动注入到被测子类实例中。
3. 精准匹配exchange的重载方法参数。

示例代码 #

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.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;

import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.when;

@ExtendWith(MockitoExtension.class)
public class AbstractClassTest {

    // 抽象类的测试子类，实现必要的抽象逻辑（如果有）
    static class TestConcreteClass extends AbstractClass<String, String> {}

    @Mock
    private RestTemplate restTemplate;

    @InjectMocks
    private TestConcreteClass testClass;

    @Test
    void testGetResponse() {
        // 精准匹配exchange的重载方法（根据你的业务场景调整参数类型）
        when(restTemplate.exchange(
                anyString(),
                any(HttpMethod.class),
                any(),
                any(Class.class)
        )).thenReturn(new ResponseEntity<>("mock-data", HttpStatus.OK));

        // 调用被测方法
        ResponseEntity<String> response = testClass.getResponse("/api/test", "request-body", HttpMethod.GET);

        // 后续验证逻辑（如断言响应状态、内容等）
    }
}

解决方案二：Spring Boot测试环境 #

步骤说明 #

如果你的测试依赖Spring容器，使用@MockBean替换容器中对应@Qualifier的RestTemplate实例，确保被测子类的@Autowired字段能拿到Mock对象。

示例代码 #

import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.http.HttpMethod;
import org.springframework.http.ResponseEntity;
import org.springframework.web.client.RestTemplate;

import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.when;

@SpringBootTest
public class ConcreteClassIntegrationTest {

    @Autowired
    private ConcreteClass concreteClass; // 假设这是AbstractClass的Spring管理子类

    @MockBean
    @Qualifier("restTemplateValue") // 必须与抽象类中@Qualifier的值一致
    private RestTemplate restTemplate;

    @Test
    void testGetResponse() {
        // Mock exchange方法
        when(restTemplate.exchange(
                anyString(),
                any(HttpMethod.class),
                any(),
                any(Class.class)
        )).thenReturn(new ResponseEntity<>(HttpStatus.OK));

        // 调用业务方法
        ResponseEntity<String> response = concreteClass.getResponse("/api/test", "request", HttpMethod.POST);

        // 验证逻辑
    }
}

关键注意事项 #

• 参数匹配精准性：如果你的业务代码中exchange使用的是ParameterizedTypeReference而非Class，需将匹配器改为any(ParameterizedTypeReference.class)，否则Mock规则不会生效。
• 避免直接调用重载方法：若测试时直接调用带RestTemplate参数的getResponse重载方法，需确保传入的是你创建的Mock实例，而非真实对象。

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