Function TestRemoteToolsetReconnectsAfterCleanClose

pkg/tools/mcp/mcp_test.go:639–675

(t *testing.T)

}

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

	pingTool := &mcp.Tool{Name: "ping"}
	mock := &failingInitClient{
		toolsToList: []*mcp.Tool{pingTool},
		waitCh:      make(chan struct{}),
	}

	ts := newTestToolset("test-remote", "remote-server", mock)
	ts.supervisor = newSupervisor(ts, lifecycle.Policy{
		Restart: lifecycle.RestartAlways,
		Backoff: lifecycle.Backoff{
			Initial:    time.Millisecond,
			Max:        2 * time.Millisecond,
			Multiplier: 2,
		},
	})

	require.NoError(t, ts.Start(t.Context()))
	require.True(t, ts.IsStarted())

	require.NoError(t, mock.Close(t.Context()))
	require.Eventually(t, func() bool {
		mock.mu.Lock()
		initCalls := mock.initCalls
		mock.mu.Unlock()
		return initCalls >= 2 && ts.IsStarted()
	}, 2*time.Second, 10*time.Millisecond, "remote toolset did not reconnect after clean close")

	toolList, err := ts.Tools(t.Context())
	require.NoError(t, err)
	require.Len(t, toolList, 1)
	assert.Equal(t, "test-remote_ping", toolList[0].Name)

	_ = ts.Stop(t.Context())
}