用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--