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