From b994cf5291888858c46609449ff0d5d940ea5a59 Mon Sep 17 00:00:00 2001 From: Chris Keogh Date: Tue, 3 Oct 2017 13:54:37 +1300 Subject: [PATCH] add more realistic policy --- moto/awslambda/responses.py | 5 ++++- tests/test_awslambda/test_lambda.py | 6 +++++- 2 files changed, 9 insertions(+), 2 deletions(-) diff --git a/moto/awslambda/responses.py b/moto/awslambda/responses.py index 972cd7a4e..94c381f5e 100644 --- a/moto/awslambda/responses.py +++ b/moto/awslambda/responses.py @@ -67,7 +67,10 @@ class LambdaResponse(BaseResponse): path = request.path if hasattr(request, 'path') else request.path_url function_name = path.split('/')[-2] if lambda_backend.has_function(function_name): - return 200, {}, json.dumps(dict(Policy='test_policy')) + policy = ("{\"Statement\":[{\"Action\":[\"lambda:InvokeFunction\"]," + "\"Resource\":\"arn:aws:lambda:us-west-2:account-id:function:helloworld\"," + "\"Effect\":\"Allow\",\"Principal\":{\"AWS\":\"account-id\"},\"Sid\":\"3\"}]}") + return 200, {}, json.dumps(dict(Policy=policy)) else: return 404, {}, "{}" diff --git a/tests/test_awslambda/test_lambda.py b/tests/test_awslambda/test_lambda.py index c7fee2745..163fa306f 100644 --- a/tests/test_awslambda/test_lambda.py +++ b/tests/test_awslambda/test_lambda.py @@ -667,4 +667,8 @@ def get_function_policy(): response = conn.get_policy( FunctionName='testFunction' ) - assert response['Policy'] == 'test_policy' \ No newline at end of file + + assert 'Policy' in response + assert isinstance(response['Policy'], str) + res = json.loads(response['Policy']) + assert res['Statement'][0]['Action'] == ['lambda:InvokeFunction'] \ No newline at end of file