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

2019-11-08 03:27:16

字体：大中小

来源：转载

供稿：网友

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;@Servicepublic 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"); }

测试结果如下: