Function TestServer_HandleRequest_TasksGet

pkg/a2a/server_test.go:139–189

(t *testing.T)

}

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

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

	agentID := "test-agent"

	// 创建一个任务
	task := NewTask("task-1", "context-1")
	task.AddMessage(Message{
		MessageID: "msg-1",
		Role:      "user",
		Parts:     []Part{{Kind: "text", Text: "Hello"}},
	})
	task.UpdateStatus(TaskStateCompleted, nil)
	require.NoError(t, taskStore.Save(agentID, task))

	// 构造 tasks/get 请求
	params := TasksGetParams{
		TaskID: task.ID,
	}
	paramsJSON, _ := json.Marshal(params)

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

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

	// 验证响应
	require.NotNil(t, resp)
	assert.Nil(t, resp.Error)
	assert.NotNil(t, resp.Result)

	// 验证返回的任务
	var result TasksGetResult
	resultBytes, _ := json.Marshal(resp.Result)
	err := json.Unmarshal(resultBytes, &result)
	require.NoError(t, err)
	assert.Equal(t, task.ID, result.Task.ID)
	assert.Equal(t, TaskStateCompleted, result.Task.Status.State)
}

func TestServer_HandleRequest_TasksCancel(t *testing.T) {
	// 创建 Actor 系统