关于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