python - 在 Pytest 中测试 AWS s3 PUT 和下载

Question

我有以下功能。

def test_download(test_args):
    mock = mock_s3()
    mock.start()
    conn = boto3.resource('s3', region_name='us-west-2')
    conn.create_bucket(Bucket=test_args.source_bucket)
    s3.Object(test_args.source_bucket, 'testing.txt').put(
        Body=open("testing.txt", 'rb'))
    handler_client = HandlerClient(test_args)
    handler_client.get_s3_file()

使用 from moto import mock_s3

我去测试的时候遇到了错误。

    def add_auth(self, request):
        if self.credentials is None:
>           raise NoCredentialsError
E           botocore.exceptions.NoCredentialsError: Unable to locate credentials at line
>       `conn.create_bucket(Bucket=test_args.source_bucket)`

我试过装饰器以及

with moto.mock_s3():

所有人都得到同样的错误。我如何解决这个问题以创建一个假桶，将一个项目放入其中。

Answer 1

不幸的是，

moto 没有模拟 boto3 的授权部分。因此，您需要设置 AWS 授权链的某些部分，例如导出 AWS_ACCESS_KEY_ID 和 AWS_SECRET_ACCESS_KEY。我们通常会在每次测试前设置一个 fixture 来更新环境:

@pytest.fixture(scope='function')
def context():
    context = attrdict.AttrMap()
    orig_env = os.environ.copy()
    os.environ['AWS_ACCESS_KEY_ID'] = 'foo'
    os.environ['AWS_SECRET_ACCESS_KEY'] = 'bar'
    context.os = {'environ': os.environ}
    yield context
    os.environ = orig_env

导入部分是设置AWS_ACCESS_KEY_ID和AWS_SECRET_ACCESS_KEY，然后yielding，然后恢复os.environ 回到原来的状态。您不必使用 context = attrdict.AttrMap 部分，我们喜欢这样，因此使用此夹具的测试函数可以访问测试 os.environ(和其他测试) 属性。