Mockito: как ожидать, что итератор будет равен?

Question

У меня есть метод, который принимать строку и итератор:

public int doSomething(String str, Iterator<String> itr) 

Я пытаюсь издеваться класс, где этот метод является, поэтому он будет возвращать мне целое число, в зависимости от переданных аргументов. Как я могу назвать when() Mockito, поэтому он ожидает, что прошедший итератор будет «равен» тому, который я укажу? Как передается итератор построить где-то динамически в системе, я не могу использовать тот же экземпляр, в when(), так что я могу только сделать копию этого как мои ожидания:

List<String> aList = new ArrayList<String>(); 
aList.add("one"); 
aList.add("two"); 

MyClass myMock = Mockito.mock(MyClass.class); 

Я попытался следующие, и ни один из им кажется, работает:

Mockito.when(myMock.doSomething("some string", aList.iterator())).thenReturn(10); 
Mockito.when(myMock.doSomething(Matchers.eq("some string"), Matchers.eq(aList.iterator()))).thenReturn(10); 

Я только удалось с помощью anyObject():

Mockito.when(myMock.doSomething(Matchers.eq("some string"), Matchers.anyObject())).thenReturn(10); 

, но тогда, конечно, я не могу установить разные результаты, отд заканчивающийся на том, что есть в итераторе ...

Answer 1

Вы можете использовать

when(myMock.doSomething(eq("some string"), argThat(new IsIteratorOfList(aList)))) 

где IsIteratorOfList является подклассом ArgumentMatcher, который проверяет, когда данные iterator'selements все добавлены в новый список, этот новый список составляет список, переданный в конструктор (aList в этом случае).

Соединитель может быть еще проще и просто проверить первый элемент или количество элементов или все, что вы хотите.

Answer 2

Вы всегда можете сделать свою собственную реализацию интерфейса Iterator.

Answer 3

Я внедрил IterableMatcher<T>. Это включает статический метод elemEq, который будет создавать один из них для вас.

Например:

doReturn(result).when(mock).method(elemEq(expected)); 

https://gist.github.com/collinsauve/1fcf924cdcced40e0bee