@Test
public void testFindTasksWithRelocationDataFetchTimeout() {

    when(relocationServiceClient.findTaskRelocationPlans(asSet(TASK1.getId(), TASK2.getId()))).thenReturn(Mono.never());
    TaskQueryResult queryResult = TaskQueryResult.newBuilder()
            .addItems(TASK1)
            .addItems(TASK2)
            .build();
    ExtTestSubscriber<TaskQueryResult> testSubscriber = new ExtTestSubscriber<>();
    taskRelocationDataInjector.injectIntoTaskQueryResult(Observable.just(queryResult)).subscribe(testSubscriber);
    testSubscriber.assertOpen();
    testScheduler.advanceTimeBy(RELOCATION_TIMEOUT_MS, TimeUnit.MILLISECONDS);
    testSubscriber.assertOnCompleted();
    TaskQueryResult result = testSubscriber.takeNext();
    assertThat(result.getItemsList()).contains(TASK1, TASK2);
}