循环中的EasyMock期望

时间:2013-04-30 19:04:56

标签: java junit easymock

是否可以为循环中调用的模拟设置不同的期望?

考虑用例,

while(condition){
   List list = Database call //Call to be mocked
   ....
   Logic based on above response
}

我们如何设置期望,例如第一次迭代返回一个对象列表,而第二次调用返回空列表?

1 个答案:

答案 0 :(得分:2)

试试这个:

expect(mockDatabase.call()).andReturn(object1).once();
expect(mockDatabase.call()).andReturn(object2).once();
replay(mockDatabase);