Function TestServer_HandleRequest_TasksCancel

pkg/a2a/server_test.go:191–237

(t *testing.T)

}

func TestServer_HandleRequest_TasksCancel(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.UpdateStatus(TaskStateWorking, nil)
	require.NoError(t, taskStore.Save(agentID, task))

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

	req := &JSONRPCRequest{
		JSONRPC: "2.0",
		ID:      "req-1",
		Method:  "tasks/cancel",
		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)

	// 验证取消标记已设置
	canceled := taskStore.IsCanceled(task.ID)
	assert.True(t, canceled)

	// 验证任务状态已更新
	updatedTask, err := taskStore.Load(agentID, task.ID)
	require.NoError(t, err)
	assert.Equal(t, TaskStateCanceled, updatedTask.Status.State)
}

func TestServer_HandleRequest_InvalidMethod(t *testing.T) {
	system := actor.NewSystem("test-a2a")