mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2026-10-27 07:52:03 +00:00
40 lines
1.4 KiB
Python
40 lines
1.4 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 seedpass.core.entry_management import EntryManager
|
|
from seedpass.core.backup import BackupManager
|
|
from seedpass.core.vault import Vault
|
|
from seedpass.core.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"
|
|
assert entry["kind"] == "ssh"
|
|
assert entry["index"] == index
|
|
assert entry["label"] == "ssh"
|
|
assert entry["notes"] == ""
|
|
assert entry["archived"] is False
|
|
assert entry["tags"] == []
|
|
assert entry["custom_fields"] == []
|
|
assert "date_added" in entry
|
|
assert "date_modified" in entry
|
|
|
|
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
|