def test_delete_s3_object_exception(self):
s3_stubber = Stubber(self.s3.meta.client)
with self.assertRaises(Exception) as cm:
with s3_stubber:
s3_obj = self.s3.Object(self.s3_bucket_name, self.s3_key_name)
delete_s3_object(s3_obj)
self.assertEquals(
cm.exception.message,
"Failed to delete infected file: {}.{}".format(
self.s3_bucket_name, self.s3_key_name),
)
def test_delete_s3_object(self):
s3_stubber = Stubber(self.s3.meta.client)
delete_object_response = {}
delete_object_expected_params = {
"Bucket": self.s3_bucket_name,
"Key": self.s3_key_name,
}
s3_stubber.add_response("delete_object", delete_object_response,
delete_object_expected_params)
with s3_stubber:
s3_obj = self.s3.Object(self.s3_bucket_name, self.s3_key_name)
delete_s3_object(s3_obj)