@@ -450,12 +450,28 @@ def test_009_execute_together(ssh, ssh2, execute_async, exec_result, run_paramet
450
450
451
451
if 0 == run_parameters ["ec" ]:
452
452
results = exec_helpers .SSHClient .execute_together (
453
- remotes = remotes , command = command , stdin = run_parameters .get ("stdin" , None )
453
+ remotes = remotes ,
454
+ command = command ,
455
+ stdin = run_parameters ["stdin" ],
456
+ open_stdout = run_parameters ["open_stdout" ],
457
+ open_stderr = run_parameters ["open_stderr" ],
454
458
)
455
459
execute_async .assert_has_calls (
456
460
(
457
- mock .call (command , stdin = run_parameters .get ("stdin" , None ), log_mask_re = None ),
458
- mock .call (command , stdin = run_parameters .get ("stdin" , None ), log_mask_re = None ),
461
+ mock .call (
462
+ command ,
463
+ stdin = run_parameters ["stdin" ],
464
+ log_mask_re = None ,
465
+ open_stdout = run_parameters ["open_stdout" ],
466
+ open_stderr = run_parameters ["open_stderr" ],
467
+ ),
468
+ mock .call (
469
+ command ,
470
+ stdin = run_parameters ["stdin" ],
471
+ log_mask_re = None ,
472
+ open_stdout = run_parameters ["open_stdout" ],
473
+ open_stderr = run_parameters ["open_stderr" ],
474
+ ),
459
475
)
460
476
)
461
477
assert results == {(host , port ): exec_result , (host2 , port ): exec_result }
@@ -472,12 +488,29 @@ def test_010_execute_together_expected(ssh, ssh2, execute_async, exec_result, ru
472
488
remotes = [ssh , ssh2 ]
473
489
474
490
results = exec_helpers .SSHClient .execute_together (
475
- remotes = remotes , command = command , stdin = run_parameters .get ("stdin" , None ), expected = [run_parameters ["ec" ]]
491
+ remotes = remotes ,
492
+ command = command ,
493
+ stdin = run_parameters ["stdin" ],
494
+ open_stdout = run_parameters ["open_stdout" ],
495
+ open_stderr = run_parameters ["open_stderr" ],
496
+ expected = [run_parameters ["ec" ]],
476
497
)
477
498
execute_async .assert_has_calls (
478
499
(
479
- mock .call (command , stdin = run_parameters .get ("stdin" , None ), log_mask_re = None ),
480
- mock .call (command , stdin = run_parameters .get ("stdin" , None ), log_mask_re = None ),
500
+ mock .call (
501
+ command ,
502
+ stdin = run_parameters .get ("stdin" , None ),
503
+ log_mask_re = None ,
504
+ open_stdout = run_parameters ["open_stdout" ],
505
+ open_stderr = run_parameters ["open_stderr" ],
506
+ ),
507
+ mock .call (
508
+ command ,
509
+ stdin = run_parameters .get ("stdin" , None ),
510
+ log_mask_re = None ,
511
+ open_stdout = run_parameters ["open_stdout" ],
512
+ open_stderr = run_parameters ["open_stderr" ],
513
+ ),
481
514
)
482
515
)
483
516
assert results == {(host , port ): exec_result , (host2 , port ): exec_result }
0 commit comments