Function TestSubmitToolResultsConcurrency

coderd/x/chatd/chatd_test.go:4729–4885

(t *testing.T)

}

func TestSubmitToolResultsConcurrency(t *testing.T) {
	t.Parallel()

	db, ps := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitLong)

	// The mock LLM returns a dynamic tool call on the first streaming
	// request, then a plain text reply on the second.
	var streamedCallCount atomic.Int32
	openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
		if !req.Stream {
			return chattest.OpenAINonStreamingResponse("Concurrency test")
		}
		if streamedCallCount.Add(1) == 1 {
			return chattest.OpenAIStreamingResponse(
				chattest.OpenAIToolCallChunk(
					"my_dynamic_tool",
					`{"input":"hello"}`,
				),
			)
		}
		return chattest.OpenAIStreamingResponse(
			chattest.OpenAITextChunks("Done.")...,
		)
	})

	user, org, model := seedChatDependenciesWithProvider(t, db, "openai-compat", openAIURL)
	server := newActiveTestServer(t, db, ps)

	// Create a chat with a dynamic tool.
	dynamicToolsJSON, err := json.Marshal([]mcpgo.Tool{{
		Name:        "my_dynamic_tool",
		Description: "A test dynamic tool.",
		InputSchema: mcpgo.ToolInputSchema{
			Type: "object",
			Properties: map[string]any{
				"input": map[string]any{"type": "string"},
			},
			Required: []string{"input"},
		},
	}})
	require.NoError(t, err)

	chat, err := server.CreateChat(ctx, chatd.CreateOptions{
		OrganizationID: org.ID,
		OwnerID:        user.ID,
		Title:          "concurrency-tool-results",
		ModelConfigID:  model.ID,
		InitialUserContent: []codersdk.ChatMessagePart{
			codersdk.ChatMessageText("Please call the dynamic tool."),
		},
		DynamicTools: dynamicToolsJSON,
	})
	require.NoError(t, err)

	// Wait for the chat to reach requires_action status.
	var chatResult database.Chat
	require.Eventually(t, func() bool {