Function TestServer_HandleRequest_MessageSend

pkg/a2a/server_test.go:82–137

(t *testing.T)

}

func TestServer_HandleRequest_MessageSend(t *testing.T) {
	// 创建 Actor 系统
	system := actor.NewSystem("test-a2a")
	defer system.Shutdown()

	// 创建 A2A 服务器
	taskStore := NewInMemoryTaskStore()
	server := NewServer(system, taskStore)

	// 创建测试 Agent
	agentID := "test-agent"
	mockAgent := &MockAgentActor{responses: []string{"Hello from agent"}}
	system.Spawn(mockAgent, agentID)

	// 构造 message/send 请求
	params := MessageSendParams{
		Message: Message{
			MessageID: "msg-1",
			Role:      "user",
			Parts:     []Part{{Kind: "text", Text: "Hello"}},
		},
		ContextID: "context-1",
	}
	paramsJSON, _ := json.Marshal(params)

	req := &JSONRPCRequest{
		JSONRPC: "2.0",
		ID:      "req-1",
		Method:  "message/send",
		Params:  json.RawMessage(paramsJSON),
	}

	// 处理请求
	ctx := context.Background()
	resp := server.HandleRequest(ctx, agentID, req)

	// 验证响应
	require.NotNil(t, resp)
	assert.Equal(t, "2.0", resp.JSONRPC)
	assert.Equal(t, "req-1", resp.ID)
	assert.Nil(t, resp.Error)
	assert.NotNil(t, resp.Result)

	// 验证任务已创建
	var result MessageSendResult
	resultBytes, _ := json.Marshal(resp.Result)
	err := json.Unmarshal(resultBytes, &result)
	require.NoError(t, err)
	assert.NotEmpty(t, result.TaskID)

	// 验证任务存储中的任务
	task, err := taskStore.Load(agentID, result.TaskID)
	require.NoError(t, err)
	assert.Equal(t, TaskStateCompleted, task.Status.State)
	assert.Len(t, task.History, 2) // 用户消息 + Agent 响应
}

func TestServer_HandleRequest_TasksGet(t *testing.T) {