feat: 增强日历功能并集成 AgentScope 代理服务

backend/tests/unit/core/agentscope/schemas/test_agent_runtime_schemas.py

@@ -0,0 +1,106 @@
+from __future__ import annotations
+
+import pytest
+from pydantic import ValidationError
+
+from core.agentscope import schemas as exported_schemas
+from core.agentscope.schemas.agent_runtime import (
+    AcceptedTaskResponse,
+    AgUiWireEvent,
+    HistorySnapshot,
+    HistorySnapshotResponse,
+    InternalRuntimeEvent,
+    ResumeCommand,
+    RunCommand,
+)
+
+
+def test_run_command_alias_roundtrip() -> None:
+    payload = {
+        "threadId": "thread-001",
+        "runId": "run-001",
+        "state": {"cursor": 1},
+        "messages": [{"role": "user", "content": "hi"}],
+        "tools": [{"name": "calendar.lookup"}],
+        "context": {"locale": "zh-CN"},
+        "forwardedProps": {"traceId": "trace-1"},
+    }
+
+    command = RunCommand.model_validate(payload)
+
+    assert command.thread_id == "thread-001"
+    assert command.run_id == "run-001"
+    assert command.forwarded_props == {"traceId": "trace-1"}
+
+    dumped = command.model_dump(mode="json", by_alias=True)
+    assert dumped["threadId"] == "thread-001"
+    assert dumped["runId"] == "run-001"
+    assert dumped["forwardedProps"] == {"traceId": "trace-1"}
+
+
+def test_history_snapshot_response_shape() -> None:
+    response = HistorySnapshotResponse(
+        threadId="thread-123",
+        snapshot=HistorySnapshot(
+            threadId="thread-123",
+            day="2026-03-11",
+            hasMore=False,
+            messages=[{"id": "msg-1"}],
+        ),
+    )
+
+    dumped = response.model_dump(mode="json", by_alias=True, exclude_none=True)
+    assert dumped["type"] == "STATE_SNAPSHOT"
+    assert dumped["threadId"] == "thread-123"
+    assert dumped["snapshot"]["scope"] == "history_day"
+    assert dumped["snapshot"]["hasMore"] is False
+    assert dumped["snapshot"]["messages"] == [{"id": "msg-1"}]
+
+
+def test_runtime_event_validation_basics() -> None:
+    internal = InternalRuntimeEvent(type="RUN_STARTED", data={"step": 1})
+    assert internal.type == "RUN_STARTED"
+    assert internal.model_dump(mode="json", by_alias=True)["data"] == {"step": 1}
+
+    wire = AgUiWireEvent(type="TEXT_MESSAGE_CONTENT", payload={"delta": "hello"})
+    dumped = wire.model_dump(mode="json", by_alias=True, exclude_none=True)
+    assert dumped["type"] == "TEXT_MESSAGE_CONTENT"
+    assert dumped["payload"] == {"delta": "hello"}
+
+    with pytest.raises(ValidationError):
+        InternalRuntimeEvent.model_validate({"threadId": "t-1", "data": {}})
+
+    with pytest.raises(ValidationError):
+        AgUiWireEvent.model_validate({"payload": {"delta": "hello"}})
+
+
+def test_task_response_and_resume_aliases() -> None:
+    accepted = AcceptedTaskResponse(
+        taskId="task-1",
+        threadId="thread-1",
+        runId="run-1",
+        created=False,
+    )
+    dumped = accepted.model_dump(mode="json", by_alias=True)
+    assert dumped["taskId"] == "task-1"
+    assert dumped["threadId"] == "thread-1"
+    assert dumped["runId"] == "run-1"
+
+    resumed = ResumeCommand.model_validate(
+        {
+            "threadId": "thread-1",
+            "runId": "run-2",
+            "messages": [],
+            "tools": [],
+            "context": {},
+        }
+    )
+    assert resumed.thread_id == "thread-1"
+    assert resumed.run_id == "run-2"
+
+
+def test_schemas_exports_include_task_and_history_models() -> None:
+    assert exported_schemas.AcceptedTaskResponse is AcceptedTaskResponse
+    assert exported_schemas.TaskAccepted is AcceptedTaskResponse
+    assert exported_schemas.TaskAcceptedResponse is AcceptedTaskResponse
+    assert exported_schemas.HistorySnapshotResponse is HistorySnapshotResponse