如何在仅适用Mockito的场景下,mock构造方法,并且实现对象的部分mock

本文讲述了在项目中避免使用PowerMockito的情况下,如何通过Mockito的mockConstructionWithAnswer方法结合类反射,实现对特定对象方法的模拟,以满足测试需求,同时避免mock静态方法带来的问题。
摘要由CSDN通过智能技术生成

背景:在开发中遇到一个问题,项目中禁止适用PowerMockito,当然了,我也非常不喜欢适用PowerMockito,除了对测试用例给予过多的便利,同时在mock一些静态方法,尤其是三方件,会有可能因为一些奇奇怪怪的原因,导致一堆mock报错,处理起来非常麻烦。

但是在这个用例中,在处理一个对象时,需要每次生成对象,都让这个对象的某个方法的表现是被替代的。
这个时候很多人会想到:

Person person = Mockito.spy(Person.class);
PowerMockito.whenNew(Person.class).withNoArguments().thenReturn(person);
Mockito.when(person.say()).thenReturn("mock for say.");

然后就可以只对每个Person对象的say方法进行mock,而其他方法执行实际代码逻辑。

接下来,尝试仅使用Mockito的方式进行这个实践

首先观察Mockito的方法,有两个方法名与构造函数有关:mockConstruction、mockConstructionWithAnswer

接下来尝试mockConstruction,发现如何mock,都无法实现上面的效果,没有被Mockito.when的方法默认是doNothing策略,导致无法达成测试用例的目的

Mockito.mockConstruction(Person.class, (person, context) -> {
      // 各种mock尝试
});

那么只能用mockConstructionWithAnswer继续进行尝试,
方法签名:

public static <T> MockedConstruction<T> mockConstructionWithAnswer(
            Class<T> classToMock, Answer defaultAnswer, Answer... additionalAnswers) {
        return mockConstruction(
                classToMock,
                context -> {
                    if (context.getCount() == 1 || additionalAnswers.length == 0) {
                        return withSettings().defaultAnswer(defaultAnswer);
                    } else if (context.getCount() >= additionalAnswers.length) {
                        return withSettings()
                                .defaultAnswer(additionalAnswers[additionalAnswers.length - 1]);
                    } else {
                        return withSettings()
                                .defaultAnswer(additionalAnswers[context.getCount() - 2]);
                    }
                },
                (mock, context) -> {});
    }

观察方法签名,显然需要我们实现一个answer进行传入,通过观察,invocation对象拥有方法的对象,以及方法的对应入参,那么可以使用类反射的机制直接突破困境,如下面的代码

Person person = Mockito.spy(Person.class);
Mockito.when(person.say()).thenReturn("mock for say.");
Mockito.mockConstructionWithAnswer(Person.class, new Answer() {
     @Override
     public Object answer(InvocationOnMock invocation) throws Throwable {
         return invocation.getMethod().invoke(person, invocation.getArguments());
     }
});
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

暖汤

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值