利用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");
    }

测试结果如下: