From 2478c44762f555e61368cccd7a3167c34ec05d40 Mon Sep 17 00:00:00 2001
From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com>
Date: Tue, 8 Jul 2025 11:02:41 -0400
Subject: [PATCH] Update retrieve entry tests for new action menu

---
 src/tests/test_custom_fields_display.py | 2 +-
 src/tests/test_manager_retrieve_totp.py | 2 +-
 src/tests/test_nostr_qr.py              | 2 +-
 src/tests/test_secret_mode.py           | 4 ++--
 4 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/tests/test_custom_fields_display.py b/src/tests/test_custom_fields_display.py
index 279e19a..04e2d45 100644
--- a/src/tests/test_custom_fields_display.py
+++ b/src/tests/test_custom_fields_display.py
@@ -42,7 +42,7 @@ def test_retrieve_entry_shows_custom_fields(monkeypatch, capsys):
             ],
         )
 
-        inputs = iter(["0", "y", "", "n"])
+        inputs = iter(["0", "y", "", ""])
         monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
 
         pm.handle_retrieve_entry()
diff --git a/src/tests/test_manager_retrieve_totp.py b/src/tests/test_manager_retrieve_totp.py
index ae0f8d2..2d300ad 100644
--- a/src/tests/test_manager_retrieve_totp.py
+++ b/src/tests/test_manager_retrieve_totp.py
@@ -43,7 +43,7 @@ def test_handle_retrieve_totp_entry(monkeypatch, capsys):
 
         entry_mgr.add_totp("Example", TEST_SEED)
 
-        inputs = iter(["0", "n"])
+        inputs = iter(["0", "n", ""])
         monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
         monkeypatch.setattr(pm.entry_manager, "get_totp_code", lambda *a, **k: "123456")
         monkeypatch.setattr(
diff --git a/src/tests/test_nostr_qr.py b/src/tests/test_nostr_qr.py
index ccb81e9..aaa594f 100644
--- a/src/tests/test_nostr_qr.py
+++ b/src/tests/test_nostr_qr.py
@@ -44,7 +44,7 @@ def test_show_qr_for_nostr_keys(monkeypatch):
         idx = entry_mgr.add_nostr_key("main")
         npub, _ = entry_mgr.get_nostr_key_pair(idx, TEST_SEED)
 
-        inputs = iter([str(idx), "n", ""])
+        inputs = iter([str(idx), "n", "", ""])
         monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
         responses = iter([True, False])
         monkeypatch.setattr(
diff --git a/src/tests/test_secret_mode.py b/src/tests/test_secret_mode.py
index e11f7a2..3a524d7 100644
--- a/src/tests/test_secret_mode.py
+++ b/src/tests/test_secret_mode.py
@@ -41,7 +41,7 @@ def test_password_retrieve_secret_mode(monkeypatch, capsys):
         pm, entry_mgr = setup_pm(tmp)
         entry_mgr.add_entry("example", 8)
 
-        inputs = iter(["0", "n"])
+        inputs = iter(["0", "n", ""])
         monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
         called = []
         monkeypatch.setattr(
@@ -90,7 +90,7 @@ def test_password_retrieve_no_secret_mode(monkeypatch, capsys):
         pm.secret_mode_enabled = False
         entry_mgr.add_entry("example", 8)
 
-        inputs = iter(["0", "n"])
+        inputs = iter(["0", "n", ""])
         monkeypatch.setattr("builtins.input", lambda *a, **k: next(inputs))
         called = []
         monkeypatch.setattr(