From 19881dbeebe4bfe717a08911e2cbb1157022c82a Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Tue, 5 Aug 2025 12:40:39 -0400
Subject: [PATCH] Test entry summary refresh on edits

---
 src/seedpass/core/manager.py                  | 10 ++--
 src/tests/test_get_entry_summaries_updates.py | 48 +++++++++++++++++++
 2 files changed, 53 insertions(+), 5 deletions(-)
 create mode 100644 src/tests/test_get_entry_summaries_updates.py

diff --git a/src/seedpass/core/manager.py b/src/seedpass/core/manager.py
index 94d7965..b3ba8d2 100644
--- a/src/seedpass/core/manager.py
+++ b/src/seedpass/core/manager.py
@@ -3813,12 +3813,12 @@ class PasswordManager:
                     print(colored("Invalid choice.", "red"))
                     continue
 
-                summaries = self.entry_manager.get_entry_summaries(
-                    filter_kind, include_archived=False
-                )
-                if not summaries:
-                    continue
                 while True:
+                    summaries = self.entry_manager.get_entry_summaries(
+                        filter_kind, include_archived=False
+                    )
+                    if not summaries:
+                        break
                     fp, parent_fp, child_fp = self.header_fingerprint_args
                     clear_header_with_notification(
                         self,
diff --git a/src/tests/test_get_entry_summaries_updates.py b/src/tests/test_get_entry_summaries_updates.py
new file mode 100644
index 0000000..8a00c5a
--- /dev/null
+++ b/src/tests/test_get_entry_summaries_updates.py
@@ -0,0 +1,48 @@
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from helpers import create_vault, TEST_SEED, TEST_PASSWORD
+
+from seedpass.core.entry_management import EntryManager
+from seedpass.core.backup import BackupManager
+from seedpass.core.config_manager import ConfigManager
+
+
+def _create_entry_manager(tmp_path: Path) -> EntryManager:
+    vault, _ = create_vault(tmp_path, TEST_SEED, TEST_PASSWORD)
+    cfg_mgr = ConfigManager(vault, tmp_path)
+    backup_mgr = BackupManager(tmp_path, cfg_mgr)
+    return EntryManager(vault, backup_mgr)
+
+
+def test_get_entry_summaries_updates_label():
+    with TemporaryDirectory() as tmpdir:
+        tmp_path = Path(tmpdir)
+        em = _create_entry_manager(tmp_path)
+        idx = em.add_entry("old", 8)
+
+        summaries = em.get_entry_summaries()
+        assert summaries == [(idx, "password", "old")]
+
+        em.modify_entry(idx, label="new")
+        summaries = em.get_entry_summaries()
+        assert summaries == [(idx, "password", "new")]
+
+
+def test_get_entry_summaries_updates_archive_restore():
+    with TemporaryDirectory() as tmpdir:
+        tmp_path = Path(tmpdir)
+        em = _create_entry_manager(tmp_path)
+        keep_idx = em.add_entry("keep", 8)
+        drop_idx = em.add_entry("drop", 8)
+
+        summaries = em.get_entry_summaries()
+        assert [s[0] for s in summaries] == [keep_idx, drop_idx]
+
+        em.archive_entry(drop_idx)
+        summaries = em.get_entry_summaries()
+        assert [s[0] for s in summaries] == [keep_idx]
+
+        em.restore_entry(drop_idx)
+        summaries = em.get_entry_summaries()
+        assert [s[0] for s in summaries] == [keep_idx, drop_idx]