TestOpenRouterProvider 测试 OpenRouter Provider
(t *testing.T)
| 118 | |
| 119 | // TestOpenRouterProvider 测试 OpenRouter Provider |
| 120 | func TestOpenRouterProvider(t *testing.T) { |
| 121 | config := &types.ModelConfig{ |
| 122 | Provider: "openrouter", |
| 123 | Model: "openai/gpt-4o", |
| 124 | APIKey: "test-key", |
| 125 | } |
| 126 | |
| 127 | provider, err := NewOpenRouterProviderSimple(config) |
| 128 | if err != nil { |
| 129 | t.Fatalf("Failed to create OpenRouter provider: %v", err) |
| 130 | } |
| 131 | |
| 132 | // 测试配置 |
| 133 | if provider.Config().Model != "openai/gpt-4o" { |
| 134 | t.Errorf("Expected model=openai/gpt-4o, got %s", provider.Config().Model) |
| 135 | } |
| 136 | |
| 137 | // 测试能力 |
| 138 | caps := provider.Capabilities() |
| 139 | if !caps.SupportStreaming { |
| 140 | t.Error("OpenRouter should support streaming") |
| 141 | } |
| 142 | if !caps.SupportToolCalling { |
| 143 | t.Error("OpenRouter should support tool calling") |
| 144 | } |
| 145 | if !caps.SupportReasoning { |
| 146 | t.Error("OpenRouter should support reasoning models") |
| 147 | } |
| 148 | if caps.MaxTokens < 100000 { |
| 149 | t.Error("OpenRouter should support large context") |
| 150 | } |
| 151 | } |
| 152 | |
| 153 | // TestMistralProvider 测试 Mistral Provider |
| 154 | func TestMistralProvider(t *testing.T) { |
nothing calls this directly
no test coverage detected