From 8ab783f403984b3ef9a6c3010a9a81c0f29c4236 Mon Sep 17 00:00:00 2001 From: xudan Date: Thu, 13 Dec 2018 06:23:08 -0500 Subject: make item 'extra_container' available for every project The item 'extra_container' used to be available for Bottlenecks only. Make it open for Dovetail framework. Change-Id: If4f2c032924447b97e401d497aca0309c0f5d847 Signed-off-by: xudan --- dovetail/container.py | 8 ++++---- dovetail/tests/unit/test_container.py | 7 +++++++ 2 files changed, 11 insertions(+), 4 deletions(-) diff --git a/dovetail/container.py b/dovetail/container.py index ed006dc9..d44035cd 100644 --- a/dovetail/container.py +++ b/dovetail/container.py @@ -168,10 +168,10 @@ class Container(object): def clean(self): cmd = ('sudo docker rm -f {}'.format(self.container_id)) dt_utils.exec_cmd(cmd, self.logger) - if self.valid_type.lower() == 'bottlenecks': - containers = dt_utils.get_value_from_dict( - 'extra_container', dt_cfg.dovetail_config[self.valid_type]) - for container in containers: + extra_containers = dt_utils.get_value_from_dict( + 'extra_container', dt_cfg.dovetail_config[self.valid_type]) + if extra_containers: + for container in extra_containers: if self.check_container_exist(container): cmd = ('sudo docker rm -f {}'.format(container)) dt_utils.exec_cmd(cmd, self.logger) diff --git a/dovetail/tests/unit/test_container.py b/dovetail/tests/unit/test_container.py index 70e01d8e..25509a70 100644 --- a/dovetail/tests/unit/test_container.py +++ b/dovetail/tests/unit/test_container.py @@ -154,6 +154,13 @@ class ContainerTesting(unittest.TestCase): call('sudo docker rm -f None', self.logger), call('sudo docker rm -f container', self.logger)]) + @patch('dovetail.container.dt_cfg') + @patch('dovetail.container.dt_utils') + def test_clean_no_extra_container(self, mock_utils, mock_config): + mock_utils.get_value_from_dict.return_value = None + self.container.clean() + mock_utils.get_value_from_dict.assert_called_once() + @patch('dovetail.container.dt_utils') def test_check_container_exist_true(self, mock_utils): container_name = 'container' -- cgit 1.2.3-korg