Я хочу протестировать функцию save_doc с помощью pytest и высмеять приведенный ниже вызов.
boto3.resource('s3').Object(bucket_name, key).put(Body=json.dumps(body))
Я пытаюсь понять, как я могу использовать декоратор «заплатки», чтобы издеваться над этим вызовом. Есть ли какие-нибудь примеры/указатели, на которые я могу посмотреть?
def save_doc(doc_id, body):
bucket_name = os.environ['bucket_name']
key = '{}{}.json'.format(os.environ['key'], doc_id)
boto3.resource('s3').Object(bucket_name, key).put(Body=json.dumps(body))
Приведенный ниже тестовый пример работает для сценария, опубликованного в вопросах.
@mock_s3
def test_save_doc(doc_id, body):
bucket_name = os.environ['bucket_name']
key = os.environ['key']
conn = boto3.resource('s3', region_name='us-east-1')
conn.create_bucket(Bucket=bucket_name)
save_doc(doc_id, body)
response = conn.Object(bucket_name, key).get()['Body'].read().decode("utf-8")
assert body == response
Пакет Moto подходит для этого варианта использования — github.com/spulec/мото.