如何使用Mockito正确模拟每个循环？

Question

我想测试我的方法，我需要模拟项目和行的向量来遍历它们中的每一个。

我准备了一个结果向量，当我从Item类（在测试中）调用getRows（）时，我总是希望得到它。

问题是测试中getRows方法的结果总是空的，所以这意味着doReturn语句在这里不能正常工作。

我做错了什么吗？

class MyServiceTest {
@Mock
private Item mockedItem;

@Mock
private Row mockedRow;

@InjectMocks
@Spy
private MyService myService;

@Test
public void testMyMethod() {
    Vector<Row> rows = new Vector<Row>();
    Row row = new Row();
    rows.add(row);

    doReturn(rows).when(mockedItem).getRows();

    ...
    //some code to test the service

    }
}

class MyService {
    protected SomeObject myMethod(Vector<Item> items) {     
        for(Item item : items) {
            for(Row row : item.getRows()) {
               //cannot get here while debugging the test

               //some code
            }
        }
    ...
    }
}
class Item {
...
    public Vector<Row> getRows(){
       //return some vector of rows
    }
}

Answer 1

< code>@InjectMock是无用的，因为被模仿的向量是被测试方法的一个参数，而不是被测试类的一个字段。< br >所以您不需要注入任何东西。< br >

您需要做的是创建一个Item的< code>Vector，其中每个元素都是一个mock，具有< code>getRows()的预期行为:返回存根行。< br >

下面是一个示例代码:

// create stubbed rows
Vector<Row> rows = new Vector<Row>();
Row row = new Row();
rows.add(row);

// mock items Vector
Vector<Item> items = new Vector<>();
for (...){ // nb element to create
  Item mockedItem = Mockito.mock(Item.class);
  doReturn(rows).when(mockedItem).getRows();
  items.add(mockedItem);
}

然后通过传递向量来调用测试的方法：

new MyService().myMethod(items);

请注意，Vector是一个旧的、有缺陷的集合
如果您不需要处理竞争条件，则将List和ArrayList作为实现