PowerMockito的基本使用解析

PowerMockito经常会结合Mockito使用，先说一下这2个的介绍：

1.Mockito和PowerMockito的简介Mockito和PowerMockito是什么东西呢？他们有什么作用呢？

Mocktio和PowerMockito都是Mock的工具类，主要是Java的类库，Mock就是伪装的意思。

他们适用于单元测试中，对于单元测试来说，我们不希望依赖于第三方的组件，比如数据库、Webservice等。在写单元测试的时候，我们如果遇到了这些需要依赖第三方的情况，我们可以使用Mock的技术，伪造出来我们自己想要的结果。

对于Java而言，mock的对象主要是Java 方法和 Java类。

下面我就介绍一下怎么使用Mockito和PowerMockito去进行Mock。

2.Mockito和PowerMockito的区别

在我看来，PowerMockito是Mockito的一种增强，他们的PowerMockito可以调用Mockito的方法，但是对于Mocktio不能Mock的对象或者方法，我们可以使用PowerMockito来实现。

比如Mockito不能用于static Method, final method, 枚举类， private method，这些我们都可以用PowerMockito来实现，当PowerMockito和mockito结合使用的时候，我们需要考虑兼容性的问题。

两者的版本需要兼容

Mockito PowerMockito

2.8.9+ 2.x 2.8.0-2.8.9 1.7.x 2.7.5 1.7.0RC4 2.4.0 1.7.0RC2 2.0.0-beta – 2.0.42-beta 1.6.5-1.7.0RC 1.10.8 – 1.10.x 1.6.2 – 2.0 1.9.5-rc1 – 1.9.5 1.5.0 – 1.5.6 1.9.0-rc1 & 1.9.0 1.4.10 – 1.4.12 1.8.5 1.3.9 – 1.4.9 1.8.4 1.3.7 & 1.3.8 1.8.3 1.3.6 1.8.1 & 1.8.2 1.3.5 1.8 1.3 1.7 1.2.5

Ref：https://github.com/powermock/powermock/wiki/Mockito

3.具体用法

本文实现实现需要构造的接口和需要返回值的接口

引入依赖

<dependencies><dependency><groupId>org.projectlombok</groupId><artifactId>lombok</artifactId></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-context-support</artifactId></dependency><dependency><groupId>org.springframework</groupId><artifactId>spring-test</artifactId><scope>test</scope></dependency><dependency><groupId>junit</groupId><artifactId>junit</artifactId><version>4.12</version><scope>test</scope></dependency><dependency><groupId>org.mockito</groupId><artifactId>mockito-all</artifactId><version>2.0.2-beta</version><scope>test</scope></dependency><dependency><groupId>org.powermock</groupId><artifactId>powermock-module-junit4</artifactId><version>1.7.4</version><scope>test</scope></dependency><dependency><groupId>org.powermock</groupId><artifactId>powermock-api-mockito</artifactId><version>1.7.4</version><scope>test</scope></dependency></dependencies>

需要Mock的类：

ProcessDB.java

package com.github.mock.simple.vo; public class ProcessDB {        public ProcessDB(String ss){        System.out.println(ss + " Enter ProcessDB ...");    }        public ProcessDB(){        System.out.println("Enter ProcessDB ...");    }        public void getResultOfConnectDBNoReturn(String ss) {        System.out.println(ss + " Enter getResultOfConnectDBNoReturn ...");    }        public String getResultOfConnectDB() {        return "haha, Really went to the database";    }} 

需要测试的类：

IUserService.java

package com.github.mock.simple.user; public interface IUserService {     public String testedMehtod(); }

UserServiceImpl.java

package com.github.mock.simple.user.impl; import org.springframework.stereotype.Service; import com.github.mock.simple.user.IUserService;import com.github.mock.simple.vo.ProcessDB; @Servicepublic class UserServiceImpl implements IUserService {     @Override    public String testedMehtod(){        System.out.println("Enter UserServiceImpl testedMehtod ...");        ProcessDB processDB = new ProcessDB("BB");        processDB.getResultOfConnectDBNoReturn("AA");        return processDB.getResultOfConnectDB();    } }

BussinessService.java

package com.github.mock.simple.user.impl; import com.github.mock.simple.vo.ProcessDB; public class BussinessService {    public String testedMehtod() {        System.out.println("Enter BussinessService testedMehtod ...");        ProcessDB processDB = new ProcessDB("BB");        processDB.getResultOfConnectDBNoReturn("AA");        return processDB.getResultOfConnectDB();    }}

测试类：

MockSpringSimpleTest.java

package com.github.mock.simple.test; import java.text.MessageFormat; import org.junit.Assert;import org.junit.Before;import org.junit.Test;import org.junit.runner.RunWith;import org.mockito.InjectMocks;import org.mockito.Mock;import org.mockito.MockitoAnnotations;import org.powermock.api.mockito.PowerMockito;import org.powermock.core.classloader.annotations.PrepareForTest;import org.powermock.modules.junit4.PowerMockRunner;import org.powermock.modules.junit4.PowerMockRunnerDelegate;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.test.context.ContextConfiguration;import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; import com.github.mock.simple.user.IUserService;import com.github.mock.simple.user.impl.BussinessService;import com.github.mock.simple.user.impl.UserServiceImpl;import com.github.mock.simple.vo.ProcessDB; @RunWith(PowerMockRunner.class)@PowerMockRunnerDelegate(SpringJUnit4ClassRunner.class)//Spring上下文@PrepareForTest({BussinessService.class,UserServiceImpl.class})@ContextConfiguration(locations = {"classpath:applicationContext-mock-inject.xml"})public class MockSpringSimpleTest {     //使用Spring上下文    @Autowired    IUserService userService;     @Mock    ProcessDB processDB;        //不使用Spring上下文时，使用该注解    @InjectMocks    private BussinessService bussinessService;     @Before    public void initMocks() throws Exception {        MockitoAnnotations.initMocks(this);        //ReflectionTestUtils.setField(userService, "processDB", processDB);        PowerMockito.whenNew(ProcessDB.class).withArguments("BB").thenReturn(processDB);        // PowerMockito.whenNew(ProcessDB.class).withNoArguments().thenReturn(processDB);    }     @Test    public void mockConnectDB() {        String aa = "haha, everything is fake";         PowerMockito.when(processDB.getResultOfConnectDB()).thenReturn(aa);        PowerMockito.doNothing().when(processDB).getResultOfConnectDBNoReturn("AA");        System.out.println(bussinessService.testedMehtod());        Assert.assertEquals("haha, everything is fake", bussinessService.testedMehtod());    }     @Test    public void mockConnectDB2() {        try {            String aa = "haha, everything is fake";            PowerMockito.when(processDB.getResultOfConnectDB()).thenReturn(aa);            PowerMockito.doNothing().when(processDB).getResultOfConnectDBNoReturn("AA");            System.out.println(userService.testedMehtod());            Assert.assertEquals("haha, everything is fake", userService.testedMehtod());        } catch (Exception ex) {            System.out.println("--- getMessage ---");            System.out.println(ex.getMessage());            System.out.println();                        System.out.println("--- toString ---");            System.out.println(ex.toString());            System.out.println();            //            System.out.println("--- printStackTrace ---");//            StringWriter stringWriter = new StringWriter();//            PrintWriter printWriter = new PrintWriter(stringWriter);//            ex.printStackTrace(printWriter);//            System.out.println(stringWriter.toString());//            System.out.println();                        System.out.println("--- printStackTrace DIY ---");            System.out.println(ex.getClass().getName() + ": " + ex.getMessage());            StringBuilder sbException = new StringBuilder();            for (StackTraceElement ele : ex.getStackTrace()) {                sbException.append(MessageFormat.format("\tat {0}.{1}({2}:{3})\n",                     ele.getClassName(), ele.getMethodName(), ele.getFileName(), ele.getLineNumber()));;            }            System.out.println(sbException);                        sbException = null;//            stringWriter = null;//            printWriter = null;        }    } }

扫描注入xml

最后applicationContext-mock-inject.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:context="http://www.springframework.org/schema/context"       xsi:schemaLocation=" http://www.springframework.org/schema/beans http://www.springframework.org/schema/beans/spring-beans.xsd   http://www.springframework.org/schema/context http://www.springframework.org/schema/context/spring-context.xsd">     <context:component-scan base-package="com.github.mock.simple"/> </beans>

对于没有实现类，但又被依赖的接口，在applicationContext-mock-inject.xml添加如下内容 (本文不需要)：

<bean name="iXxService" class="org.mockito.Mockito" factory-method="mock">    <constructor-arg value="com.github.mock.simple.api.IXxService"/></bean>

同时在测试类里面添加下面的代码：

@MockiXxService iXxService;

在 @Before里面添加下面的代码

ReflectionTestUtils.setField(userService, "iXxService", iXxService);

测试结果

PowerMockito的使用技巧

当IT中有些依赖组件无法正常集成，需要mock支持测试，可以使用power mockito。

特别注意：

当对一个对象进行powermockito，应该在prepare方法，统一mock这个对象。然后在其他方法，分别进行调用when，否则，多个方法内进行mock，会出错。

比如有个 Service处于IT case的底层，普通的mock根本mock不进去，但我们又不能为了集成测试，为这个testcase单独开一个口子，注入mock对象。power mockito强大的mock能力在这里可以用上。

比如：

我的mock对象impalaService它在schmaMessagehandler类里new出来的，则需要加上注解。

首先在test 类的开头，加上注解头部，头部类是mock对象所在类。

@RunWith(PowerMockRunner.class)@PrepareForTest({HttpClient.class,SchemaMessageHandler.class})

其次：

PooledImpalaService impalaService = PowerMockito.mock(PooledImpalaService.class);PowerMockito.whenNew(PooledImpalaService.class).withArguments((ConfigurationanyObject()).thenReturn(impalaService);doNothing().when(impalaService).createTable(anyString(),(Schema) anyObject());

使用powermockito，注意在用any（）参数时候，比如

doNothing().when(impalaService).createTable(anyString(),(Schema) anyObject());

参数列表中，只要一个使用了any（），any****，则所有参数都要用any相关的参数，否则mock不成功。

总的来说，在it当中，只有你想mock一个对象，一定可以，比如你在A类中用到了B类，那么在prepareForTest中增加A类的注解。

如下：

@PrepareForTest({A.class})然后，在it中 声明一个B类，B b = PowerMockito.mock(B.class);这时候，就可以指定b的方法的返回值，或 PowerMockit.doNothing().when(b).方法名()，让该方法什么也不做。

最后，再讲A实例化。PowerMockit是讲究mock设置顺序的。一定要注意。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。

人生的路无需苛求。只要你迈步，路就在你的脚下延伸。