如何在 easymock 中模拟应返回其参数之一的方法？

Question

public Object doSomething(Object o); 我想模拟它。它应该只返回它的参数。我尝试过：

Capture<Object> copyCaptcher = new Capture<Object>();
expect(mock.doSomething(capture(copyCaptcher)))
        .andReturn(copyCatcher.getValue());

但没有成功，我只得到一个 AssertionError 作为 java.lang.AssertionError: Nothing capture Yet 。有什么想法吗？

Answer 1

好吧，最简单的方法是在 IAnswer 实现中使用 Capture...当然，在内联执行此操作时，您必须将其声明为 final。

MyService mock = createMock(MyService.class);

final Capture<ParamAndReturnType> myCapture = new Capture<ParamAndReturnType>();
expect(mock.someMethod(capture(myCapture))).andAnswer(
    new IAnswer<ParamAndReturnType>() {
        @Override
        public ParamAndReturnType answer() throws Throwable {
            return myCapture.getValue();
        }
    }
);
replay(mock)

这可能是最准确的方法，无需依赖某些上下文信息。这每次都对我有用。

Answer 2

我一直在寻找相同的行为，最后写了以下内容：

import org.easymock.EasyMock;
import org.easymock.IAnswer;

/**
 * Enable a Captured argument to be answered to an Expectation.
 * For example, the Factory interface defines the following
 * <pre>
 *  CharSequence encode(final CharSequence data);
 * </pre>
 * For test purpose, we don't need to implement this method, thus it should be mocked.
 * <pre>
 * final Factory factory = mocks.createMock("factory", Factory.class);
 * final ArgumentAnswer<CharSequence> parrot = new ArgumentAnswer<CharSequence>();
 * EasyMock.expect(factory.encode(EasyMock.capture(new Capture<CharSequence>()))).andAnswer(parrot).anyTimes();
 * </pre>
 * Created on 22 juin 2010.
 * @author Remi Fouilloux
 *
 */
public class ArgumentAnswer<T> implements IAnswer<T> {

    private final int argumentOffset;

    public ArgumentAnswer() {
        this(0);
    }

    public ArgumentAnswer(int offset) {
        this.argumentOffset = offset;
    }

    /**
     * {@inheritDoc}
     */
    @SuppressWarnings("unchecked")
    public T answer() throws Throwable {
        final Object[] args = EasyMock.getCurrentArguments();
        if (args.length < (argumentOffset + 1)) {
            throw new IllegalArgumentException("There is no argument at offset " + argumentOffset);
        }
        return (T) args[argumentOffset];
    }

}

我在类的 javadoc 中写了一个快速的“如何做”。

希望这有帮助。

Answer 3

捕获用于测试随后传递给模拟的值。如果你只需要一个模拟来返回一个参数（或从参数计算出的一些值），你只需要实现 IAnswer。

如果您想要一种可重用的方式将参数 X 传回，请参阅“Remi Fouilloux”的实现，但请忽略他在示例中对 Capture 的使用。

如果您只想像“does_the_trick”的示例那样内联它，那么 Capture 在这里是一个转移注意力的事情。这是简化版本：

MyService mock = createMock(MyService.class);
expect(mock.someMethod(anyObject(), anyObject()).andAnswer(
    new IAnswer<ReturnType>() {
        @Override
        public ReturnType answer() throws Throwable {
            // you could do work here to return something different if you needed.
            return (ReturnType) EasyMock.getCurrentArguments()[0]; 
        }
    }
);
replay(mock)

Answer 4

基于@does_the_trick并使用lambdas，您现在可以编写以下内容：

MyService mock = EasyMock.createMock(MyService.class);

final Capture<ParamAndReturnType> myCapture = EasyMock.newCapture();
expect(mock.someMethod(capture(myCapture))).andAnswer(() -> myCapture.getValue());

或者按照@thetoolman的建议不进行捕获

expect(mock.someMethod(capture(myCapture)))
.andAnswer(() -> (ParamAndReturnType)EasyMock.getCurrentArguments()[0]);

Answer 5

嗯，如果我正确理解你的问题，我认为你可能过于复杂化了。

<代码>
对象 someObject = .... ;
Expect(mock.doSomething(someObject)).andReturn(someObject);
应该

可以正常工作。请记住，您正在提供预期的参数和返回值。因此在两个作品中使用相同的对象。