def test_module(self, open_mock, mock_subproc_popen):
#
# setup
#
self._open_mock(open_mock)
process_mock = mock.Mock()
attrs = {'communicate.return_value': ('ouput', 'ok'),
'poll.return_value': 0}
process_mock.configure_mock(**attrs)
mock_subproc_popen.return_value = process_mock
template = self.useFixture(ep.PuppetExPlanDownloable()).execution_plan
script = list(template['Scripts'].values())[0]
self.puppet_executor.load('path',
script['Options'])
self.puppet_executor.run('test')
def test_module_error(self, open_mock, mock_subproc_popen):
#
# setup
#
self._open_mock(open_mock)
process_mock = mock.Mock()
attrs = {
'communicate.return_value': ('ouput', 'error'),
'poll.return_value': 2
}
process_mock.configure_mock(**attrs)
mock_subproc_popen.return_value = process_mock
template = self.useFixture(ep.PuppetExPlanDownloable()).execution_plan
self.puppet_executor.load('path',
template['Scripts'].values()[0]['Options'])
self.assertRaises(ex.CustomException, self.puppet_executor.run, 'test')