Add wait_for_job test

Ensure that wait_for_job accepts a jobId.