agent_id_test.go

 1package config
 2
 3import (
 4	"testing"
 5
 6	"github.com/stretchr/testify/assert"
 7	"github.com/stretchr/testify/require"
 8)
 9
10func TestConfig_AgentIDs(t *testing.T) {
11	cfg := &Config{
12		Options: &Options{
13			DisabledTools: []string{},
14		},
15	}
16	cfg.SetupAgents()
17
18	t.Run("Coder agent should have correct ID", func(t *testing.T) {
19		coderAgent, ok := cfg.Agents[AgentCoder]
20		require.True(t, ok)
21		assert.Equal(t, AgentCoder, coderAgent.ID, "Coder agent ID should be '%s'", AgentCoder)
22	})
23
24	t.Run("Task agent should have correct ID", func(t *testing.T) {
25		taskAgent, ok := cfg.Agents[AgentTask]
26		require.True(t, ok)
27		assert.Equal(t, AgentTask, taskAgent.ID, "Task agent ID should be '%s'", AgentTask)
28	})
29}