Add initial message metadata support

georgeyk · Feb 9, 2017 · bd96115 · bd96115
1 parent 38a05a0
commit bd96115
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/loafer/ext/aws/message_translators.py b/loafer/ext/aws/message_translators.py
@@ -11,13 +11,13 @@ def translate(self, message):
         except (KeyError, TypeError):
             logger.error('missing Body key in SQS message. It really came from SQS ?'
                          '\nmessage={!r}'.format(message))
-            return {'content': None}
+            return {'content': None, 'metadata': {}}
 
         try:
-            return {'content': json.loads(body)}
+            return {'content': json.loads(body), 'metadata': {}}
         except json.decoder.JSONDecodeError as exc:
             logger.error('error={!r} message={!r}'.format(exc, message))
-            return {'content': None}
+            return {'content': None, 'metadata': {}}
 
 
 class SNSMessageTranslator:
@@ -29,10 +29,10 @@ def translate(self, message):
             logger.error(
                 'Missing Body or Message key in SQS message. It really came from SNS ?'
                 '\nmessage={!r}'.format(message))
-            return {'content': None}
+            return {'content': None, 'metadata': {}}
 
         try:
-            return {'content': json.loads(message)}
+            return {'content': json.loads(message), 'metadata': {}}
         except (json.decoder.JSONDecodeError, TypeError) as exc:
             logger.error('error={!r} message={!r}'.format(exc, message))
-            return {'content': None}
+            return {'content': None, 'metadata': {}}
diff --git a/loafer/message_translators.py b/loafer/message_translators.py
@@ -7,4 +7,4 @@ class StringMessageTranslator:
 
     def translate(self, message):
         logger.debug('{!r} will translate {!r}'.format(type(self).__name__, message))
-        return {'content': str(message)}
+        return {'content': str(message), 'metadata': {}}
diff --git a/tests/test_message_translator.py b/tests/test_message_translator.py
@@ -4,10 +4,10 @@
 def test_translate():
     translator = StringMessageTranslator()
     message = translator.translate(1)
-    assert message == {'content': '1'}
+    assert message == {'content': '1', 'metadata': {}}
 
     message = translator.translate('test')
-    assert message == {'content': 'test'}
+    assert message == {'content': 'test', 'metadata': {}}
 
     message = translator.translate(None)
-    assert message == {'content': 'None'}
+    assert message == {'content': 'None', 'metadata': {}}