feat(proxy): 集中代理 URL 验证并实现全局 fail-fast
提取 proxyurl.Parse() 公共包,将分散在 6 处的代理 URL 验证逻辑 统一收敛,确保无效代理配置在创建时立即失败,永不静默回退直连。 主要变更: - 新增 proxyurl 包:统一 TrimSpace → url.Parse → Host 校验 → Scheme 白名单 - socks5:// 自动升级为 socks5h://,防止 DNS 泄漏(大小写不敏感) - antigravity: http.ProxyURL → proxyutil.ConfigureTransportProxy 支持 SOCKS5 - openai_oauth: 删除 newOpenAIOAuthHTTPClient,收编至 httpclient.GetClient - 移除未使用的 ProxyStrict 字段(fail-fast 已是全局默认行为) - 补充 15 个 proxyurl 测试 + pricing/usage fail-fast 测试
This commit is contained in:
@@ -19,7 +19,7 @@ type PricingServiceSuite struct {
|
||||
|
||||
func (s *PricingServiceSuite) SetupTest() {
|
||||
s.ctx = context.Background()
|
||||
client, ok := NewPricingRemoteClient("").(*pricingRemoteClient)
|
||||
client, ok := NewPricingRemoteClient("", false).(*pricingRemoteClient)
|
||||
require.True(s.T(), ok, "type assertion failed")
|
||||
s.client = client
|
||||
}
|
||||
@@ -140,6 +140,22 @@ func (s *PricingServiceSuite) TestFetchPricingJSON_ContextCancel() {
|
||||
require.Error(s.T(), err)
|
||||
}
|
||||
|
||||
func TestNewPricingRemoteClient_InvalidProxy_NoFallback(t *testing.T) {
|
||||
client := NewPricingRemoteClient("://bad", false)
|
||||
_, ok := client.(*pricingRemoteClientError)
|
||||
require.True(t, ok, "should return error client when proxy is invalid and fallback disabled")
|
||||
|
||||
_, err := client.FetchPricingJSON(context.Background(), "http://example.com")
|
||||
require.Error(t, err)
|
||||
require.Contains(t, err.Error(), "proxy client init failed")
|
||||
}
|
||||
|
||||
func TestNewPricingRemoteClient_InvalidProxy_WithFallback(t *testing.T) {
|
||||
client := NewPricingRemoteClient("://bad", true)
|
||||
_, ok := client.(*pricingRemoteClient)
|
||||
require.True(t, ok, "should fallback to direct client when allowed")
|
||||
}
|
||||
|
||||
func TestPricingServiceSuite(t *testing.T) {
|
||||
suite.Run(t, new(PricingServiceSuite))
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user