Mockito使用

最近用Mockito写了好多测试，之所以选中Mockito，而不是EasyMock，或者JMock，是因为我看了官方的对比文档：Mockito对比EasyMock，从文档中就可以看出来，Mockito的语法很自然，而EashMock之类的什么record，play模型显得有点多余。Mockito的所有功能都在Mockito这个类中，里面的函数按功能可分为几类：
1. 参数匹配 — 就是Matchers里的那些any开头的函数
2. Mock
3. 打桩（stub）功能 — 就是那些do什么开头的函数
4. 验证模型， 包括atLeast，atLeastOnce，atMost，only，times等。

大多数函数都很直观，看名字就能知道它的功能，下面是一些稍微复杂的用法：

1。 如何匹配变长参数？

[java]  view plain copy

1. public interface OrderDao  
2. {  
3.     public void saveOrders(Order ...order);  
4. }  
5. public class OrderServiceImpl   
6. {  
7.     OrderDao orderDao;  
8.       
9.     public void addOrders(Order ...o)  
10.     {  
11.         orderDao.saveOrders(o);  
12.     }  
13. }  

变长参数其实就是一个参数数组，所以可以用any(类[].class)类匹配它，像这样

[java]  view plain copy

1. public class OrderServiceImplTest  
2. {  
3.     @Test  
4.     public void addOrder()  
5.     {  
6.         OrderDao dao = mock(OrderDao.class);  
7.         OrderServiceImpl orderService = new OrderServiceImpl(dao);  
8.         Order o = new Order(1);  
9.           
10.         //when(dao.)  
11.         orderService.addOrders(o);  
12.         verify(dao,times(1)).saveOrders(any(Order[].class));  
13.     }  
14. }  

Mockito本来有个anyVararg就是作这个的，不过我用了发现不行，有点奇怪。

2. Mock和Spy的区别
如果你mock了一个类，那么这个类的所有的函数都被Mockito改写了（如果是没有返回值的函数，则什么都不作，如果是有返回值，会返回默认值，比如布尔型的话返回false，List的话会返回一个空的列表，int的话会返回0等等），如果你Spy了一个类，那么所有的函数都没有被改变，除了那些被你打过桩的函数。看例子：

[java]  view plain copy

1. public class TestServiceImpl  
2. {  
3.     public int getOrderCounts()  
4.     {  
5.         return 10;  
6.     }  
7. }  
8. @Test  
9.     public void MockVsSpy()  
10.     {  
11.         TestServiceImpl service = mock(TestServiceImpl.class);  
12.         //输出0，因为该函数被Mockito改写了  
13.         System.out.println("Order counts of mock object" + service.getOrderCounts());  
14.         when(service.getOrderCounts()).thenReturn(2);  
15.         //输出2， 因为我们给这个函数打了桩  
16.         System.out.println("Order counts of mock object AFTER stubs " + service.getOrderCounts());  
17.           
18.         service = new TestServiceImpl();  
19.         service = spy(service);  
20.         //输出10， 因为Mockito spy 不会改写已有的函数  
21.         System.out.println("Order counts of spy object" + service.getOrderCounts());  
22.         when(service.getOrderCounts()).thenReturn(2);  
23.         //输出2， 因为我们给这个函数打了桩  
24.         System.out.println("Order counts of spy object AFTER stubs " + service.getOrderCounts());  
25.     }  

3. 如何写自定义的参数匹配器

看下面的例子

[java]  view plain copy

1. public class Account  
2. {  
3.     private String    name;  
4.     private String    adddress;  
5.   
6.     public Account(String name, String address)  
7.     {  
8.         this.name = name;  
9.         this.adddress = address;  
10.     }  
11.     ...get/set 函数  
12. }  
13. public interface AccountDao  
14. {  
15.     public void addAccount(Account a);  
16. }  
17.   
18. public class AccountServiceImpl  
19. {  
20.     AccountDao dao;  
21.       
22.     public AccountServiceImpl(AccountDao dao)  
23.     {  
24.         this.dao = dao;  
25.     }  
26.       
27.     public void addAccount(String name, String address)  
28.     {  
29.         dao.addAccount(new Account(name, address));  
30.     }  
31. }  
32.   
33.   
34. public class AccountServiceImplTest  
35. {  
36.     @Test  
37.     public void addAccount()  
38.     {  
39.         AccountDao dao = mock(AccountDao.class);  
40.         AccountServiceImpl service = new AccountServiceImpl(dao);  
41.           
42.         service.addAccount("obama", "white house");  
43.           
44.         verify(dao).addAccount(new Account("obama", "white house"));  
45.     }  
46. }  

上面的例子会失败，因为Mockito在做参数匹配时是根据equals函数的结果来判断两个参数是不是一样的。而我们的Account类并没有对equals作特殊的实现，所以会失败。修正的方法有三个，一个是改写Account类的equals函数。一个是用Mockito的反射相等匹配，就是把最后一句改成

[java]  view plain copy

1. verify(dao).addAccount(refEq(new Account("obama", "white house")));  

最后一种方法是写一个自定义的参数匹配器，如果Account的代码不是你控制的，那么你就只能选这种方法了。这时候最后一句要改成这样：

[java]  view plain copy

1. verify(dao).addAccount(argThat(new ArgumentMatcher<Account>()  
2.         {  
3.             @Override  
4.             public boolean matches(Object argument)  
5.             {  
6.                 Account person = (Account)argument;  
7.                 return person.getName().equal