青年IT男

个人从事金融行业,就职过易极付、思建科技等重庆一流技术团队,目前就职于某网约车平台负责整个支付系统建设。自身对金融行业有强烈的爱好。同时也实践大数据、数据存储、自动化集成和部署、分布式微服务、人工智能等领域。

JMockit常见用法 -Mock Dubbo消费Bean

JMockit常见用法 -Mock Dubbo消费Bean

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的方案:

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

如果想对某几个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(); } }
0
1028826685@qq.com