Function TestManager_ToolsStartupGate

agent/x/agentmcp/manager_internal_test.go:282–560

(t *testing.T)

}

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

	if os.Getenv("TEST_MCP_FAKE_SERVER") == "1" {
		runFakeMCPServer()
		return
	}

	t.Run("MissingBeforeStartupCanAppearBeforeSettlement", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitLong)
		logger := slogtest.Make(t, nil).Leveled(slog.LevelDebug)
		dir := t.TempDir()
		configPath := filepath.Join(dir, ".mcp.json")

		m := NewManager(ctx, logger, agentexec.DefaultExecer, nil)
		t.Cleanup(func() { _ = m.Close() })

		type result struct {
			tools []workspacesdk.MCPToolInfo
			err   error
		}
		done := make(chan result, 1)
		go func() {
			tools, err := m.Tools(ctx, []string{configPath})
			done <- result{tools: tools, err: err}
		}()

		_, entry := fakeMCPServerConfig(t, "srv")
		writeMCPConfig(t, dir, map[string]mcpServerEntry{"srv": entry})
		m.MarkStartupSettled()

		select {
		case got := <-done:
			require.NoError(t, got.err)
			require.Len(t, got.tools, 1)
			assert.Contains(t, got.tools[0].Name, "echo")
		case <-ctx.Done():
			t.Fatalf("Tools did not return after startup settled: %v", ctx.Err())
		}
	})

	t.Run("MissingAfterStartupReturnsEmptyAndMarksFirstSync", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitLong)
		logger := slogtest.Make(t, nil).Leveled(slog.LevelDebug)
		dir := t.TempDir()
		configPath := filepath.Join(dir, ".mcp.json")

		m := NewManager(ctx, logger, agentexec.DefaultExecer, nil)
		m.MarkStartupSettled()
		t.Cleanup(func() { _ = m.Close() })

		tools, err := m.Tools(ctx, []string{configPath})
		require.NoError(t, err)
		assert.Empty(t, tools)

		m.mu.RLock()