Return 模拟方法的参数作为结果

Question

最简单的example:I有以下方法：

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

我想将此方法模拟为 return 作为结果传递的参数。例如：

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

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

Answer 1

这样说：

public String testMethod(String arg){
    return arg;
}

Answer 2

通用-通用？您可以使用以下

public <T> T testMethod(T arg) {
    return arg;
}

然后你可以将它与任何类型一起使用，

String s = testMethod("someString");
Integer i = testMethod(0);

Answer 3

你可以自己写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());

Answer 4

您可以实现自己的 Answer:

Answer<String> returnArgAnswer = new Answer<String>() {
    public String answer(InvocationOnMock invocationOnMock) throws Throwable {
        return (String) invocationOnMock.getArguments()[0];
    }
};

SomeClass x = mock(SomeClass.class);
Mockito.when(x.testMethod(anyString())).thenAnswer(returnArgAnswer);

然后，使用任何字符串调用 x.testMethod 将 return 该字符串。

Answer 5

这已经内置于 Mockito 1.9.5 及更高版本中，如 returnsFirstArg。

when(mock.testMethod(anyString())).thenAnswer(returnsFirstArg());

您可能希望避免直接使用 ClonesArguments，因为它在内部包 org.mockito.internal.stubbing.answers 中。 Mockito 的静态工厂类是获取 Mockito 的预定义答案和匹配器的首选方式。

Return 模拟方法的参数作为结果

Return argument of mocked method as a result

java

junit

mockito