Function TestProviderFactory

pkg/provider/openai_compatible_test.go:318–363

TestProviderFactory 测试工厂模式

(t *testing.T)

// TestProviderFactory 测试工厂模式
func TestProviderFactory(t *testing.T) {
	factory := NewMultiProviderFactory()

	testCases := []struct {
		provider string
		model    string
		apiKey   string
		baseURL  string
	}{
		{"openai", "gpt-4o", "test-key", ""},
		{"groq", "llama-3.3-70b-versatile", "test-key", ""},
		{"ollama", "llama3.2", "", ""},
		{"openrouter", "openai/gpt-4o", "test-key", ""},
		{"mistral", "mistral-large-latest", "test-key", ""},
		{"doubao", "ep-xxxxx", "test-key", ""},
		{"moonshot", "moonshot-v1-128k", "test-key", ""},
		{"gemini", "gemini-2.0-flash-exp", "test-key", ""},
		{"custom", "gpt-4o", "test-key", "https://api.example.com/v1"},
	}

	for _, tc := range testCases {
		t.Run(tc.provider, func(t *testing.T) {
			config := &types.ModelConfig{
				Provider: tc.provider,
				Model:    tc.model,
				APIKey:   tc.apiKey,
				BaseURL:  tc.baseURL,
			}

			provider, err := factory.Create(config)
			if err != nil {
				t.Fatalf("Failed to create provider %s: %v", tc.provider, err)
			}

			if provider == nil {
				t.Fatalf("Provider %s is nil", tc.provider)
			}

			// 验证基本功能
			caps := provider.Capabilities()
			if !caps.SupportStreaming {
				t.Errorf("Provider %s should support streaming", tc.provider)
			}
		})
	}
}

// TestMessageConversion 测试消息转换
func TestMessageConversion(t *testing.T) {