李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
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"))
|
|
}
|