Sub2API v1.0 - AI API 网关（二开初始版本，基于上游 Wei-Shaw/sub2api）

backend/internal/service/grok_upstream_headers_test.go

@@ -0,0 +1,86 @@
+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"))
+}