moto_mock_aws_decorator_s3_pytest_example.py

1import boto3
2import pytest
3from moto import mock_aws
4
5@mock_aws
6def test_my_model_save():
7    conn = boto3.resource("s3", region_name="us-east-1")
8    # We need to create the bucket since this is all in-memory
9    conn.create_bucket(Bucket="mybucket")
10
11    model_instance = MyModel("steve", "is awesome")
12    model_instance.save()
13
14    body = conn.Object("mybucket", "steve").get()["Body"].read().decode("utf-8")
15
16    assert body == "is awesome"
17
18class MyModel:
19    def __init__(self, name, value):
20        self.name = name
21        self.value = value
22
23    def save(self):
24        s3 = boto3.client("s3", region_name="us-east-1")
25        s3.put_object(Bucket="mybucket", Key=self.name, Body=self.value)