catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

add awslambda.add_permission

Browse source
This commit is contained in:
Chris Keogh 2017-10-03 15:23:00 +13:00 • committed by dbfr3qs
commit 9bb07e6b6e
3 changed files with 65 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

46
tests/test_awslambda/test_lambda.py
View file

@ -646,6 +646,39 @@ def test_get_function_created_with_zipfile():
},
)
@mock_lambda
def add_function_permission():
conn = boto3.client('lambda', 'us-west-2')
zip_content = get_test_zip_file1()
result = conn.create_function(
FunctionName='testFunction',
Runtime='python2.7',
Role='test-iam-role',
Handler='lambda_function.handler',
Code={
'ZipFile': zip_content,
},
Description='test lambda function',
Timeout=3,
MemorySize=128,
Publish=True,
)
response = conn.add_permission(
FunctionName='testFunction',
StatementId='1',
Action="lambda:InvokeFunction",
Principal='432143214321',
SourceArn="arn:aws:lambda:us-west-2:account-id:function:helloworld",
SourceAccount='123412341234',
EventSourceToken='blah',
Qualifier='2'
)
assert 'Statement' in response
res = json.loads(response['Statement'])
assert res['Action'] == "lambda:InvokeFunction"
@mock_lambda
def get_function_policy():
conn = boto3.client('lambda', 'us-west-2')
@ -664,6 +697,17 @@ def get_function_policy():
Publish=True,
)
response = conn.add_permission(
FunctionName='testFunction',
StatementId='1',
Action="lambda:InvokeFunction",
Principal='432143214321',
SourceArn="arn:aws:lambda:us-west-2:account-id:function:helloworld",
SourceAccount='123412341234',
EventSourceToken='blah',
Qualifier='2'
)
response = conn.get_policy(
FunctionName='testFunction'
)
@ -671,4 +715,4 @@ def get_function_policy():
assert 'Policy' in response
assert isinstance(response['Policy'], str)
res = json.loads(response['Policy'])
assert res['Statement'][0]['Action'] == ['lambda:InvokeFunction']
assert res['Statement'][0]['Action'] == 'lambda:InvokeFunction'