java - 结果返回模拟方法的参数

最简单的例子:我有以下方法:

public String testMethod(String arg){
    .....
}

我想模拟这个方法来返回传递的参数。例如:

testMethod("aString") returns "aString"
testMethod("anotherString") returns "anotherString"

我知道我可以对这种行为进行硬编码，但我希望它是通用的。

最佳答案

您可以编写自己的 Answer :

when(mock.testMethod(anyString())).thenAnswer(new Answer<String>() {
    @Override
    public String answer(InvocationOnMock invocation) {
        return invocation.getArgumentAt(0, String.class);
    }
});

或者:

when(mock.testMethod(anyString()))
    .thenAnswer(AdditionalAnswers.<String>returnsFirstArg());

关于java - 结果返回模拟方法的参数，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/31012664/