在我的单元测试中,我使用EasyMock来创建模拟对象。 在我的测试代码中,我有类似的东西
EasyMock.expect(mockObject.someMethod(anyObject())).andReturn(1.5);
因此,现在EasyMock将接受对someMethod()
的任何调用。有没有办法获得传递给mockObject.someMethod()
的实际值,或者我需要为所有可能的情况编写EasyMock.expect()
语句?
答案 0 :(得分:24)
您可以使用Capture
类来预期和捕获参数值:
Capture capturedArgument = new Capture();
EasyMock.expect(mockObject.someMethod(EasyMock.capture(capturedArgument)).andReturn(1.5);
Assert.assertEquals(expectedValue, capturedArgument.getValue());
请注意,Capture
是泛型类型,您可以使用参数类对其进行参数化:
Capture<Integer> integerArgument = new Capture<Integer>();
更新
如果您想在expect
定义中为不同参数返回不同的值,可以使用andAnswer
方法:
EasyMock.expect(mockObject.someMethod(EasyMock.capture(integerArgument)).andAnswer(
new IAnswer<Integer>() {
@Override
public Integer answer() {
return integerArgument.getValue(); // captured value if available at this point
}
}
);
正如评论中所指出的,另一种选择是在getCurrentArguments()
内使用answer
来电:
EasyMock.expect(mockObject.someMethod(anyObject()).andAnswer(
new IAnswer<Integer>() {
@Override
public Integer answer() {
return (Integer) EasyMock.getCurrentArguments()[0];
}
}
);