def test_finalize_success(self, log_mock):
"""FirehoseClient - Finalize, Success"""
request_id = 'success_id'
stream_name = 'stream_name'
count = 3
response = {'ResponseMetadata': {'RequestId': request_id}}
FirehoseClient._finalize(response, stream_name, count)
log_mock.assert_called_with(
'Successfully sent %d message(s) to firehose %s with RequestId \'%s\'',
count, stream_name, request_id)
def test_finalize_failures(self, failure_mock):
"""FirehoseClient - Finalize, With Failures"""
response = {
'FailedPutCount':
1,
'RequestResponses': [{
'RecordId': 'rec_id_01'
}, {
'ErrorCode': 10,
'ErrorMessage': 'foo'
}, {
'RecordId': 'rec_id_03'
}]
}
FirehoseClient._finalize(response, 'stream_name', 3)
failure_mock.assert_called_with(1)
