x/cloudbase-init
Code Issues Proposed changes

Merge pull request #88 from trobert2/bug_repair

Browse Source 
Adds fix for heat directory bug
This commit is contained in:
Alessandro Pilotti 2014-09-19 13:17:22 +03:00
commit 7a9c6d23d3
2 changed files with 26 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
cloudbaseinit/plugins/windows/userdataplugins/heat.py
View File

@ -38,14 +38,15 @@ class HeatPlugin(base.BaseUserDataPlugin):
def __init__(self):
super(HeatPlugin, self).__init__("text/x-cfninitdata")
def _check_heat_config_dir(self):
if not os.path.exists(CONF.heat_config_dir):
os.makedirs(CONF.heat_config_dir)
def _check_dir(self, file_name):
dir_name = os.path.dirname(file_name)
if not os.path.exists(dir_name):
os.makedirs(dir_name)
def process(self, part):
self._check_heat_config_dir()
file_name = os.path.join(CONF.heat_config_dir, part.get_filename())
self._check_dir(file_name)
with open(file_name, 'wb') as f:
f.write(part.get_payload())

28
cloudbaseinit/tests/plugins/windows/userdataplugins/test_heat.py
View File

@ -15,6 +15,7 @@
# under the License.
import mock
import os
import unittest
from oslo.config import cfg
@ -31,25 +32,36 @@ class HeatUserDataHandlerTests(unittest.TestCase):
@mock.patch('os.path.exists')
@mock.patch('os.makedirs')
def test_check_heat_config_dir(self, mock_makedirs, mock_exists):
@mock.patch('os.path.dirname')
def test_check_heat_config_dir(self, mock_dirname, mock_makedirs,
mock_exists):
mock_exists.return_value = False
self._heat._check_heat_config_dir()
mock_exists.assert_called_once_with(CONF.heat_config_dir)
mock_makedirs.assert_called_once_with(CONF.heat_config_dir)
fake_path = mock.sentinel.fake_path
fake_dir = mock.sentinel.fake_dir
mock_dirname.return_value = fake_dir
self._heat._check_dir(file_name=fake_path)
mock_dirname.assert_called_once_with(fake_path)
mock_exists.assert_called_once_with(fake_dir)
mock_makedirs.assert_called_once_with(fake_dir)
@mock.patch('cloudbaseinit.plugins.windows.userdatautils'
'.execute_user_data_script')
@mock.patch('cloudbaseinit.plugins.windows.userdataplugins.heat'
'.HeatPlugin._check_heat_config_dir')
def _test_process(self, mock_check_heat_config_dir,
mock_execute_user_data_script, filename):
'.HeatPlugin._check_dir')
def _test_process(self, mock_check_dir, mock_execute_user_data_script,
filename):
mock_part = mock.MagicMock()
mock_part.get_filename.return_value = filename
with mock.patch('six.moves.builtins.open', mock.mock_open(),
create=True) as handle:
response = self._heat.process(mock_part)
handle().write.assert_called_once_with(mock_part.get_payload())
mock_check_heat_config_dir.assert_called_once_with()
path = os.path.join(CONF.heat_config_dir, filename)
mock_check_dir.assert_called_once_with(path)
mock_part.get_filename.assert_called_with()
if filename == self._heat._heat_user_data_filename:
mock_execute_user_data_script.assert_called_with(