From 9f73b218acf87fc056558049693cac3e3e0a9d4e Mon Sep 17 00:00:00 2001
From: Liam Young <liam.young@canonical.com>
Date: Fri, 11 Oct 2019 11:40:10 +0000
Subject: [PATCH] Ensure units are idle before prepare_series_upgrade

Ensure all units are idel before running prepare_series_upgrade.
Closes issue #88
---
 zaza/openstack/utilities/generic.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/zaza/openstack/utilities/generic.py b/zaza/openstack/utilities/generic.py
index dd9e599..7eb175c 100644
--- a/zaza/openstack/utilities/generic.py
+++ b/zaza/openstack/utilities/generic.py
@@ -368,6 +368,7 @@ def series_upgrade(unit_name, machine_num,
     application = unit_name.split('/')[0]
     set_dpkg_non_interactive_on_unit(unit_name)
     dist_upgrade(unit_name)
+    model.block_until_all_units_idle()
     logging.info("Prepare series upgrade on {}".format(machine_num))
     model.prepare_series_upgrade(machine_num, to_series=to_series)
     logging.info("Waiting for workload status 'blocked' on {}"