From e7837bcfbeb99c164c78cfb934b333437e1eda3b Mon Sep 17 00:00:00 2001 From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com> Date: Sun, 13 Jul 2025 22:37:42 -0400 Subject: [PATCH] Add full sync roundtrip test --- src/tests/test_full_sync_roundtrip_new.py | 64 +++++++++++++++++++++++ 1 file changed, 64 insertions(+) create mode 100644 src/tests/test_full_sync_roundtrip_new.py diff --git a/src/tests/test_full_sync_roundtrip_new.py b/src/tests/test_full_sync_roundtrip_new.py new file mode 100644 index 0000000..2d0c36d --- /dev/null +++ b/src/tests/test_full_sync_roundtrip_new.py @@ -0,0 +1,64 @@ +import asyncio +from pathlib import Path +from tempfile import TemporaryDirectory + +from helpers import create_vault, dummy_nostr_client + +from password_manager.entry_management import EntryManager +from password_manager.backup import BackupManager +from password_manager.config_manager import ConfigManager +from password_manager.manager import PasswordManager, EncryptionMode + + +def _init_pm(dir_path: Path, client) -> PasswordManager: + vault, enc_mgr = create_vault(dir_path) + cfg_mgr = ConfigManager(vault, dir_path) + backup_mgr = BackupManager(dir_path, cfg_mgr) + entry_mgr = EntryManager(vault, backup_mgr) + + pm = PasswordManager.__new__(PasswordManager) + pm.encryption_mode = EncryptionMode.SEED_ONLY + pm.encryption_manager = enc_mgr + pm.vault = vault + pm.entry_manager = entry_mgr + pm.backup_manager = backup_mgr + pm.config_manager = cfg_mgr + pm.nostr_client = client + pm.fingerprint_dir = dir_path + pm.is_dirty = False + return pm + + +def test_full_sync_roundtrip(dummy_nostr_client, monkeypatch): + client, relay = dummy_nostr_client + with TemporaryDirectory() as tmpdir: + base = Path(tmpdir) + dir_a = base / "A" + dir_b = base / "B" + dir_a.mkdir() + dir_b.mkdir() + + pm_a = _init_pm(dir_a, client) + pm_b = _init_pm(dir_b, client) + + # Manager A publishes initial snapshot + pm_a.entry_manager.add_entry("site1", 12) + pm_a.sync_vault() + manifest_id = relay.manifests[-1].id + + # Manager B retrieves snapshot + pm_b.sync_index_from_nostr_if_missing() + entries = pm_b.entry_manager.list_entries() + assert [e[1] for e in entries] == ["site1"] + + # Manager A publishes delta with second entry + pm_a.entry_manager.add_entry("site2", 12) + delta_bytes = pm_a.vault.get_encrypted_index() or b"" + monkeypatch.setattr("nostr.client.time.time", lambda: 1) + asyncio.run(client.publish_delta(delta_bytes, manifest_id)) + + # Manager B fetches delta and updates + pm_b.sync_index_from_nostr() + pm_b.entry_manager.clear_cache() + labels = [e[1] for e in pm_b.entry_manager.list_entries()] + assert sorted(labels) == ["site1", "site2"]