File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -240,12 +240,12 @@ def raise_channel_exception_once(*args, **kwargs):
240240 raise_channel_exception_once .should_raise_channel_exception = True
241241
242242 def generate_fake_client ():
243- fake_client = Fake ('SSHClient' , allows_any_call = True ) # TODO expect_call=True ?
244- fake_client .provides ('connect' ).calls (raise_channel_exception_once )
243+ fake_client = Fake ('SSHClient' , allows_any_call = True )
244+ fake_client .expects ('connect' ).calls (raise_channel_exception_once )
245245 return fake_client
246246
247247 fake_ssh = Fake ('ssh' , allows_any_call = True )
248- fake_ssh .provides ('SSHClient' ).calls (generate_fake_client )
248+ fake_ssh .expects ('SSHClient' ).calls (generate_fake_client )
249249
250250 # We need the real exceptions here to preserve the inheritence structure
251251 # and for except clauses because python3 is picky about that
You can’t perform that action at this time.
0 commit comments