Spring Boot REST Controller集成测试返回406而不是500

Question

我有一个控制器，其endpoint可以生成压缩数据的字节流。下面代码中的MyDtO和ZipService类是自定义类，它们的功能类似于POJO，我想将其内容添加到zip中，服务将获取POJO的字节并将其写入ZipoutStream，然后通过封装在带有相应HttpStatus和标头。“快乐路径”工作正常，正在按预期生成zip文件。

@GetMapping(path = "/{id}/export", produces = "application/zip")
public ResponseEntity<byte[]> export(@ApiParam(required = true) @PathVariable(value = "id") String id) throws IOException {
    try {
        MyDTO myDTO = myService.getDTO(id);
        byte[] zippedData = zipService.createZip(myDTO.getBytes());
        HttpHeaders httpHeaders = new HttpHeaders();
        httpHeaders.add(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\""name.zip\"");
        return new ResponseEntity<>(zipDTO.getData(), httpHeaders, HttpStatus.OK);
    } catch (ZipException e) {
        return new ResponseEntity(e.getRestError(), HttpStatus.INTERNAL_SERVER_ERROR);
    }
}

当我想测试抛出自定义ZipException的情况时，问题出在我的联调类中（如果压缩数据存在问题，则可能会发生这种情况）。我们在组织中必须遵守的标准之一是，每个自定义异常都需要扩展Exception并拥有一个名为RestError的自定义对象，该对象具有表示自定义错误代码和消息的String变量。

@JsonInclude(JsonInclude.Include.NON_EMPTY)
public class RestError {

    private String code;
    private String message;

    //Constructors

    //Getters and setters
}

此对象似乎导致集成测试出现问题

@Test
public void myIntegrationTest() throws Exception {
    MyDTO myDTO = new MyDTO();
    RestError restError = new RestError("Custom error code", "Custom error message")
    ZipException zipException = new ZipException(restError);
    given(myService.getDTO("id")).willReturn(myDTO);
    given(zipService.createZip(any(), any())).willThrow(zipException);

    mockMvc.perform(get("/{id}/export", "id").accept(MediaType.ALL)
            .andDo(print())
            .andExpect(status().is5xxServerError());
}

在这种情况下，我预计HttpStatus将达到500，但MockMvc将达到406-内容不可接受。我把测试弄得一团糟，这样它就可以接受和期望任何/所有数据，但每次仍会遇到406错误。我知道这与异常的RestError对象有关，因为如果我从控制器返回的ResponseEntity中取出它，那么就会返回预期的响应状态。在此方面的任何帮助都将不胜感激。

Answer 1

从@GetMap（path="/{id}/export"，产生="Application/zip"）中删除生成项并将其更改为@GetMap（path="/{id}/export"）

由于这个原因，在测试用例执行期间，返回了406错误，如果删除它，您将得到您要抛出的确切错误。

但是，请检查zip文件是否按预期下载。