如何使用Java和Spark模拟多部分/表单数据？

Question

我正在使用 multipart/form-data 接收一个文件，如下所示（使用 Spark，而不是 SpringBoot）：

 @Override
public Object handle(Request request, Response response) throws Exception {
    request.attribute("org.eclipse.jetty.multipartConfig", new MultipartConfigElement(""));

    Part filePart = request.raw().getPart("file");
    String regex = request.raw().getParameter("regex");
    String fileName = filePart.getSubmittedFileName();
    byte[] fileBytes = filePart.getInputStream().readAllBytes();

问题是，我想对这个控制器进行单元测试，为了做到这一点，我需要一种方法来模拟请求以在内部包含多部分/表单数据，或者至少一种使用“当... theReturn”来模仿那部分代码...... 有什么想法吗？

提前致谢！

Answer 1

所以我设法找到了这个问题的答案，我想也许我可以通过回答它来帮助其他人：

@Test
public void whenValidRegex_returnOkAndTotalAmount() throws Exception {
    final Part file = mock(MultiPartInputStreamParser.MultiPart.class);
    final Request mock = mock(Request.class); //Spark request
    final Response mockResponse = mock(Response.class); //Spark response
    final HttpServletRequest httpServletRequest = mock(HttpServletRequest.class); //Javax servlet

    when(mock.raw()).thenReturn(httpServletRequest);
    when(file.getSubmittedFileName()).thenReturn("file.pdf");
    when(mock.raw().getParameter("regex")).thenReturn(String.valueOf("SOME REGEX"));
    when(mock.params("numPage")).thenReturn(String.valueOf("1"));
    when(file.getInputStream()).thenReturn(IOUtils.toInputStream("ARGENTINIAN PESOS", Charset.defaultCharset())); //Here you are mocking the input stream you might get from the part file.
    when(mock.raw().getPart("file")).thenReturn(file);

现在您已经模拟了多部分/表单数据，您可以继续测试模拟服务调用等。 请忽略我的特定代码中的内容，例如“页码”模拟，您不需要它。 希望这对其他人有帮助。 再见！