(t *testing.T)
| 343 | func (m *mockProvider) BaseConfig() base.Config { return base.Config{} } |
| 344 | |
| 345 | func TestModelOverride(t *testing.T) { |
| 346 | t.Parallel() |
| 347 | |
| 348 | defaultModel := &mockProvider{id: modelsdev.NewID("openai", "gpt-4o")} |
| 349 | overrideModel := &mockProvider{id: modelsdev.NewID("anthropic", "claude-sonnet-4-0")} |
| 350 | |
| 351 | a := New("root", "test", WithModel(defaultModel)) |
| 352 | |
| 353 | // Initially should return the default model |
| 354 | assert.Equal(t, "openai/gpt-4o", a.Model(t.Context()).ID().String()) |
| 355 | assert.False(t, a.HasModelOverride()) |
| 356 | |
| 357 | // Set an override |
| 358 | a.SetModelOverride(overrideModel) |
| 359 | assert.True(t, a.HasModelOverride()) |
| 360 | assert.Equal(t, "anthropic/claude-sonnet-4-0", a.Model(t.Context()).ID().String()) |
| 361 | |
| 362 | // ConfiguredModels still reflects the originally configured models |
| 363 | configuredModels := a.ConfiguredModels() |
| 364 | require.Len(t, configuredModels, 1) |
| 365 | assert.Equal(t, "openai/gpt-4o", configuredModels[0].ID().String()) |
| 366 | |
| 367 | // Clear the override |
| 368 | a.SetModelOverride(nil) |
| 369 | assert.False(t, a.HasModelOverride()) |
| 370 | assert.Equal(t, "openai/gpt-4o", a.Model(t.Context()).ID().String()) |
| 371 | } |
| 372 | |
| 373 | func TestSetModelOverride_ReturnsSnapshotOfStoredValue(t *testing.T) { |
| 374 | // SetModelOverride must return a snapshot of the value it just stored, |
nothing calls this directly
no test coverage detected