def test_create_s3_event_multiple_buckets(mock_get_properties,
mock_create_s3_event,
mock_remove_perms, mock_boto3,
mock_arn, mock_perms):
"""Try to create a lambda with two S3 triggers from different buckets."""
triggers = TRIGGERS_BUCKET_A + TRIGGERS_BUCKET_B
properties = get_properties_with_triggers(triggers)
mock_get_properties.return_value = properties
mock_arn.return_value = 'fake_arn'
events = LambdaEvent(app='test_app',
env='test_env',
region='us-east-1',
prop_path='other')
events.create_lambda_events()
s3_calls = [
mock.call(app_name='test_app',
env='test_env',
region='us-east-1',
bucket='my.shared.bucket',
triggers=TRIGGERS_BUCKET_A),
mock.call(app_name='test_app',
env='test_env',
region='us-east-1',
bucket='my.other.shared.bucket',
triggers=TRIGGERS_BUCKET_B)
]
mock_create_s3_event.assert_has_calls(s3_calls, any_order=True)
def test_create_s3_event_multiple_filters(mock_get_properties, mock_create_s3_event, mock_remove_perms, mock_boto3, mock_arn, mock_perms):
"""Try to create a lambda with two s3 triggers from the same bucket."""
triggers = TRIGGERS_MULTIPLE_FILTER
properties = get_properties_with_triggers(TRIGGERS_MULTIPLE_FILTER)
mock_get_properties.return_value = properties
mock_arn.return_value = 'fake_arn'
events = LambdaEvent(app='test_app', env='test_env', region='us-east-1', prop_path='other')
events.create_lambda_events()
mock_create_s3_event.assert_called_with(app_name='test_app', env='test_env', region='us-east-1', bucket='my.shared.bucket', triggers=triggers)