105 lines
3.2 KiB
Python
105 lines
3.2 KiB
Python
from __future__ import annotations
|
|
|
|
from types import SimpleNamespace
|
|
|
|
import pytest
|
|
|
|
from src.services import conversation_service as svc
|
|
|
|
|
|
def test_build_state_files_only_parsed_and_with_content():
|
|
attachments = [
|
|
{
|
|
"status": "parsed",
|
|
"file_path": "/attachments/a.md",
|
|
"markdown": "line1\nline2",
|
|
"uploaded_at": "2026-02-20T00:00:00+00:00",
|
|
},
|
|
{
|
|
"status": "pending",
|
|
"file_path": "/attachments/b.md",
|
|
"markdown": "ignored",
|
|
},
|
|
{
|
|
"status": "parsed",
|
|
"file_path": "/attachments/c.md",
|
|
"markdown": "",
|
|
},
|
|
]
|
|
|
|
files = svc._build_state_files(attachments)
|
|
|
|
assert list(files.keys()) == ["/attachments/a.md"]
|
|
assert files["/attachments/a.md"]["content"] == ["line1", "line2"]
|
|
assert files["/attachments/a.md"]["created_at"] == "2026-02-20T00:00:00+00:00"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_thread_attachment_state_updates_graph(monkeypatch: pytest.MonkeyPatch):
|
|
captured: dict = {}
|
|
fake_state = SimpleNamespace(
|
|
values={
|
|
"files": {
|
|
"/attachments/old.md": {"content": ["old"]},
|
|
"/work/result.md": {"content": ["keep"]},
|
|
}
|
|
}
|
|
)
|
|
|
|
class FakeGraph:
|
|
async def aget_state(self, config):
|
|
captured["read_config"] = config
|
|
return fake_state
|
|
|
|
async def aupdate_state(self, *, config, values):
|
|
captured["write_config"] = config
|
|
captured["write_values"] = values
|
|
|
|
class FakeAgent:
|
|
async def get_graph(self):
|
|
return FakeGraph()
|
|
|
|
monkeypatch.setattr(svc.agent_manager, "get_agent", lambda _agent_id: FakeAgent())
|
|
|
|
attachments = [
|
|
{
|
|
"status": "parsed",
|
|
"file_path": "/attachments/resume.md",
|
|
"markdown": "hello\nworld",
|
|
"uploaded_at": "2026-02-20T00:00:00+00:00",
|
|
}
|
|
]
|
|
await svc._sync_thread_attachment_state(
|
|
thread_id="thread-1",
|
|
user_id="u1",
|
|
agent_id="ChatbotAgent",
|
|
attachments=attachments,
|
|
)
|
|
|
|
assert captured["read_config"] == {"configurable": {"thread_id": "thread-1", "user_id": "u1"}}
|
|
assert captured["write_config"] == {"configurable": {"thread_id": "thread-1", "user_id": "u1"}}
|
|
assert captured["write_values"]["attachments"] == attachments
|
|
assert "/attachments/resume.md" in captured["write_values"]["files"]
|
|
assert "/attachments/old.md" not in captured["write_values"]["files"]
|
|
assert "/work/result.md" in captured["write_values"]["files"]
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_sync_thread_attachment_state_skips_when_agent_missing(monkeypatch: pytest.MonkeyPatch):
|
|
warnings: list[str] = []
|
|
fake_logger = SimpleNamespace(
|
|
warning=lambda message: warnings.append(message),
|
|
)
|
|
|
|
monkeypatch.setattr(svc, "logger", fake_logger)
|
|
monkeypatch.setattr(svc.agent_manager, "get_agent", lambda _agent_id: None)
|
|
|
|
await svc._sync_thread_attachment_state(
|
|
thread_id="thread-1",
|
|
user_id="u1",
|
|
agent_id="MissingAgent",
|
|
attachments=[],
|
|
)
|
|
|
|
assert any("agent not found" in msg for msg in warnings)
|