syntax fix and add unit test

This commit is contained in:
Chris MacNaughton
2018-06-20 17:09:34 +02:00
parent 063aadd4df
commit 57ee4a3662
2 changed files with 15 additions and 1 deletions
+14
View File
@@ -142,6 +142,20 @@ class TestModel(ut_utils.BaseTestCase):
self.assertEqual(model.get_juju_model(), 'envmodel')
self.get_current_model.assert_not_called()
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.get(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 test_get_juju_model_noenv(self):
self.patch_object(model.os, 'environ')
self.patch_object(model, 'get_current_model')
+1 -1
View File
@@ -52,7 +52,7 @@ def get_juju_model():
CURRENT_MODEL = os.environ["JUJU_MODEL"]
except KeyError:
try:
CURRENT_MODEL = os.environ["MODEL_NAME"]:
CURRENT_MODEL = os.environ["MODEL_NAME"]
except KeyError:
# If unset connect get the current active model
CURRENT_MODEL = get_current_model()