fdcbf7aacf
提取 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 测试
129 lines
3.8 KiB
Go
129 lines
3.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"io"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type ClaudeUsageServiceSuite struct {
|
|
suite.Suite
|
|
srv *httptest.Server
|
|
fetcher *claudeUsageService
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TearDownTest() {
|
|
if s.srv != nil {
|
|
s.srv.Close()
|
|
s.srv = nil
|
|
}
|
|
}
|
|
|
|
// usageRequestCapture holds captured request data for assertions in the main goroutine.
|
|
type usageRequestCapture struct {
|
|
authorization string
|
|
anthropicBeta string
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TestFetchUsage_Success() {
|
|
var captured usageRequestCapture
|
|
|
|
s.srv = newLocalTestServer(s.T(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
captured.authorization = r.Header.Get("Authorization")
|
|
captured.anthropicBeta = r.Header.Get("anthropic-beta")
|
|
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = io.WriteString(w, `{
|
|
"five_hour": {"utilization": 12.5, "resets_at": "2025-01-01T00:00:00Z"},
|
|
"seven_day": {"utilization": 34.0, "resets_at": "2025-01-08T00:00:00Z"},
|
|
"seven_day_sonnet": {"utilization": 56.0, "resets_at": "2025-01-08T00:00:00Z"}
|
|
}`)
|
|
}))
|
|
|
|
s.fetcher = &claudeUsageService{
|
|
usageURL: s.srv.URL,
|
|
allowPrivateHosts: true,
|
|
}
|
|
|
|
resp, err := s.fetcher.FetchUsage(context.Background(), "at", "")
|
|
require.NoError(s.T(), err, "FetchUsage")
|
|
require.Equal(s.T(), 12.5, resp.FiveHour.Utilization, "FiveHour utilization mismatch")
|
|
require.Equal(s.T(), 34.0, resp.SevenDay.Utilization, "SevenDay utilization mismatch")
|
|
require.Equal(s.T(), 56.0, resp.SevenDaySonnet.Utilization, "SevenDaySonnet utilization mismatch")
|
|
|
|
// Assertions on captured request data
|
|
require.Equal(s.T(), "Bearer at", captured.authorization, "Authorization header mismatch")
|
|
require.Equal(s.T(), "oauth-2025-04-20", captured.anthropicBeta, "anthropic-beta header mismatch")
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TestFetchUsage_NonOK() {
|
|
s.srv = newLocalTestServer(s.T(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusUnauthorized)
|
|
_, _ = io.WriteString(w, "nope")
|
|
}))
|
|
|
|
s.fetcher = &claudeUsageService{
|
|
usageURL: s.srv.URL,
|
|
allowPrivateHosts: true,
|
|
}
|
|
|
|
_, err := s.fetcher.FetchUsage(context.Background(), "at", "")
|
|
require.Error(s.T(), err)
|
|
require.ErrorContains(s.T(), err, "status 401")
|
|
require.ErrorContains(s.T(), err, "nope")
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TestFetchUsage_BadJSON() {
|
|
s.srv = newLocalTestServer(s.T(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("Content-Type", "application/json")
|
|
_, _ = io.WriteString(w, "not-json")
|
|
}))
|
|
|
|
s.fetcher = &claudeUsageService{
|
|
usageURL: s.srv.URL,
|
|
allowPrivateHosts: true,
|
|
}
|
|
|
|
_, err := s.fetcher.FetchUsage(context.Background(), "at", "")
|
|
require.Error(s.T(), err)
|
|
require.ErrorContains(s.T(), err, "decode response failed")
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TestFetchUsage_ContextCancel() {
|
|
s.srv = newLocalTestServer(s.T(), http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
// Never respond - simulate slow server
|
|
<-r.Context().Done()
|
|
}))
|
|
|
|
s.fetcher = &claudeUsageService{
|
|
usageURL: s.srv.URL,
|
|
allowPrivateHosts: true,
|
|
}
|
|
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel() // Cancel immediately
|
|
|
|
_, err := s.fetcher.FetchUsage(ctx, "at", "")
|
|
require.Error(s.T(), err, "expected error for cancelled context")
|
|
}
|
|
|
|
func (s *ClaudeUsageServiceSuite) TestFetchUsage_InvalidProxyReturnsError() {
|
|
s.fetcher = &claudeUsageService{
|
|
usageURL: "http://example.com",
|
|
allowPrivateHosts: true,
|
|
}
|
|
|
|
_, err := s.fetcher.FetchUsage(context.Background(), "at", "://bad-proxy-url")
|
|
require.Error(s.T(), err)
|
|
require.ErrorContains(s.T(), err, "create http client failed")
|
|
}
|
|
|
|
func TestClaudeUsageServiceSuite(t *testing.T) {
|
|
suite.Run(t, new(ClaudeUsageServiceSuite))
|
|
}
|