Function TestModelFallbackManager_RetrySuccess

pkg/agent/model_fallback_test.go:265–315

(t *testing.T)

}

func TestModelFallbackManager_RetrySuccess(t *testing.T) {
	factory := NewMockProviderFactory()

	// 第一次失败，第二次成功
	factory.SetProvider("openai/gpt-4", &MockProvider{
		name:       "openai/gpt-4",
		shouldFail: true,
		failCount:  1, // 只失败一次
	})

	deps := &Dependencies{
		ProviderFactory: factory,
	}

	fallbacks := []*ModelFallback{
		{
			Config: &types.ModelConfig{
				Provider: "openai",
				Model:    "gpt-4",
			},
			MaxRetries: 2,
			Enabled:    true,
			Priority:   1,
		},
	}

	manager, err := NewModelFallbackManager(fallbacks, deps)
	if err != nil {
		t.Fatalf("Failed to create manager: %v", err)
	}

	ctx := context.Background()
	messages := []types.Message{
		{Role: "user", Content: "Hello"},
	}

	resp, err := manager.Complete(ctx, messages, nil)
	if err != nil {
		t.Fatalf("Complete failed: %v", err)
	}

	if resp.Message.Content != "mock response from openai/gpt-4" {
		t.Errorf("Expected response from gpt-4, got: %s", resp.Message.Content)
	}

	// 检查统计信息
	stats := manager.GetStats()
	if stats.FallbackCount != 0 {
		t.Errorf("Expected 0 fallbacks (retry succeeded), got: %d", stats.FallbackCount)
	}
}

func TestModelFallbackManager_EnableDisableModel(t *testing.T) {
	factory := NewMockProviderFactory()