import json from typing import List from unittest import SkipTest from uuid import uuid4 import boto3 import pytest from moto import mock_aws, settings from moto.core import DEFAULT_ACCOUNT_ID as ACCOUNT_ID from moto.s3.models import FakeBucket, FakeKey from moto.s3.notifications import ( S3NotificationEvent, _detail_type, _send_event_bridge_message, ) REGION_NAME = "us-east-1" @pytest.mark.parametrize( "event_names, expected_event_message", [ ( [ S3NotificationEvent.OBJECT_CREATED_PUT_EVENT, S3NotificationEvent.OBJECT_CREATED_POST_EVENT, S3NotificationEvent.OBJECT_CREATED_COPY_EVENT, S3NotificationEvent.OBJECT_CREATED_COMPLETE_MULTIPART_UPLOAD_EVENT, ], "Object Created", ), ( [ S3NotificationEvent.OBJECT_REMOVED_DELETE_EVENT, S3NotificationEvent.OBJECT_REMOVED_DELETE_MARKER_CREATED_EVENT, ], "Object Deleted", ), ([S3NotificationEvent.OBJECT_RESTORE_POST_EVENT], "Object Restore Initiated"), ( [S3NotificationEvent.OBJECT_RESTORE_COMPLETED_EVENT], "Object Restore Completed", ), ( [S3NotificationEvent.OBJECT_RESTORE_DELETE_EVENT], "Object Restore Expired", ), ( [S3NotificationEvent.LIFECYCLE_TRANSITION_EVENT], "Object Storage Class Changed", ), ([S3NotificationEvent.INTELLIGENT_TIERING_EVENT], "Object Access Tier Changed"), ([S3NotificationEvent.OBJECT_ACL_EVENT], "Object ACL Updated"), ([S3NotificationEvent.OBJECT_TAGGING_PUT_EVENT], "Object Tags Added"), ([S3NotificationEvent.OBJECT_TAGGING_DELETE_EVENT], "Object Tags Deleted"), ], ) def test_detail_type(event_names: List[str], expected_event_message: str): for event_name in event_names: assert _detail_type(event_name) == expected_event_message def test_detail_type_unknown_event(): with pytest.raises(ValueError) as ex: _detail_type("unknown event") assert ( str(ex.value) == "unsupported event `unknown event` for s3 eventbridge notification (https://docs.aws.amazon.com/AmazonS3/latest/userguide/EventBridge.html)" ) @mock_aws def test_send_event_bridge_message(): # setup mocks events_client = boto3.client("events", region_name=REGION_NAME) logs_client = boto3.client("logs", region_name=REGION_NAME) rule_name = "test-rule" events_client.put_rule( Name=rule_name, EventPattern=json.dumps({"account": [ACCOUNT_ID]}) ) log_group_name = "/test-group" logs_client.create_log_group(logGroupName=log_group_name) mocked_bucket = FakeBucket(str(uuid4()), ACCOUNT_ID, REGION_NAME) mocked_key = FakeKey( "test-key", bytes("test content", encoding="utf-8"), ACCOUNT_ID ) # do nothing if event target does not exists. _send_event_bridge_message( ACCOUNT_ID, mocked_bucket, S3NotificationEvent.OBJECT_CREATED_PUT_EVENT, mocked_key, ) assert ( len(logs_client.filter_log_events(logGroupName=log_group_name)["events"]) == 0 ) # do nothing even if an error is raised while sending events. events_client.put_targets( Rule=rule_name, Targets=[ { "Id": "test", "Arn": f"arn:aws:logs:{REGION_NAME}:{ACCOUNT_ID}:log-group:{log_group_name}", } ], ) _send_event_bridge_message(ACCOUNT_ID, mocked_bucket, "unknown-event", mocked_key) assert ( len(logs_client.filter_log_events(logGroupName=log_group_name)["events"]) == 0 ) if not settings.TEST_DECORATOR_MODE: raise SkipTest(("Doesn't quite work right with the Proxy or Server")) # an event is correctly sent to the log group. _send_event_bridge_message( ACCOUNT_ID, mocked_bucket, S3NotificationEvent.OBJECT_CREATED_PUT_EVENT, mocked_key, ) events = logs_client.filter_log_events(logGroupName=log_group_name)["events"] assert len(events) == 1 event_msg = json.loads(events[0]["message"]) assert event_msg["detail-type"] == "Object Created" assert event_msg["source"] == "aws.s3" assert event_msg["region"] == REGION_NAME assert event_msg["resources"] == [f"arn:aws:s3:::{mocked_bucket.name}"] event_detail = event_msg["detail"] assert event_detail["bucket"] == {"name": mocked_bucket.name} assert event_detail["object"]["key"] == mocked_key.name assert event_detail["reason"] == "ObjectCreated"