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

来源：互联网发布：本港台直播软件编辑：程序博客网时间：2024/05/20 00:15

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));
   }
}

0 0