From b98039819ea5c6468f0da87e07a6a0758576af5b Mon Sep 17 00:00:00 2001
From: trobert2 <rtingirica@cloudbasesolutions.com>
Date: Fri, 7 Feb 2014 17:47:31 +0200
Subject: [PATCH] Modified test for heat handler plugin
Added a new test for the new method _check_heat_config_dir and
modiffied the test for the changed process method.
---
.../windows/userdataplugins/test_heat.py | 27 ++++++++++++++-----
1 file changed, 21 insertions(+), 6 deletions(-)
diff --git a/cloudbaseinit/tests/plugins/windows/userdataplugins/test_heat.py b/cloudbaseinit/tests/plugins/windows/userdataplugins/test_heat.py
index 5c543cd4..1015a185 100644
--- a/cloudbaseinit/tests/plugins/windows/userdataplugins/test_heat.py
+++ b/cloudbaseinit/tests/plugins/windows/userdataplugins/test_heat.py
@@ -28,14 +28,29 @@ class HeatUserDataHandlerTests(unittest.TestCase):
def setUp(self):
self._heat = heat.HeatPlugin()
+ @mock.patch('os.path.exists')
+ @mock.patch('os.makedirs')
+ def test_check_heat_config_dir(self, 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)
+
@mock.patch('cloudbaseinit.plugins.windows.userdatautils'
'.execute_user_data_script')
- def _test_process(self, mock_execute_user_data_script, filename):
+ @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):
mock_part = mock.MagicMock()
mock_part.get_filename.return_value = filename
- response = self._heat.process(mock_part)
+ with mock.patch('__builtin__.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()
mock_part.get_filename.assert_called_with()
- if filename:
+ if filename == self._heat._heat_user_data_filename:
mock_execute_user_data_script.assert_called_with(
mock_part.get_payload())
self.assertEqual(response, mock_execute_user_data_script())
@@ -43,7 +58,7 @@ class HeatUserDataHandlerTests(unittest.TestCase):
self.assertTrue(response is None)
def test_process(self):
- self._test_process(filename='cfn-userdata')
+ self._test_process(filename=self._heat._heat_user_data_filename)
- def test_process_content_not_supported(self):
- self._test_process(filename=None)
+ def test_process_content_other_data(self):
+ self._test_process(filename='other data')