Dubbo是我们常用的RPC框架,在写单元测试需要调用Dubbo消费Bean时,如何模拟Dubbo消费Bean的行为呢?

   就拿发邮件来说,通常,在代码中,我们是调用邮件的Dubbo服务来完成发送邮件的目的,于是我们会在Spring配置好的发邮件的Dubbo消费Bean,
dubbo-consumer.xml

<?xml version="1.0" encoding="utf-8"?>
<beans xmlns="http://www.springframework.org/schema/beans"
	xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
	xmlns:dubbo="http://dubbo.apache.org/schema/dubbo"
	xsi:schemaLocation="http://www.springframework.org/schema/beans        
	http://www.springframework.org/schema/beans/spring-beans-4.3.xsd        
	http://dubbo.apache.org/schema/dubbo        
	http://dubbo.apache.org/schema/dubbo/dubbo.xsd">

	<dubbo:application name="consumer-of-dubbo-mock-test" />

	<dubbo:registry address="zookeeper://your-zk-address:2181" />
	<!-- 生成远程服务代理,可以和本地bean一样使用mailService -->
	<dubbo:reference id="mailService" interface="cn.jmockit.demos.MailService" />
</beans>



   熟悉Dubbo的朋友都知道,上面xml配置是Dubbo的基本配置,配置了dubbo服务的zookeeper地址,还配置了名叫mailService的Dubbo消费Bean,用于在应用程序中发送邮件。
 
   我们在运行单元测试时,如果zookeeper连不上或者mailService的服务提供者不存在,则会导致Spring初始化失败, 而且我们也不希望真正发送邮件(除非是为了测试发送邮件)。于是我们希望对mailService进行Mock。

  下面给出一种Mock Dubbo消费Bean的方案:

  1. 在spring初始化前,对所有Dubbo消费Bean的进行Mock,即<dubbo>标签里的interface都返回本地默认实现。

  2. 如果想对某几个Dubbo消费Bean进行Mock,则自定义Dubbo消费Bean的实现即可。



   请看测试代码:

//dubbo消费bean Mock 
@SuppressWarnings({ "unchecked", "rawtypes" })
@ContextConfiguration(locations = { "/META-INF/dubbo-consumer.xml" })
@RunWith(SpringJUnit4ClassRunner.class)
public class DubboConsumerBeanMockingTest {
	// 这里要@BeforeClass,因为要抢在spring加载dubbo前,对dubbo的消费工厂bean
	// ReferenceBean进行mock,不然dubbo可能因为连上不zk或无法找不
	// 服务的提供者等原因而无法初始化的,进而,单元测试运行不下去
	@BeforeClass
	public static void mockDubbo() {
		// 你准备mock哪个消费bean
		// 比如要对dubbo-consumber.xml里配置的cn.jmockit.demos.MailService这个消费bean进行mock
		Map<String, Object> mockMap = new HashMap<String, Object>();
		mockMap.put("cn.jmockit.demos.MailService", new MockUp(MailService.class) {
			// 在这里书写对这个消费bean进行mock的mock逻辑,想mock哪个方法,就自行添加,注意方法一定要加上@Mock注解哦
			@Mock
			public boolean sendMail(long userId, String content) {
				// 单元测试时,不需要调用邮件服务器发送邮件,这里统一mock邮件发送成功
				return true;
			}
		}.getMockInstance());
		// 如果要Mock其它的消费bean,自行添加,mockMap.put.....如上
		new DubboConsumerBeanMockUp(mockMap);
	}

	// 现在你使用的dubbo消费bean就是本地mock过的了,并不是指向远程dubbo服务的bean了
	@Resource
	MailService mailService;

	@Test
	public void testSendMail() {
		long userId = 123456;
		String content = "test mail content";
		Assert.isTrue(mailService.sendMail(userId, content));
	}
}



   上述代码,最关键的就是DubboConsumerBeanMockUp类了,这个类Mock了所有的Dubbo消费Bean.

   源代码如下:

//dubbo消费bean的MockUp(伪类)
@SuppressWarnings("rawtypes")
public class DubboConsumerBeanMockUp extends MockUp<ReferenceBean> {
	// 自定义的消费bean mock对象
	private Map<String, Object> mockMap;

	public DubboConsumerBeanMockUp() {
	}

	public DubboConsumerBeanMockUp(Map<String, Object> mockMap) {
		this.mockMap = mockMap;
	}

	// 对ReferenceBean的getObject方法的Mock
	@SuppressWarnings("unchecked")
	@Mock
	public Object getObject(Invocation inv) throws Exception {
		ReferenceBean ref = inv.getInvokedInstance();
		String interfaceName = ref.getInterface();
		Object mock = mockMap.get(interfaceName);
		if (mock != null) {
			return mock;
		}
		return (new MockUp(Class.forName(interfaceName)) {
		}).getMockInstance();
	}
}