mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2025-09-09 15:58:48 +00:00
47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
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()
|