Function TestReload

agent/x/agentmcp/reload_internal_test.go:231–408

(t *testing.T)

}

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

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

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

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

		err := m.Reload(ctx, []string{configPath})
		require.NoError(t, err)

		tools := m.cachedTools()
		require.Len(t, tools, 1, "should have one tool from the fake server")
		assert.Contains(t, tools[0].Name, "echo")

		// Snapshot should be fresh.
		assert.False(t, m.SnapshotChanged([]string{configPath}))
	})

	t.Run("ReloadAfterClose", func(t *testing.T) {
		t.Parallel()
		ctx := testutil.Context(t, testutil.WaitLong)
		logger := slogtest.Make(t, nil).Leveled(slog.LevelDebug)

		m := NewManager(ctx, logger, agentexec.DefaultExecer, nil)
		require.NoError(t, m.Close())

		err := m.Reload(ctx, []string{"/nonexistent"})
		require.Error(t, err, "reload after close should fail")
	})

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

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

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

		// Launch multiple concurrent reloads.
		const numCallers = 5
		var wg sync.WaitGroup
		errs := make([]error, numCallers)
		for i := range numCallers {
			wg.Go(func() {
				errs[i] = m.Reload(ctx, []string{configPath})
			})