Поток уже прооперирован или закрыт в Мокито?

Question

Я хочу протестировать метод возврата потока.

class Foo {
    public void getClicks(boolean somecondition) {
        Stream s1 = computeAndGetAllClicks(new Date());
        if (somecondition) {
            Stream s2 = computeAndGetAllClicks(new Date());
        }
        return Stream.concat(s1, s2);
    }

    Stream computeAndGetAllClicks(Date d) {
        // return stream
    }
}

Теперь я написал следующий тест для случая, когда somecondition = true

    final ArgumentCaptor<Date> argumentCaptorTodayDate = ArgumentCaptor.forClass(Date.class);

    doReturn(Arrays.asList(new Click.Builder().build()).stream())
    .when(fooInstance)
    .addAtlasLineItem(computeAndGetAllClicks(argumentCaptorTodayDate.capture))

   final ArgumentCaptor<Date> argumentCaptorEndDate = ArgumentCaptor.forClass(Date.class);


    doReturn(Arrays.asList(new Click.Builder().build()).stream())
    .when(fooInstance)
    .addAtlasLineItem(computeAndGetAllClicks(argumentCaptorEndDate.capture))

    fooInstance.getClicks(true);

Однако я получаю исключение:java.lang.IllegalStateException: поток уже был обработан или закрыт.

Как мне обойти это ??

Answer 1

вы можете использовать thenAnswer и создавать новый поток каждый раз, когда он вызывается:

doAnswer(invocationOnMock->Arrays.asList(new Click.Builder().build()).stream())