def testMakeFailedRunSample(self, sample_mock):
error_msg = 'error'
pkb.MakeFailedRunSample(error_msg, stages.PROVISION)
sample_mock.assert_called_once()
sample_mock.assert_called_with(
'Run Failed', 1, 'Run Failed', {
'error_message': error_msg,
'run_stage': stages.PROVISION,
'flags': '{}'
})
def testMakeFailedRunSample(self, sample_mock):
error_msg = 'error'
spec = mock.MagicMock()
spec.vms = []
spec.failed_substatus = None
pkb.MakeFailedRunSample(spec, error_msg, stages.PROVISION)
sample_mock.assert_called_once()
sample_mock.assert_called_with('Run Failed', 1, 'Run Failed', {
'error_message': error_msg,
'run_stage': stages.PROVISION,
'flags': '{}'
})
def testMakeFailedRunSampleWithTruncation(self, sample_mock):
error_msg = 'This is a long error message that should be truncated.'
pkb.FLAGS.failed_run_samples_error_length = 7
pkb.MakeFailedRunSample(error_msg, stages.PROVISION)
sample_mock.assert_called_once()
sample_mock.assert_called_with(
'Run Failed', 1, 'Run Failed', {
'error_message': "This is",
'run_stage': stages.PROVISION,
'flags': '{}'
})
def testMakeFailedRunSampleWithTruncation(self, sample_mock):
error_msg = 'This is a long error message that should be truncated.'
spec = mock.MagicMock()
spec.vms = []
spec.failed_substatus = 'QuotaExceeded'
pkb.FLAGS.failed_run_samples_error_length = 7
pkb.MakeFailedRunSample(spec, error_msg, stages.PROVISION)
sample_mock.assert_called_once()
sample_mock.assert_called_with('Run Failed', 1, 'Run Failed', {
'error_message': 'This is',
'run_stage': stages.PROVISION,
'flags': '{}',
'failed_substatus': 'QuotaExceeded'
})