spring - 如何测试返回 zip 文件的 Spring Controller ？

Question

我有一个返回 zip 文件的 Controller 。我想将 zip 文件与预期的 zip 进行比较，但我不确定如何从我的结果中获取文件。这是我到目前为止所拥有的:

public class FileControllerTest extends ControllerTest {

  @InjectMocks
  private FileController controller;

  @Autowired
  private WebApplicationContext context;

  private MockMvc mvc;

  @Before
  public void initTests() throws IOException {
    MockitoAnnotations.initMocks(this);
    mvc = MockMvcBuilders.webAppContextSetup(context).build();
  }

  @Test
  public void shouldReturnZip() throws Exception {
    MvcResult result = mvc
        .perform(get(SERVER + FileController.REQUEST_MAPPING + "/zip").accept("application/zip"))
        .andExpect(status().isOk()).andExpect(content().contentType("application/zip"))
        .andDo(MockMvcResultHandlers.print()).andReturn();

  }
}

Answer 1

你可以从 MvcResult .getResponse().getContentAsByteArray() 得到一个字节数组.
从那里您可以转换 ByteArrayInputStream成File或 ZipFile用于比较。