def test_get_juju_model_alt(self): self.patch_object(model.os, 'environ') self.patch_object(model, 'get_current_model') self.get_current_model.return_value = 'modelsmodel' def _get_env(key): return _env[key] self.environ.__getitem__.side_effect = _get_env _env = {"MODEL_NAME": 'envmodel'} # JUJU_ENV environment variable set self.assertEqual(model.get_juju_model(), 'envmodel') self.get_current_model.assert_not_called()
def setUpClass(cls, application_name=None, model_alias=None): """Run setup for test class to create common resources.""" cls.model_aliases = model.get_juju_model_aliases() if model_alias: cls.model_name = cls.model_aliases[model_alias] else: cls.model_name = model.get_juju_model() cls.test_config = lifecycle_utils.get_charm_config(fatal=False) if application_name: cls.application_name = application_name else: cls.application_name = cls.test_config['charm_name'] cls.lead_unit = model.get_lead_unit_name(cls.application_name, model_name=cls.model_name) logging.debug('Leader unit is {}'.format(cls.lead_unit))
from zaza.openstack.charm_tests.vault import ( setup as vault_setup, utils as vault_utils ) import zaza.openstack.utilities.cert import utils.mojo_utils as mojo_utils import logging if __name__ == "__main__": # NOTE(ajkavanagh): Set the jujulib Connection frame size to 4GB to # cope with all the outputs from large models that have had lots of status # changes. juju.client.connection.Connection.MAX_FRAME_SIZE = 2**32 cli_utils.setup_logging() target_model = model.get_juju_model() wl_statuses = copy.deepcopy(openstack.WORKLOAD_STATUS_EXCEPTIONS) os_version = openstack.get_current_os_versions( 'designate').get('designate') if os_version <= 'pike': # Remove the memcached relation to disable designate. This is a # workaround for Bug #1848307 logging.info("Removing designate memcached relation") model.remove_relation( 'designate', 'coordinator-memcached', 'memcached:cache') wl_statuses['designate'] = { 'workload-status-message': """'coordinator-memcached' missing""", 'workload-status': 'blocked'} logging.info("Waiting for statuses with exceptions ...")