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