Mockito 的使用

转自：Mockito 中文文档 ( 2.0.26 beta )

转自：手把手教你 Mockito 的使用

参数匹配器

Argument Matcher(参数匹配器)

Mockito通过equals()方法，来对方法参数进行验证。但是有时候我们需要更加灵活的参数需求，比如，匹配任何的String类型的参数等等。参数匹配器就是一个能够满足这些需求的工具。

Mockito框架中的Matchers类内建了很多参数匹配器，我们常用的Mockito对象就是继承自Matchers。比如anyInt()匹配任何int类型的参数，anyString()匹配任何字符串...

@Test 
public void argumentMatchersTest(){ 
    List<String> mock = mock(List.class); 
    when(mock.get(anyInt())).thenReturn("Hello").thenReturn("World"); 
    String result=mock.get(100)+" "+mock.get(200); 
    verify(mock,times(2)).get(anyInt()); 
    assertEquals("Hello World",result); 
} 

首先mock了List接口，然后用迭代的方式模拟了get方法的返回值，这里用了anyInt()参数匹配器来匹配任何的int类型的参数。所以当第一次调用get方法时输入任意参数为100方法返回”Hello”，第二次调用时输入任意参数200返回值”World”。

这里需要注意：

如果使用了参数匹配器，那么所有的参数需要由匹配器来提供，否则将会报错。假如我们使用参数匹配器stubbing了mock对象的方法，那么在verify的时候也需要使用它。如：

@Test 
public void argumentMatchersTest(){ 
    Map mapMock = mock(Map.class); 
    when(mapMock.put(anyInt(), anyString())).thenReturn("world"); 
    mapMock.put(1, "hello"); 
    verify(mapMock).put(anyInt(), eq("hello")); 
}   

在最后的验证时如果只输入字符串”hello”是会报错的，必须使用Matchers类内建的eq方法。如果将anyInt()换成1进行验证也需要用eq(1)。

自定义匹配器-ArgumentMatcher抽象类

自定义参数匹配器的时候需要继承ArgumentMatcher抽象类，它实现了Hamcrest框架的Matcher接口，定义了describeTo方法，所以我们只需要实现matches方法在其中定义规则即可。
下面自定义的参数匹配器是匹配size大小为2的List：

class IsListOfTwoElements extends ArgumentMatcher<List> { 
    @ 
    public boolean matches(Object list) {  
        return ((List) list).size() == 2;  
    }  
}  
  
@Test  
public void argumentMatchersTest(){  
    List mock = mock(List.class);  
    when(mock.addAll(argThat(new IsListOfTwoElements()))).thenReturn(true);  
       
    mock.addAll(Arrays.asList("one", "two", "three"));  
    verify(mock).addAll(argThat(new IsListOfTwoElements()));  
}

argThat(Matcher<T> matcher)方法用来应用自定义的规则，可以传入任何实现Matcher接口的实现类。上例中在stubbing和verify addAll方法时通过argThat(Matcher<T> matcher)，传入了自定义的参数匹配器IsListOfTwoElements用来匹配size大小为2的List。因为例子中传入List的元素为三个，所以测试将失败。

较复杂的参数匹配将会降低测试代码的可读性。有时实现参数对象的equals()方法是个不错的选择（Mockito默认使用equals()方法进行参数匹配），它可以使测试代码更为整洁。另外，有些场景使用参数捕获器（ArgumentCaptor）要比自定义参数匹配器更加合适。 

如何捕获 mock 方法的调用参数

Mockito以java代码风格的形式来验证参数值 : 即通过使用equals()函数。这也是我们推荐用于参数匹配的方式，因为这样会使得测试代码更简单、简洁。在某些情况下，当验证交互之后要检测真实的参数值时这将变得有用。例如 ：

@Test
public void captureNonGenericArgument() {
  UserDao userDao = Mockito.mock(UserDao.class);
  UserService  userService = new UserService(userDao);
 
  userService.saveUser(new User(null, "Yanbin"));
  
  ArgumentCaptor<User> argumentCaptor = ArgumentCaptor.forClass(User.class);
  verify(userDao, times(1)).save(argumentCaptor.capture());
 
   assertEquals("Yanbin", argumentCaptor.getValue().name);
   assertEquals("Chicago", argumentCator.getValue().city); //可断言捕获参数的更多特征

从面对被捕获参数 argumentCaptor.getValue() 的断言可看出它比 argThat() 的优势，argThat() 无法告诉我们不匹配的细节

警告 : 我们建议使用没有测试桩的ArgumentCaptor来验证，因为使用含有测试桩的ArgumentCaptor会降低测试代码的可读性，因为captor是在断言代码块之外创建的。另一个好处是它可以降低本地化的缺点，因为如果测试桩函数没有被调用，那么参数就不会被捕获。总之，ArgumentCaptor与自定义的参数匹配器相关(可以查看ArgumentMatcher类的文档 )。这两种技术都能用于检测外部传递到Mock对象的参数。然而，使用ArgumentCaptor在以下的情况下更合适 :

• 自定义不能被重用的参数匹配器
• 你仅需要断言参数值

我们同样可以在打桩的时候捕获参数，如

ArgumentCaptor<User> argumentCaptor = argumentCaptor.forClass(User.class);
when(userDao.findUserLike(argumentCaptor.capture)).thenReturn(Mockito.mock(User.class));
 
assertEquals("Yanbin", argumentCaptor.getValue().name);

 不能以这种方式在打桩的时候捕获参数：

when(userDao.findUserLike(argumentCaptor.capture)).thenReturn(getUser(argumentCaptor.getValue()));

否则会报错：

出错位置在打桩的地方。记住打桩并不等于异步调用，它返回的是个固定值！