@Test
public void testMigrationOfTasksWithPolicy() {

    Job<BatchJobExt> job = TestDataFactory.newBatchJob("job1", 1, newSelfManagedDisruptionBudget(1_000));
    relocationConnectorStubs.addJob(job);
    relocationConnectorStubs.place(TestDataFactory.REMOVABLE_INSTANCE_GROUP, jobOperations.getTasks().get(0));
    Map<String, TaskRelocationPlan> result = step.collectTasksThatMustBeRelocated();
    assertThat(result).hasSize(1);
}