mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-16 07:33:06 +08:00
IanShaw027
7c62382d04
从 personal-dev 取优并接线到 feat/grok-complete-integration： - 调度阈值：grok_sched_* 写入、ApplyAccountSchedulingThreshold、 account_scheduling_thresholds 设置与 Settings UI、网关过滤 - 配额头：x-rate-limit 别名、相对秒 reset、tier/entitlement 扩展 - 批量 usage：GetUsageBatch + POST /usage/batch + AccountsView 合并加载 - CLI 身份：xai.cli_identity 统一 pin/UA；service 层 applyGrokCLIHeaders 与 transport 最终改写对齐；media eligibility 模块落地 - UsageCell：月度 cents→USD 与 30d 进度条展示 保留既有 failure taxonomy / sticky / free recovery 等 HEAD cools。
87 lines
3.1 KiB
Go
87 lines
3.1 KiB
Go
package service
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/tlsfingerprint"
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/xai"
|
|
)
|
|
|
|
func TestApplyDefaultGrokUpstreamHeadersUsesCLIUserAgent(t *testing.T) {
|
|
t.Setenv(xai.CLIVersionEnv, "")
|
|
|
|
req, err := http.NewRequest(http.MethodGet, "https://api.x.ai/v1/responses", nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("User-Agent", "claude-code/1.2.3")
|
|
req.Header.Set("x-grok-client-version", "none")
|
|
|
|
applyDefaultGrokUpstreamHeaders(req)
|
|
|
|
require.Equal(t, xai.CLIUserAgent(xai.CLIClientVersion), req.Header.Get("User-Agent"))
|
|
require.Equal(t, xai.CLIClientVersion, req.Header.Get("x-grok-client-version"))
|
|
require.Equal(t, xai.CLIClientIdentifier, req.Header.Get("x-grok-client-identifier"))
|
|
}
|
|
|
|
func TestApplyDefaultGrokUpstreamHeadersHonorsCLIVersionOverride(t *testing.T) {
|
|
t.Setenv(xai.CLIVersionEnv, "0.2.95")
|
|
|
|
req, err := http.NewRequest(http.MethodGet, "https://api.x.ai/v1/responses", nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("User-Agent", "codex_cli_rs/0.144.0")
|
|
|
|
applyDefaultGrokUpstreamHeaders(req)
|
|
|
|
require.Equal(t, "0.2.95", req.Header.Get("x-grok-client-version"))
|
|
require.Equal(t, xai.CLIUserAgent("0.2.95"), req.Header.Get("User-Agent"))
|
|
require.Equal(t, "grok-shell", req.Header.Get("x-grok-client-identifier"))
|
|
}
|
|
|
|
func TestResolveGrokUpstreamUserAgentNeverPassthrough(t *testing.T) {
|
|
t.Setenv(xai.CLIVersionEnv, "")
|
|
gin.SetMode(gin.TestMode)
|
|
|
|
rec := httptest.NewRecorder()
|
|
c, _ := gin.CreateTestContext(rec)
|
|
c.Request = httptest.NewRequest(http.MethodPost, "/v1/responses", nil)
|
|
c.Request.Header.Set("User-Agent", "claude-cli/2.0.0 (Mac OS; arm64)")
|
|
|
|
require.Equal(t, xai.CLIUserAgent(xai.CLIClientVersion), resolveGrokUpstreamUserAgent(c))
|
|
require.Equal(t, xai.CLIUserAgent(xai.CLIClientVersion), resolveGrokUpstreamUserAgent(nil))
|
|
}
|
|
|
|
func TestApplyGrokRuntimeHeadersKeepsCLIUserAgent(t *testing.T) {
|
|
t.Setenv(xai.CLIVersionEnv, "")
|
|
|
|
req, err := http.NewRequest(http.MethodPost, "https://cli-chat-proxy.grok.com/v1/responses", nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("User-Agent", "claude-code/9.9.9")
|
|
|
|
applyGrokRuntimeHeaders(req, openAITLSFingerprintRuntime{
|
|
UpstreamUserAgent: "codex_cli_rs/0.144.0",
|
|
UpstreamOriginator: "codex_cli_rs",
|
|
})
|
|
|
|
require.Equal(t, xai.CLIUserAgent(xai.CLIClientVersion), req.Header.Get("User-Agent"))
|
|
require.Equal(t, "codex_cli_rs", req.Header.Get("Originator"))
|
|
require.Equal(t, xai.CLIClientVersion, req.Header.Get("x-grok-client-version"))
|
|
}
|
|
|
|
func TestApplyGrokTLSProfileHeadersAlwaysUsesCLIUserAgent(t *testing.T) {
|
|
t.Setenv(xai.CLIVersionEnv, "")
|
|
|
|
req, err := http.NewRequest(http.MethodPost, "https://api.x.ai/v1/responses", nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("User-Agent", "grok-native/1.0")
|
|
|
|
// HEAD Profile is TLS-only; Originator/UserAgent HTTP fields are not present.
|
|
applyGrokTLSProfileHeaders(req, &tlsfingerprint.Profile{Name: "chrome"})
|
|
|
|
require.Equal(t, xai.CLIUserAgent(xai.CLIClientVersion), req.Header.Get("User-Agent"))
|
|
require.Equal(t, xai.CLIClientVersion, req.Header.Get("x-grok-client-version"))
|
|
}
|