depot/third_party/nixpkgs/pkgs/tools/misc/parquet-tools/moto5.patch

28 lines
610 B
Diff

diff --git a/tests/fixtures/aws.py b/tests/fixtures/aws.py
index 7eea4bd..9fb3345 100644
--- a/tests/fixtures/aws.py
+++ b/tests/fixtures/aws.py
@@ -1,15 +1,17 @@
import boto3
-from moto import mock_s3
import pytest
+try:
+ # Moto 4.x
+ from moto import mock_s3
+except ImportError:
+ # Moto 5.x
+ from moto import mock_aws as mock_s3
@pytest.fixture
def aws_session():
- mock_s3_server = mock_s3()
- mock_s3_server.start()
- yield boto3.Session()
- mock_s3_server.stop()
-
+ with mock_s3():
+ yield boto3.Session()
@pytest.fixture
def aws_s3_bucket(aws_session):