EasyMock:获取EasyMock.anyObject()的真实参数值?

时间:2014-01-08 20:25:50

标签: java unit-testing easymock

在我的单元测试中,我使用EasyMock来创建模拟对象。 在我的测试代码中,我有类似的东西

EasyMock.expect(mockObject.someMethod(anyObject())).andReturn(1.5);

因此,现在EasyMock将接受对someMethod()的任何调用。有没有办法获得传递给mockObject.someMethod()的实际值,或者我需要为所有可能的情况编写EasyMock.expect()语句?

1 个答案:

答案 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];
        }
    }
);