File tree Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Expand file tree Collapse file tree 1 file changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -280,7 +280,9 @@ def cancel(self):
280
280
281
281
class TestResumableBidiRpc (object ):
282
282
def test_initial_state (self ):
283
- bidi_rpc = bidi .ResumableBidiRpc (None , lambda _ : True )
283
+ callback = mock .Mock ()
284
+ callback .return_value = True
285
+ bidi_rpc = bidi .ResumableBidiRpc (None , callback )
284
286
285
287
assert bidi_rpc .is_active is False
286
288
@@ -380,7 +382,9 @@ def test_recv_recover_already_recovered(self):
380
382
grpc .StreamStreamMultiCallable ,
381
383
instance = True ,
382
384
side_effect = [call_1 , call_2 ])
383
- bidi_rpc = bidi .ResumableBidiRpc (start_rpc , lambda _ : True )
385
+ callback = mock .Mock ()
386
+ callback .return_value = True
387
+ bidi_rpc = bidi .ResumableBidiRpc (start_rpc , callback )
384
388
385
389
bidi_rpc .open ()
386
390
You can’t perform that action at this time.
0 commit comments