Function TestDynamicToolCallPausesAndResumes

coderd/x/chatd/chatd_test.go:4287–4474

(t *testing.T)

}

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

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

	// Track streaming calls to the mock LLM.
	var streamedCallCount atomic.Int32
	var streamedCallsMu sync.Mutex
	streamedCalls := make([]chattest.OpenAIRequest, 0, 2)

	openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
		// Non-streaming requests are title generation. Return a
		// simple title.
		if !req.Stream {
			return chattest.OpenAINonStreamingResponse("Dynamic tool test")
		}

		// Capture the full request for later assertions.
		streamedCallsMu.Lock()
		streamedCalls = append(streamedCalls, chattest.OpenAIRequest{
			Messages: append([]chattest.OpenAIMessage(nil), req.Messages...),
			Tools:    append([]chattest.OpenAITool(nil), req.Tools...),
			Stream:   req.Stream,
		})
		streamedCallsMu.Unlock()

		if streamedCallCount.Add(1) == 1 {
			// First call: the LLM invokes our dynamic tool.
			return chattest.OpenAIStreamingResponse(
				chattest.OpenAIToolCallChunk(
					"my_dynamic_tool",
					`{"input":"hello world"}`,
				),
			)
		}
		// Second call: the LLM returns a normal text response.
		return chattest.OpenAIStreamingResponse(
			chattest.OpenAITextChunks("Dynamic tool result received.")...,
		)
	})

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

	// Dynamic tools do not need a workspace connection, but the
	// chatd server always builds workspace tools. Use an active
	// server without an agent connection, so the built-in tools
	// are never invoked because the only tool call targets our
	// dynamic tool.
	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{