mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-10-14 03:52:26 +08:00
feeeei
05d7946af8
- 绑定缓存的请求与缓存管理接口收到 429 / 5xx 时，只有上游网关账号的响应原样回写
（UpstreamFailoverError.BoundUpstreamPassthrough / GeminiCachedContentUpstreamError.UpstreamGateway），
官方账号仍走错误透传规则与通用映射，不向客户端暴露配额明细
- 记录新增 channel_usage（迁移 244）保存创建时的渠道计费口径，PATCH 延长按同一口径预检、报价与记账
- 创建 / 延长费用行的 request_id 改为固定前缀 gcache:{create|patch}:{id}，不被客户端或网关请求 id 覆盖，
按缓存 ID 即可检索一个缓存的全部费用行
- 调度快照凭据投影补齐 base_url / auth_mode / 上游显式缓存声明，显式缓存候选过滤一次算全，
不再逐个选号后剔除
78 lines
3.2 KiB
Go
78 lines
3.2 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestResolveUsageBillingRequestID_ForcedWebSearchBeatsClientID(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, "web_search:uuid-1")
|
|
require.Equal(t, "web_search:uuid-1", got)
|
|
}
|
|
|
|
func TestResolveUsageBillingRequestID_ClientWinsOverPlainUpstream(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, "resp_abc")
|
|
require.Equal(t, "client:client-shared-id", got)
|
|
}
|
|
|
|
func TestIsForcedUsageBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.True(t, isForcedUsageBillingRequestID("web_search:x"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok-video:task-1"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok_audio:up-1"))
|
|
require.True(t, isForcedUsageBillingRequestID("grok_realtime:sess-1"))
|
|
require.True(t, isForcedUsageBillingRequestID(GeminiCachedContentCreateUsageRequestID("abc")))
|
|
require.False(t, isForcedUsageBillingRequestID("resp_abc"))
|
|
}
|
|
|
|
func TestGeminiCachedContentUsageRequestIDs(t *testing.T) {
|
|
t.Parallel()
|
|
publicID, err := NewGeminiCachedContentPublicID()
|
|
require.NoError(t, err)
|
|
create := GeminiCachedContentCreateUsageRequestID(publicID)
|
|
patch := GeminiCachedContentPatchUsageRequestID(publicID)
|
|
require.Equal(t, "gcache:create:"+publicID, create)
|
|
require.True(t, strings.HasPrefix(patch, "gcache:patch:"+publicID+":"))
|
|
require.NotEqual(t, patch, GeminiCachedContentPatchUsageRequestID(publicID), "每次延长各自唯一")
|
|
require.LessOrEqual(t, len(create), 64, "usage_logs.request_id 列长 64")
|
|
require.LessOrEqual(t, len(patch), 64, "usage_logs.request_id 列长 64")
|
|
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
require.Equal(t, create, resolveUsageBillingRequestID(ctx, create), "缓存费用行不被客户端请求 id 覆盖")
|
|
require.Equal(t, patch, resolveUsageBillingRequestID(ctx, patch))
|
|
}
|
|
|
|
func TestStableGrokAudioBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, "grok_audio:up-1", StableGrokAudioBillingRequestID("up-1"))
|
|
require.Equal(t, "grok_audio:up-1", StableGrokAudioBillingRequestID("grok_audio:up-1"))
|
|
got := StableGrokAudioBillingRequestID("")
|
|
require.True(t, strings.HasPrefix(got, "grok_audio:"))
|
|
require.Greater(t, len(got), len("grok_audio:"))
|
|
}
|
|
|
|
func TestStableGrokRealtimeBillingRequestID(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, "grok_realtime:s1", StableGrokRealtimeBillingRequestID("s1"))
|
|
require.Equal(t, "grok_realtime:s1", StableGrokRealtimeBillingRequestID("grok_realtime:s1"))
|
|
got := StableGrokRealtimeBillingRequestID("")
|
|
require.True(t, strings.HasPrefix(got, "grok_realtime:"))
|
|
}
|
|
|
|
func TestResolveUsageBillingRequestID_ForcedGrokAudioBeatsClientID(t *testing.T) {
|
|
t.Parallel()
|
|
ctx := context.WithValue(context.Background(), ctxkey.ClientRequestID, "client-shared-id")
|
|
got := resolveUsageBillingRequestID(ctx, StableGrokAudioBillingRequestID("up-9"))
|
|
require.Equal(t, "grok_audio:up-9", got)
|
|
}
|