在編寫單元測試時 ,必須牢記不要依賴外部組件。 為了避免這種情況,我們使用了模擬框架,對我來說,最容易使用的是Mockito 。
在本文中,我們將看到在Mockito中使用的一種“高級”技術,可以使用Answer接口在模擬方法上返回相同的參數實例。
在本文中,我們將看到在Mockito中使用的一種“高級”技術,可以使用Answer接口在模擬方法上返回相同的參數實例。
假設我們正在為管理Person和Job類的類編寫單元測試,并且在操作中它使用DAO類在Person和Job之間插入關系類(M:N)稱為PersonJob 。
例如,正在測試的類如下所示:
public PersonJob createPersonJob(Person person, Job job) {.. some job .. PersonJob personJob = new PersonJob(person, job);return this.personJobDao.create(personJob);}
因此,在這種情況下,您似乎需要嘲笑personJobDao 。
讓我們創建模擬并記錄交互:
public class WhenAJobIsAssignedToPerson {@Testpublic void relationship_should_be_made_persistent() {PersonJobDao personJobDao = mock(PersonJobDao.class);when(personJobDao.create(any(PersonJob.class))).thenReturn(???)PersonJobManager personJobManager = new PersonJobManager();personJobManager.setPersonJobDao(personJobDao);Person person = new Person();Job job = new Job();PersonJob personJob = personJobManager.createPersonJob(person, job);assertThat(personJob.getPerson(), is(person));assertThat(personJob.getJob(), is(job));}}
是的,您可以看到您不知道要返回什么,因為實例是由被測類創建的,并且在測試方法中,您也不知道哪個實例是由createPersonJob方法創建的。 要解決此問題,您需要使用thenAnswer而不是thenReturn方法:
public class WhenAJobIsAssignedToPerson {@Testpublic void relationship_should_be_made_persistent() {PersonJobDao personJobDao = mock(PersonJobDao.class);when(personJobDao.create(any(PersonJob.class))).thenAnswer(new Answer<PersonJob>() {public PersonJob answer(InvocationOnMock invocation)throws Throwable {return (PersonJob) invocation.getArguments()[0];}});PersonJobManager personJobManager = new PersonJobManager();personJobManager.setPersonJobDao(personJobDao);Person person = new Person();Job job = new Job();PersonJob personJob = personJobManager.createPersonJob(person, job);assertThat(personJob.getPerson(), is(person));assertThat(personJob.getJob(), is(job));}}
需要注意的是回答接口需要您實現答案的方法,這在我們的情況下,簡單地返回personJobDao的第一個參數(PersonJob實例)。 創建方法。
現在,我們可以安心編寫斷言,而不必擔心返回的實例。
參考:在One Jar To Rule Them All博客中與 JCG合作伙伴 Alex Soto的Mockito進行回答 。
翻譯自: https://www.javacodegeeks.com/2012/07/answering-with-mockito.html