Function TestConnectAll_DeterministicOrder

coderd/x/chatd/mcpclient/mcpclient_test.go:238–336

(t *testing.T)

}

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

	t.Run("AcrossServers", func(t *testing.T) {
		t.Parallel()
		ctx := context.Background()
		logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})

		ts1 := newTestMCPServer(t, makeTool("zebra"))
		ts2 := newTestMCPServer(t, makeTool("alpha"))
		ts3 := newTestMCPServer(t, makeTool("middle"))

		tools, cleanup := mcpclient.ConnectAll(
			ctx,
			logger,
			[]database.MCPServerConfig{
				makeConfig("srv3", ts3.URL),
				makeConfig("srv1", ts1.URL),
				makeConfig("srv2", ts2.URL),
			},
			nil,
			uuid.Nil, nil,
			nil,
		)
		t.Cleanup(cleanup)

		require.Len(t, tools, 3)
		// Sorted by full prefixed name (slug__tool), so slug
		// order determines the sequence, not the tool name.
		assert.Equal(t,
			[]string{"srv1__zebra", "srv2__alpha", "srv3__middle"},
			toolNames(tools),
		)
	})

	t.Run("WithMultiToolServer", func(t *testing.T) {
		t.Parallel()
		ctx := context.Background()
		logger := slogtest.Make(t, &slogtest.Options{IgnoreErrors: true})

		multi := newTestMCPServer(t, makeTool("zeta"), makeTool("beta"))
		other := newTestMCPServer(t, makeTool("gamma"))

		tools, cleanup := mcpclient.ConnectAll(
			ctx,
			logger,
			[]database.MCPServerConfig{
				makeConfig("zzz", multi.URL),
				makeConfig("aaa", other.URL),
			},
			nil,
			uuid.Nil, nil,
			nil,
		)
		t.Cleanup(cleanup)

		require.Len(t, tools, 3)
		assert.Equal(t,