def test_skip_if_permission_already_granted(stubbed_session):
lambda_client = stubbed_session.stub('lambda')
policy = {
'Id': 'default',
'Statement': [
{'Action': 'lambda:InvokeFunction',
'Condition': {
'ArnLike': {
'AWS:SourceArn': 'rule-arn',
}
},
'Effect': 'Allow',
'Principal': {'Service': 'events.amazonaws.com'},
'Resource': 'resource-arn',
'Sid': 'statement-id'},
],
'Version': '2012-10-17'
}
lambda_client.get_policy(
FunctionName='function-arn').returns({'Policy': json.dumps(policy)})
stubbed_session.activate_stubs()
awsclient = TypedAWSClient(stubbed_session)
awsclient.add_permission_for_scheduled_event(
'rule-arn', 'function-arn')
stubbed_session.verify_stubs()
def test_skip_if_permission_already_granted(stubbed_session):
lambda_client = stubbed_session.stub('lambda')
policy = {
'Id': 'default',
'Statement': [
{'Action': 'lambda:InvokeFunction',
'Condition': {
'ArnLike': {
'AWS:SourceArn': 'rule-arn',
}
},
'Effect': 'Allow',
'Principal': {'Service': 'events.amazonaws.com'},
'Resource': 'resource-arn',
'Sid': 'statement-id'},
],
'Version': '2012-10-17'
}
lambda_client.get_policy(
FunctionName='function-arn').returns({'Policy': json.dumps(policy)})
stubbed_session.activate_stubs()
awsclient = TypedAWSClient(stubbed_session)
awsclient.add_permission_for_scheduled_event(
'rule-arn', 'function-arn')
stubbed_session.verify_stubs()
def test_add_permission_for_scheduled_event(stubbed_session):
lambda_client = stubbed_session.stub('lambda')
lambda_client.get_policy(FunctionName='function-arn').returns(
{'Policy': '{}'})
lambda_client.add_permission(
Action='lambda:InvokeFunction',
FunctionName='function-arn',
StatementId=stub.ANY,
Principal='events.amazonaws.com',
SourceArn='rule-arn'
).returns({})
stubbed_session.activate_stubs()
awsclient = TypedAWSClient(stubbed_session)
awsclient.add_permission_for_scheduled_event(
'rule-arn', 'function-arn')
stubbed_session.verify_stubs()
def test_add_permission_for_scheduled_event(stubbed_session):
lambda_client = stubbed_session.stub('lambda')
lambda_client.get_policy(FunctionName='function-arn').returns(
{'Policy': '{}'})
lambda_client.add_permission(
Action='lambda:InvokeFunction',
FunctionName='function-arn',
StatementId=stub.ANY,
Principal='events.amazonaws.com',
SourceArn='rule-arn'
).returns({})
stubbed_session.activate_stubs()
awsclient = TypedAWSClient(stubbed_session)
awsclient.add_permission_for_scheduled_event(
'rule-arn', 'function-arn')
stubbed_session.verify_stubs()
