mirror of
https://github.com/PR0M3TH3AN/SeedPass.git
synced 2026-11-02 10:52:05 +00:00
google-labs-jules[bot]andPR0M3TH3AN
c933bcd43f
- Fixes `AssertionError: assert {'path': '\\tmp\\exported.md'} == {'path': '/tmp/exported.md'}` by checking against both `\tmp\exported.md` and `/tmp/exported.md` in `src/tests/test_api_new_endpoints.py`
- Fixes `AssertionError: assert '/tmp/doc.txt' in '\\tmp\\doc.txt\n'` by checking for both variations in `src/tests/test_cli_entry_add_commands.py`
- Runs `black` to format unformatted tests and CLI code to fix lint/formatting failures.
Co-authored-by: PR0M3TH3AN <[email protected]>
103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
from types import SimpleNamespace
|
|
|
|
import typer
|
|
from typer.testing import CliRunner
|
|
|
|
from seedpass import cli
|
|
from seedpass.cli import app
|
|
from seedpass.cli import common as cli_common
|
|
from seedpass.core.entry_types import EntryType
|
|
|
|
runner = CliRunner()
|
|
|
|
|
|
def test_cli_vault_unlock(monkeypatch):
|
|
called = {}
|
|
|
|
def unlock_vault(pw):
|
|
called["pw"] = pw
|
|
return 0.5
|
|
|
|
pm = SimpleNamespace(unlock_vault=unlock_vault, select_fingerprint=lambda fp: None)
|
|
monkeypatch.setattr(cli_common, "PasswordManager", lambda: pm)
|
|
monkeypatch.setattr(cli.typer, "prompt", lambda *a, **k: "pw")
|
|
result = runner.invoke(app, ["vault", "unlock"])
|
|
assert result.exit_code == 0
|
|
assert "Unlocked in" in result.stdout
|
|
assert "seedpass --help" in result.stdout
|
|
assert called["pw"] == "pw"
|
|
|
|
|
|
def test_cli_vault_unlock_with_env_broker(monkeypatch):
|
|
called = {}
|
|
broker_called = {}
|
|
|
|
def unlock_vault(pw):
|
|
called["pw"] = pw
|
|
return 0.25
|
|
|
|
pm = SimpleNamespace(unlock_vault=unlock_vault, select_fingerprint=lambda fp: None)
|
|
monkeypatch.setattr(cli_common, "PasswordManager", lambda: pm)
|
|
monkeypatch.setattr(
|
|
"seedpass.cli.vault.resolve_broker_password",
|
|
lambda **kwargs: broker_called.update(kwargs) or "broker-pw",
|
|
)
|
|
monkeypatch.setattr(
|
|
cli.typer,
|
|
"prompt",
|
|
lambda *a, **k: (_ for _ in ()).throw(
|
|
AssertionError("prompt should not be called")
|
|
),
|
|
)
|
|
result = runner.invoke(app, ["vault", "unlock", "--auth-broker", "env"])
|
|
assert result.exit_code == 0
|
|
assert called["pw"] == "broker-pw"
|
|
assert broker_called["broker"] == "env"
|
|
|
|
|
|
def test_cli_entry_add_search_sync(monkeypatch):
|
|
calls = {}
|
|
|
|
def add_entry(label, length, username=None, url=None):
|
|
calls["add"] = (label, length, username, url)
|
|
return 1
|
|
|
|
def search_entries(q, kinds=None):
|
|
calls["search"] = (q, kinds)
|
|
return [(1, "Label", None, None, False, EntryType.PASSWORD)]
|
|
|
|
def start_background_vault_sync():
|
|
calls["sync"] = True
|
|
return {"manifest_id": "m", "chunk_ids": [], "delta_ids": []}
|
|
|
|
pm = SimpleNamespace(
|
|
entry_manager=SimpleNamespace(
|
|
add_entry=add_entry, search_entries=search_entries
|
|
),
|
|
start_background_vault_sync=start_background_vault_sync,
|
|
sync_vault=lambda: {"manifest_id": "m", "chunk_ids": [], "delta_ids": []},
|
|
select_fingerprint=lambda fp: None,
|
|
)
|
|
monkeypatch.setattr(cli_common, "PasswordManager", lambda: pm)
|
|
|
|
# entry add
|
|
result = runner.invoke(app, ["entry", "add", "Label"])
|
|
assert result.exit_code == 0
|
|
assert "1" in result.stdout
|
|
assert calls["add"] == ("Label", 12, None, None)
|
|
assert calls.get("sync") is True
|
|
|
|
# entry search
|
|
result = runner.invoke(
|
|
app, ["entry", "search", "lab", "--kind", "password", "--kind", "totp"]
|
|
)
|
|
assert result.exit_code == 0
|
|
assert "Label" in result.stdout
|
|
assert calls["search"] == ("lab", ["password", "totp"])
|
|
|
|
# nostr sync
|
|
result = runner.invoke(app, ["nostr", "sync"])
|
|
assert result.exit_code == 0
|
|
assert "manifest" in result.stdout.lower()
|
|
assert calls.get("sync") is True
|