def test_multipass(): """Ensure the correct calls are made to attempt multiple passwords.""" runner = Bladerunner() sshc = Mock() with patch.object(runner, "login", return_value=("fake", 1)) as p_login: assert runner._multipass(sshc, "hunter11", 123) == ("fake", 1) p_login.assert_called_once_with(sshc, "hunter11", 123)
def test_multipass_failure(): """Ensure all passwords are tried before returning failure.""" runner = Bladerunner() sshc = Mock() with patch.object(runner, "login", return_value=("fake", -4)) as p_login: ret = runner._multipass(sshc, ["hunter1", "hunter2"], 12) assert p_login.mock_calls == [ call(sshc, "hunter1", 12), call(sshc, "hunter2", 12), ] assert ret == (None, -4)