add wait_for_running helper

Signed-off-by: Adam Stokes <adam.stokes@ubuntu.com>