File tree 2 files changed +15
-6
lines changed
2 files changed +15
-6
lines changed Original file line number Diff line number Diff line change @@ -163,11 +163,15 @@ async def retry_target_stream(
163
163
164
164
# TODO: Remove this conditional once the minimum supported Python version is 3.11
165
165
if sys .version_info [:3 ] >= (3 , 11 , 0 ):
166
- await cast (AsyncGenerator ["_Y" , None ], target_iterator ).athrow (
167
- sys .exception ()) # type: ignore
166
+ await cast (
167
+ AsyncGenerator ["_Y" , None ], target_iterator
168
+ ).athrow (
169
+ sys .exception ()
170
+ ) # type: ignore
168
171
else :
169
- await cast (AsyncGenerator ["_Y" , None ], target_iterator ).athrow (
170
- * sys .exc_info ())
172
+ await cast (
173
+ AsyncGenerator ["_Y" , None ], target_iterator
174
+ ).athrow (* sys .exc_info ())
171
175
else :
172
176
raise
173
177
return
Original file line number Diff line number Diff line change 16
16
import re
17
17
18
18
import mock
19
+ import pytest
19
20
import requests .exceptions
20
21
21
22
from google .api_core import exceptions
@@ -136,7 +137,8 @@ def test_constructor_options(self):
136
137
assert retry_ ._timeout == 4
137
138
assert retry_ ._on_error is _some_function
138
139
139
- def test_with_timeout (self ):
140
+ @pytest .mark .parametrize ("use_deadline" , [True , False ])
141
+ def test_with_timeout (self , use_deadline ):
140
142
retry_ = self ._make_one (
141
143
predicate = mock .sentinel .predicate ,
142
144
initial = 1 ,
@@ -145,9 +147,12 @@ def test_with_timeout(self):
145
147
timeout = 4 ,
146
148
on_error = mock .sentinel .on_error ,
147
149
)
148
- new_retry = retry_ .with_timeout (42 )
150
+ new_retry = (
151
+ retry_ .with_timeout (42 ) if not use_deadline else retry_ .with_deadline (42 )
152
+ )
149
153
assert retry_ is not new_retry
150
154
assert new_retry ._timeout == 42
155
+ assert new_retry .timeout == 42 if not use_deadline else new_retry .deadline == 42
151
156
152
157
# the rest of the attributes should remain the same
153
158
assert new_retry ._predicate is retry_ ._predicate
You can’t perform that action at this time.
0 commit comments