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)