init

tests/test_hook.py

@@ -0,0 +1,90 @@
+import json
+from urllib.error import URLError
+
+from laya_router.hook import handle
+
+
+def _ok_response():
+    body = {
+        "level": "complex",
+        "level_confidence": 0.81,
+        "needs_plan": 0.9,
+        "needs_repo": 0.72,
+        "needs_web": 0.2,
+        "needs_deep_reasoning": 0.88,
+    }
+
+    class Response:
+        def __enter__(self):
+            return self
+
+        def __exit__(self, *args):
+            return False
+
+        def read(self):
+            return json.dumps(body).encode()
+
+    return Response()
+
+
+def test_cursor_is_a_noop():
+    text = handle({"prompt": "帮我改这个函数"}, {"CURSOR_VERSION": "1.0"}, "claude")
+    assert text == "{}"
+
+
+def test_slash_command_is_skipped():
+    assert handle({"prompt": "/hooks"}, {}, "codex") == "{}"
+
+
+def test_filesystem_path_is_not_treated_as_slash_command(monkeypatch, tmp_path):
+    monkeypatch.setenv("LAYA_ROUTER_HOME", str(tmp_path))
+    monkeypatch.setattr("laya_router.hook.call_laya", lambda settings, body: _payload())
+    text = handle({"prompt": "/Users/richard/proj 里的入口在哪"}, {}, "codex")
+    assert json.loads(text)["hookSpecificOutput"]["hookEventName"] == "UserPromptSubmit"
+
+
+def test_fail_open_when_server_is_down(monkeypatch, tmp_path):
+    monkeypatch.setenv("LAYA_ROUTER_HOME", str(tmp_path))
+
+    def boom(settings, body):
+        raise URLError("connection refused")
+
+    monkeypatch.setattr("laya_router.hook.call_laya", boom)
+    assert handle({"prompt": "解释这个报错"}, {}, "claude") == "{}"
+    assert (tmp_path / "logs" / "hook.log").is_file()
+
+
+def test_success_output(monkeypatch, tmp_path):
+    monkeypatch.setenv("LAYA_ROUTER_HOME", str(tmp_path))
+    monkeypatch.setattr(
+        "laya_router.hook.call_laya",
+        lambda settings, body: {
+            "level": "simple",
+            "level_confidence": 0.66,
+            "needs_plan": 0.1,
+            "needs_repo": 0.2,
+            "needs_web": 0.05,
+            "needs_deep_reasoning": 0.08,
+        },
+    )
+    payload = handle(
+        {"prompt": "把这个变量改名", "session_id": "s1", "cwd": "/tmp/repo"},
+        {},
+        "codex",
+    )
+    parsed = json.loads(payload)
+    context = parsed["hookSpecificOutput"]["additionalContext"]
+    assert "Task complexity: simple" in context
+    assert "uncalibrated" in context
+    assert "hint only" in context
+
+
+def _payload():
+    return {
+        "level": "normal",
+        "level_confidence": 0.5,
+        "needs_plan": 0.4,
+        "needs_repo": 0.4,
+        "needs_web": 0.1,
+        "needs_deep_reasoning": 0.3,
+    }