我在用 mock 编写单元测试时遇到了问题。有一个我需要模拟的对象有很多 getter,我确实在代码中调用它们。但是,这些不是我的单元测试的目的。那么,有没有一种方法可以模拟所有方法,而不是一个一个地模拟它们。
这是代码示例:
public class ObjectNeedToMock{
private String field1;
...
private String field20;
private int theImportantInt;
public String getField1(){return this.field1;}
...
public String getField20(){return this.field20;}
public int getTheImportantInt(){return this.theImportantInt;}
}
这是我需要测试的服务类
public class Service{
public void methodNeedToTest(ObjectNeedToMock objectNeedToMock){
String stringThatIdontCare1 = objectNeedToMock.getField1();
...
String stringThatIdontCare20 = objectNeedToMock.getField20();
// do something with the field1 to field20
int veryImportantInt = objectNeedToMock.getTheImportantInt();
// do something with the veryImportantInt
}
}
在测试类中,测试方法就像
@Test
public void testMethodNeedToTest() throws Exception {
ObjectNeedToMock o = mock(ObjectNeedToMock.class);
when(o.getField1()).thenReturn(anyString());
....
when(o.getField20()).thenReturn(anyString());
when(o.getTheImportantInt()).thenReturn("1"); //This "1" is the only thing I care
}
那么,有没有一种方法可以避免将无用的“field1”的所有“when”都写到“field20”
最佳答案
您可以控制模拟的默认答案。创建模拟时,请使用:
Mockito.mock(ObjectNeedToMock.class, new Answer() {
@Override
public Object answer(InvocationOnMock invocation) throws Throwable {
/*
Put your default answer logic here.
It should be based on type of arguments you consume and the type of arguments you return.
i.e.
*/
if (String.class.equals(invocation.getMethod().getReturnType())) {
return "This is my default answer for all methods that returns string";
} else {
return RETURNS_DEFAULTS.answer(invocation);
}
}
}));
关于java - Mockito 模拟所有方法调用和返回,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/26154657/