junit mockito Base基类

编写单元测试时我们都习惯性减少重复代码

以下基于spring mvc框架，需要手动pom导包

BaseTest类用于启动上下文进行debug调试

MockBaseTset类用于不启动上下文进行打桩mock

pom.xml

      <dependency> 		  <groupId>org.mockito</groupId> 		  <artifactId>mockito-all</artifactId> 		  <version>1.9.5</version> 		  <scope>test</scope> 	  </dependency>  	  <dependency> 		  <groupId>org.powermock</groupId> 		  <artifactId>powermock-api-mockito</artifactId> 		  <version>1.6.1</version> 		  <scope>test</scope> 	  </dependency> 	  <dependency> 		  <groupId>org.powermock</groupId> 		  <artifactId>powermock-module-junit4</artifactId> 		  <version>1.6.1</version> 		  <scope>test</scope> 	  </dependency> 	  	  <dependency> 		  <groupId>org.springframework</groupId> 		  <artifactId>spring-test</artifactId> 		  <version> 4.2.5.RELEASE</version> 		  <scope>test</scope> 		  <exclusions> 			  <exclusion> 				  <groupId>commons-logging</groupId> 				  <artifactId>commons-logging</artifactId> 			  </exclusion> 		  </exclusions> 	  </dependency> 	  <dependency> 		  <groupId>junit</groupId> 		  <artifactId>junit</artifactId> 		  <scope>test</scope> 	  </dependency>

BaseTest.java

import com.thinkive.gateway.v2.result.Result; import org.codehaus.jackson.map.ObjectMapper; import org.junit.runner.RunWith; import org.powermock.core.classloader.annotations.PowerMockIgnore; import org.springframework.http.HttpHeaders; 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.MvcResult; import org.springframework.test.web.servlet.request.MockMvcRequestBuilders; import org.springframework.test.web.servlet.result.MockMvcResultMatchers; import org.springframework.util.LinkedMultiValueMap; import org.springframework.util.MultiValueMap;  import java.util.HashMap; import java.util.Map;  /**  * @description: 单元测试基类  */  @RunWith(SpringJUnit4ClassRunner.class) //解决@Slf4j报错 @PowerMockIgnore({"javax.script.*","javax.management.*"})  @ContextConfiguration({"classpath*:dubbo_cfg/*.xml", "classpath*:spring-cfg.xml"}) //不用启动 Servlet 容器，就可以获取一个 Web 应用上下文 @WebAppConfiguration   public class BaseTest extends Mockito{     private static final ObjectMapper MAPPER = new ObjectMapper();     protected Map<String,String> params = new HashMap<>();     protected Map<String,String> headers = new HashMap<>();     protected  MockMvc mockMvc;      protected String prefix;      /**      * POST请求      * @param url      * @throws Exception      */     protected JsonResult postHttp(String url) throws Exception {         MultiValueMap<String, String> multiValueParamMap = new LinkedMultiValueMap<>();         if (params.size() > 0) {             multiValueParamMap.setAll(params);         }         HttpHeaders httpHeaders = new HttpHeaders();         if (headers.size() > 0) {             for (String key : headers.keySet()) {                 httpHeaders.set(key, headers.get(key));             }         }         params.clear();         MvcResult mvcResult = mockMvc.perform(MockMvcRequestBuilders                         .post(prefix + url)                         .params(multiValueParamMap)                         .headers(httpHeaders))                 .andExpect(MockMvcResultMatchers.status().isOk())                 .andReturn();         mvcResult.getResponse().setCharacterEncoding("UTF-8");         mvcResult.getResponse().setContentType("application/json;charset=UTF-8");         String result = mvcResult.getResponse().getContentAsString();         return MAPPER.readValue(result, JsonResult.class);     }      /**      * GET请求      * @param url url后缀      * @return JsonResult      * @throws Exception      */     protected JsonResult getHttp(String url) throws Exception {         MultiValueMap<String, String> multiValueParamMap = new LinkedMultiValueMap<>();         if (params.size() > 0) {             multiValueParamMap.setAll(params);         }         HttpHeaders httpHeaders = new HttpHeaders();         if (headers.size() > 0) {             for (String key : headers.keySet()) {                 httpHeaders.set(key, headers.get(key));             }         }         MvcResult mvcResult = mockMvc.perform(MockMvcRequestBuilders                         .get(prefix + url)                         .params(multiValueParamMap)                         .headers(httpHeaders))                 .andExpect(MockMvcResultMatchers.status().isOk())                 .andReturn();         mvcResult.getResponse().setCharacterEncoding("UTF-8");         mvcResult.getResponse().setContentType("application/json;charset=UTF-8");         String result = mvcResult.getResponse().getContentAsString();         return MAPPER.readValue(result, JsonResult.class);     }  } 

MockBaseTest.java

@RunWith(MockitoJUnitRunner.class) @ActiveProfiles("dev") @PowerMockIgnore("javax.management.*") public class MockBaseTest extends BaseTest{  }

示例：

public class CustControllerTest extends MockBaseTest {  	@InjectMocks //用于创建一个实例，调用真实的方法 	private CustController custController;  	@Mock 	private CustService custService;  	@Before 	public void init(){ 		this.prefix = "/api/cust"; 		//MockitoAnnotations.initMocks(this); 		//初始化MVC 		this.mockMvc = MockMvcBuilders.standaloneSetup(custController).build(); 		     } 	@Test 	public void testInfoList() throws Exception{ 		 		CustInfo custInfo = new CustInfo(); 		custInfo.setIdNo("4304221955xxxxxxxx"); 		when(custService.findCustInfo(anyLong())).thenReturn(custInfo);         //添加请求参数         params.put("id","1");         //发送post请求 		JsonResult jsonResult = postHttp("/info"); 		String idNo = (List<Map>)((Map)jsonResult.get("info")).get("idNo"); 		Assert.assertNotNull(idNo); 		Assert.assertEquals(custInfo.getIdNo(),idNo); 	}  }