Mockito bean多层嵌套的部分mock使用

96 阅读 0 评论 64 点赞

我是靠谱客的博主喜悦雪碧，最近开发中收集的这篇文章主要介绍Mockito bean多层嵌套的部分mock使用，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

Jmock和Mockito是常用的mock工具，Jmock功能相当强大，也比较容易用。Mockito 在处理多层嵌套时比较麻烦，下面说明了这种场景的主要用法。不过建议不是用jmock。

package com.jd.test;

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.anyLong;
import static org.mockito.Mockito.when;

import javax.annotation.Resource;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.MockitoAnnotations;
import org.mockito.Spy;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;

import com.jd.test.service.CustomerBusiness;
import com.jd.test.service.FeeBusiness;
import com.jd.test.service.OrderBusiness;
import com.jd.test.service.ProductBusiness;
import com.jd.test.service.impl.FeeBusinessImpl;

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(locations = "/spring-config.xml")
public class MockitoTest {

	
	@Resource
	@InjectMocks
	private ProductBusiness injectProductBusiness;
	
	@Mock
	private OrderBusiness orderBusiness;
	
	@Resource
	@InjectMocks
	private CustomerBusiness customerBusiness;
	
	@Spy
	private FeeBusiness feeBusiness = new FeeBusinessImpl();
	

	@Before
	public void setup() {
		MockitoAnnotations.initMocks(this);
	}
	
	/**
	 * 自动注入mock管理的bean. 解决二层和多层bean嵌套的部分mock场景。即injectProductBusiness->customerBusiness->feeBusiness 时，如果想mock feeBusiness中的某个方法. 
	 * 
	 * 	@Resource
		@InjectMocks
		private ProductBusiness injectProductBusiness;
		
		@Resource
		@InjectMocks
		private CustomerBusiness customerBusiness;
		
		@Spy
		private FeeBusiness feeBusiness = new FeeBusinessImpl();
	 * 
	 *  @InjectMocks 标签会自动填充带@Spy和@Mock标签的bean.
	 *  @Spy 被它代理的bean,默认执行原生方法。但可以在后期mock想要的方法。
	 *  @Mock 相当于mockito帮助简单的实例化bean，因此无法执行原生方法。适用于整个bean都mock,如DAO。
	 *   
	 *  同时可以结合spring一起管理bean.对bean的管理应该是spring先进行一系列的如初始化bean操作，然后mockito会引用spring生成的bean,并对bean里的指定field进行重新注入。以达到实现部分mock功能 。
	 */
	@Test
	public void injectClass() {
		
		when(feeBusiness.getFee(anyLong())).thenReturn("spy inject");
		
		assertEquals(injectProductBusiness.getCustomerName(1), "productBusinessimpl->null");
		assertEquals(injectProductBusiness.getCustomerPhone(2),"productBusinessimpl->CustomerBusinessImpl->2");
		assertEquals(injectProductBusiness.getOrderFee(3),"productBusinessimpl->CustomerBusinessImpl->spy inject");
		
		
	}

}

相关类：

package com.jd.test.service.impl;

import javax.annotation.Resource;

import org.springframework.stereotype.Component;

import com.jd.test.domain.Order;
import com.jd.test.service.OrderBusiness;

@Component
public class OrderBusinessImpl implements OrderBusiness {

	public Order getOrder(long orderid) {
		
		Order order = new Order();
		order.setAddress("北京市");
		order.setOrderid(orderid);
		
		return order;
	}

	public String getOrderAddress(long orderid) {
		return "address:" + orderid;
	}

	public String getCustomerName(long orderid) {
		return "name:" + orderid;
	}

}

package com.jd.test.service.impl;

import javax.annotation.Resource;

import org.springframework.stereotype.Component;

import com.jd.test.service.CustomerBusiness;
import com.jd.test.service.FeeBusiness;

@Component
public class CustomerBusinessImpl implements CustomerBusiness{

	@Resource
	private FeeBusiness feeBusiness;
	
	public String getCustomerPhone(long orderid) {
		return "CustomerBusinessImpl->" + orderid;
	}
	
	public String getOrderFee(long id){
		return "CustomerBusinessImpl->" + feeBusiness.getFee(id);
	}

}

package com.jd.test.service.impl;

import org.springframework.stereotype.Component;

import com.jd.test.service.FeeBusiness;

@Component
public class FeeBusinessImpl implements FeeBusiness{

	public String getFee(long id) {
		
		return "FeeBusinessImpl-->" + id;
	}

	public String getAmount(long id) {
		return "FeeBusinessImpl-->" + id;
	}

}