def test_process_msg_logchutes(self, mock_process): """ Tests the process_msg function for LogChutes. """ self.kwargs['method'].routing_key = 'logchutes' with patch('receiver.receiver.create_doc_obj', return_value=self.mock_doc_obj) as mock_create: process_msg(**self.kwargs) mock_create.assert_called_once_with(self.decoded_msg) mock_process.assert_called_once_with(self.mock_doc_obj)
def test_process_msg_exception(self): """ Tests the process_msg function when an exception is raised. """ logging.disable(logging.NOTSET) with patch('receiver.receiver.logging.getLogger', return_value=LOGGER): with patch('receiver.receiver.json.loads', side_effect=Exception('foo')): with LogCapture() as log_capture: process_msg(**self.kwargs) log_capture.check(( 'receiver', 'ERROR', 'An error occurred while processing the message \'{"@uuid": "12345", ' '"collection": "elasticsearch.test_index.test_logs", "message": ' '"foobar"}\':\n' ' foo'), )