用Spring提供的test jar包帮助测试用到HTTP request的类
Spring提供的test jar(本文用的是org.springframework.test-3.0.1.RELEASE.jar)包里有一个MockMultipartHttpServletRequest类,可以用来帮助测试通过HTTP request上传文件的类,通过MockMultipartHttpServletRequest.addFile(MockMultipartFile)方法将文件'上传'到request里去,不幸的是这个类却不能工作,Spring承诺在3.1中解决.
本文通过使用Spring提供的另一个类MockHttpServletRequest,通过将文件的内容'拼'成符合上传文件时http的请求内容,加的request的contents里去达到上传文件的目的.
测试方法:
@Test
public void testPerform() {
String bondary = "XX12181285033829";//http内容里各个参数间的分割标志,可以是任意字符串
String endline = EnvConstant.NEWLINE_DELIMETER;//"/r/n";//换行符
MockHttpServletRequest request = ContentsUtil.getDefaultMockedRequest(bondary);
String filePath = "testdata/request/CopyBill_1.csv";
//ContentsUtil.getHttpFileContents会返回符合文件上传时候的http内容
String httpFilePart = ContentsUtil.getHttpFileContents(bondary, endline, "file", "CopyBill_1.csv", "text/plain", filePath);
request.setContent(httpFilePart.getBytes());
//加入参数
request.setParameter("bm_cb_dtCategory", "Copy Bill requestuest");
request.setParameter("bm_cb_SRID", "SR0001");
System.out.println(httpFilePart.toString());
ActionCopyBill ActionCopyBill = new ActionCopyBill();
MockHttpServletResponse response = new MockHttpServletResponse();
HttpSession session = request.getSession(true);//如果被测试的方法要用到session值,需要如下填充
session.setAttribute(WebKeys.LOGIN_USERID, "1");
ActionCopyBill.perform(request, response);
String ritContents = ContentsUtil.getContentsFromFile("testdata/rep/CopyBill_1.rsp");
String respContents = ContentsUtil.getContentsFromOutputStream(response);
Assert.assertEquals(respContents, ritContents);
}
用到的帮助类:getHttpFileContents方法将文件'拼'成http请求内容.
public class ContentsUtil {
public static InputStream readFromdefaultClspath(String fileName) {
InputStream stream = ClassLoader.getSystemResourceAsStream(fileName);
return stream;
}
public static String getContentsFromFile(String fileName) {
InputStream stream = readFromdefaultClspath(fileName);
StringBuffer strbf = new StringBuffer();
try {
BufferedReader in = new BufferedReader(new InputStreamReader(stream));
String str;
while ((str = in.readLine()) != null) {
strbf.append(str);
strbf.append(EnvConstant.NEWLINE_DELIMETER);
}
in.close();
} catch (IOException e) {
e.printStackTrace();
}
return strbf.toString();
}
public static String getContentsFromStream(InputStream stream) {
StringBuffer strbf = new StringBuffer();
try {
BufferedReader in = new BufferedReader(new InputStreamReader(stream));
String str;
while ((str = in.readLine()) != null) {
strbf.append(str);
}
in.close();
} catch (IOException e) {
e.printStackTrace();
}
return strbf.toString();
}
public static String getContentsFromOutputStream(MockHttpServletResponse response) {
try {
return response.getContentAsString();
} catch (IOException e) {
e.printStackTrace();
}
return null;
}
public static MockHttpServletRequest getDefaultMockedRequest(String bondary) {
MockHttpServletRequest req = new MockHttpServletRequest();
req.setCharacterEncoding("text/plain");
req.setMethod("POST");
req.setContentType("multipart/form-data; boundary=" + bondary);
req.addHeader("Content-type", "multipart/form-data");
return req;
}
public static String getHttpFileContents(String bondary, String newline, String formField,
String filename, String contentType, String filePath) {
String fileContents = ContentsUtil.getContentsFromFile(filePath);
StringBuilder sb = new StringBuilder();
sb.append(newline);
sb.append("--");
sb.append(bondary);
sb.append(newline);
sb.append("Content-Disposition: form-data; name=/"");
sb.append(formField);
sb.append("/"; filename=/"");
sb.append(filename);
sb.append(newline);
sb.append("Content-Type: ");
sb.append(contentType);
sb.append(newline);
sb.append(newline);
sb.append(fileContents);
sb.append("--");
sb.append(bondary);
sb.append("--");
sb.append(newline);
return sb.toString();
}
}
http内容里的文件部分:
--XX12181285033829
Content-Disposition: form-data; name="file"; filename="CopyBill_1.csv
Content-Type: text/plain
Account No,Invoice ID,Invoice Date,Thoroughfare Number,Thoroughfare,SubPremise,Premises Name,Post Town,Post Code
GP00000181,001,22/02/2009,118,COTSWOLD GARDENS,subPremise_004,premisesName_004,LONDON,NW2 1PN
--XX12181285033829--