From bd134d882c8c16a71a8b00f2e09f4bd13acab76d Mon Sep 17 00:00:00 2001 From: tengqm Date: Fri, 17 Mar 2017 09:39:01 -0400 Subject: [PATCH] Add get_env and get_files function to heat driver These resources are added to SDK for retrieving stack data. Change-Id: I02c89bd929330edd3a99bff8dadc0fdf17dda088 Depends-On: Ic4ea1c58b8c43b2e4217b4a96195b338a3f4310b --- senlin/drivers/openstack/heat_v1.py | 8 ++++++++ senlin/tests/unit/drivers/test_heat_v1.py | 8 ++++++++ 2 files changed, 16 insertions(+) diff --git a/senlin/drivers/openstack/heat_v1.py b/senlin/drivers/openstack/heat_v1.py index cb807c880..ad7dead17 100644 --- a/senlin/drivers/openstack/heat_v1.py +++ b/senlin/drivers/openstack/heat_v1.py @@ -52,6 +52,14 @@ class HeatClient(base.DriverBase): def stack_check(self, stack_id): return self.conn.orchestration.check_stack(stack_id) + @sdk.translate_exception + def stack_get_environment(self, stack_id): + return self.conn.orchestration.get_stack_environment(stack_id) + + @sdk.translate_exception + def stack_get_files(self, stack_id): + return self.conn.orchestration.get_stack_files(stack_id) + @sdk.translate_exception def stack_get_template(self, stack_id): return self.conn.orchestration.get_stack_template(stack_id) diff --git a/senlin/tests/unit/drivers/test_heat_v1.py b/senlin/tests/unit/drivers/test_heat_v1.py index f78f666da..37a3c9ab5 100644 --- a/senlin/tests/unit/drivers/test_heat_v1.py +++ b/senlin/tests/unit/drivers/test_heat_v1.py @@ -72,6 +72,14 @@ class TestHeatV1(base.SenlinTestCase): self.hc.stack_check('stack_id') self.orch.check_stack.assert_called_once_with('stack_id') + def test_stack_get_environment(self): + self.hc.stack_get_environment('stack_id') + self.orch.get_stack_environment.assert_called_once_with('stack_id') + + def test_stack_get_files(self): + self.hc.stack_get_files('stack_id') + self.orch.get_stack_files.assert_called_once_with('stack_id') + def test_stack_get_template(self): self.hc.stack_get_template('stack_id') self.orch.get_stack_template.assert_called_once_with('stack_id')