mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-08 23:38:49 +00:00
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
import sys
|
|
from pathlib import Path
|
|
from tempfile import TemporaryDirectory
|
|
|
|
from helpers import create_vault, TEST_SEED, TEST_PASSWORD
|
|
|
|
sys.path.append(str(Path(__file__).resolve().parents[1]))
|
|
|
|
from password_manager.entry_management import EntryManager
|
|
from password_manager.backup import BackupManager
|
|
from password_manager.vault import Vault
|
|
from password_manager.config_manager import ConfigManager
|
|
|
|
|
|
def test_add_and_retrieve_ssh_key_pair():
|
|
with TemporaryDirectory() as tmpdir:
|
|
tmp_path = Path(tmpdir)
|
|
vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
|
|
cfg_mgr = ConfigManager(vault, tmp_path)
|
|
backup_mgr = BackupManager(tmp_path, cfg_mgr)
|
|
entry_mgr = EntryManager(vault, backup_mgr)
|
|
|
|
index = entry_mgr.add_ssh_key("ssh", TEST_SEED)
|
|
entry = entry_mgr.retrieve_entry(index)
|
|
assert entry == {
|
|
"type": "ssh",
|
|
"kind": "ssh",
|
|
"index": index,
|
|
"label": "ssh",
|
|
"notes": "",
|
|
}
|
|
|
|
priv1, pub1 = entry_mgr.get_ssh_key_pair(index, TEST_SEED)
|
|
priv2, pub2 = entry_mgr.get_ssh_key_pair(index, TEST_SEED)
|
|
assert priv1 == priv2
|
|
assert pub1 == pub2
|