测试框架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