@Test
public void testFindTaskWithRelocationDeadline() {

    long deadlineTimestamp = titusRuntime.getClock().wallTime() + 1_000;
    when(relocationServiceClient.findTaskRelocationPlan(TASK1.getId())).thenReturn(
            Mono.just(Optional.of(newRelocationPlan(TASK1, deadlineTimestamp)))
    );
    Task merged = taskRelocationDataInjector.injectIntoTask(TASK1.getId(), Observable.just(TASK1)).toBlocking().first();
    assertThat(merged.getMigrationDetails().getNeedsMigration()).isTrue();
    assertThat(merged.getMigrationDetails().getDeadline()).isEqualTo(deadlineTimestamp);
}