def test_get_password():
"""Ensure we're prompting the user with getpass."""
class FakeSettings(object):
usePassword = True
password = []
setsecond_password = False
second_password = []
setjumpbox_password = False
jump_host = []
jump_pass = []
with patch.object(getpass, "getpass", return_value="mock") as patched:
settings = get_passwords(FakeSettings())
patched.assert_called_once_with("Password: "******"password", "second_password", "jump_pass"]:
assert getattr(settings, config) == "mock", "password should be 'mock'"
def test_get_jump_password():
"""Ensure the jumpbox password is prompted for."""
class FakeSettings(object):
usePassword = True
password = "******"
setsecond_password = False
second_password = "******"
setjumpbox_password = True
jump_host = ["somehost"]
jump_pass = []
with patch.object(getpass, "getpass", return_value="mock") as patched:
settings = get_passwords(FakeSettings())
patched.assert_called_once_with("Jumpbox password: "******"mock"
assert settings.password == "hunter7"
assert settings.second_password == "hunter8"
def test_get_password():
"""Ensure we're prompting the user with getpass."""
class FakeSettings(object):
usePassword = True
password = []
setsecond_password = False
second_password = []
setjumpbox_password = False
jump_host = []
jump_pass = []
with patch.object(getpass, "getpass", return_value="mock") as patched:
settings = get_passwords(FakeSettings())
patched.assert_called_once_with("Password: "******"password", "second_password", "jump_pass"]:
assert getattr(settings, config) == "mock", "password should be 'mock'"
def test_get_jump_password():
"""Ensure the jumpbox password is prompted for."""
class FakeSettings(object):
usePassword = True
password = "******"
setsecond_password = False
second_password = "******"
setjumpbox_password = True
jump_host = ["somehost"]
jump_pass = []
with patch.object(getpass, "getpass", return_value="mock") as patched:
settings = get_passwords(FakeSettings())
patched.assert_called_once_with("Jumpbox password: "******"mock"
assert settings.password == "hunter7"
assert settings.second_password == "hunter8"
