Function newHeaderRecordingServer

coderd/x/chatd/mcpclient/coder_headers_test.go:26–46

newHeaderRecordingServer creates a streamable HTTP MCP server with a single "ping" tool. Every request's headers are appended to the returned slice so tests can assert which headers were forwarded.

(t *testing.T)

// single "ping" tool. Every request's headers are appended to the
// returned slice so tests can assert which headers were forwarded.
func newHeaderRecordingServer(t *testing.T) (*httptest.Server, *sync.Mutex, *[]http.Header) {
	t.Helper()
	var (
		mu      sync.Mutex
		headers []http.Header
	)
	srv := mcpserver.NewMCPServer("hdr-server", "1.0.0")
	srv.AddTools(mcpserver.ServerTool{
		Tool: mcp.NewTool("ping", mcp.WithDescription("records the request headers")),
		Handler: func(_ context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
			mu.Lock()
			headers = append(headers, req.Header.Clone())
			mu.Unlock()
			return mcp.NewToolResultText("ok"), nil
		},
	})
	httpSrv := mcpserver.NewStreamableHTTPServer(srv)
	ts := httptest.NewServer(httpSrv)
	t.Cleanup(ts.Close)
	return ts, &mu, &headers
}

// TestConnectAll_ForwardCoderHeaders_DefaultOff is a regression guard
// that the Coder identity headers are NOT sent when the option is