利用Mockito进行单元测试[0]

Mockito简介

略

Mockito主要功能

利用Mockito可以对需要调用外部接口或需要查询数据库的方法进行单元测试，可以根据需要设定外部接口(或数据库)的返回。

重点是：测试代码的可读性高，写起来。。。比较方便。

例程

什么都不如一段代码能介绍的东西多(对于我这种语文不好的人而言),以下是一个最简单的Mockito的例程。

下面这段代码实现了调用百度的一个公共API：

package cn.webank.cnc.afs.test;

/**
* Created by narcli on 2016/12/8.
*/

import org.springframework.stereotype.Service;

import java.io.BufferedReader;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

@Service
public class WeatherService {
public String apikey = "36f74c013ead3bb1dbc0e29349ead738";

public String getWeather(String city) {
String httpUrl = "http://apis.baidu.com/apistore/weatherservice/weather";
String httpArg = "citypinyin=" + city;
return unicodeToString(request(httpUrl, httpArg));
}

public String request(String httpUrl, String httpArg) {
BufferedReader reader = null;
String result = null;
StringBuffer sbf = new StringBuffer();
httpUrl = httpUrl + "?" + httpArg;
try {
URL url = new URL(httpUrl);
HttpURLConnection connection = (HttpURLConnection) url.openConnection();
connection.setRequestMethod("GET");
connection.setRequestProperty("apikey", apikey);
connection.connect();
InputStream is = connection.getInputStream();
reader = new BufferedReader(new InputStreamReader(is, "UTF-8"));
String strRead = null;
while ((strRead = reader.readLine()) != null) {
sbf.append(strRead);
sbf.append("\r\n");
}
reader.close();
result = sbf.toString();
} catch (Exception e) {
e.printStackTrace();
}
return result;
}

public String unicodeToString(String str) {
Pattern pattern = Pattern.compile("(\\\\u(\\p{XDigit}{4}))");
Matcher matcher = pattern.matcher(str);
char ch;
while (matcher.find()) {
ch = (char) Integer.parseInt(matcher.group(2), 16);
str = str.replace(matcher.group(1), ch + "");
}
return str;
}
}

[基本都是复制粘贴的]

这段代码可以实现去调用百度的接口，查询一个城市的天气状况。

很显然，在办公机可以顺利运行的这段代码，放到开发机上就只能等报错了。

使用Mockito，就可以“模拟”外部接口的返回，在开发机上实现对这个接口的测试！

首先，你需要一个依赖：

<dependency>
<groupId>org.mockito</groupId>
<artifactId>mockito-all</artifactId>
<version>1.9.5</version>
</dependency>

然后，你需要利用注解搞到一个”假的”WeatherService：

@Mock
private WeatherService mockService;

然后在测试之前告诉编译器，这是假的。[在测试前初始化]

@Before
public void myBefore() {
MockitoAnnotations.initMocks(this);
}

之后就可以“打桩”设置返回，然后模拟外部接口调用了：

@Test
public void TestWeacherBeiJing() {
Mockito.when(mockService.getWeather("beijing")).thenReturn("test");
String a = mockService.getWeather("beijing");
System.out.println(a);
Assert.assertEquals(a, "test");
}

控制台输出如下:

test

如果需要固定这个接口的返回，去测试别的功能，还可以按下面的方式做：

@Before
public void myBefore() {
MockitoAnnotations.initMocks(this);
Mockito.when(mockService.getWeather(anyString())).thenReturn("test");
}
@Test
public void TestWeacherBeiJing() {
String a = mockService.getWeather("beijing");
System.out.println(a);
Assert.assertEquals(a, "test");
}
@Test
public void TestWeacherShenZhen() {
String a = mockService.getWeather("shenzhen");
System.out.println(a);
Assert.assertEquals(a, "test");
}
@Test
public void TestWeacherGuangZhou() {
String a = mockService.getWeather("guangzhou");
System.out.println(a);
Assert.assertEquals(a, "test");
}

测试结果
b792
如下: