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。 如何匹配变长参数？

1 2 3 4 5 6 7 8 9 10 11 12 13

public interface OrderDao { public void saveOrders ( Order . . . order ) ; } public class OrderServiceImpl {      OrderDao orderDao ;           public void addOrders ( Order . . . o )      {          orderDao . saveOrders ( o ) ;      } }

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14

public class OrderServiceImplTest {      @Test public void addOrder ( ) {      OrderDao dao = mock ( OrderDao . class ) ;      OrderServiceImpl orderService = new OrderServiceImpl ( dao ) ;      Order o = new Order ( 1 ) ;           //when(dao.)      orderService . addOrders ( o ) ;      verify ( dao , times ( 1 ) ) . saveOrders ( any ( Order [ ] . class ) ) ; } }

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

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25

public class TestServiceImpl { public int getOrderCounts ( ) {      return 10 ; } } @Test public void MockVsSpy ( ) {      TestServiceImpl service = mock ( TestServiceImpl . class ) ; //输出0，因为该函数被Mockito改写了      System . out . println ( "Order counts of mock object" + service . getOrderCounts ( ) ) ;      when ( service . getOrderCounts ( ) ) . thenReturn ( 2 ) ; //输出2， 因为我们给这个函数打了桩      System . out . println ( "Order counts of mock object AFTER stubs " + service . getOrderCounts ( ) ) ;           service = new TestServiceImpl ( ) ;      service = spy ( service ) ; //输出10， 因为Mockito spy 不会改写已有的函数      System . out . println ( "Order counts of spy object" + service . getOrderCounts ( ) ) ;      when ( service . getOrderCounts ( ) ) . thenReturn ( 2 ) ; //输出2， 因为我们给这个函数打了桩      System . out . println ( "Order counts of spy object AFTER stubs " + service . getOrderCounts ( ) ) ; }

3. 如何写自定义的参数匹配器
看下面的例子

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46

public class Account { private String      name ;      private String      adddress ;      public Account ( String name , String address ) {      this . name = name ;      this . adddress = address ; } . . . get / set 函数 } public interface AccountDao { public void addAccount ( Account a ) ; } public class AccountServiceImpl { AccountDao dao ; public AccountServiceImpl ( AccountDao dao ) {      this . dao = dao ; } public void addAccount ( String name , String address ) {      dao . addAccount ( new Account ( name , address ) ) ; } } public class AccountServiceImplTest {      @Test public void addAccount ( ) {      AccountDao dao = mock ( AccountDao . class ) ;      AccountServiceImpl service = new AccountServiceImpl ( dao ) ;           service . addAccount ( "obama" , "white house" ) ;           verify ( dao ) . addAccount ( new Account ( "obama" , "white house" ) ) ; } }

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

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

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

1 2 3 4 5 6 7 8 9

verify ( dao ) . addAccount ( argThat ( new ArgumentMatcher <Account> ( )          {              @Override              public boolean matches ( Object argument )              {                  Account person = ( Account ) argument ;                  return person . getName ( ) . equals ( "obama" ) && person . getAddress ( ) . equals ( "white house" ) ? true : false ;              }          } ) ) ;

This entry was posted in  Uncategorized by  linmingren. Bookmark the  permalink.