From cb11ddd1e9519ff3567882057188c7726fbb5895 Mon Sep 17 00:00:00 2001
From: Liam Young <liam.young@canonical.com>
Date: Thu, 29 Mar 2018 06:07:12 +0000
Subject: [PATCH] Fix skip decorator

---
 zaza/charm_tests/test_utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/zaza/charm_tests/test_utils.py b/zaza/charm_tests/test_utils.py
index e5e2bd7..81bbf6d 100644
--- a/zaza/charm_tests/test_utils.py
+++ b/zaza/charm_tests/test_utils.py
@@ -1,11 +1,14 @@
 import logging
 import zaza.model
 
+import zaza.charm_lifecycle.utils as utils
+
 
 def skipIfNotHA(service_name):
     def _skipIfNotHA_inner_1(f):
         def _skipIfNotHA_inner_2(*args, **kwargs):
-            if len(zaza.model.get_app_ips(service_name)) > 1:
+            ips = zaza.model.get_app_ips(utils.get_juju_model(), service_name)
+            if len(ips) > 1:
                 return f(*args, **kwargs)
             else:
                 logging.warn("Skipping HA test for non-ha service {}".format(