Files
sub2api/backend/internal/service/credentials_sanitize_test.go
李建琦 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
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

51 lines
1.5 KiB
Go

package service
import (
"testing"
"github.com/stretchr/testify/require"
)
func TestSanitizeStoredCredentials_StripsEphemeralSSOSecrets(t *testing.T) {
creds := map[string]any{
"access_token": "at",
"refresh_token": "rt",
"password": "secret",
"sso_token": "sso",
"sso": "cookie-sso",
"sso-rw": "rw",
"clearTextPassword": "plain",
"cookie": "jar",
"base_url": "https://api.x.ai",
}
out := SanitizeStoredCredentials(PlatformGrok, creds)
require.Equal(t, "at", out["access_token"])
require.Equal(t, "rt", out["refresh_token"])
require.Equal(t, "https://api.x.ai", out["base_url"])
require.NotContains(t, out, "password")
require.NotContains(t, out, "sso_token")
require.NotContains(t, out, "sso")
require.NotContains(t, out, "sso-rw")
require.NotContains(t, out, "clearTextPassword")
require.NotContains(t, out, "cookie")
}
func TestSanitizeStoredCredentials_AlwaysStripsCookie(t *testing.T) {
// Bulk paths may pass empty platform; cookie must never persist next to tokens.
for _, platform := range []string{PlatformOpenAI, PlatformGrok, ""} {
creds := map[string]any{
"cookie": "session",
"password": "x",
"api_key": "k",
}
out := SanitizeStoredCredentials(platform, creds)
require.Equal(t, "k", out["api_key"], platform)
require.NotContains(t, out, "password", platform)
require.NotContains(t, out, "cookie", platform)
}
}
func TestSanitizeStoredCredentials_NilSafe(t *testing.T) {
require.Nil(t, SanitizeStoredCredentials(PlatformGrok, nil))
}