def test_run_safely_to_parralel():
"""Ensure after the first success the rest are run in parallel."""
runner = Bladerunner({
"username": "******",
"password": "******",
"progressbar": True,
"port": 2244,
})
# we have to set up the progressbar ourselves here, normally it'd happen in
# run once we know the len of servers to run on.
runner.progress = ProgressBar(3, runner.options)
with patch.object(runner, "connect", return_value=("ok", 0)) as p_connect:
with patch.object(runner, "send_commands", return_value=[]) as p_send:
with patch.object(runner, "close") as p_close:
with patch.object(runner, "_run_parallel_no_check") as p_run:
with patch.object(runner.progress, "update") as p_update:
runner._run_parallel_safely(["1st", "2nd", "3rd"])
p_connect.assert_called_once_with(
"1st",
"broguy",
"hunter14",
2244,
)
p_send.assert_called_once_with("ok", "1st")
p_close.assert_called_once_with("ok", True)
# if the progressbar is used we should update it once for the check run
assert p_update.called
p_run.assert_called_once_with(["2nd", "3rd"])
def test_run_safely_to_parralel():
"""Ensure after the first success the rest are run in parallel."""
runner = Bladerunner({
"username": "******",
"password": "******",
"progressbar": True,
"port": 2244,
})
# we have to set up the progressbar ourselves here, normally it'd happen in
# run once we know the len of servers to run on.
runner.progress = ProgressBar(3, runner.options)
with patch.object(runner, "connect", return_value=("ok", 0)) as p_connect:
with patch.object(runner, "send_commands", return_value=[]) as p_send:
with patch.object(runner, "close") as p_close:
with patch.object(runner, "_run_parallel_no_check") as p_run:
with patch.object(runner.progress, "update") as p_update:
runner._run_parallel_safely(["1st", "2nd", "3rd"])
p_connect.assert_called_once_with(
"1st",
"broguy",
"hunter14",
2244,
)
p_send.assert_called_once_with("ok", "1st")
p_close.assert_called_once_with("ok", True)
# if the progressbar is used we should update it once for the check run
assert p_update.called
p_run.assert_called_once_with(["2nd", "3rd"])
def test_run_safely_to_serial():
"""Ensure we only carry on with parallel no check on good first login."""
runner = Bladerunner({
"username": "******",
"password": "******",
"port": 202,
})
with patch.object(runner, "connect", return_value=(None, -2)) as p_connect:
with patch.object(runner, "_run_serial", return_value=[]) as p_run:
ret = runner._run_parallel_safely(["one", "two", "three"])
p_connect.assert_called_once_with(
"one",
"dudebro",
"hunter99",
202,
)
p_run.assert_called_once_with(["two", "three"])
assert ret == [{"name": "one", "results": [("login", runner.errors[1])]}]
def test_run_safely_to_serial():
"""Ensure we only carry on with parallel no check on good first login."""
runner = Bladerunner({
"username": "******",
"password": "******",
"port": 202,
})
with patch.object(runner, "connect", return_value=(None, -2)) as p_connect:
with patch.object(runner, "_run_serial", return_value=[]) as p_run:
ret = runner._run_parallel_safely(["one", "two", "three"])
p_connect.assert_called_once_with(
"one",
"dudebro",
"hunter99",
202,
)
p_run.assert_called_once_with(["two", "three"])
assert ret == [{"name": "one", "results": [("login", runner.errors[1])]}]
