Skip to content

Commit b1c2b3b

Browse files
committed
lint fix
1 parent 72fa58c commit b1c2b3b

2 files changed

Lines changed: 10 additions & 4 deletions

File tree

packages/google-cloud-storage/tests/unit/asyncio/test_async_appendable_object_writer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -529,16 +529,18 @@ async def test_append_persists_metadata_on_resumption(self, mock_appendable_writ
529529

530530
# Setup a fast retry policy to fail quickly in test
531531
from google.api_core.retry_async import AsyncRetry
532+
532533
fast_retry = AsyncRetry(
533534
predicate=lambda e: True,
534535
initial=0.01,
535536
maximum=0.01,
536537
multiplier=1.0,
537-
deadline=0.1
538+
deadline=0.1,
538539
)
539540

540541
# Act - append (should trigger retry and use stored metadata)
541542
from google.api_core.exceptions import RetryError
543+
542544
try:
543545
await writer.append(b"data", retry_policy=fast_retry)
544546
except RetryError:

packages/google-cloud-storage/tests/unit/asyncio/test_async_multi_range_downloader.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -742,8 +742,7 @@ async def test_download_ranges_persists_metadata_on_resumption(
742742

743743
# Act - Create MRD with metadata
744744
mrd = await AsyncMultiRangeDownloader.create_mrd(
745-
mock_client, _TEST_BUCKET_NAME, _TEST_OBJECT_NAME,
746-
metadata=test_metadata
745+
mock_client, _TEST_BUCKET_NAME, _TEST_OBJECT_NAME, metadata=test_metadata
747746
)
748747

749748
# Assert first open used metadata
@@ -752,6 +751,7 @@ async def test_download_ranges_persists_metadata_on_resumption(
752751

753752
# Setup resumption trigger
754753
from google.api_core import exceptions as core_exceptions
754+
755755
retryable_exc = core_exceptions.ServiceUnavailable("Retry me")
756756

757757
# Mock multiplexer to trigger retry
@@ -761,6 +761,7 @@ async def test_download_ranges_persists_metadata_on_resumption(
761761
# Configure reopen_stream to execute the factory
762762
async def fake_reopen_stream(broken_gen, stream_factory):
763763
await stream_factory()
764+
764765
mrd._multiplexer.reopen_stream = AsyncMock(side_effect=fake_reopen_stream)
765766

766767
mrd._multiplexer.send = AsyncMock(side_effect=retryable_exc)
@@ -771,23 +772,26 @@ async def fake_reopen_stream(broken_gen, stream_factory):
771772

772773
# We need mock_queue.get to return a StreamError to propagate exception to retry manager
773774
from google.cloud.storage.asyncio._stream_multiplexer import _StreamError
775+
774776
mock_queue.get.return_value = _StreamError(retryable_exc, generation=1)
775777

776778
mock_random_int.return_value = 123
777779

778780
# Setup a fast retry policy to fail quickly in test
779781
from google.api_core.retry_async import AsyncRetry
782+
780783
fast_retry = AsyncRetry(
781784
predicate=lambda e: True,
782785
initial=0.01,
783786
maximum=0.01,
784787
multiplier=1.0,
785-
deadline=0.1
788+
deadline=0.1,
786789
)
787790

788791
# Act - download ranges (should trigger retry and use stored metadata)
789792
buffer = BytesIO()
790793
from google.api_core.exceptions import RetryError
794+
791795
try:
792796
await mrd.download_ranges([(0, 10, buffer)], retry_policy=fast_retry)
793797
except RetryError:

0 commit comments

Comments
 (0)