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)