Mockito配合powermock工具mock构造函数

更新时间:2019-04-18 10:21:23点击次数:309次
背景
Mockito的强大就不多说明了,这是Mockito2的官方文档:点击这里查看,但有些场景依然无法满足,这时候可以借助powermock,powermock是对Mockito的一些能力的扩展,如支持静态方法 ,私有方法等等场景的mock。PowerMock内部使用了Mockito的API,可以保证同时使用两种mock框架。

现有类BeanUtils如下方法代码:

public String getName() {
//其它逻辑
        String name = new BeanObject().getName();
        return name;
    }
我需要测试BeanUtils类getName()接口,只需确认其中逻辑符合预想,最后调用new BeanObject().getName()的方法的返回值并不关心,也可能BeanObject的getName方法需要传入其它参数,或者这个方法内部有更复杂的逻辑需要回避。

那么我需要mock BeanObejct类的getName方法。

解决方法是mock (BeanObject的mock实例的getName)方法。那这样需要保证每次new BeanObject()的时候构造的都是我mock的那个BeanObject实例。

实现代码
就详细的说下测试实现吧。

1. 先引入测试相关的maven依赖

    <dependencies>
     <dependency>
            <groupId>junit</groupId>
            <artifactId>junit</artifactId>
            <version>4.12</version>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.mockito</groupId>
            <artifactId>mockito-core</artifactId>
            <version>2.8.9</version>
            <scope>test</scope>
        </dependency>
 
        <!-- https://mvnrepository.com/artifact/org.powermock/powermock-module-junit4 -->
        <dependency>
            <groupId>org.powermock</groupId>
            <artifactId>powermock-module-junit4</artifactId>
            <version>1.7.4</version>
            <scope>test</scope>
        </dependency>
 
        <!-- https://mvnrepository.com/artifact/org.powermock/powermock-api-mockito2 -->
        <dependency>
            <groupId>org.powermock</groupId>
            <artifactId>powermock-api-mockito2</artifactId>
            <version>1.7.4</version>
            <scope>test</scope>
        </dependency>
   </dependencies>
2. 贴下这几个类的代码

public class BeanUtils {
 
    public String getName() {
        String name = new BeanObject().getName();
        return name;
    }
}
public class BeanObject {
 
    private String name = "test_bean";
 
    public String getName() {
        return name;
    }
 
}
@RunWith(PowerMockRunner.class)
@PrepareForTest({BeanUtils.class})
public class TestBeanUtils {
 
    @Test
    public void testGetName() throws Exception {
 
        BeanObject beanObject = Mockito.mock(BeanObject.class);
        PowerMockito.whenNew(BeanObject.class).withNoArguments().thenReturn(beanObject);
        Mockito.when(beanObject.getName()).thenReturn("mock_test");
        System.out.println(new BeanUtils().getName());
    }
}
PowerMock的使用
借助上面代码TestBeanUtils简单说下配置:

1. 首先在测试类上使用注解@RunWith(PowerMockRunner.class)

2. 使用注解@PrepareForTest.(这个注解也可作用在方法,取决于你想要的作用范围 )

@PrepareForTest这个注解是告诉PowerMock框架,我们需要对测试类作字节码级的处理。这些类可以是final、私有、静态等方法 或者实例的时候返回mock对象。注意我这里声明要处理的是BeanUtils类,是需要测试的类而不BeanObject类。

要mock BeanObject的构造方法,我们可以使用PowerMockito的whenNew(),这是个链式调用,之后调用withNoArguments,见名知义,就是无参构造,当然还有withAnyArguments()、withArguments(...)等根据实际需要,然后返回的使用mock的BeanObject实例。

这样每次new BeanObject的时候,构造出来的都是那个mock的BeanObject实例。

然后,再用下面这行代码,mock它的getName方法,便达到预期结果了。

Mockito.when(beanObject.getName()).thenReturn("mock_test");

本站文章版权归原作者及原出处所有 。内容为作者个人观点, 并不代表本站赞同其观点和对其真实性负责,本站只提供参考并不构成任何投资及应用建议。本站是一个个人学习交流的平台,网站上部分文章为转载,并不用于任何商业目的,我们已经尽可能的对作者和来源进行了通告,但是能力有限或疏忽,造成漏登,请及时联系我们,我们将根据著作权人的要求,立即更正或者删除有关内容。本站拥有对此声明的最终解释权。

  • 项目经理 点击这里给我发消息
  • 项目经理 点击这里给我发消息
  • 项目经理 点击这里给我发消息
  • 项目经理 点击这里给我发消息