syntax fix and add unit test
This commit is contained in:
@@ -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
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user