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)