Dubbo是我们常用的RPC框架，在写单元测试需要调用Dubbo消费Bean时，如何模拟Dubbo消费Bean的行为呢？
就拿发邮件来说，通常，在代码中，我们是调用邮件的Dubbo服务来完成发送邮件的目的，于是我们会在Spring配置好的发邮件的Dubbo消费Bean,
dubbo-consumer.xml
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
<?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的方案：
-
在spring初始化前，对所有Dubbo消费Bean的进行Mock，即<dubbo>标签里的interface都返回本地默认实现。
-
如果想对某几个Dubbo消费Bean进行Mock，则自定义Dubbo消费Bean的实现即可。
请看测试代码：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
|
//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.
源代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
//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(); }} |