在boto3中，我如何使用Stubbermockdownload_fileobj這是一個(gè)資源方法？例如：import boto3from botocore.stub import Stubbers3 = boto3.resource('s3')def foo(s3): with open('filename', 'wb') as data: s3.download_fileobj('mybucket', 'mykey', data)def test_foo(): s3_test = boto3.resource('s3') s3_stub = Stubber(s3_test.meta.client) s3_stub.add_response() # something here with s3_stub: foo(s3_test)
1 回答
千巷貓影
TA貢獻(xiàn)1829條經(jīng)驗(yàn) 獲得超7個(gè)贊
get_fileobj 轉(zhuǎn)換為“head_object”和“get_object”調(diào)用。這是一個(gè)對(duì)兩個(gè)調(diào)用進(jìn)行存根的基本代碼片段。
bucket_name = 'mybucket'
key = 'mykey'
content = 'This is the content'
expected_params = {
'Bucket': bucket_name,
'Key': key,
}
# Head object
response = {
'ContentLength': 10,
'ContentType': 'utf-8',
'ResponseMetadata': {
'Bucket': bucket_name,
}
}
s3_stub.add_response('head_object', response, expected_params)
# Get object
data = BytesIO()
data.write(content.encode("utf-8")
data.seek(0)
response = {
'ContentLength': len(content),
'ContentType': 'utf-8',
'Body': data,
'ResponseMetadata': {
'Bucket': bucket_name,
}
}
s3_stub.add_response('get_object', response, expected_params)
添加回答
舉報(bào)
0/150
提交
取消