Add support for loading kolla config from ENV

Implements: blueprint zookeeper

Change-Id: Ibabcb5d6d304be9290633338c513f16651d4479a

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/test_set_config.py b/tests/test_set_config.py
index c1e7f7927..1e8ac9ce6 100644
--- a/tests/test_set_config.py
+++ b/tests/test_set_config.py
@@ -45,3 +45,20 @@ class LoadFromFile(base.BaseTestCase):
                 mock.call().__enter__(),
                 mock.call().write(u'/bin/true'),
                 mock.call().__exit__(None, None, None)], mo.mock_calls)
+
+
+class LoadFromEnv(base.BaseTestCase):
+
+    def test_load_ok(self):
+        in_config = json.dumps({'command': '/bin/true',
+                                'config_files': {}})
+
+        mo = mock.mock_open()
+        with mock.patch.object(set_configs, 'open', mo):
+            with mock.patch.dict('os.environ', {'KOLLA_CONFIG': in_config}):
+                set_configs.load_config()
+                self.assertEqual([mock.call('/run_command', 'w+'),
+                                  mock.call().__enter__(),
+                                  mock.call().write(u'/bin/true'),
+                                  mock.call().__exit__(None, None, None)],
+                                 mo.mock_calls)