web-dev-qa-db-ja.com

Mockitoを使用して、メソッドが特定の引数で呼び出されたことを確認するにはどうすればよいですか?

Mockito 1.9.0を使用しています。メソッドが1回だけ呼び出されたこと、および渡されたフィールドの1つに特定の値が含まれていることを確認するにはどうすればよいですか?私のJUnitテストでは、

@Before
public void setupMainProg() { 
    // Initialize m_orderSvc, m_opportunitySvc, m_myprojectOrgSvc
    ...
    m_prog = new ProcessOrdersWorker(m_orderSvc, m_opportunitySvc, m_myprojectOrgSvc);
}   // setupMainProg

@Test
public void testItAll() throws GeneralSecurityException, IOException { 
    m_prog.work();  
}

メソッド「work」は、「m_orderSvc」(オブジェクトに渡される引数の1つ)のメソッドを呼び出します。 「m_orderSvc」には、メンバーフィールド「m_contractsDao」が含まれています。 「m_contractsDao.save」が一度だけ呼び出され、渡された引数に特定の値が含まれていることを確認したい。

これは少しわかりにくいかもしれません。質問を明確にする方法を教えてください。喜んでそうします。

51
Dave

まず、モックm_contractsDaoを作成してセットアップする必要があります。クラスがContractsDaoであると仮定すると:

ContractsDao mock_contractsDao = mock(ContractsDao.class);
when(mock_contractsDao.save(any(String.class))).thenReturn("Some result");

次に、m_orderSvcにモックを挿入し、メソッドを呼び出します。

m_orderSvc.m_contractsDao = mock_contractsDao;
m_prog = new ProcessOrdersWorker(m_orderSvc, m_opportunitySvc, m_myprojectOrgSvc);
m_prog.work(); 

最後に、モックが適切に呼び出されたことを確認します。

verify(mock_contractsDao, times(1)).save("Parameter I'm expecting");
58
mamboking

マンボキングの答えを基に構築:

_ContractsDao mock_contractsDao = mock(ContractsDao.class);
when(mock_contractsDao.save(anyString())).thenReturn("Some result");

m_orderSvc.m_contractsDao = mock_contractsDao;
m_prog = new ProcessOrdersWorker(m_orderSvc, m_opportunitySvc, m_myprojectOrgSvc);
m_prog.work(); 
_

引数が特定の値を含むかどうかを確認するためにリクエストに対処する場合、引数が文字列であり、文字列が引数にサブストリングが含まれています。このためにあなたがすることができます:

_ArgumentCaptor<String> savedCaptor = ArgumentCaptor.forClass(String.class);
verify(mock_contractsDao).save(savedCaptor.capture());
assertTrue(savedCaptor.getValue().contains("substring I want to find");
_

その仮定が間違っていて、save()への引数が何らかのコレクションである場合、それはわずかに異なるだけです:

_ArgumentCaptor<Collection<MyType>> savedCaptor = ArgumentCaptor.forClass(Collection.class);
verify(mock_contractsDao).save(savedCaptor.capture());
assertTrue(savedCaptor.getValue().contains(someMyTypeElementToFindInCollection);
_

Hamcrestマッチャーの使用方法を知っている場合は、 ArgumentMatchers をチェックインすることもできます。

23
Kevin Welker

これはより良い解決策です。

verify(mock_contractsDao, times(1)).save(Mockito.eq("Parameter I'm expecting"));
10
thug-gamer