Add wait_for_host_or_timeout in infra/hosts