如何使用 MockHttpServletRequest 对文件上传进行单元测试？

Question

我有一个 Spring (3.0) 控制器，其方法将 HttpServletRequest 作为参数之一，因为它正在处理（多个）文件上传。

@RequestMapping(value = "/classified/{idClassified}/dealer/{idPerson}/upload",
    method = RequestMethod.POST)
@ResponseBody
public final String uploadClassifiedPicture(
    @PathVariable int idClassified,
    @PathVariable int idPerson, 
    @RequestParam String token, 
    HttpServletRequest request);

如何进行单元测试？我知道我可以创建一个 MockHttpServletRequest，但我不知道如何将一个或多个文件传递给它。

MockHttpServletRequest request = new MockHttpServletRequest("POST", 
     "/classified/38001/dealer/54/upload?token=dfak241adf");

Answer 1

我建议稍微更改方法签名，以使上传的文件成为普通参数（类型为 MultipartFile （不是 CommonsMultipartFile)):

@RequestMapping(value = "/classified/{idClassified}/dealer/{idPerson}/upload",
    method = RequestMethod.POST)
@ResponseBody
public final String uploadClassifiedPicture(
    @PathVariable int idClassified,
    @PathVariable int idPerson, 
    @RequestParam String token, 
    @RequestParam MultipartFile content);

那么你可以使用 MockMultipartFile在您的测试中：

final String fileName = "test.txt";
final byte[] content = "Hallo Word".getBytes();
MockMultipartFile mockMultipartFile =
       new MockMultipartFile("content", fileName, "text/plain", content);

uploadClassifiedPicture(1, 1, "token", mockMultipartFile);

---

如果您不想更改方法签名，那么您可以使用 MockMultipartHttpServletRequest反而。

它有一个方法 addFile（MultipartFile 文件）。当然，所需参数可以是 MockMultipartFile。

Answer 2

您还可以使用 MockMvc 对象以及 MockMvcRequestBuilders 向控制器发送测试文件上传请求：

@Test
public void testSendNotEmptyFile() throws Exception {

          mvc.perform(MockMvcRequestBuilders.fileUpload("Your controller URL")
            .file("file", "Test Content".getBytes())
            .contentType(MediaType.MULTIPART_FORM_DATA)
            .accept(MediaType.APPLICATION_JSON))
        .andExpect(status().isOk());
}