mirror of
https://github.com/hansjone/oclaw.git
synced 2026-10-09 03:03:15 +08:00
oliver
420abac9f1
- Rename platform/ to svc/ to avoid shadowing stdlib platform. - Replace from oclaw.* with from svc/runtime/interfaces; update -m CLI paths. - tests/conftest: prepend repo root to sys.path (no parent-folder package name). - CI: paths and offline_eval script under repo root. - Ops scripts: PYTHONPATH must be repo root for python -m runtime.* (fixes gateway/WhatsApp sidecar startup). - Fix default oclaw.json path in tabular/file attachment limits; stabilize attachment test config. Co-authored-by: Cursor <cursoragent@cursor.com>
130 lines
4.5 KiB
Python
130 lines
4.5 KiB
Python
from __future__ import annotations
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from svc.persistence.sqlite_store import OclawTask
|
|
from runtime.workers.wiki.main import _capture_after_turn, _dedup_merge, _process_task
|
|
|
|
|
|
def _task(task_id: str = "t1") -> OclawTask:
|
|
return OclawTask(
|
|
id=task_id,
|
|
tenant_id="tenant-1",
|
|
session_id="session-1",
|
|
task_type="wiki_capture",
|
|
status="claimed",
|
|
payload="{}",
|
|
result="{}",
|
|
attempt_count=1,
|
|
claimed_by="worker",
|
|
lease_expires_at=None,
|
|
last_error="",
|
|
created_at="2026-01-01T00:00:00Z",
|
|
updated_at="2026-01-01T00:00:00Z",
|
|
finished_at=None,
|
|
)
|
|
|
|
|
|
def test_capture_and_dedup_merge_are_incremental(tmp_path: Path) -> None:
|
|
wiki_root = tmp_path / "wiki"
|
|
task = _task("t-cap")
|
|
payload = {
|
|
"session_id": "s-a",
|
|
"turn_uuid": "turn-1",
|
|
"user_text": "hello",
|
|
"assistant_text": "world",
|
|
}
|
|
cap = _capture_after_turn(wiki_root=wiki_root, task=task, payload=payload)
|
|
assert cap["ok"] is True
|
|
assert str(cap["staged"]).endswith(".md")
|
|
|
|
merge1 = _dedup_merge(wiki_root=wiki_root)
|
|
assert merge1["ok"] is True
|
|
assert int(merge1["merged_count"]) == 1
|
|
|
|
# A second staging file with identical content should be treated as duplicate.
|
|
payload2 = {
|
|
"session_id": "s-a",
|
|
"turn_uuid": "turn-2",
|
|
"user_text": "hello",
|
|
"assistant_text": "world",
|
|
}
|
|
_capture_after_turn(wiki_root=wiki_root, task=task, payload=payload2)
|
|
merge2 = _dedup_merge(wiki_root=wiki_root)
|
|
assert merge2["ok"] is True
|
|
assert int(merge2["merged_count"]) == 0
|
|
assert int(merge2["duplicate_count"]) >= 1
|
|
|
|
|
|
def test_process_task_writes_index_and_lint_report(tmp_path: Path) -> None:
|
|
wiki_root = tmp_path / "wiki"
|
|
plugin_cfg = {"wiki_root": str(wiki_root)}
|
|
payload = {
|
|
"kind": "captureAfterTurn",
|
|
"session_id": "s-b",
|
|
"turn_uuid": "turn-3",
|
|
"user_text": "query",
|
|
"assistant_text": "answer",
|
|
}
|
|
task = _task("t-proc")
|
|
task = OclawTask(**{**task.__dict__, "payload": json.dumps(payload, ensure_ascii=False)})
|
|
|
|
def _wiki_apply(args: dict) -> dict:
|
|
rel = str(args.get("path") or "").replace("\\", "/").strip("/")
|
|
fp = wiki_root / rel
|
|
fp.parent.mkdir(parents=True, exist_ok=True)
|
|
action = str(args.get("action") or "").strip()
|
|
content = str(args.get("content") or "")
|
|
if action == "write":
|
|
fp.write_text(content, encoding="utf-8")
|
|
elif action == "append":
|
|
old = fp.read_text(encoding="utf-8") if fp.exists() else ""
|
|
sep = "" if not old or old.endswith("\n") else "\n"
|
|
fp.write_text(old + sep + content, encoding="utf-8")
|
|
return {"ok": True}
|
|
|
|
def _wiki_lint(_args: dict) -> dict:
|
|
return {"ok": True, "issues": []}
|
|
|
|
out = _process_task(task=task, handlers={"wiki_apply": _wiki_apply, "wiki_lint": _wiki_lint}, plugin_cfg=plugin_cfg)
|
|
assert out["ok"] is True
|
|
assert (wiki_root / ".oclaw" / "index.json").exists()
|
|
assert (wiki_root / ".oclaw" / "LINT_REPORT.md").exists()
|
|
|
|
|
|
def test_dedup_merge_routes_into_topic_pages(tmp_path: Path) -> None:
|
|
wiki_root = tmp_path / "wiki"
|
|
task = _task("t-topic")
|
|
payload = {
|
|
"session_id": "s-topic",
|
|
"turn_uuid": "turn-topic",
|
|
"user_text": "please check router vlan",
|
|
"assistant_text": "network fix applied",
|
|
}
|
|
_capture_after_turn(wiki_root=wiki_root, task=task, payload=payload)
|
|
out = _dedup_merge(wiki_root=wiki_root)
|
|
assert out["ok"] is True
|
|
assert int(out["merged_count"]) >= 1
|
|
assert int((out.get("topic_counts") or {}).get("network", 0)) >= 1
|
|
assert (wiki_root / "topics" / "auto-network.md").exists()
|
|
assert (wiki_root / ".oclaw" / "topic-index.json").exists()
|
|
|
|
|
|
def test_dedup_merge_honors_configured_topic_rules(tmp_path: Path) -> None:
|
|
wiki_root = tmp_path / "wiki"
|
|
task = _task("t-custom-topic")
|
|
payload = {
|
|
"session_id": "s-custom",
|
|
"turn_uuid": "turn-custom",
|
|
"user_text": "please use terraform module",
|
|
"assistant_text": "terraform plan applied",
|
|
}
|
|
_capture_after_turn(wiki_root=wiki_root, task=task, payload=payload)
|
|
out = _dedup_merge(
|
|
wiki_root=wiki_root,
|
|
topic_rules=[{"topic": "iac", "keywords": ["terraform", "ansible"]}],
|
|
)
|
|
assert out["ok"] is True
|
|
assert int((out.get("topic_counts") or {}).get("iac", 0)) >= 1
|
|
assert (wiki_root / "topics" / "auto-iac.md").exists()
|