@@ -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