Mockito无法Mock RestTemplate的exchange方法问题求助

问题原因 #

1. RestTemplate实例完全隔离 #

业务方法validateMemberData中直接通过new RestTemplate()创建实例，测试代码中Mock的是另一个独立的RestTemplate对象，两者毫无关联，导致Mock配置无法作用到业务逻辑里的实例。

2. 静态方法isSelfServicePortalServerAvailable未Mock #

如果这个方法默认返回false，业务代码会直接跳过RestTemplate调用分支，根本不会走到你Mock的exchange方法逻辑。

解决步骤 #

步骤1：改造业务类，让RestTemplate可替换 #

将业务类中硬编码的new RestTemplate()改为可配置的静态字段，提供测试专用的设置方法：

public class SelfServiceValidation {
    // 原静态字段，默认初始化RestTemplate
    private static RestTemplate restTemplate = new RestTemplate();
    // 其他原有字段...

    // 新增测试专用方法，用于替换RestTemplate实例
    static void setRestTemplate(RestTemplate template) {
        restTemplate = template;
    }

    public static MemberDataValidationResponse validateMemberData(MemberValidationRequest memberValidationRequest)
            throws Exception {
        MemberDataValidationResponse memberValidateResponse = new MemberDataValidationResponse();
        try {
            if (isSelfServicePortalServerAvailable(selfServicePortalURL)) {
                // ... 原有代码
                // 替换掉new RestTemplate()，使用静态字段
                // RestTemplate restTemplate = new RestTemplate();
                // ... 原有代码
                ResponseEntity<MemberDataValidationResponse> response = restTemplate.exchange(
                        builder.buildAndExpand().toUri(), HttpMethod.POST, request, MemberDataValidationResponse.class);
                // ... 原有代码
            } else {
                log.info(SELF_SERVICE_PORTAL_UNAVAILABLE_MESSAGE);
            }
        } catch (Exception cause) {
            throw new RuntimeException(cause);
        }
        return memberValidateResponse;
    }

    // 原有静态方法isSelfServicePortalServerAvailable...
}

步骤2：调整测试代码，正确Mock依赖 #

import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.Mock;
import org.mockito.MockedStatic;
import org.mockito.MockitoAnnotations;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.*;
import static org.junit.jupiter.api.Assertions.*;

public class SelfServiceValidationTest {

    @Mock
    private RestTemplate restTemplate;

    @BeforeEach
    void setUp() {
        // 初始化Mock对象
        MockitoAnnotations.openMocks(this);
        // 将Mock的RestTemplate设置到业务类的静态字段中
        SelfServiceValidation.setRestTemplate(restTemplate);
    }

    @Test
    public void testValidateMemberData_Success() throws Exception {
        MemberValidationRequest memberRequest = new MemberValidationRequest();
        memberRequest.setCorelationIdentifier("123");

        MemberDataValidationResponse mockResponse = new MemberDataValidationResponse();
        mockResponse.setStatus("200");

        // Mock exchange方法，匹配业务代码中的参数类型
        when(restTemplate.exchange(any(Uri.class), eq(HttpMethod.POST), any(HttpEntity.class), eq(MemberDataValidationResponse.class)))
                .thenReturn(ResponseEntity.ok(mockResponse));

        MemberDataValidationResponse response = null;

        // 同时Mock两个静态方法：SelfServiceDaoHelper和isSelfServicePortalServerAvailable
        try (MockedStatic<SelfServiceDaoHelper> mockDaoHelper = mockStatic(SelfServiceDaoHelper.class);
             MockedStatic<SelfServiceValidation> mockValidation = mockStatic(SelfServiceValidation.class)) {

            // Mock auditApiRequestData返回1
            mockDaoHelper.when(() -> SelfServiceDaoHelper.auditApiRequestData(any(MapSqlParameterSource.class)))
                    .thenReturn(1);

            // Mock isSelfServicePortalServerAvailable返回true，确保走RestTemplate分支
            mockValidation.when(() -> SelfServiceValidation.isSelfServicePortalServerAvailable(anyString()))
                    .thenReturn(true);

            // 调用真实的业务方法，避免静态Mock覆盖逻辑
            mockValidation.when(() -> SelfServiceValidation.validateMemberData(any(MemberValidationRequest.class)))
                    .thenCallRealMethod();

            // 执行测试
            response = SelfServiceValidation.validateMemberData(memberRequest);
        }

        // 断言结果
        assertNotNull(response);
        assertEquals("200", response.getStatus());
        // 验证exchange方法是否被调用
        verify(restTemplate, times(1)).exchange(any(Uri.class), eq(HttpMethod.POST), any(HttpEntity.class), eq(MemberDataValidationResponse.class));
    }
}

关键说明 #

• RestTemplate实例对齐：通过静态字段+测试专用set方法，让业务代码使用测试中Mock的实例，确保Mock配置生效。
• 静态方法Mock：必须MockisSelfServicePortalServerAvailable返回true，否则不会进入RestTemplate调用分支；同时确保SelfServiceDaoHelper的Mock作用域覆盖业务方法调用。
• 真实方法调用：MockSelfServiceValidation的静态方法时，通过thenCallRealMethod()确保执行真实业务逻辑，避免被Mock覆盖。

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