def test_password(capsys):
process('keygen --password the password')
out, err = capsys.readouterr()
assert out.rstrip() == 'Created private RSA key {!r}'.format(get_default_key_path())
assert not err
load_key(get_default_key_path(), password='******')
def test_out_path(capsys):
key_path = os.path.join(tempfile.gettempdir(), 'key.private')
try:
os.remove(key_path)
except OSError:
pass
assert not os.path.isfile(key_path)
process('keygen --out {}'.format(key_path))
out, err = capsys.readouterr()
assert out.rstrip() == 'Created private RSA key {!r}'.format(key_path)
assert not err
assert os.path.isfile(key_path)
load_key(key_path)
os.remove(key_path)
def test_password_file(capsys):
pw_file = os.path.join(tempfile.gettempdir(), 'password.tmp')
with open(pw_file, mode='wt') as fp:
fp.write('the password')
process('keygen --password {}'.format(pw_file))
out, err = capsys.readouterr()
assert not err
assert out.splitlines() == [
'Reading the key password from the file',
'Created private RSA key {!r}'.format(get_default_key_path())
]
load_key(get_default_key_path(), password='******')
os.remove(pw_file)
def test_algorithm(algorithm, capsys):
process('keygen {}'.format(algorithm))
out, err = capsys.readouterr()
assert out.rstrip() == 'Created private {} key {!r}'.format(algorithm.upper(), get_default_key_path())
assert not err
load_key(get_default_key_path())
def test_no_args(capsys):
process('keygen')
out, err = capsys.readouterr()
assert out.rstrip() == 'Created private RSA key {!r}'.format(get_default_key_path())
assert not err
load_key(get_default_key_path())