bugfix: handling when success properties not set

Author: m-karo

emf/common/integrations/rabbit.py

@@ -262,7 +262,7 @@ def _process_messages(self, basic_deliver, properties, body):
                 try:
                     logger.info(f"Handling message with handler: {message_handler.__class__.__name__}")
                     body, properties = message_handler.handle(body, properties=properties, channel=None)
-                    if not properties.headers['success']: # stop processing next handlers if message success was set to false
+                    if not properties.headers.get('success', True): # stop processing next handlers if message success was set to false
                         break
 
                 except Exception as error:
@@ -582,7 +582,7 @@ def _process_messages(self, basic_deliver, properties, body):
                 for message_handler in self.message_handlers:
                     logger.info(f"Handling message with handler: {message_handler.__class__.__name__}")
                     body, properties = message_handler.handle(body, properties=properties, channel=self._channel)
-                    if not properties.headers['success']: # stop processing next handlers if message success was set to false
+                    if not properties.headers.get('success', True): # stop processing next handlers if message success was set to false
                         break
             except Exception as error:
                 logger.error(f"Message handling failed: {error}", exc_info=True)