Function TestToolInjectionOrder

aibridge/mcp/mcp_test.go:297–333

(t *testing.T)

}

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

	// Setup.
	logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: false}).Leveled(slog.LevelDebug)
	ctx, cancel := context.WithTimeout(t.Context(), testutil.WaitLong)
	t.Cleanup(cancel)

	// Given: a MCP mock server offering a set of tools.
	mcpSrv := httptest.NewServer(createMockMCPSrv(t))
	t.Cleanup(mcpSrv.Close)

	tracer := otel.Tracer("forTesting")
	// When: creating two MCP server proxies, both listing the same tools by name but under different server namespaces.
	proxy, err := mcp.NewStreamableHTTPServerProxy("coder", mcpSrv.URL, nil, nil, nil, logger, tracer)
	require.NoError(t, err)
	proxy2, err := mcp.NewStreamableHTTPServerProxy("shmoder", mcpSrv.URL, nil, nil, nil, logger, tracer)
	require.NoError(t, err)

	// Then: initialize both proxies.
	require.NoError(t, proxy.Init(ctx))
	require.NoError(t, proxy2.Init(ctx))

	// Then: validate that their tools are separately sorted stably.
	validateToolOrder(t, proxy)
	validateToolOrder(t, proxy2)

	// When: creating a manager which contains both MCP server proxies.
	mgr := mcp.NewServerProxyManager(map[string]mcp.ServerProxier{
		"coder":   proxy,
		"shmoder": proxy2,
	}, otel.GetTracerProvider().Tracer("test"))
	require.NoError(t, mgr.Init(ctx))

	// Then: the tools from both servers should be collectively sorted stably.
	validateToolOrder(t, mgr)
}

func validateToolOrder(t *testing.T, proxy mcp.ServerProxier) {
	t.Helper()