Add wait_for_job to the v16 test utils

Adds wait_for_job to the v16 test utils so server tests run again on v16.