Python FileProcessor._zip_bytes_io示例

def test_zip_bytes_io(s3):
    file_processor = FileProcessor({}, {}, s3)

    actual = file_processor._zip_bytes_io(pytest.bucket_origin, pytest.key)

    assert type(actual) is BytesIO
    with ZipFile(actual) as zip_obj:
        assert zip_obj.namelist() == [pytest.file_zipped]
        with zip_obj.open(pytest.file_zipped) as file_obj:
            assert file_obj.read() == pytest.body_zipped.encode()

示例#2

显示文件

文件： test_file_processor.py 项目： knakayama/aws-sam-ci-cd

def test_upload_file_obj(s3, s3_prefix):
    bytes_io = BytesIO()
    key = f'{s3_prefix}/{pytest.file_zipped}'
    os.environ['BUCKET_PROCESSED'] = pytest.bucket_processed

    file_processor = FileProcessor({}, {}, s3)
    zip_bytes_io = file_processor._zip_bytes_io(pytest.bucket_origin,
                                                pytest.key)
    file_processor._upload_file_obj(zip_bytes_io)
    s3.download_fileobj(pytest.bucket_processed, key, bytes_io)

    assert type(bytes_io) is BytesIO
    assert bytes_io.getvalue() == pytest.body_zipped.encode()