From c13ea8f282302714fc9ddfd0561eab24d01817f3 Mon Sep 17 00:00:00 2001 From: Chris MacNaughton Date: Wed, 20 Jun 2018 17:24:21 +0200 Subject: [PATCH] .get will never throw the exception we expect on mismatch --- unit_tests/test_zaza_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/unit_tests/test_zaza_model.py b/unit_tests/test_zaza_model.py index 0b8e37f..95bd7cb 100644 --- a/unit_tests/test_zaza_model.py +++ b/unit_tests/test_zaza_model.py @@ -134,7 +134,7 @@ class TestModel(ut_utils.BaseTestCase): self.get_current_model.return_value = 'modelsmodel' def _get_env(key): - return _env.get(key) + return _env[key] self.environ.__getitem__.side_effect = _get_env _env = {"JUJU_MODEL": 'envmodel'} @@ -148,7 +148,7 @@ class TestModel(ut_utils.BaseTestCase): self.get_current_model.return_value = 'modelsmodel' def _get_env(key): - return _env.get(key) + return _env[key] self.environ.__getitem__.side_effect = _get_env _env = {"MODEL_NAME": 'envmodel'}