@@ -195,9 +195,9 @@ async def test_state_lookup(self, mock_appendable_writer):
195195 writer ._is_stream_open = True
196196 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
197197
198- mock_appendable_writer ["mock_stream" ]. recv . return_value = (
199- storage_type . BidiWriteObjectResponse ( persisted_size = 100 )
200- )
198+ mock_appendable_writer [
199+ "mock_stream"
200+ ]. recv . return_value = storage_type . BidiWriteObjectResponse ( persisted_size = 100 )
201201
202202 size = await writer .state_lookup ()
203203
@@ -431,9 +431,9 @@ async def test_flush_resets_counters(self, mock_appendable_writer):
431431 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
432432 writer .bytes_appended_since_last_flush = 100
433433
434- mock_appendable_writer ["mock_stream" ]. recv . return_value = (
435- storage_type . BidiWriteObjectResponse ( persisted_size = 200 )
436- )
434+ mock_appendable_writer [
435+ "mock_stream"
436+ ]. recv . return_value = storage_type . BidiWriteObjectResponse ( persisted_size = 200 )
437437
438438 await writer .flush ()
439439
@@ -474,9 +474,9 @@ async def test_finalize_lifecycle(self, mock_appendable_writer):
474474 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
475475
476476 resource = storage_type .Object (size = 999 )
477- mock_appendable_writer ["mock_stream" ]. recv . return_value = (
478- storage_type . BidiWriteObjectResponse ( resource = resource )
479- )
477+ mock_appendable_writer [
478+ "mock_stream"
479+ ]. recv . return_value = storage_type . BidiWriteObjectResponse ( resource = resource )
480480
481481 res = await writer .finalize ()
482482
@@ -534,9 +534,9 @@ async def test_finalize_with_checksum(self, mock_appendable_writer):
534534 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
535535
536536 resource = storage_type .Object (size = 999 )
537- mock_appendable_writer ["mock_stream" ]. recv . return_value = (
538- storage_type . BidiWriteObjectResponse ( resource = resource )
539- )
537+ mock_appendable_writer [
538+ "mock_stream"
539+ ]. recv . return_value = storage_type . BidiWriteObjectResponse ( resource = resource )
540540
541541 checksum = 12345678
542542 res = await writer .finalize (full_object_checksum = checksum )
@@ -655,9 +655,7 @@ async def test_finalize_custom_retry_policy(self, mock_appendable_writer):
655655 writer ._is_stream_open = True
656656 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
657657
658- custom_policy = AsyncRetry (
659- predicate = lambda exc : isinstance (exc , exceptions .InternalServerError )
660- )
658+ custom_policy = AsyncRetry (predicate = lambda exc : isinstance (exc , exceptions .InternalServerError ))
661659 resource = storage_type .Object (size = 999 )
662660 mock_appendable_writer ["mock_stream" ].recv .return_value = (
663661 storage_type .BidiWriteObjectResponse (resource = resource )
@@ -667,9 +665,7 @@ async def test_finalize_custom_retry_policy(self, mock_appendable_writer):
667665 assert res == resource
668666
669667 @pytest .mark .asyncio
670- async def test_close_with_finalize_and_custom_retry_policy (
671- self , mock_appendable_writer
672- ):
668+ async def test_close_with_finalize_and_custom_retry_policy (self , mock_appendable_writer ):
673669 from google .api_core .retry_async import AsyncRetry
674670
675671 writer = self ._make_one (mock_appendable_writer ["mock_client" ])
@@ -701,3 +697,6 @@ async def test_close_retry_on_transient_error(self, mock_appendable_writer):
701697 assert writer .persisted_size == 999
702698 assert mock_appendable_writer ["mock_stream" ].send .await_count == 2
703699 assert not writer ._is_stream_open
700+
701+
702+
0 commit comments