Function TestWithPromptFiles

pkg/teamloader/teamloader_test.go:500–544

(t *testing.T)

}

func TestWithPromptFiles(t *testing.T) {
	t.Setenv("OPENAI_API_KEY", "dummy")

	tests := []struct {
		name           string
		cliPromptFiles []string
		expected       []string
	}{
		{
			name:           "no CLI prompt files",
			cliPromptFiles: nil,
			expected:       []string{}, // basic.yaml has no add_prompt_files
		},
		{
			name:           "single CLI prompt file",
			cliPromptFiles: []string{"AGENTS.md"},
			expected:       []string{"AGENTS.md"},
		},
		{
			name:           "multiple CLI prompt files",
			cliPromptFiles: []string{"AGENTS.md", "CLAUDE.md"},
			expected:       []string{"AGENTS.md", "CLAUDE.md"},
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			agentSource, err := config.Resolve("testdata/basic.yaml", nil)
			require.NoError(t, err)

			var opts []Opt
			if len(tt.cliPromptFiles) > 0 {
				opts = append(opts, WithPromptFiles(tt.cliPromptFiles))
			}

			team, err := Load(t.Context(), agentSource, &config.RuntimeConfig{}, withTestProviderRegistry(opts...)...)
			require.NoError(t, err)

			rootAgent, err := team.Agent("root")
			require.NoError(t, err)

			assert.Equal(t, tt.expected, rootAgent.AddPromptFiles())
		})
	}
}

func TestWithPromptFilesMergesWithConfig(t *testing.T) {
	t.Setenv("OPENAI_API_KEY", "dummy")