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