def test_s3_event_object_bucket_key_missing(self):
event = {"Records": [{"s3": {"bucket": {}, "object": {}}}]}
with self.assertRaises(Exception) as cm:
event_object(event)
self.assertEquals(
cm.exception.message,
"Unable to retrieve object from event.\n{}".format(event),
)
def test_s3_event_object_missing_key(self):
event = {
"Records": [{
"s3": {
"bucket": {
"name": self.s3_bucket_name
}
}
}]
}
with self.assertRaises(Exception) as cm:
event_object(event)
self.assertEquals(cm.exception.message, "No key found in event!")
def test_s3_event_test_object(self):
event = {'Service': 'Amazon S3',
'Event': 's3:TestEvent',
'Time': '2020-05-27T12:08:55.740Z',
'Bucket': 'test.bucket',
'RequestId': '123123123123123123123123',
'HostId': 'asdfasdfasdfasdfasdfasdfasdfasdfasfdasdfasdf'}
s3_obj = event_object(event)
self.assertIsNone(s3_obj)
def test_s3_event_object(self):
event = {
"Records": [
{
"s3": {
"bucket": {"name": self.s3_bucket_name},
"object": {"key": self.s3_key_name},
}
}
]
}
s3_obj = event_object(event)
expected_s3_object = self.s3.Object(self.s3_bucket_name, self.s3_key_name)
self.assertEquals(s3_obj, expected_s3_object)
def test_s3_event_object(self):
body = {
"bucket": {
"name": self.s3_bucket_name
},
"object": {
"key": self.s3_key_name
},
}
s3_event = {"Records": [{"EventSource": "aws:s3", "s3": body}]}
s3_obj = event_object(s3_event)
expected_s3_object = self.s3.Object(self.s3_bucket_name,
self.s3_key_name)
self.assertEquals(s3_obj, expected_s3_object)
def test_sns_event_object(self):
event = {
"Records": [
{
"s3": {
"bucket": {"name": self.s3_bucket_name},
"object": {"key": self.s3_key_name},
}
}
]
}
sns_event = {"Records": [{"Sns": {"Message": json.dumps(event)}}]}
s3_obj = event_object(sns_event, self.s3)
expected_s3_object = self.s3.Object(self.s3_bucket_name, self.s3_key_name)
self.assertEquals(s3_obj, expected_s3_object)
def test_sqs_event_object(self):
body = {
"bucket": {
"name": self.s3_bucket_name
},
"object": {
"key": self.s3_key_name
},
}
sqs_event = {
"Records": [{
"EventSource": "aws:sqs",
"body": json.dumps(body)
}]
}
s3_obj = event_object(sqs_event)
expected_s3_object = self.s3.Object(self.s3_bucket_name,
self.s3_key_name)
self.assertEqual(s3_obj, expected_s3_object)
def test_s3_event_object_no_records(self):
event = {"Records": []}
with self.assertRaises(Exception) as cm:
event_object(event)
self.assertEquals(cm.exception.message,
"No records found in event!")
