From 78368c0e2f3e18a6527fd541c31f4a86dabb95bd Mon Sep 17 00:00:00 2001 From: thePR0M3TH3AN <53631862+PR0M3TH3AN@users.noreply.github.com> Date: Wed, 16 Jul 2025 12:57:19 -0400 Subject: [PATCH] Fix seed setup test to patch masked input --- src/tests/test_manager_seed_setup.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/tests/test_manager_seed_setup.py b/src/tests/test_manager_seed_setup.py index 73b91d9..e8277c9 100644 --- a/src/tests/test_manager_seed_setup.py +++ b/src/tests/test_manager_seed_setup.py @@ -24,9 +24,14 @@ def test_setup_existing_seed_words(monkeypatch): m = Mnemonic("english") phrase = m.generate(strength=128) words = phrase.split() - inputs = iter(words + ["y"] * len(words)) - monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(inputs)) - monkeypatch.setattr(builtins, "input", lambda *_: next(inputs)) + word_iter = iter(words) + monkeypatch.setattr( + "password_manager.manager.masked_input", + lambda *_: next(word_iter), + ) + # Ensure prompt_seed_words uses the patched function + monkeypatch.setattr(seed_prompt, "masked_input", lambda *_: next(word_iter)) + monkeypatch.setattr(builtins, "input", lambda *_: "y") pm = PasswordManager.__new__(PasswordManager) monkeypatch.setattr(pm, "_finalize_existing_seed", lambda seed: seed)