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(