From 799519913603cc4880b293aad67246aa00958e0e Mon Sep 17 00:00:00 2001
From: Liam Young <liam.young@canonical.com>
Date: Thu, 9 Jan 2020 20:17:19 +0000
Subject: [PATCH] Run complete-cluster-series-upgrade

After perfomring a series upgrade on rabbitmq run
complete-cluster-series-upgrade
---
 zaza/openstack/charm_tests/series_upgrade/tests.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/zaza/openstack/charm_tests/series_upgrade/tests.py b/zaza/openstack/charm_tests/series_upgrade/tests.py
index 3d378cb..107ac80 100644
--- a/zaza/openstack/charm_tests/series_upgrade/tests.py
+++ b/zaza/openstack/charm_tests/series_upgrade/tests.py
@@ -109,6 +109,15 @@ class SeriesUpgradeTest(unittest.TestCase):
                 files=self.files,
                 post_upgrade_functions=post_upgrade_functions)
 
+            if "rabbitmq-server" in applications[application]["charm"]:
+                logging.info(
+                    "Running complete-cluster-series-upgrade action on leader")
+                model.run_action_on_leader(
+                    'rabbitmq-server',
+                    'complete-cluster-series-upgrade',
+                    action_params={})
+                model.block_until_all_units_idle()
+
 
 class OpenStackSeriesUpgrade(SeriesUpgradeTest):
     """OpenStack Series Upgrade.