def test_execute_user_data_script(self, mock_get_command, _): mock_get_command.return_value.return_value = ( mock.sentinel.output, mock.sentinel.error, -1 ) retval = userdatautils.execute_user_data_script( mock.sentinel.user_data) self.assertEqual(-1, retval)
def process(self, part): file_name = os.path.join(CONF.heat_config_dir, part.get_filename()) self._check_dir(file_name) encoding.write_file(file_name, part.get_payload()) if part.get_filename() == self._heat_user_data_filename: return userdatautils.execute_user_data_script(part.get_payload())
def process(self, part): self._check_heat_config_dir() file_name = os.path.join(CONF.heat_config_dir, part.get_filename()) with open(file_name, 'wb') as f: f.write(part.get_payload()) if part.get_filename() == self._heat_user_data_filename: return userdatautils.execute_user_data_script(part.get_payload())
def _process_non_multi_part(self, user_data): if user_data.startswith(b'#cloud-config'): user_data_plugins = factory.load_plugins() cloud_config_plugin = user_data_plugins.get('text/cloud-config') ret_val = cloud_config_plugin.process(user_data) else: ret_val = userdatautils.execute_user_data_script(user_data) return self._get_plugin_return_value(ret_val)
def _process_non_multi_part(self, user_data): if user_data.startswith('#cloud-config'): user_data_plugins = factory.load_plugins() cloud_config_plugin = user_data_plugins.get('text/cloud-config') ret_val = cloud_config_plugin.process(user_data) else: ret_val = userdatautils.execute_user_data_script(user_data) return self._get_plugin_return_value(ret_val)
def _test_execute_user_data_script(self, mock_get_os_utils, mock_path_expandvars, mock_path_exists, mock_os_remove, mock_gettempdir, mock_re_search, fake_user_data): mock_osutils = mock.MagicMock() mock_gettempdir.return_value = 'fake_temp' uuid.uuid4 = mock.MagicMock(return_value='randomID') match_instance = mock.MagicMock() path = os.path.join('fake_temp', 'randomID') args = None powershell = False mock_get_os_utils.return_value = mock_osutils mock_path_exists.return_value = True if fake_user_data == '^rem cmd\s': side_effect = [match_instance] number_of_calls = 1 extension = '.cmd' args = [path + extension] shell = True elif fake_user_data == '^#!/usr/bin/env\spython\s': side_effect = [None, match_instance] number_of_calls = 2 extension = '.py' args = ['python.exe', path + extension] shell = False elif fake_user_data == '#!': side_effect = [None, None, match_instance] number_of_calls = 3 extension = '.sh' args = ['bash.exe', path + extension] shell = False elif fake_user_data == '#ps1_sysnative\s': side_effect = [None, None, None, match_instance] number_of_calls = 4 extension = '.ps1' sysnative = True powershell = True elif fake_user_data == '#ps1_x86\s': side_effect = [None, None, None, None, match_instance] number_of_calls = 5 extension = '.ps1' shell = False sysnative = False powershell = True else: side_effect = [None, None, None, None, None] number_of_calls = 5 mock_re_search.side_effect = side_effect with mock.patch('cloudbaseinit.plugins.windows.userdatautils.open', mock.mock_open(), create=True): response = userdatautils.execute_user_data_script(fake_user_data) mock_gettempdir.assert_called_once_with() self.assertEqual(mock_re_search.call_count, number_of_calls) if args: mock_osutils.execute_process.assert_called_with(args, shell) mock_os_remove.assert_called_once_with(path + extension) self.assertEqual(response, None) elif powershell: mock_osutils.execute_powershell_script.assert_called_with( path + extension, sysnative) mock_os_remove.assert_called_once_with(path + extension) self.assertEqual(response, None) else: self.assertEqual(response, 0)
def _test_execute_user_data_script(self, mock_get_os_utils, mock_path_expandvars, mock_path_exists, mock_path_isdir, mock_os_remove, mock_gettempdir, mock_re_search, fake_user_data, directory_exists): mock_osutils = mock.MagicMock() mock_gettempdir.return_value = 'fake_temp' uuid.uuid4 = mock.MagicMock(return_value='randomID') match_instance = mock.MagicMock() path = os.path.join('fake_temp', 'randomID') args = None mock_get_os_utils.return_value = mock_osutils if fake_user_data == '^rem cmd\s': side_effect = [match_instance] number_of_calls = 1 extension = '.cmd' args = [path+extension] shell = True elif fake_user_data == '^#!/usr/bin/env\spython\s': side_effect = [None, match_instance] number_of_calls = 2 extension = '.py' args = ['python.exe', path+extension] shell = False elif fake_user_data == '#!': side_effect = [None, None, match_instance] number_of_calls = 3 extension = '.sh' args = ['bash.exe', path+extension] shell = False elif fake_user_data == '#ps1\s': side_effect = [None, None, None, match_instance] number_of_calls = 4 extension = '.ps1' args = ['powershell.exe', '-ExecutionPolicy', 'RemoteSigned', '-NonInteractive', path+extension] shell = False else: side_effect = [None, None, None, None, match_instance] number_of_calls = 5 extension = '.ps1' shell = False if directory_exists: args = [mock_path_expandvars('%windir%\\sysnative\\' 'WindowsPowerShell\\v1.0\\' 'powershell.exe'), '-ExecutionPolicy', 'RemoteSigned', '-NonInteractive', path+extension] mock_path_isdir.return_value = True else: mock_path_isdir.return_value = False mock_re_search.side_effect = side_effect with mock.patch('cloudbaseinit.plugins.windows.userdatautils.open', mock.mock_open(), create=True): response = userdatautils.execute_user_data_script(fake_user_data) mock_gettempdir.assert_called_once_with() self.assertEqual(mock_re_search.call_count, number_of_calls) if args: mock_osutils.execute_process.assert_called_with(args, shell) if not directory_exists: self.assertEqual(response, 0) else: self.assertEqual(response, None)
def test_execute_user_data_script(self, mock_get_command, _): mock_get_command.return_value.return_value = (mock.sentinel.output, mock.sentinel.error, -1) retval = userdatautils.execute_user_data_script( mock.sentinel.user_data) self.assertEqual(-1, retval)
def _process_non_multi_part(self, user_data): ret_val = userdatautils.execute_user_data_script(user_data) return self._get_plugin_return_value(ret_val)
def test_execute_user_data_script_no_commands(self, _): retval = userdatautils.execute_user_data_script(b"unknown") self.assertEqual(0, retval)
def test_execute_user_data_script_fails(self, mock_get_command, _): mock_get_command.return_value.side_effect = ValueError retval = userdatautils.execute_user_data_script( mock.sentinel.user_data) self.assertEqual(0, retval)
def process(self, part): if part.get_filename() == self._heat_user_data_filename: return userdatautils.execute_user_data_script(part.get_payload()) else: LOG.info("Heat content not supported: %s" % part.get_filename())
def _test_execute_user_data_script(self, mock_uuid4, mock_get_os_utils, mock_path_expandvars, mock_path_exists, mock_os_remove, mock_gettempdir, mock_re_search, fake_user_data): mock_osutils = mock.MagicMock() mock_gettempdir.return_value = 'fake_temp' mock_uuid4.return_value = 'randomID' match_instance = mock.MagicMock() path = os.path.join('fake_temp', 'randomID') args = None powershell = False mock_get_os_utils.return_value = mock_osutils mock_path_exists.return_value = True if fake_user_data == '^rem cmd\s': side_effect = [match_instance] number_of_calls = 1 extension = '.cmd' args = [path + extension] shell = True elif fake_user_data == '^#!/usr/bin/env\spython\s': side_effect = [None, match_instance] number_of_calls = 2 extension = '.py' args = ['python.exe', path + extension] shell = False elif fake_user_data == '#!': side_effect = [None, None, match_instance] number_of_calls = 3 extension = '.sh' args = ['bash.exe', path + extension] shell = False elif fake_user_data == '#ps1_sysnative\s': side_effect = [None, None, None, match_instance] number_of_calls = 4 extension = '.ps1' sysnative = True powershell = True elif fake_user_data == '#ps1_x86\s': side_effect = [None, None, None, None, match_instance] number_of_calls = 5 extension = '.ps1' shell = False sysnative = False powershell = True else: side_effect = [None, None, None, None, None] number_of_calls = 5 mock_re_search.side_effect = side_effect with mock.patch('cloudbaseinit.plugins.windows.userdatautils.open', mock.mock_open(), create=True): response = userdatautils.execute_user_data_script(fake_user_data) mock_gettempdir.assert_called_once_with() self.assertEqual(number_of_calls, mock_re_search.call_count) if args: mock_osutils.execute_process.assert_called_with(args, shell) mock_os_remove.assert_called_once_with(path + extension) self.assertEqual(None, response) elif powershell: mock_osutils.execute_powershell_script.assert_called_with( path + extension, sysnative) mock_os_remove.assert_called_once_with(path + extension) self.assertEqual(None, response) else: self.assertEqual(0, response)