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_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_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_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_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)