Yes, wait_for_ajax() should work for this.