Update 2026-05-13 16:43:53

dataclaw-api/tests/test_nanobot_project_resolution.py

@@ -0,0 +1,110 @@
+import asyncio
+from types import SimpleNamespace
+
+from app.core.nanobot import NanobotIntegration
+from app.context import current_session_id
+
+
+class _DummySessions:
+    def __init__(self) -> None:
+        self.saved = []
+        self._session = SimpleNamespace(messages=[])
+
+    def get_or_create(self, _session_id: str):
+        return self._session
+
+    def save(self, session) -> None:
+        self.saved.append(session)
+
+
+class _DummyAgent:
+    def __init__(self) -> None:
+        self.sessions = _DummySessions()
+        self.provider = SimpleNamespace(default_model="demo-model")
+        self.model = "demo-model"
+
+    async def process_direct(self, *_args, **_kwargs):
+        return "ok"
+
+
+def test_process_message_project_id_fallback_from_session_alias(monkeypatch) -> None:
+    service = NanobotIntegration()
+    base_agent = _DummyAgent()
+    custom_agent = _DummyAgent()
+    service.agent = base_agent
+    service._started = True
+
+    captured: dict[str, object] = {}
+
+    async def fake_get_or_create_model_agent(model_id, target_config, project_id):
+        captured["project_id"] = project_id
+        return custom_agent
+
+    monkeypatch.setattr(service, "_get_or_create_model_agent", fake_get_or_create_model_agent)
+    monkeypatch.setattr("app.core.nanobot.get_llm_configs", lambda: [])
+    monkeypatch.setattr("app.core.nanobot.get_active_llm_config", lambda: None)
+    monkeypatch.setattr(
+        "app.core.session_alias_store.session_alias_store.get_alias_meta",
+        lambda _session_id: {"project_id": 77},
+    )
+
+    response = asyncio.run(service.process_message("hello", session_id="api:s1"))
+
+    assert response == "ok"
+    assert captured["project_id"] == 77
+
+
+def test_process_message_project_id_prefers_request_value(monkeypatch) -> None:
+    service = NanobotIntegration()
+    base_agent = _DummyAgent()
+    custom_agent = _DummyAgent()
+    service.agent = base_agent
+    service._started = True
+
+    captured: dict[str, object] = {}
+
+    async def fake_get_or_create_model_agent(model_id, target_config, project_id):
+        captured["project_id"] = project_id
+        return custom_agent
+
+    monkeypatch.setattr(service, "_get_or_create_model_agent", fake_get_or_create_model_agent)
+    monkeypatch.setattr("app.core.nanobot.get_llm_configs", lambda: [])
+    monkeypatch.setattr("app.core.nanobot.get_active_llm_config", lambda: None)
+    monkeypatch.setattr(
+        "app.core.session_alias_store.session_alias_store.get_alias_meta",
+        lambda _session_id: {"project_id": 88},
+    )
+
+    response = asyncio.run(service.process_message("hello", session_id="api:s2", project_id=9))
+
+    assert response == "ok"
+    assert captured["project_id"] == 9
+
+
+def test_register_custom_tools_always_contains_subagent_tools() -> None:
+    service = NanobotIntegration()
+    names: list[str] = []
+
+    class _ToolRegistry:
+        def register(self, tool) -> None:
+            names.append(tool.name)
+
+    fake_agent = SimpleNamespace(tools=_ToolRegistry())
+    service._register_custom_tools(fake_agent, project_id=None)
+
+    assert "list_subagents" in names
+    assert "invoke_subagent" in names
+
+
+def test_subagent_tool_resolves_project_from_session_alias(monkeypatch) -> None:
+    from app.tools.subagent import _resolve_project_id
+
+    token = current_session_id.set("api:subagent-test")
+    try:
+        monkeypatch.setattr(
+            "app.tools.subagent.session_alias_store.get_alias_meta",
+            lambda _session_id: {"project_id": 66},
+        )
+        assert _resolve_project_id(None) == 66
+    finally:
+        current_session_id.reset(token)