最近用Mockito寫了好多測(cè)試,之所以選中Mockito,而不是EasyMock,或者JMock,是因?yàn)槲铱戳斯俜降膶?duì)比文檔:Mockito對(duì)比EasyMock,從文檔中就可以看出來(lái),Mockito的語(yǔ)法很自然,而EashMock之類的什么record,play模型顯得有點(diǎn)多余。Mockito的所有功能都在Mockito這個(gè)類中,里面的函數(shù)按功能可分為幾類: 大多數(shù)函數(shù)都很直觀,看名字就能知道它的功能,下面是一些稍微復(fù)雜的用法: 1。 如何匹配變長(zhǎng)參數(shù)?
public interface OrderDao { public void saveOrders(Order ...order); } public class OrderServiceImpl { OrderDao orderDao; public void addOrders(Order ...o) { orderDao.saveOrders(o); } } 變長(zhǎng)參數(shù)其實(shí)就是一個(gè)參數(shù)數(shù)組,所以可以用any(類[].class)類匹配它,像這樣
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本來(lái)有個(gè)anyVararg就是作這個(gè)的,不過(guò)我用了發(fā)現(xiàn)不行,有點(diǎn)奇怪。
2. Mock和Spy的區(qū)別 public class TestServiceImpl { public int getOrderCounts() { return 10; } } @Test public void MockVsSpy() { TestServiceImpl service = mock(TestServiceImpl.class); //輸出0,因?yàn)樵摵瘮?shù)被Mockito改寫了 System.out.println("Order counts of mock object" + service.getOrderCounts()); when(service.getOrderCounts()).thenReturn(2); //輸出2, 因?yàn)槲覀兘o這個(gè)函數(shù)打了樁 System.out.println("Order counts of mock object AFTER stubs " + service.getOrderCounts()); service = new TestServiceImpl(); service = spy(service); //輸出10, 因?yàn)镸ockito spy 不會(huì)改寫已有的函數(shù) System.out.println("Order counts of spy object" + service.getOrderCounts()); when(service.getOrderCounts()).thenReturn(2); //輸出2, 因?yàn)槲覀兘o這個(gè)函數(shù)打了樁 System.out.println("Order counts of spy object AFTER stubs " + service.getOrderCounts()); } 3. 如何寫自定義的參數(shù)匹配器 看下面的例子 public class Account { private String name; private String adddress; public Account(String name, String address) { this.name = name; this.adddress = address; } ...get/set 函數(shù) } 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")); } } 上面的例子會(huì)失敗,因?yàn)镸ockito在做參數(shù)匹配時(shí)是根據(jù)equals函數(shù)的結(jié)果來(lái)判斷兩個(gè)參數(shù)是不是一樣的。而我們的Account類并沒有對(duì)equals作特殊的實(shí)現(xiàn),所以會(huì)失敗。修正的方法有三個(gè),一個(gè)是改寫Account類的equals函數(shù)。一個(gè)是用Mockito的反射相等匹配,就是把最后一句改成 |
|