Function TestExploreChatUsesPersistedMCPSnapshot

coderd/x/chatd/chatd_test.go:785–969

(t *testing.T)

}

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

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

	externalMCP := mcpserver.NewMCPServer("external-snapshot-mcp", "1.0.0")
	externalMCP.AddTools(mcpserver.ServerTool{
		Tool: mcpgo.NewTool("echo",
			mcpgo.WithDescription("Echoes the input"),
			mcpgo.WithString("input",
				mcpgo.Description("The input string"),
				mcpgo.Required(),
			),
		),
		Handler: func(_ context.Context, req mcpgo.CallToolRequest) (*mcpgo.CallToolResult, error) {
			input, _ := req.GetArguments()["input"].(string)
			return mcpgo.NewToolResultText("echo: " + input), nil
		},
	})
	externalMCPServer := httptest.NewServer(mcpserver.NewStreamableHTTPServer(externalMCP))
	defer externalMCPServer.Close()

	secondMCP := mcpserver.NewMCPServer("second-mcp", "1.0.0")
	secondMCP.AddTools(mcpserver.ServerTool{
		Tool: mcpgo.NewTool("echo",
			mcpgo.WithDescription("Echoes the input"),
			mcpgo.WithString("input",
				mcpgo.Description("The input string"),
				mcpgo.Required(),
			),
		),
		Handler: func(_ context.Context, req mcpgo.CallToolRequest) (*mcpgo.CallToolResult, error) {
			input, _ := req.GetArguments()["input"].(string)
			return mcpgo.NewToolResultText("echo: " + input), nil
		},
	})
	secondMCPServer := httptest.NewServer(mcpserver.NewStreamableHTTPServer(secondMCP))
	defer secondMCPServer.Close()

	var (
		requestsMu sync.Mutex
		requests   []recordedOpenAIRequest
	)
	openAIURL := chattest.NewOpenAI(t, func(req *chattest.OpenAIRequest) chattest.OpenAIResponse {
		if !req.Stream {
			return chattest.OpenAINonStreamingResponse("ok")
		}

		requestsMu.Lock()
		requests = append(requests, recordOpenAIRequest(req))
		requestsMu.Unlock()

		return chattest.OpenAIStreamingResponse(
			chattest.OpenAITextChunks("done")...,
		)
	})