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():
"""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)
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)
