嘲讽boto3 S3客户端方法Python
我试图从boto3 s3客户端对象中嘲讽singluar方法来抛出异常。 但我需要所有其他方法来正常工作。
这样，我可以在执行upload_part_copy时发生错误时测试单个异常测试
第一次尝试
import boto3
from mock import patch
with patch('botocore.client.S3.upload_part_copy', side_effect=Exception('Error Uploading')) as mock:
client = boto3.client('s3')
# Should return actual result
o = client.get_object(Bucket='my-bucket', Key='my-key')
# Should return mocked exception
e = client.upload_part_copy()
但是，这给出了以下错误：
ImportError: No module named S3
第二次尝试
在查看botocore.client.py源代码后，我发现它正在做一些巧妙的事情，并且upload_part_copy方法不存在。 我发现它似乎调用BaseClient._make_api_call所以我试图嘲笑它
import boto3
from mock import patch
with patch('botocore.client.BaseClient._make_api_call', side_effect=Exception('Error Uploading')) as mock:
client = boto3.client('s3')
# Should return actual result
o = client.get_object(Bucket='my-bucket', Key='my-key')
# Should return mocked exception
e = client.upload_part_copy()
这会抛出一个异常，但会抛出一个异常......但是在我想避免的get_object上。
关于如何仅在upload_part_copy方法上抛出异常的任何想法？
Botocore有一个客户端stubber可以用于这个目的：docs。
以下是一个错误的例子：
import boto3
from botocore.stub import Stubber
client = boto3.client('s3')
stubber = Stubber(client)
stubber.add_client_error('upload_part_copy')
stubber.activate()
# Will raise a ClientError
client.upload_part_copy()
下面是一个正常响应的例子。此外，现在可以在上下文中使用stubber。 请注意，stubber会尽可能验证您提供的响应是否与服务实际返回的内容匹配，这一点很重要。 这并不完美，但它会保护你免于插入总的无意义答复。
import boto3
from botocore.stub import Stubber
client = boto3.client('s3')
stubber = Stubber(client)
list_buckets_response = {
"Owner": {
"DisplayName": "name",
"ID": "EXAMPLE123"
},
"Buckets": [{
"CreationDate": "2016-05-25T16:55:48.000Z",
"Name": "foo"
}]
}
expected_params = {}
stubber.add_response('list_buckets', list_buckets_response, expected_params)
with stubber:
response = client.list_buckets()
assert response == list_buckets_response
只要我在这里发布，我设法提出了一个解决方案。 这里是希望它有助于:)
import botocore
from botocore.exceptions import ClientError
from mock import patch
import boto3
orig = botocore.client.BaseClient._make_api_call
def mock_make_api_call(self, operation_name, kwarg):
if operation_name == 'UploadPartCopy':
parsed_response = {'Error': {'Code': '500', 'Message': 'Error Uploading'}}
raise ClientError(parsed_response, operation_name)
return orig(self, operation_name, kwarg)
with patch('botocore.client.BaseClient._make_api_call', new=mock_make_api_call):
client = boto3.client('s3')
# Should return actual result
o = client.get_object(Bucket='my-bucket', Key='my-key')
# Should return mocked exception
e = client.upload_part_copy()
Jordan Philips也利用botocore.stub.Stubber类发布了一个很好的解决方案。 虽然是一个更清洁的解决方案，但我无法嘲笑特定的操作。
下面是一个简单的python unittest的例子，可以用来伪造client = boto3.client（'ec2'） api call ...
import boto3
class MyAWSModule():
def __init__(self):
client = boto3.client('ec2')
tags = client.describe_tags(DryRun=False)
class TestMyAWSModule(unittest.TestCase):
@mock.patch("boto3.client.get_tags")
@mock.patch("boto3.client")
def test_open_file_with_existing_file(self, mock_boto_client, mock_describe_tags):
mock_boto_client.return_value = mock_get_tags_response
my_aws_module = MyAWSModule()
mock_boto_client.assert_call_once('ec2')
mock_describe_tags.assert_call_once_with(DryRun=False)
mock_get_tags_response = {
'Tags': [
{
'ResourceId': 'string',
'ResourceType': 'customer-gateway',
'Key': 'string',
'Value': 'string'
},
],
'NextToken': 'string'
}
希望这有助于。
链接地址: http://www.djcxy.com/p/27925.html