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])]}