模拟方法不返回预期结果

Question

我想mock接受对象参数的方法mockedMethod(Object Object)。此对象参数是从TestedMethod动态创建的。MockedMethod不会返回预期的结果，尽管这两个对象具有相同的数据，但只是不同的哈希代码。

public void testedMethod() {
    Object object = initObject(); // this object is initialized dynamically
    Long result = mockedMethod(object);
    ...
}

@Test
public void test() {
    Object object = new Object();
    when(api.mockedMethod(object).thenReturn(3L);
    testedMethod();
}

有没有什么方法来模拟方法，使它能够以预期的结果响应？

Answer 1

似乎你有两个选择：

您可以使用any()或notnull()，正如geoand在注释中提到的那样。它们定义任何参数(any)或任何非空参数(Notnull)的行为。

when(api.mockedMethod(Mockito.any()).thenReturn(3L);
// or
when(api.mockedMethod(Mockito.notNull()).thenReturn(3L);

这些静态方法返回参数匹配器，并且可以在从匹配器对象继承的Mockito对象上找到（以保持API之间的一点点分离）。

如果需要检查对象是否匹配某些自定义条件，可以使用自定义ArgumentMatcher对象。这些对象（也称为Hamcrest匹配器，因为它们直接来自Hamcrest库）只有一个布尔方法，如果对象匹配条件，则返回true。

/** Returns true if the passed object has FooBar anywhere in its class name. */
class HasFooBarInClassName extends ArgumentMatcher<Object> {
  @Override public boolean matches(Object object) {
    return object.getClass().getName().contains("FooBar");
  }
}

when(api.mockedMethod(argThat(new HasFooBarInClassName())).thenReturn(3L);

有时，您的代码需要“捕获”对象以进行额外的验证或处理，这是Mockito无法单独完成的。您可以使用argumentcaptor来帮助您：

ArgumentCaptor<Object> captor = ArgumentCaptor.for(Object.class);
when(api.mockedMethod(any()).thenReturn(3L);
testedMethod();
verify(api).mockedMethod(captor.capture());

// Now you can check the object, as needed, using your own methods.
checkObjectAsNeeded(captor.getValue());