def test_run_single_success():
"""Ensure the calls to send commands on a single host."""
runner = Bladerunner({
"username": "******",
"password": "******",
"progressbar": True,
"port": 2012,
})
runner.progress = ProgressBar(1, 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.progress, "update") as p_update:
runner._run_single("nowhere")
p_connect.assert_called_once_with(
"nowhere",
"dudeguybro",
"hunter40",
2012,
)
p_send.assert_called_once_with("ok", "nowhere")
p_close.assert_called_once_with("ok", True)
# if the progressbar is used we should tick it once for the check run
assert p_update.called
def test_run_single_success():
"""Ensure the calls to send commands on a single host."""
runner = Bladerunner({
"username": "******",
"password": "******",
"progressbar": True,
"port": 2012,
})
runner.progress = ProgressBar(1, 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.progress, "update") as p_update:
runner._run_single("nowhere")
p_connect.assert_called_once_with(
"nowhere",
"dudeguybro",
"hunter40",
2012,
)
p_send.assert_called_once_with("ok", "nowhere")
p_close.assert_called_once_with("ok", True)
# if the progressbar is used we should tick it once for the check run
assert p_update.called
def test_run_single_error():
"""Ensure an error is passed back during run_single on connect errors."""
runner = Bladerunner({
"username": "******",
"password": "******",
"progressbar": True,
"port": 2212,
})
runner.progress = ProgressBar(1, runner.options)
with patch.object(runner, "connect", return_value=(None, -3)) as p_connect:
with patch.object(runner.progress, "update") as p_update:
ret = runner._run_single("nowhere")
p_connect.assert_called_once_with(
"nowhere",
"dudeguy",
"hunter111",
2212,
)
# if the progressbar should tick regardless of success
assert p_update.called
assert ret == {"name": "nowhere", "results": [("login", runner.errors[2])]}
def test_run_single_error():
"""Ensure an error is passed back during run_single on connect errors."""
runner = Bladerunner(
username="******",
password="******",
progressbar=True,
port=2212,
)
runner.progress = ProgressBar(1, runner.options)
with patch.object(runner, "connect", return_value=(None, -3)) as p_connect:
with patch.object(runner.progress, "update") as p_update:
ret = runner._run_single("nowhere")
p_connect.assert_called_once_with(
"nowhere",
"dudeguy",
"hunter111",
2212,
)
# if the progressbar should tick regardless of success
assert p_update.called
assert ret == {"name": "nowhere", "results": [("login", runner.errors[2])]}
