Spring测试:为数据文件发送多个部分

Question

在我们的Spring Boot应用程序中，我们从请求中获得一个多部分形式的数据文件，如下所示:

ServletFileUpload upload = new ServletFileUpload();
try {
FileItemIterator iterStream = upload.getItemIterator(request);
while (iterStream.hasNext()) {
FileItemStream item = iterStream.next();
if (item.getFieldName().equals("file")) {
return upload(
item.getName(),
item.getContentType(),
item.openStream()
);
}
}

现在我正在写一个集成测试下载文件。我使用mockMvc发送请求。我已经尝试了很多不同的选项:

File dataFile = resourceLoader.getResource("classpath:data/fakeload/test.pdf").getFile();
MockMultipartFile file = new MockMultipartFile("test", "test.pdf",MediaType.APPLICATION_PDF_VALUE, FileUtils.readFileToByteArray(dataFile));

mockMvc.perform(multipart("/api/project-others/documents")/*.file("test.pdf",FileUtils.readFileToByteArray(dataFile))*/.file(file)
.with(getAuthentication("AUTH"))
.contentType(MediaType.MULTIPART_FORM_DATA_VALUE+";boundary=----WebKitFormBoundarylRihI4R4f6S5eHA2")
.requestAttr("file",FileUtils.readFileToByteArray(dataFile))
//.content(createFileContent(FileUtils.readFileToByteArray(dataFile),"WebKitFormBoundarylRihI4R4f6S5eHA2",MediaType.APPLICATION_PDF_VALUE,"test.pdf"))
.content(file.getBytes())
.param("projectId","1")
.param("docId","1")
)
.andExpect(status().isOk())
.andDo(print());
}
private byte[] createFileContent(byte[] data, String boundary, String contentType, String fileName){
String start = "--" + boundary + "rn Content-Disposition: form-data; name="file"; filename=""+fileName+""rn"
+ "Content-type: "+contentType+"rnrn";;
String end = "rn--" + boundary + "--"; // correction suggested @butfly
return ArrayUtils.addAll(start.getBytes(),ArrayUtils.addAll(data,end.getBytes()));
}

但是iterStream.hasNext()总是false(iterStream为空)。我错过了什么?什么错了吗?

Answer 1

我想明白了，我的错误是边界的值边界=---- webkitformboundarylrihi4r4f655eha2 -是错误的border like = q1w2e3r4t5y6u7i809 -right

右测试代码:

@Test
public void when_creating_document_expect_created_doc() throws Exception {
FileInputStream dataFile = new FileInputStream(resourceLoader.getResource("classpath:data/fakeload/test.pdf").getFile());
byte[] dataBytes = dataFile.readAllBytes();
postCreateDocument(USER_NAME, 1L, 1L, dataBytes)
.andExpect(status().isOk())
.andExpect(jsonPath("$.id").value(3))
.andDo(print());
;
}
private ResultActions postCreateDocument(String username, Long projectOtherId, Long propertyId,
byte[] bytes) throws Exception {
String boundary = "q1w2e3r4t5y6u7i8o9";
var request = MockMvcRequestBuilders
.multipart("/api/project-others/documents")
.content(createFileContent(bytes, boundary, MediaType.APPLICATION_PDF_VALUE, "test.pdf"))
.param("projectOtherId", String.valueOf(projectOtherId))
.param("docTypeId", String.valueOf(propertyId))
.contentType("multipart/form-data; boundary=" + boundary)
.accept(MediaType.APPLICATION_JSON);
if (username != null) {
request.with(getAuthentication(username));
}
return mockMvc.perform(request);
}
public byte[] createFileContent(byte[] data, String boundary, String contentType, String fileName) {
String start = "--" + boundary + "rn Content-Disposition: form-data; name="file"; filename="" + fileName + ""rn"
+ "Content-type: " + contentType + "rnrn";
String end = "rn--" + boundary + "--";
return ArrayUtils.addAll(start.getBytes(), ArrayUtils.addAll(data, end.getBytes()));
}
}