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,93 @@
|
||||
package service
|
||||
|
||||
// 投影漏列回归(service 半程):认证快照 build → L2 JSON 序列化
|
||||
// → 反序列化 → 还原 apiKey.Group → 请求 ctx → 利润门解析,全链路保真。
|
||||
// repository 半程(真实 GetByKeyForAuth 投影)见
|
||||
// internal/repository/api_key_repo_profit_projection_integration_test.go。
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"testing"
|
||||
|
||||
"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func profitAuthTestAPIKey() *APIKey {
|
||||
groupID := int64(50)
|
||||
return &APIKey{
|
||||
ID: 82,
|
||||
UserID: 40,
|
||||
GroupID: &groupID,
|
||||
Name: "profit-auth-roundtrip",
|
||||
Status: StatusActive,
|
||||
User: &User{
|
||||
ID: 40,
|
||||
Email: "profit@test.local",
|
||||
Status: StatusActive,
|
||||
Concurrency: 5,
|
||||
},
|
||||
Group: &Group{
|
||||
ID: groupID,
|
||||
Name: "VIP-roundtrip",
|
||||
Platform: PlatformOpenAI,
|
||||
Status: StatusActive,
|
||||
Hydrated: true,
|
||||
RateMultiplier: 0.06,
|
||||
SubscriptionType: SubscriptionTypeStandard,
|
||||
PeakRateEnabled: false,
|
||||
ProfitControlEnabled: true,
|
||||
ProfitMinMargin: 0.2,
|
||||
ProfitSafetyBuffer: 0.05,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
// 快照构建 → L2 JSON 往返 → 还原 → 装门:利润字段必须全程保真,阈值与
|
||||
// 计费同源(0.06 × (1−0.25) = 0.045)。
|
||||
func TestAPIKeyAuthSnapshotProfitControlRoundtrip(t *testing.T) {
|
||||
svc := &APIKeyService{}
|
||||
apiKey := profitAuthTestAPIKey()
|
||||
|
||||
snapshot := svc.snapshotFromAPIKey(context.Background(), apiKey)
|
||||
require.NotNil(t, snapshot)
|
||||
require.Equal(t, apiKeyAuthSnapshotVersion, snapshot.Version)
|
||||
require.Equal(t, 20, snapshot.Version, "v20 起认证快照携带分组长上下文与模型定价字段")
|
||||
|
||||
// 模拟 L2 缓存的完整 JSON 往返(与 apiKeyCache.SetAuthCache/GetAuthCache 同构)。
|
||||
payload, err := json.Marshal(&APIKeyAuthCacheEntry{Snapshot: snapshot})
|
||||
require.NoError(t, err)
|
||||
var restored APIKeyAuthCacheEntry
|
||||
require.NoError(t, json.Unmarshal(payload, &restored))
|
||||
|
||||
materialized, used, err := svc.applyAuthCacheEntry(apiKey.Key, &restored)
|
||||
require.NoError(t, err)
|
||||
require.True(t, used)
|
||||
require.NotNil(t, materialized.Group)
|
||||
require.True(t, materialized.Group.Hydrated)
|
||||
require.True(t, materialized.Group.ProfitControlEnabled)
|
||||
require.InDelta(t, 0.2, materialized.Group.ProfitMinMargin, 1e-12)
|
||||
require.InDelta(t, 0.05, materialized.Group.ProfitSafetyBuffer, 1e-12)
|
||||
require.InDelta(t, 0.06, materialized.Group.RateMultiplier, 1e-12)
|
||||
|
||||
// 中间件语义:materialized.Group 进请求 ctx → 门必须按快照配置装上。
|
||||
ctx := context.WithValue(context.Background(), ctxkey.Group, materialized.Group)
|
||||
gwSvc := &OpenAIGatewayService{}
|
||||
gate := gwSvc.resolveOpenAIProfitControlGate(ctx, materialized.GroupID)
|
||||
require.NotNil(t, gate, "还原后的认证分组必须能装门(投影漏列时本断言最先失败)")
|
||||
require.InDelta(t, 0.06*(1-0.25), gate.threshold, 1e-12)
|
||||
}
|
||||
|
||||
// 旧版本快照(v16 及更早,无利润字段保真保证)必须被淘汰回源,不得复用。
|
||||
func TestAPIKeyAuthSnapshotOldVersionEvicted(t *testing.T) {
|
||||
svc := &APIKeyService{}
|
||||
snapshot := svc.snapshotFromAPIKey(context.Background(), profitAuthTestAPIKey())
|
||||
require.NotNil(t, snapshot)
|
||||
snapshot.Version = 16
|
||||
|
||||
materialized, used, err := svc.applyAuthCacheEntry("sk-old", &APIKeyAuthCacheEntry{Snapshot: snapshot})
|
||||
require.NoError(t, err)
|
||||
require.False(t, used, "版本不匹配的缓存条目必须淘汰并回源重建")
|
||||
require.Nil(t, materialized)
|
||||
}
|
||||
Reference in New Issue
Block a user