def test_run_calls_a_new_process_timeout(self, mock_run_command):
entry_point = 'path'
timeout = 1
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: None,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
mock_run_command.return_value = (0, "", "", True)
result = runner.run({})
self.assertTrue(mock_run_command.called)
self.assertEqual(LIVEACTION_STATUS_FAILED, result[0])
def test_run_calls_a_new_process_failure(self, mock_run_command, mock_quote_unix):
timeout = 1
runner = csr.get_runner()
runner.runner_parameters = {
csr.RUNNER_INPUTS: None,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
mock_run_command.return_value = (1, "", "", False)
mock_quote_unix.return_value = ""
result = runner.run({})
mock_quote_unix.assert_called_with(tests_config.CONF.cloudslang.home_dir)
self.assertTrue(mock_run_command.called)
self.assertEqual(LIVEACTION_STATUS_FAILED, result[0])
def test_pre_run_sets_attributes(self):
entry_point = 'path'
inputs = {'a': 1}
timeout = 10
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: inputs,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
self.assertEqual(runner.entry_point, entry_point)
self.assertEqual(runner._inputs, inputs)
self.assertEqual(runner._timeout, timeout)
def test_inputs_are_save_to_file_properly(self, mock_yaml_dump, mock_run_command):
entry_point = 'path'
inputs = {'a': 1}
timeout = 1
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: inputs,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
mock_run_command.return_value = (0, "", "", True)
mock_yaml_dump.return_value = ""
result = runner.run({})
self.assertTrue(mock_run_command.called)
mock_yaml_dump.assert_called_with(inputs, default_flow_style=False)
self.assertEqual(LIVEACTION_STATUS_FAILED, result[0])
def test_inputs_provided_via_inputs_runner_parameter(self, mock_run_command):
entry_point = 'path'
inputs = {'a': 1}
timeout = 1
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: inputs,
csr.RUNNER_TIMEOUT: timeout,
}
runner._write_inputs_to_a_temp_file = mock.Mock()
runner._write_inputs_to_a_temp_file.return_value = None
mock_run_command.return_value = (0, "", "", False)
runner.pre_run()
runner.run({})
runner._write_inputs_to_a_temp_file.assert_called_with(inputs=inputs)
def test_temp_file_deletes_when_exception_occurs(self, mock_os_remove, mock_run_command):
entry_point = 'path'
inputs = {'a': 1}
timeout = 1
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: inputs,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
mock_run_command.return_value = (0, "", "", True)
mock_run_command.side_effect = IOError('Boom!')
with self.assertRaisesRegex(IOError, "Boom!"):
runner.run({})
self.assertTrue(mock_os_remove.called)
# lets really remove it now
os.remove(mock_os_remove.call_args[0][0])
def test_prepare_command(self):
entry_point = 'flow_path'
inputs = None
timeout = 1
runner = csr.get_runner()
runner.entry_point = entry_point
runner.runner_parameters = {
csr.RUNNER_INPUTS: inputs,
csr.RUNNER_TIMEOUT: timeout,
}
runner.pre_run()
# No inputs
result = runner._prepare_command(has_inputs=False, inputs_file_path=None)
expected = '/opt/cslang/bin/cslang run --f flow_path --cp /opt/cslang'
self.assertEqual(result, expected)
# Inputs
result = runner._prepare_command(has_inputs=True, inputs_file_path='inputs_file')
expected = '/opt/cslang/bin/cslang run --f flow_path --cp /opt/cslang --if inputs_file'
self.assertEqual(result, expected)
def test_runner_creation(self):
runner = csr.get_runner()
self.assertTrue(runner)
self.assertTrue(runner.runner_id)