Function TestModelFallbackManager_BasicFallback

pkg/agent/model_fallback_test.go:129–200

(t *testing.T)

}

func TestModelFallbackManager_BasicFallback(t *testing.T) {
	// 创建 mock factory
	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",
	})

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

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