单元测试用例常用基础设置

package com.palmchat.controller;

import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.fileUpload;
import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;

import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

import javax.annotation.Resource;

import org.junit.After;
import org.junit.Before;
import org.junit.runner.RunWith;
import org.springframework.mock.web.MockMultipartFile;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockHttpServletRequestBuilder;
import org.springframework.test.web.servlet.request.MockMultipartHttpServletRequestBuilder;
import org.springframework.test.web.servlet.setup.MockMvcBuilders;
import org.springframework.web.context.WebApplicationContext;
import org.springframework.test.context.junit4.AbstractTransactionalJUnit4SpringContextTests;

@RunWith(SpringJUnit4ClassRunner.class)
@WebAppConfiguration

@ContextConfiguration(locations = { "classpath:/application-context.xml" })

//AbstractTransactionalJUnit4SpringContextTests实现回滚

public abstract class CommonControllerTest extends AbstractTransactionalJUnit4SpringContextTests {
    @Resource
    private WebApplicationContext webApplicationContext;

    protected MockMvc mockMvc;
    
    protected MockHttpServletRequestBuilder makePostByParams(String uri, Map<String, Object> params) {
        
        List<String> fileKeys = params.keySet().stream().filter(key -> params.get(key) instanceof MockMultipartFile).collect(Collectors.toList());
        
        MockHttpServletRequestBuilder mockHttpServletRequestBuilder;
        
        if(fileKeys.size() == 0) {
            mockHttpServletRequestBuilder = post(uri);
        } else {
            MockMultipartHttpServletRequestBuilder mockMultipartHttpServletRequestBuilder = fileUpload(uri);
            fileKeys.stream().forEach(key -> mockMultipartHttpServletRequestBuilder.file((MockMultipartFile)params.get(key)));
            fileKeys.stream().forEach(key -> params.remove(key));
            mockHttpServletRequestBuilder = mockMultipartHttpServletRequestBuilder;
        }
        
        params.keySet().stream().forEach(e -> mockHttpServletRequestBuilder.param(e, params.get(e).toString()));
        
        return mockHttpServletRequestBuilder;
    }
    
    @Before
    public void setup() {
        this.mockMvc = MockMvcBuilders.webAppContextSetup(this.webApplicationContext).build();
    }
    
    @After
    public void after() {
    }
    
    @SuppressWarnings("unchecked")
    protected <T> T fetchReturnObj(String uri, Map<String, Object> params, String key) throws Exception {
        return (T)this.mockMvc.perform(makePostByParams(uri, params)).andReturn().getRequest().getAttribute(key);
    }
    
    protected void requestParamsException(String uri, Map<String, Object> params) throws Exception {
//        this.mockMvc.perform(makePostByParams(uri, params)).andExpect(status().isOk())
//                .andExpect(request().attribute("code", -3));
    }
}