Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
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
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
This commit is contained in:
@@ -0,0 +1,103 @@
|
||||
package service
|
||||
|
||||
import (
|
||||
"regexp"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestBuildOAuthMetadataUserID_FallbackWithoutAccountUUID(t *testing.T) {
|
||||
svc := &GatewayService{}
|
||||
|
||||
parsed := &ParsedRequest{
|
||||
Model: "claude-sonnet-4-5",
|
||||
Stream: true,
|
||||
MetadataUserID: "",
|
||||
}
|
||||
|
||||
account := &Account{
|
||||
ID: 123,
|
||||
Type: AccountTypeOAuth,
|
||||
Extra: map[string]any{}, // intentionally missing account_uuid / claude_user_id
|
||||
}
|
||||
|
||||
fp := &Fingerprint{ClientID: "deadbeef"} // should be used as user id in legacy format
|
||||
|
||||
got := svc.buildOAuthMetadataUserID(parsed, account, fp)
|
||||
require.NotEmpty(t, got)
|
||||
|
||||
// Legacy format: user_{client}_account__session_{uuid}
|
||||
re := regexp.MustCompile(`^user_[a-zA-Z0-9]+_account__session_[a-f0-9-]{36}$`)
|
||||
require.True(t, re.MatchString(got), "unexpected user_id format: %s", got)
|
||||
}
|
||||
|
||||
func TestBuildOAuthMetadataUserID_UsesAccountUUIDWhenPresent(t *testing.T) {
|
||||
svc := &GatewayService{}
|
||||
|
||||
parsed := &ParsedRequest{
|
||||
Model: "claude-sonnet-4-5",
|
||||
Stream: true,
|
||||
MetadataUserID: "",
|
||||
}
|
||||
|
||||
account := &Account{
|
||||
ID: 123,
|
||||
Type: AccountTypeOAuth,
|
||||
Extra: map[string]any{
|
||||
"account_uuid": "acc-uuid",
|
||||
"claude_user_id": "clientid123",
|
||||
"anthropic_user_id": "",
|
||||
},
|
||||
}
|
||||
|
||||
got := svc.buildOAuthMetadataUserID(parsed, account, nil)
|
||||
require.NotEmpty(t, got)
|
||||
|
||||
// New format: user_{client}_account_{account_uuid}_session_{uuid}
|
||||
re := regexp.MustCompile(`^user_clientid123_account_acc-uuid_session_[a-f0-9-]{36}$`)
|
||||
require.True(t, re.MatchString(got), "unexpected user_id format: %s", got)
|
||||
}
|
||||
|
||||
// TestBuildOAuthMetadataUserID_SessionIDStableAcrossTurns 验证伪装路径合成的
|
||||
// metadata.user_id 在同一会话多轮请求间保持不变(session_id 稳定),贴近真实 Claude Code
|
||||
// 进程级稳定的 session。账号 / 指纹 / UA 版本均相同,唯一可能变化的就是 session_id,
|
||||
// 因此直接比较完整 user_id 字符串即可判定 session_id 是否稳定。
|
||||
func TestBuildOAuthMetadataUserID_SessionIDStableAcrossTurns(t *testing.T) {
|
||||
svc := &GatewayService{}
|
||||
account := &Account{ID: 777, Type: AccountTypeOAuth, Extra: map[string]any{"account_uuid": "acc-uuid"}}
|
||||
fp := &Fingerprint{ClientID: "clientid777", UserAgent: "claude-cli/2.1.161 (external, cli)"}
|
||||
|
||||
mustParse := func(body string) *ParsedRequest {
|
||||
parsed, err := ParseGatewayRequest(NewRequestBodyRef([]byte(body)), PlatformAnthropic)
|
||||
require.NoError(t, err)
|
||||
return parsed
|
||||
}
|
||||
|
||||
round1 := mustParse(`{"model":"claude-sonnet-4-5","system":"sys","messages":[` +
|
||||
`{"role":"user","content":"first question"}]}`)
|
||||
round2 := mustParse(`{"model":"claude-sonnet-4-5","system":"sys","messages":[` +
|
||||
`{"role":"user","content":"first question"},` +
|
||||
`{"role":"assistant","content":"answer 1"},` +
|
||||
`{"role":"user","content":"second question"}]}`)
|
||||
round3 := mustParse(`{"model":"claude-sonnet-4-5","system":"sys","messages":[` +
|
||||
`{"role":"user","content":"first question"},` +
|
||||
`{"role":"assistant","content":"answer 1"},` +
|
||||
`{"role":"user","content":"second question"},` +
|
||||
`{"role":"assistant","content":"answer 2"},` +
|
||||
`{"role":"user","content":"third question"}]}`)
|
||||
|
||||
id1 := svc.buildOAuthMetadataUserID(round1, account, fp)
|
||||
id2 := svc.buildOAuthMetadataUserID(round2, account, fp)
|
||||
id3 := svc.buildOAuthMetadataUserID(round3, account, fp)
|
||||
|
||||
require.NotEmpty(t, id1)
|
||||
require.Equal(t, id1, id2, "session_id 应随对话增长保持不变")
|
||||
require.Equal(t, id2, id3, "session_id 应跨所有轮次保持不变")
|
||||
|
||||
// 不同的首条 user 消息应派生出不同的 session_id(不同会话)。
|
||||
other := mustParse(`{"model":"claude-sonnet-4-5","system":"sys","messages":[` +
|
||||
`{"role":"user","content":"a completely different opener"}]}`)
|
||||
idOther := svc.buildOAuthMetadataUserID(other, account, fp)
|
||||
require.NotEqual(t, id1, idOther, "不同首条消息应派生不同 session_id")
|
||||
}
|
||||
Reference in New Issue
Block a user