/deploy sit

podaac/lambda_handler/lambda_handler.py

@@ -244,6 +244,14 @@ def process(self):
         granules = self.input['granules']
         append_output = {}
 
+        if self.kwargs.get('context', None):
+            try:
+                aws_request_id = self.kwargs.get('context').aws_request_id
+                message = f"aws_request_id: {aws_request_id} collection: {self.config.get('collection')}"
+                self.logger.info(message)
+            except AttributeError:
+                pass
+
         self.download_palette_files(config_file_path)
 
         for granule in granules:
@@ -390,6 +398,12 @@ def image_generate(self, file_, config_file, palette_dir, granule_id):
 
         return uploaded_files
 
+
+    @classmethod
+    def handler(cls, event, context=None, path=None, noclean=False):
+        """ General event handler """
+        return cls.run(path=path, noclean=noclean, context=context, **event)
+
     @classmethod
     def run(cls, *args, **kwargs):
         """ Run this payload with the given Process class """

tests/test_lambda_handler.py

@@ -179,6 +179,10 @@ def test_get_config_no_setting():
         # exception expected
         pass
 
+class Context:
+    def __init__(self, aws_request_id):
+        self.aws_request_id = aws_request_id
+
 
 @mock_s3
 @patch('requests.get')
@@ -233,7 +237,8 @@ def test_lambda_handler_cumulus(mocked_get):
             is_valid_shema = validate(instance=files, schema=file_schema)
             assert is_valid_shema is None
 
-    output = lambda_handler.handler(event, {})
+    context = Context("fake_request_id")
+    output = lambda_handler.handler(event, context)
 
     for granule in output.get('payload').get('granules'):
         is_valid_shema = validate(instance=granule.get('files'), schema=file_schema)