李建琦
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
68 lines
2.0 KiB
Go
68 lines
2.0 KiB
Go
package xai
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveCLIVersionDefaultsToPinnedClientVersion(t *testing.T) {
|
|
t.Setenv(CLIVersionEnv, "")
|
|
// Default advertise pin is CLIClientVersion; CLIStableVersion is only the floor.
|
|
require.Equal(t, CLIClientVersion, ResolveCLIVersion())
|
|
require.True(t, IsSupportedCLIVersion(CLIClientVersion))
|
|
require.True(t, IsSupportedCLIVersion(CLIStableVersion))
|
|
}
|
|
|
|
func TestResolveCLIVersionAcceptsValidOverride(t *testing.T) {
|
|
t.Setenv(CLIVersionEnv, "0.2.95-alpha.1")
|
|
require.Equal(t, "0.2.95-alpha.1", ResolveCLIVersion())
|
|
}
|
|
|
|
func TestResolveCLIVersionRejectsUnsafeOrTooOld(t *testing.T) {
|
|
for _, version := range []string{
|
|
"0.2.92",
|
|
"0.2.93-beta.1",
|
|
"0.2.95\r\nX-Injected: true",
|
|
"0.2.093",
|
|
"0.3",
|
|
"1",
|
|
} {
|
|
t.Run(version, func(t *testing.T) {
|
|
t.Setenv(CLIVersionEnv, version)
|
|
require.Equal(t, CLIClientVersion, ResolveCLIVersion())
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestApplyCLIProxyHeaders(t *testing.T) {
|
|
t.Setenv(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", "sub2api-grok/1.0")
|
|
|
|
ApplyCLIProxyHeaders(req)
|
|
|
|
require.Equal(t, CLIClientVersion, req.Header.Get("x-grok-client-version"))
|
|
require.Equal(t, CLIClientIdentifier, req.Header.Get("x-grok-client-identifier"))
|
|
require.Equal(t, CLITokenAuth, req.Header.Get("X-XAI-Token-Auth"))
|
|
require.Equal(t, CLIUserAgent(CLIClientVersion), req.Header.Get("User-Agent"))
|
|
}
|
|
|
|
func TestApplyCLIProxyHeadersLeavesAPIHostUnchanged(t *testing.T) {
|
|
t.Setenv(CLIVersionEnv, "0.2.95")
|
|
|
|
req, err := http.NewRequest(http.MethodPost, "https://api.x.ai/v1/responses", nil)
|
|
require.NoError(t, err)
|
|
req.Header.Set("User-Agent", "sub2api-grok/1.0")
|
|
|
|
ApplyCLIProxyHeaders(req)
|
|
|
|
require.Empty(t, req.Header.Get("x-grok-client-version"))
|
|
require.Empty(t, req.Header.Get("x-grok-client-identifier"))
|
|
require.Empty(t, req.Header.Get("X-XAI-Token-Auth"))
|
|
require.Equal(t, "sub2api-grok/1.0", req.Header.Get("User-Agent"))
|
|
}
|