关于Spring中Controller的单元测试

第一次写对Controller的单元测试，在不了解的情况下被自己的无知给坑了，下面给出写法。

package com.alipay.marketingservice.action;import com.alibaba.fastjson.JSON;import com.alipay.marketingservice.util.ConstantManager;import com.alipay.opbiservice.vo.UserVo;import org.junit.Before;import org.junit.Test;import org.junit.runner.RunWith;import org.springframework.mock.web.MockHttpSession;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.TestExecutionListeners;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;import org.springframework.test.context.support.DependencyInjectionTestExecutionListener;import org.springframework.test.context.web.WebAppConfiguration;import org.springframework.test.web.servlet.MockMvc;import org.springframework.test.web.servlet.MvcResult;import org.springframework.test.web.servlet.ResultActions;import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;import org.springframework.test.web.servlet.setup.MockMvcBuilders;import javax.annotation.Resource;import java.util.ArrayList;import java.util.Calendar;import java.util.List;import java.util.Map;@WebAppConfiguration@RunWith(SpringJUnit4ClassRunner.class)@TestExecutionListeners( { DependencyInjectionTestExecutionListener.class })@ContextConfiguration(locations = "classpath:META-INF/spring/*.xml")public class ExportActionTest {    @Resource    private MerchantAction merchantAction;    @Resource    private ExportAction exportAction;    private MockMvc mockMvc;    private MockMvc mockMvc2;    private MockHttpSession session;    @Before    public void setUp(){        mockMvc = MockMvcBuilders.standaloneSetup(merchantAction).build();        mockMvc2 = MockMvcBuilders.standaloneSetup(exportAction).build();        session = new MockHttpSession();    }    @Test    public void test()throws Exception{        //自定义request跳过权限检查        UserVo userVo = new UserVo();        List permission = new ArrayList();        //在这里添加需要访问的权限接口        permission.add("***_***");        permission.add("***_get***");        userVo.setPermission(permission);        //设置session        session.setAttribute("id", JSON.toJSONString(userVo));        //设置时间        Calendar calendar = Calendar.getInstance();        calendar.set(2017, 7, 1);        Long beginDate = calendar.getTimeInMillis();        calendar.set(2017, 9, 1);        Long endDate = calendar.getTimeInMillis();        //通过接口获取商家名        ResultActions resultActions = this.mockMvc.perform(MockMvcRequestBuilders.post("/***/get***.action").session(session).servletPath("/***/get***.action"));        MvcResult mvcResult = resultActions.andReturn();        String result1 = mvcResult.getResponse().getContentAsString();        Map map0 = (Map) JSON.parse(result1);        Map map = (Map) map0.get("data");        if(map == null){            System.out.println("没获取到商家");        }        Map<String, Object> map2 = ((List<Map<String, Object>>)map.get("mer**")).get(0);        String id = (String) map2.get("id");        String name = (String) map2.get("name");        String merchant = id + "," + name;        //设置导出选项        //调用方法        ResultActions resultActions2 = this.mockMvc2.perform(MockMvcRequestBuilders.post("/***/***.action").servletPath("/***/***.action").session(session).param("beginDate", beginDate.toString())                .param("en**", endDate.toString()).param("mer**", merchant)                .param("op**[]", "***", "***", "***", "***", "***", "***", "***", "***"));        MvcResult mvcResult2 = resultActions2.andReturn();        String result2 = mvcResult2.getResponse().getContentAsString();        String resultExcel = new String(result2.getBytes("ISO-8859-1"), "UTF-8");        System.out.println(resultExcel);    }}

这里使用的是MockMVC，其实我一开始还画蛇添足自定义了request，并设置值，后来发现MockMVC中应该内置了request，具体源码我还没看，等之后闲下来的时候再慢慢整理，(*^_^*)

private MockHttpServletRequest request;private MockHttpServletResponse response;request = new MockHttpServletRequest();request.setCharacterEncoding("UTF-8");request.setServletPath("/***/***.action");response = new MockHttpServletResponse();

这里注意的是不要这样对session设置值

request.getSession().setAttribute(ConstantManager.SESSION_USER, JSON.toJSONString(userVo));

这里可以对session， servletPath等request属性进行设置值

 ResultActions resultActions2 = this.mockMvc2.perform(MockMvcRequestBuilders.post("/***/***.action").servletPath("/***/***.action").session(session).param("beginDate", beginDate.toString())                .param("en**", endDate.toString()).param("mer**", merchant)                .param("op**[]", "***", "***", "***", "***", "***", "***", "***", "***"));

设置数组参数的部分源码，只要往后加就行了。

public MockHttpServletRequestBuilder param(String name, String... values) {addToMultiValueMap(this.parameters, name, values);return this;}

好啦，就是这么多，更多的内容根据自己的需求自己探索，结果总是让人快乐的。O(∩_∩)O