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' })