如何使用ArgumentCaptor与Mockito.when()。thenReturn()

时间:2016-12-15 15:11:11

标签: java unit-testing junit arguments mockito

我有一个用例,我希望捕获方法中传递的值,并使用此值来定义Mockito行为。

像这样:

@InjectMocks
private ClassUnderTest classUnderTest;

@Mock
private MockedClass mockedClass;

@Captor
private ArgumentCaptor<ArgumentUsedInMockedClass> captor;

@Test
public void testMethod() {
    Result result = new Result();
    result.setResultId(RESULT_ID);

    Mockito.verify(mockedClass).doSomething(captor.capture());

    Mockito.when(mockedClass.doSomething(captor.getValue())).thenReturn(result);

    Assert.assertTrue(classUnderTest
            .doSomething(foo, bar)
            .equals(result));
}

但是我收到了这个错误:

[junit] Wanted but not invoked:
[junit] mockedClass.doSomething(
[junit]     <Capturing argument>
[junit] ); 
[junit] Actually, there were zero interactions with this mock.

在我的doSomething()函数中,foo和bar用于生成ArgumentUsedInMockedClass类型的参数。现在,这个ArgumentUsedInMockedClass类型没有正确定义equals(),因此如果我试图直接使用

,它就会失败

Mockito.when(mockedClass.doSomething(argumentUsedInMockedClass)).thenReturn(result);在测试方法中生成argumentUsedInMockedClass,即使它的参数相同。我正在尝试捕获在此ClassUnderTest中创建的确切对象来定义mockito行为,但它看起来像一个循环,其中第一个

Assert.assertTrue(classUnderTest
            .doSomething(foo, bar)
            .equals(result));

必须碰巧实际捕获,但为了工作mockedClass应该定义依赖于它的行为。

如何解决这种情况或使用其他测试方法?

1 个答案:

答案 0 :(得分:3)

您收到错误是因为您尝试验证方法是否在实际调用之前被调用,这发生在assertTrue

您可以像这样重写测试:

@Test
public void testMethod() {
    Result result = new Result();
    result.setResultId(RESULT_ID);

    // Define the behavior
    Mockito.when(mockedClass.doSomething(any(ArgumentUsedInMockedClass.class)).thenReturn(result);

    Assert.assertTrue(classUnderTest
            .doSomething(foo, bar)
            .equals(result));
    Mockito.verify(mockedClass).doSomething(captor.capture());
    ArgumentUsedInMockedClass argument = captor.getValue();
    // Do some further verification/assertion with argument
}