Function TestModel_LogsSelection

pkg/agent/agent_test.go:497–530

(t *testing.T)

}

func TestModel_LogsSelection(t *testing.T) {
	// Not parallel: it swaps the process-global default logger via slog.SetDefault.
	var buf concurrent.Buffer
	handler := slog.NewTextHandler(&buf, &slog.HandlerOptions{Level: slog.LevelInfo})
	prev := slog.Default()
	slog.SetDefault(slog.New(handler))
	t.Cleanup(func() { slog.SetDefault(prev) })

	model1 := &mockProvider{id: modelsdev.NewID("anthropic", "claude-sonnet-4-0")}
	model2 := &mockProvider{id: modelsdev.NewID("openai", "gpt-4o")}

	a := New("scanner", "test", WithModel(model1), WithModel(model2))

	// Verify basic selection logging
	selected := a.Model(t.Context())
	logOutput := buf.String()

	assert.Contains(t, logOutput, "Model selected")
	assert.Contains(t, logOutput, "agent=scanner")
	assert.Contains(t, logOutput, selected.ID().String())
	assert.Contains(t, logOutput, "pool_size=2")

	// Verify override scenario logs correct pool_size
	buf.Reset()
	override := &mockProvider{id: modelsdev.NewID("google", "gemini-2.0-flash")}
	a.SetModelOverride(override)

	selected = a.Model(t.Context())
	logOutput = buf.String()

	assert.Equal(t, "google/gemini-2.0-flash", selected.ID().String())
	assert.Contains(t, logOutput, "google/gemini-2.0-flash")
	assert.Contains(t, logOutput, "pool_size=1")
}

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