Mockito 在 RestTemplate.exchange Springboot 中返回“java.lang.IllegalArgumentException: URI 不是绝对的”

Question

Mockito 在 RestTemplate.exchange 中返回“java.lang.IllegalArgumentException：URI 不是绝对的”。 我不确定为什么会发生这种情况，因为看起来我是 mocking restTemplate 正确并且因为我看到了那个异常，所以 RestTemplate 似乎不是模拟。

这是我的 class

@Component
public class RestTemplateWrapper {
  private static final Logger LOGGER = LoggerFactory.getLogger(RestTemplateWrapper.class);
  
  public <T> ResponseEntity<T> callWebServiceGET(String url,HttpEntity<?> httpEntity,
      ParameterizedTypeReference<T> parameterizedTypeReference) {

    RestTemplate restTemplate = new RestTemplate();
    ResponseEntity<T> response=null;
    LOGGER.trace("Entered callWebServiceGET");
    LOGGER.info("Calling WebService {}", url);
    try {
      response=restTemplate.exchange(url, HttpMethod.GET, httpEntity, parameterizedTypeReference);
    } catch (HttpClientErrorException e) {
      if (HttpStatus.NOT_FOUND.equals(e.getStatusCode())) {
        LOGGER.error("Service Unavailable - Code 404 returned. " + url + e.getMessage());
      } else if (HttpStatus.UNAUTHORIZED.equals(e.getStatusCode())) {
        LOGGER.error("Token Expired- Code 401 returned. " + e.getMessage());
      } else if (HttpStatus.BAD_REQUEST.equals(e.getStatusCode())) {
        LOGGER.error("Bad Input, 400 returned.{} {} ", url , e.getMessage(), e);
      } else {
        LOGGER.error("WEB Service Failure. " + e.getMessage());
      }
    }
    return response;
  }

}

这是我的测试用例：

@PrepareForTest({RestTemplateWrapper.class})
public class RestTemplateWrapperTest  {
  
  @Mock
  private RestTemplate mockRestTemplate;
  @InjectMocks
  private RestTemplateWrapper webUtils;
  @Before
  public void setUp() throws Exception {
    MockitoAnnotations.initMocks(this);
  }
  @Test
  public void callWebServiceGET_OK()  {
    HttpEntity<String> httpEntity= new ResponseEntity<>(HttpStatus.OK);
    ResponseEntity<String> entityResponse=new ResponseEntity<>("MOCK_RESPONSE", HttpStatus.OK);
    when(mockRestTemplate.exchange(eq("/objects/get-objectA"), eq(HttpMethod.GET), eq(httpEntity),any(
        ParameterizedTypeReference.class))).thenReturn(
        entityResponse);
    ResponseEntity<String> mockResponse= webUtils.callWebServiceGET("",null,  new ParameterizedTypeReference<String>(){
    });
    //verify(mockRestTemplate,times(1)).exchange(Matchers.anyString(), Matchers.any(), Matchers.any());
     Assert.assertEquals("MOCK_RESPONSE",mockResponse.getBody());
  }
  
}

响应：

URI is not absolute
java.lang.IllegalArgumentException: URI is not absolute
    at java.net.URI.toURL(URI.java:1088)
    at org.springframework.http.client.SimpleClientHttpRequestFactory.createRequest(SimpleClientHttpRequestFactory.java:145)
    at org.springframework.http.client.support.HttpAccessor.createRequest(HttpAccessor.java:87)
    at org.springframework.web.client.RestTemplate.doExecute(RestTemplate.java:727)
    at org.springframework.web.client.RestTemplate.execute(RestTemplate.java:666)
    at org.springframework.web.client.RestTemplate.exchange(RestTemplate.java:604)
    at com.project.di.tp.purchaseorderservice.utils.RestTemplateWrapper.callWebServiceGET(RestTemplateWrapper.java:29)
    at com.project.di.tp.purchaseorderservice.utils.RestTemplateWrapperTest.callWebServiceGET_OK(RestTemplateWrapperTest.java:51)

关于如何解决这个问题的任何想法？ 我一直在尝试 4 个小时。

Answer 1

尽管在JavaDocs中没有明确说明，但您必须在那里提供绝对值 URL。

这是因为您无处提供基数 URL，而相对数 URL 将相对于该基数。 您也无法在浏览器中输入"/objects/get-objectA"作为 URL。

所以我建议您对第一个参数使用类似"http://example.com/objects/get-objectA"的东西：

when(mockRestTemplate.exchange(
        eq("http://example.com/bla"), 
        eq(HttpMethod.GET), 
        isNull(HttpEntity.class), 
        any(ParameterizedTypeReference.class))).
    thenReturn(entityResponse);

ResponseEntity<String> mockResponse =
    webUtils.callWebServiceGET(
        "http://example.com/bla", 
        null, 
        new ParameterizedTypeReference<String>(){});

请注意，使用给定参数调用webUtils.callWebServiceGET不会使 Mockito 返回想要的答案，因此我在对绝对 URL 的调用中更改了一个 URL，您期望在Mockito.when中更改预期参数成为类型化的null （类型化以匹配方法签名）。

更新：

正如您自己已经发现的那样，您的Mockito.when不起作用，因为您没有在测试方法中使用测试创建的模拟，而是在每次调用callWebServiceGET时创建一个新的RestTemplate实例。 （不知道为什么没早点看到，sorry！）

我建议您使用构造函数将RestTemplate注入到经过测试的 class 中：

private final RestTemplate restTemplate;
public RestTemplateWrapper(RestTemplate restTemplate) {
  this.restTemplate = restTemplate;
}

// remove the following line in the method callWebServiceGET:
// RestTemplate restTemplate = new RestTemplate();

使用此代码，Spring 将自动将模拟的RestTemplate注入到测试中，但要运行生产代码，您需要添加一个 bean 以提供用于注入的RestTemplate 。

将其添加到 Configuration class 中，您还可以在其中定义其他 bean：

@Bean
public RestTemplate restTemplate(RestTemplateBuilder builder) {
   // Do any additional configuration here
   return builder.build();
}

（在How to autowire RestTemplate using annotations的答案中找到此代码片段）

作为测试的一般建议：尽量避免在任何要测试的代码中使用new运算符，而是使用注入。 如果您需要创建多个实例（例如在一个循环中等），请尝试注入一个为您创建实例的工厂 - 因此在测试中您可以模拟该工厂。

Answer 2

我找到了解决方案，似乎问题是我的 class 是 RestTemplateWrapper 正在 callWebServiceGET 中创建一个实例，因此 mockito 不能模拟 object。如果在方法之外设置 object，它可以工作，但我不想这样做。

有什么方法可以模拟方法内部的 object 吗？