[英]Return argument of mocked method as a result
最简单的例子:我有以下方法:
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());
您可以实现自己的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
将返回该字符串。
它已经在Mockito 1.9.5和更高版本中内置,作为returnsFirstArg
。
when(mock.testMethod(anyString())).thenAnswer(returnsFirstArg());
您可能希望避免直接使用ClonesArguments
,因为它在内部包org.mockito.internal.stubbing.answers
。 Mockito的静态工厂类是获取Mockito的预定义答案和匹配器的首选方法。
这样说:
public String testMethod(String arg){
return arg;
}
泛型? 您可以使用以下内容
public <T> T testMethod(T arg) {
return arg;
}
然后,您可以将其用于任何类型,
String s = testMethod("someString");
Integer i = testMethod(0);
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.