Mockito 单元测试中模拟 RestClientException 异常

import org.junit.jupiter.api.Test;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpMethod;
import org.springframework.web.client.RestClientException;
import org.springframework.web.client.RestTemplate;

import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.ArgumentMatchers.*;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.verify;

import org.springframework.http.ResponseEntity;
import org.springframework.http.HttpStatus;

import org.junit.jupiter.api.extension.ExtendWith;

@ExtendWith(MockitoExtension.class)
public class IdpFacadeClientTest {

    @Mock
    private RestTemplate restTemplate;

    @InjectMocks
    private IdpFacadeClient idpFacadeClient;

    // 假设的 DTO 类，用于更新记录
    static class UpdateRecordsDto {
        private String firstName;

        public String getFirstName() {
            return firstName;
        }

        public void setFirstName(String firstName) {
            this.firstName = firstName;
        }
    }

    private String idpFacadeUpdateUrl = "http://example.com/update";


    @Test
    void test_update_customer_in_ath0_server_error(){
        //given
        UpdateRecordsDto updateRecordsDto = new UpdateRecordsDto();
        updateRecordsDto.setFirstName("testFirstName");

        //when
        doThrow(new RestClientException("Idp facade API error"))
                .when(restTemplate).exchange(anyString(), eq(HttpMethod.PATCH), any(HttpEntity.class), eq(String.class));

        //then
        assertThrows(RestClientException.class, () -> idpFacadeClient.updateCustomerFirstNameInIdp(updateRecordsDto));
    }


    private boolean verifyUpdatedCustomerBody(HttpEntity<?> entity) {
        // 实现根据 entity 内容进行验证的逻辑
        return true; // 示例，始终返回 true
    }

    // 模拟的 IdpFacadeClient 类
    class IdpFacadeClient {
        public String updateCustomerFirstNameInIdp(UpdateRecordsDto customerUpdateRecordsDto) {
            String response = null;
            try {
                ResponseEntity<String> restResponse = restTemplate.exchange(idpFacadeUpdateUrl, HttpMethod.PATCH, new HttpEntity<>(customerUpdateRecordsDto),  String.class);
                if (restResponse.getStatusCode() == HttpStatus.OK) {
                    //log.info("customer first name update success {} with status {}", customerUpdateRecordsDto.getFirstName(), response.getStatusCode());
                }else {
                    //log.error("Customer first name update failed in IDP for {} with status {}", customerUpdateRecordsDto.getFirstName(), response.getStatusCode());
                }
            } catch (RestClientException e) {
                throw new RestClientException("Idp facade API error for user first name " + customerUpdateRecordsDto.getFirstName(), e);
            }
            return response;
        }
    }
}