Mockito入门及配合Junit进行单元测试

一、背景

最近一段时间在给写的框架补单元测试。发现有一些方法做单元测试不好做，主要是其内部依赖了许多外部环境，如Zookeeper、DB、Http请求等等。
如果为了这些单元测试单独去搞一套环境，很是费事。再者代码每天在Jenkins上跑单元测试，环境不一定相通。

如果您面临上述相同的问题，可以继续向下看。

二、一个例子

2.1 需要做单元测试的代码

假设我们有下面这样一个类，其中的isAdmin方法主要是从redis中获取数据，然后进行对比判断。现实中，这里可能有很多复杂的逻辑，这里只做抛砖引玉。

public class RedisDemo {    private Jedis jedis;    public void setUp() {        jedis = new Jedis("127.0.0.1", 6379);        jedis.connect();    }    public boolean isAdmin(String user) {        String ret = jedis.get("name");        if (ret.equals(user)) {            return true;        }        return false;    }    public static void main(String[] args) {        RedisDemo demo = new RedisDemo();        demo.setUp("127.0.0.1", 6379);        boolean isAdmin = demo.isAdmin("test");        System.out.println(isAdmin);    }}

2.2 单元测试

@Test    public void test() {        RedisDemo redisDemo = new RedisDemo();        redisDemo.setUp("127.0.0.1", 6379);        Assert.assertSame(true, redisDemo.isAdmin("admin"));    }

我们通常使用上述的代码对其进行单元测试，看似简单。但是需要我们搭建一套redis环境，并且初始化数据。如果只有redis还好，假如项目中依赖的外部环境有ZK、redis、MySQL、http还有其他部门的一些接口，在这样做的话，其他不说，光维护这一套环境就很麻烦了。

那应该怎么简化操作呢？继续向下看。

三、使用Mockito进行测试

3.1 引入依赖

    <dependencies>        <dependency>            <groupId>org.mockito</groupId>            <artifactId>mockito-all</artifactId>            <version>1.8.5</version>            <scope>test</scope>        </dependency>        <dependency>            <groupId>junit</groupId>            <artifactId>junit</artifactId>            <version>4.12</version>            <scope>test</scope>        </dependency>    </dependencies>

3.2 编写单元测试

public class RedisTest {    @Rule    public ExpectedException exception = ExpectedException.none(); // 不允许有异常    @Test    public void mockTest() throws Exception {        Jedis mockJedis = Mockito.mock(Jedis.class);        Mockito.when(mockJedis.get("name")).thenReturn("admin");        RedisDemo redisDemo = new RedisDemo();        Field daoField = redisDemo.getClass().getDeclaredField("jedis");        daoField.setAccessible(true);        daoField.set(redisDemo, mockJedis);        Assert.assertSame(true, redisDemo.isAdmin("admin"));    }}

这里使用了Mockito.mock()这个API进行Mock，当程序调用jedis的方法时，并不会真正触发jedis的方法，而是返回一个设定的值。

四、总结

本文只是对这个模块构造很复杂的测试，分享了一个工具。当然，Mockito的功能不仅仅如此，更详细的文档请参考：
https://static.javadoc.io/org.mockito/mockito-core/2.11.0/org/mockito/Mockito.html