如何使用泛型模拟 ResponseEntity<?>？

Question

我需要模拟一个服务。我在模拟课程时在ResponseEntity<?> resp 中得到空值。

需要模拟的方法：

public List<Expression> getExpression(String expressView, Localdate date) {

    List<Expression> =new ArrayList<>();
    Map<String, Object> uri = new HashMap<>();
    UriComponenetsBuilder build = 
        UriComponentsBuilder.fromHttpUrl("someUrl" + "/" + expressView);
    build.queryParam(someParameter, someParameter);
    build.queryParam(someParameter, someParameter);
    build.queryParam(someParameter, someParameter);

    HttpHeaders header = new HttpHeaders();
    header.setContentType(MediaType.APPLICATION_JSON);
    RestClient client = new RestClient(
        build.build().encode.toUriString, HttpMethod.GET.Uri, header
    );

    ResponseEntity<?> resp = restC.SomeMethod(client);

    if (resp = !null) {
        //it goes to these line
    }

  }

在我的模拟方法中：

when(restC.SomeMethod(client)).thenReturn(resp);

所以上面的方法调用一个服务获取一些数据，获取expressView的值并保存为列表。当我嘲笑 when(restC.SomeMethod(client)).thenReturn(resp); 方法时，它点击了 URL，但我作为响应得到的值 resp 为 null 。 所以在这里我将resp 的值设为null。我在邮递员中检查了 URL(someUrl) 它返回的值。

如何模拟ResponseEntity<?>？

谢谢。

Answer 1

首先，创建一个ResponseEntity 对象：

HttpHeaders header = new HttpHeaders();
header.setContentType(MediaType.APPLICATION_JSON);

ResponseEntity<?> responseEntity = new ResponseEntity<>(
    "some response body",
    header, 
    HttpStatus.OK
);

然后构建一个 mock 以返回 responseEntity 对象：

when(restC.SomeMethod(client)).thenReturn(responseEntity);

---

注意点：

避免在@Service class 中使用ResponseEntity。您应该在@RestController class 中使用ResponseEntity。

你可以使用@Autowired注解Inject你的@Service class，比如：

@RestController
public class YourControllerClass {

    @Autowired
    private YourServiceClass yourServiceClass;

或者使用constructor，比如：

@RestController
public class YourControllerClass {

    private YourServiceClass yourServiceClass;

    public YourControllerClass(YourServiceClass yourServiceClass) {
        this.yourServiceClass= yourServiceClass;
    }

所以：

@Service class 将处理 business or data objects 和 @RestController class 将处理 Response 和 Request 对象。这样我们就有了Single Responsibility的原则。

---

一些不错的链接：

• Spring MVC - Using RequestEntity and ResponseEntity
• How do I mock a REST template
• Building REST services with Spring
• Understanding the Single Responsibility Principle

希望这会有所帮助！