mirror of
https://github.com/kuhyx/testsAndMisc.git
synced 2026-11-10 22:47:02 +01:00
Krzysztof kuhy Rudnicki
86565bdef6
Indexes the arch-wiki-docs dump that is already on disk and exposes it
to the knowledge-rag MCP server, so Arch questions can be asked by
symptom ("why doesn't my systemd user unit see DISPLAY under i3")
rather than by page title, with no network round-trip at query time.
The one Arch-specific MCP server in the wild fetches pages over the
network and bundles pacman operations that duplicate the local yay-mcp.
This does neither: it only searches, against a local corpus.
install.sh installs its own dependencies, wires the MCP config and
registers a pacman hook so the index follows package updates. The
generated index is machine-specific and regenerable, so it is
gitignored rather than committed.
_reindex_entry runs under the knowledge-rag interpreter, where
mcp_server is importable and this repo's packages are not. Resolving
that import with importlib.import_module states the deferral outright
instead of a function-level import statement that reads as an oversight,
and the backend failure path now re-raises as SystemExit, so the
exit-code translation is explicit rather than a swallowed exception.
Both replace lint suppressions; the package carries none.
pylint runs in its own venv, so beautifulsoup4 and markdownify are
declared there too — being in meta/requirements.txt does not make them
resolvable to the hook.
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01YQLwdquDCNmWtvgi3Lk72J
85 lines
2.6 KiB
Python
85 lines
2.6 KiB
Python
"""Tests for archwiki_rag._reindex_entry.
|
|
|
|
The entry point normally runs under the knowledge-rag interpreter, where
|
|
``mcp_server`` is importable. Here a stub module is injected into
|
|
``sys.modules`` so the three exit paths can be exercised in-process.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import sys
|
|
import types
|
|
from typing import TYPE_CHECKING, cast
|
|
from unittest.mock import MagicMock
|
|
|
|
import pytest
|
|
|
|
from python_pkg.archwiki_rag import _reindex_entry
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
|
|
|
|
@pytest.fixture
|
|
def fake_mcp_server() -> Iterator[MagicMock]:
|
|
"""Install a stub ``mcp_server.server`` module for the duration of a test.
|
|
|
|
Yields:
|
|
MagicMock: The stubbed ``get_orchestrator`` callable.
|
|
"""
|
|
package = types.ModuleType("mcp_server")
|
|
server = types.ModuleType("mcp_server.server")
|
|
get_orchestrator = MagicMock()
|
|
server.get_orchestrator = get_orchestrator
|
|
package.server = server
|
|
|
|
sys.modules["mcp_server"] = package
|
|
sys.modules["mcp_server.server"] = server
|
|
try:
|
|
yield get_orchestrator
|
|
finally:
|
|
del sys.modules["mcp_server.server"]
|
|
del sys.modules["mcp_server"]
|
|
|
|
|
|
class TestMain:
|
|
def test_reports_missing_knowledge_rag(
|
|
self,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
# A None entry in sys.modules makes `import` raise ImportError, which
|
|
# is exactly what happens for real when this module runs outside the
|
|
# knowledge-rag venv.
|
|
sys.modules["mcp_server"] = cast("types.ModuleType", None)
|
|
try:
|
|
assert _reindex_entry.main() == 1
|
|
finally:
|
|
del sys.modules["mcp_server"]
|
|
|
|
assert "not importable" in capsys.readouterr().err
|
|
|
|
def test_reports_reindex_failure(
|
|
self,
|
|
fake_mcp_server: MagicMock,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
fake_mcp_server.return_value.reindex_all.side_effect = RuntimeError("chroma")
|
|
|
|
# A backend failure is translated into a non-zero exit rather than a
|
|
# traceback, so the subprocess caller sees a clean exit code.
|
|
with pytest.raises(SystemExit) as excinfo:
|
|
_reindex_entry.main()
|
|
|
|
assert excinfo.value.code == 1
|
|
assert "reindex failed: chroma" in capsys.readouterr().err
|
|
|
|
def test_success_reports_stats(
|
|
self,
|
|
fake_mcp_server: MagicMock,
|
|
capsys: pytest.CaptureFixture[str],
|
|
) -> None:
|
|
fake_mcp_server.return_value.reindex_all.return_value = {"indexed": 7}
|
|
|
|
assert _reindex_entry.main() == 0
|
|
assert "reindex complete" in capsys.readouterr().out
|