parquet-tools: support Moto 5.x

This makes the test fixtures compatible with the new Moto API.

Close #301032

Link: https://github.com/ktrueda/parquet-tools/issues/54
Link: https://github.com/ktrueda/parquet-tools/pull/55
This commit is contained in:
edef 2024-04-03 11:55:29 +00:00
parent d8fe5e6c92
commit 51912b15f8
2 changed files with 34 additions and 0 deletions

View File

@ -18,6 +18,12 @@ buildPythonApplication rec {
hash = "sha256-2jIwDsxB+g37zV9hLc2VNC5YuZXTpTmr2aQ72AeHYJo=";
};
patches = [
# support Moto 5.x
# https://github.com/ktrueda/parquet-tools/pull/55
./moto5.patch
];
postPatch = ''
substituteInPlace tests/test_inspect.py \
--replace "parquet-cpp-arrow version 5.0.0" "parquet-cpp-arrow version ${pyarrow.version}" \

View File

@ -0,0 +1,28 @@
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):