测试框架Mockito使用笔记

Mockito，测试框架，语法简单，功能强大！
静态、私有、构造等方法测试需要配合PowerMock，PowerMock有Mockito和EasyMock两个版本，语法相同，本文只介绍Mockito.

测试对像：ArticleManager

ArticleManager 类

public class ArticleManager {

    void setDatabase(ArticleDatabase database) {... }

    String get(String val){

        database.get(val);

    }

    String add(String val){

        database.set(val);

    }

}

ArticleDatabase 类

public class ArticleDatabase {

    String get(String val){...}

    String add(String val){...}

}

注解

@Mock

创建Mock对像

@Mock

private ArticleDatabase ds;

实例ds是一个虚拟对像，ds对像本身的所有方法返回值都为Null。

@InjectMocks

@InjectMocks

private ArticleManager articleManager;

将现有Mock对像(ds)注入到articleManager，支持方法包括constructor, setter， property 和 Spring的@Autowired !

@Captor

参数调用截取器

//定义@Captor，如果要获取多个参数, 需要定义多个@Captor

@Captor

private ArgumentCaptor<String> argumentString;

articleManager.add("content");

verify(ds, times(1)).add(argumentString.capture());

//打印参数值

logger.debug(argumentString.getValue());

@Spy

Mock类的部分真实或定制后调用。没有定制的方法就调用真实的方法。

@Spy

private ArticleManager articleManager2;

//定制方法，这个地方用`when`还会调用真实的方法后返回定制结果

//见：http://docs.mockito.googlecode.com/hg/latest/org/mockito/Spy.html

doReturn("0").when(articleManager2).get(0);

articleManager2.get(0);

verify(articleManager2).get(0);

//调用真实的方法

articleManager2.get(1)

verify(articleManager2).get(1);

@Spy真实调用也可以用@Mock对像thenCallRealMethod方法调用：

when(articleManager2.get(2)).thenCallRealMethod();

Stubbing

特别制定Mock对像某方法的返回值、类型或异常等，默认是Null，

when(ds.get(0)).thenReturn("first");

when(ds.get(1)).thenThrow(new RuntimeException());

verify

对某方法返回结果的校验，包括调用次数，返回值等

articleManager.add("one");

verify(ds).add("one");

参考：@Captor

模糊匹配

在Stubbing或verify里可用模糊匹配来适配多种情况的调用，Mockito在包org.mockito.Matchers下提供有anyString(), anyInt()或者直接anyObject()等大量模糊匹配函数，如果有需要还可自己写，实现(org.hamcrest.Matcher)。例：

when(articleManager.get(anyString())).toReturn("element");

System.out.println(articleManager.get("1"));

verify(articleManager).get(anyString());

在使用校验函数verify时，参数如果有模糊匹配则所有参数都要用模糊匹配，如：

//正确的写法：

verify(mock).articleManager(anyInt(), anyString(), eq("third argument"));

//错误用法：

verify(mock).articleManager(anyInt(), anyString(), "third argument");

更多模糊匹配函数请参考：http://mockito.googlecode.com/svn/branches/1.8.5/javadoc/org/mockito/Matchers.html

参考地址：

mockito home
mockito docs