Merge pull request #1138 from dbfr3qs/master

Add invoke_async to aws_lambda

moto/awslambda/responses.py

@@ -39,6 +39,13 @@ class LambdaResponse(BaseResponse):
         else:
             raise ValueError("Cannot handle request")
 
+    def invoke_async(self, request, full_url, headers):
+        self.setup_class(request, full_url, headers)
+        if request.method == 'POST':
+            return self._invoke_async(request, full_url)
+        else:
+            raise ValueError("Cannot handle request")
+
     def tag(self, request, full_url, headers):
         self.setup_class(request, full_url, headers)
         if request.method == 'GET':
@@ -65,6 +72,20 @@ class LambdaResponse(BaseResponse):
         else:
             return 404, response_headers, "{}"
 
+    def _invoke_async(self, request, full_url):
+        response_headers = {}
+        lambda_backend = self.get_lambda_backend(full_url)
+
+        path = request.path if hasattr(request, 'path') else request.path_url
+        function_name = path.split('/')[-3]
+        if lambda_backend.has_function(function_name):
+            fn = lambda_backend.get_function(function_name)
+            fn.invoke(self.body, self.headers, response_headers)
+            response_headers['Content-Length'] = str(0)
+            return 202, response_headers, ""
+        else:
+            return 404, response_headers, "{}"
+
     def _list_functions(self, request, full_url, headers):
         lambda_backend = self.get_lambda_backend(full_url)
         return 200, {}, json.dumps({

moto/awslambda/urls.py

@@ -11,5 +11,6 @@ url_paths = {
     '{0}/(?P<api_version>[^/]+)/functions/?$': response.root,
     '{0}/(?P<api_version>[^/]+)/functions/(?P<function_name>[\w_-]+)/?$': response.function,
     '{0}/(?P<api_version>[^/]+)/functions/(?P<function_name>[\w_-]+)/invocations/?$': response.invoke,
+    '{0}/(?P<api_version>[^/]+)/functions/(?P<function_name>[\w_-]+)/invoke-async/?$': response.invoke_async,
     '{0}/(?P<api_version>[^/]+)/tags/(?P<resource_arn>.+)': response.tag
 }

tests/test_awslambda/test_lambda.py

@@ -555,3 +555,27 @@ def test_tags_not_found():
         Resource='arn:aws:lambda:123456789012:function:not-found',
         TagKeys=['spam']
     ).should.throw(botocore.client.ClientError)
+
+@mock_lambda
+def test_invoke_async_function():
+    conn = boto3.client('lambda', 'us-west-2')
+    conn.create_function(
+        FunctionName='testFunction',
+        Runtime='python2.7',
+        Role='test-iam-role',
+        Handler='lambda_function.handler',
+        Code={
+            'ZipFile': get_test_zip_file1(),
+        },
+        Description='test lambda function',
+        Timeout=3,
+        MemorySize=128,
+        Publish=True,
+    )
+
+    success_result = conn.invoke_async(
+        FunctionName='testFunction', 
+        InvokeArgs=json.dumps({ 'test': 'event' })
+        )
+
+    success_result['Status'].should.equal(202)