Add unit tests for utilities

src/tests/test_password_prompt.py

@@ -0,0 +1,34 @@
+import builtins
+from itertools import cycle
+
+import pytest
+
+from utils import password_prompt
+
+
+def test_prompt_new_password(monkeypatch):
+    responses = cycle(["goodpass", "goodpass"])
+    monkeypatch.setattr(
+        password_prompt.getpass, "getpass", lambda prompt: next(responses)
+    )
+    result = password_prompt.prompt_new_password()
+    assert result == "goodpass"
+
+
+def test_prompt_new_password_retry(monkeypatch):
+    seq = iter(["pass1", "pass2", "passgood", "passgood"])
+    monkeypatch.setattr(password_prompt.getpass, "getpass", lambda prompt: next(seq))
+    result = password_prompt.prompt_new_password()
+    assert result == "passgood"
+
+
+def test_prompt_existing_password(monkeypatch):
+    monkeypatch.setattr(password_prompt.getpass, "getpass", lambda prompt: "mypassword")
+    assert password_prompt.prompt_existing_password() == "mypassword"
+
+
+def test_confirm_action_yes_no(monkeypatch):
+    monkeypatch.setattr(builtins, "input", lambda _: "Y")
+    assert password_prompt.confirm_action()
+    monkeypatch.setattr(builtins, "input", lambda _: "n")
+    assert not password_prompt.confirm_action()