mirror of
https://github.com/pewdiepie-archdaemon/odysseus.git
synced 2026-06-16 09:45:24 -04:00
Afonso Coutinho
af61b2d4e6
Adds focused regression coverage for the research CLI complete-to-done status alias.
58 lines
2.3 KiB
Python
58 lines
2.3 KiB
Python
"""`odysseus-research list --status complete` must match completed runs.
|
|
|
|
Completed research runs are persisted with status "done" (research_handler),
|
|
but the user-facing CLI value is the friendlier "complete". The CLI offered
|
|
"complete" yet filtered `status != args.status`, so `--status complete` never
|
|
matched any record. The fix keeps "complete" as the CLI value and maps it to
|
|
the stored "done" at filter time, so the on-disk corpus stays the source of
|
|
truth and the documented CLI surface keeps working.
|
|
"""
|
|
import importlib.machinery
|
|
import importlib.util
|
|
import json
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
|
|
|
|
def _load_cli():
|
|
path = ROOT / "scripts" / "odysseus-research"
|
|
loader = importlib.machinery.SourceFileLoader("odysseus_research_cli_status", str(path))
|
|
spec = importlib.util.spec_from_loader(loader.name, loader)
|
|
module = importlib.util.module_from_spec(spec)
|
|
loader.exec_module(module)
|
|
return module
|
|
|
|
|
|
def test_complete_is_a_valid_status_choice():
|
|
cli = _load_cli()
|
|
parser = cli._build_parser()
|
|
ns = parser.parse_args(["list", "--status", "complete"])
|
|
assert ns.status == "complete"
|
|
|
|
|
|
def test_filter_returns_completed_runs(tmp_path, monkeypatch):
|
|
cli = _load_cli(); cli._DATA_DIR = tmp_path
|
|
(tmp_path / "r1.json").write_text(json.dumps({"query": "q1", "status": "done"}))
|
|
(tmp_path / "r2.json").write_text(json.dumps({"query": "q2", "status": "running"}))
|
|
emitted = []
|
|
monkeypatch.setattr(cli, "emit", lambda value, args: emitted.append(value))
|
|
# CLI "complete" must map to the stored "done" and match r1.
|
|
cli.cmd_list(SimpleNamespace(status="complete", limit=50))
|
|
ids = [r["id"] for r in emitted[0]]
|
|
assert ids == ["r1"] # only the completed run
|
|
|
|
|
|
def test_verbatim_status_still_filters(tmp_path, monkeypatch):
|
|
cli = _load_cli(); cli._DATA_DIR = tmp_path
|
|
(tmp_path / "r1.json").write_text(json.dumps({"query": "q1", "status": "done"}))
|
|
(tmp_path / "r2.json").write_text(json.dumps({"query": "q2", "status": "running"}))
|
|
emitted = []
|
|
monkeypatch.setattr(cli, "emit", lambda value, args: emitted.append(value))
|
|
cli.cmd_list(SimpleNamespace(status="running", limit=50))
|
|
ids = [r["id"] for r in emitted[0]]
|
|
assert ids == ["r2"] # verbatim choices pass through unchanged
|