mirror of
https://github.com/lllin000/PaperForge.git
synced 2026-07-22 17:00:23 +00:00
45de80da5d
- Updated 35 files: from paperforge_lite.* → from paperforge.* - Verified: pytest passes (172 passed, 2 skipped)
157 lines
4.8 KiB
Python
157 lines
4.8 KiB
Python
# Tests for paperforge CLI worker dispatch
|
|
# These tests prove the locked command surface without invoking real workers.
|
|
|
|
import pytest
|
|
import json
|
|
import sys
|
|
from pathlib import Path
|
|
from unittest.mock import patch, MagicMock
|
|
|
|
# ----------------------------------------------------------------------
|
|
# Worker function stubs — capture calls for assertion
|
|
# ----------------------------------------------------------------------
|
|
CAPTURED_CALLS: list[tuple[str, Path]] = []
|
|
|
|
|
|
def stub_run_status(vault: Path) -> int:
|
|
CAPTURED_CALLS.append(("run_status", vault))
|
|
return 0
|
|
|
|
|
|
def stub_run_selection_sync(vault: Path) -> int:
|
|
CAPTURED_CALLS.append(("run_selection_sync", vault))
|
|
return 0
|
|
|
|
|
|
def stub_run_index_refresh(vault: Path) -> int:
|
|
CAPTURED_CALLS.append(("run_index_refresh", vault))
|
|
return 0
|
|
|
|
|
|
def stub_run_deep_reading(vault: Path, verbose: bool = False) -> int:
|
|
CAPTURED_CALLS.append(("run_deep_reading", vault))
|
|
return 0
|
|
|
|
|
|
def stub_run_ocr(vault: Path) -> int:
|
|
CAPTURED_CALLS.append(("run_ocr", vault))
|
|
return 0
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
# Test fixtures
|
|
# ----------------------------------------------------------------------
|
|
@pytest.fixture
|
|
def clean_captured():
|
|
"""Clear captured calls before each test."""
|
|
CAPTURED_CALLS.clear()
|
|
yield
|
|
CAPTURED_CALLS.clear()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_vault(tmp_path):
|
|
"""Create a minimal mock vault with paperforge.json."""
|
|
pf_cfg = {
|
|
"system_dir": "99_System",
|
|
"resources_dir": "03_Resources",
|
|
"literature_dir": "Literature",
|
|
"control_dir": "LiteratureControl",
|
|
"base_dir": "05_Bases",
|
|
}
|
|
(tmp_path / "paperforge.json").write_text(json.dumps(pf_cfg), encoding="utf-8")
|
|
return tmp_path
|
|
|
|
|
|
# ----------------------------------------------------------------------
|
|
# Worker dispatch tests
|
|
# ----------------------------------------------------------------------
|
|
def test_status_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'status']) calls run_status."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_status", stub_run_status):
|
|
argv = ["--vault", str(mock_vault), "status"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_status", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_selection_sync_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'selection-sync']) calls run_selection_sync."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_selection_sync", stub_run_selection_sync):
|
|
argv = ["--vault", str(mock_vault), "selection-sync"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_selection_sync", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_index_refresh_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'index-refresh']) calls run_index_refresh."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_index_refresh", stub_run_index_refresh):
|
|
argv = ["--vault", str(mock_vault), "index-refresh"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_index_refresh", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_deep_reading_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'deep-reading']) calls run_deep_reading."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_deep_reading", stub_run_deep_reading):
|
|
argv = ["--vault", str(mock_vault), "deep-reading"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_deep_reading", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_ocr_run_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'ocr', 'run']) calls run_ocr."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_ocr", stub_run_ocr):
|
|
argv = ["--vault", str(mock_vault), "ocr", "run"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_ocr", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_ocr_alias_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'ocr']) calls run_ocr (alias for 'ocr run')."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "run_ocr", stub_run_ocr):
|
|
argv = ["--vault", str(mock_vault), "ocr"]
|
|
cli.main(argv)
|
|
|
|
assert ("run_ocr", mock_vault) in CAPTURED_CALLS
|
|
|
|
|
|
def test_ocr_doctor_dispatch(clean_captured, mock_vault):
|
|
"""main(['--vault', vault, 'ocr', 'doctor']) calls _cmd_ocr_doctor."""
|
|
import importlib
|
|
import paperforge.cli as cli
|
|
importlib.reload(cli)
|
|
|
|
with patch.object(cli, "_cmd_ocr_doctor", lambda vault, args: 0):
|
|
argv = ["--vault", str(mock_vault), "ocr", "doctor"]
|
|
code = cli.main(argv)
|
|
|
|
assert code == 0
|