| 408 | } |
| 409 | |
| 410 | func TestResolveSources_DefaultAliasOverride(t *testing.T) { |
| 411 | home := t.TempDir() |
| 412 | t.Setenv("HOME", home) |
| 413 | |
| 414 | // Create an agent file |
| 415 | agentFile := filepath.Join(t.TempDir(), "custom-agent.yaml") |
| 416 | require.NoError(t, os.WriteFile(agentFile, []byte(`agents: |
| 417 | root: |
| 418 | model: openai/gpt-4o |
| 419 | description: Custom agent for sources |
| 420 | `), 0o644)) |
| 421 | |
| 422 | // Set up alias for "default" |
| 423 | cfg, err := userconfig.Load() |
| 424 | require.NoError(t, err) |
| 425 | require.NoError(t, cfg.SetAlias("default", &userconfig.Alias{Path: agentFile})) |
| 426 | require.NoError(t, cfg.Save()) |
| 427 | |
| 428 | // ResolveSources with "default" should return the aliased file |
| 429 | sources, err := ResolveSources("default", nil) |
| 430 | require.NoError(t, err) |
| 431 | require.Len(t, sources, 1) |
| 432 | |
| 433 | // The key should be the filename without extension |
| 434 | source, ok := sources["custom-agent"] |
| 435 | require.True(t, ok, "expected source key 'custom-agent', got keys: %v", sources) |
| 436 | |
| 437 | // Verify it reads the custom content |
| 438 | data, err := source.Read(t.Context()) |
| 439 | require.NoError(t, err) |
| 440 | assert.Contains(t, string(data), "Custom agent for sources") |
| 441 | } |
| 442 | |
| 443 | func TestResolveSources_EmptyWithDefaultAliasOverride(t *testing.T) { |
| 444 | home := t.TempDir() |