import sys
from pathlib import Path
import pytest
from helpers import create_vault, TEST_SEED, TEST_PASSWORD

sys.path.append(str(Path(__file__).resolve().parents[1]))

from password_manager.migrations import LATEST_VERSION


def setup(tmp_path: Path):
    vault, enc_mgr = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
    return enc_mgr, vault


def test_migrate_v0_to_v2(tmp_path: Path):
    enc_mgr, vault = setup(tmp_path)
    legacy = {"passwords": {"0": {"website": "a", "length": 8}}}
    enc_mgr.save_json_data(legacy)
    data = vault.load_index()
    assert data["schema_version"] == LATEST_VERSION
    expected_entry = {"website": "a", "length": 8, "type": "password", "notes": ""}
    assert data["entries"]["0"] == expected_entry


def test_migrate_v1_to_v2(tmp_path: Path):
    enc_mgr, vault = setup(tmp_path)
    legacy = {"schema_version": 1, "passwords": {"0": {"website": "b", "length": 10}}}
    enc_mgr.save_json_data(legacy)
    data = vault.load_index()
    assert data["schema_version"] == LATEST_VERSION
    expected_entry = {
        "website": "b",
        "length": 10,
        "type": "password",
        "notes": "",
    }
    assert data["entries"]["0"] == expected_entry


def test_error_on_future_version(tmp_path: Path):
    enc_mgr, vault = setup(tmp_path)
    future = {"schema_version": LATEST_VERSION + 1, "entries": {}}
    enc_mgr.save_json_data(future)
    with pytest.raises(ValueError):
        vault.load_index()
