Function TestModelFallbackManager_AllModelsFail

pkg/agent/model_fallback_test.go:202–263

(t *testing.T)

}

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

	// 所有模型都失败
	factory.SetProvider("openai/gpt-4", &MockProvider{
		name:       "openai/gpt-4",
		shouldFail: true,
		failCount:  999,
	})

	factory.SetProvider("anthropic/claude-3", &MockProvider{
		name:       "anthropic/claude-3",
		shouldFail: true,
		failCount:  999,
	})

	deps := &Dependencies{
		ProviderFactory: factory,
	}

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

	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"},
	}

	_, err = manager.Complete(ctx, messages, nil)
	if err == nil {
		t.Fatal("Expected error when all models fail")
	}

	// 检查统计信息
	stats := manager.GetStats()