def test_batch_send_failure(self): proc = FirehoseProcessor("test") proc.MAX_RECORD_BATCH = 1 # Setup the mock proc._client.put_record_batch.return_value = dict(FailedPutCount=1) # Start and log proc.start() proc.process("a decently larger message") proc.stop() assert len(self.mock_boto.mock_calls) == 4 assert len(proc._client.put_record_batch.mock_calls) == 3
def test_message_max_batch(self): proc = FirehoseProcessor("test") proc.MAX_RECORD_BATCH = 1 # Setup the mock proc._client.put_record_batch.return_value = dict(FailedPutCount=0) # Start and log proc.start() proc.process("a decently larger message") proc.stop() eq_(len(self.mock_boto.mock_calls), 2) eq_(len(proc._client.put_record_batch.mock_calls), 1)