李建琦
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
603 lines
26 KiB
Go
603 lines
26 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// 夹具模型必须同时满足两个条件，否则测的就不是想测的那条规则：
|
|
// 1. 两者价格不同——否则"更便宜才采纳"的断言退化成恒真；
|
|
// 2. 两者都能被 HasIdentifiedTokenPricing 确定性识别（即价格表里的精确条目），
|
|
// 否则请求会先被"响应模型必须可识别"这道更靠前的门挡掉，成本比较根本走不到。
|
|
//
|
|
// claude-opus-4 / gpt-5.1 之类的名字不满足条件 2（前者不是 fallback 精确键，
|
|
// 后者与 gpt-5.5 共用同一条 gpt-5.4 价格因而也不满足条件 1）。
|
|
const (
|
|
anthropicCheapFixtureModel = "claude-sonnet-4"
|
|
anthropicPriceyFixtureModel = "claude-opus-4.8"
|
|
openAICheapFixtureModel = "gpt-5.4-nano"
|
|
openAIPriceyFixtureModel = "gpt-5.5"
|
|
)
|
|
|
|
// orderedResponseBillingModels 返回 (cheaper, pricier) 及各自成本，按当前价格表排序，
|
|
// 使断言不依赖两个具体模型的价格大小关系（价格表调整时测试仍然自洽）。
|
|
func orderedResponseBillingModels(t *testing.T, svc *BillingService, tokens UsageTokens, a, b string) (string, string, *CostBreakdown, *CostBreakdown) {
|
|
t.Helper()
|
|
costA, err := svc.CalculateCost(a, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
costB, err := svc.CalculateCost(b, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
require.NotEqual(t, costA.TotalCost, costB.TotalCost, "fixture prices for %s and %s must differ", a, b)
|
|
require.True(t, svc.HasIdentifiedTokenPricing(a), "fixture model %s must be identifiable in the pricing table", a)
|
|
require.True(t, svc.HasIdentifiedTokenPricing(b), "fixture model %s must be identifiable in the pricing table", b)
|
|
if costA.TotalCost < costB.TotalCost {
|
|
return a, b, costA, costB
|
|
}
|
|
return b, a, costB, costA
|
|
}
|
|
|
|
// --- Anthropic gateway (GatewayService.RecordUsage) ---
|
|
|
|
func TestGatewayServiceRecordUsage_ResponseModelBillsCheaperResponseModel(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newGatewayRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{})
|
|
tokens := UsageTokens{InputTokens: 100, OutputTokens: 50}
|
|
cheaper, pricier, cheaperCost, _ := orderedResponseBillingModels(t, svc.billingService, tokens, anthropicCheapFixtureModel, anthropicPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &RecordUsageInput{
|
|
Result: &ForwardResult{
|
|
RequestID: "gateway_response_model_downgrade",
|
|
Usage: ClaudeUsage{InputTokens: 100, OutputTokens: 50},
|
|
Model: pricier,
|
|
UpstreamResponseModel: cheaper, // upstream declared a runtime downgrade
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 501, Quota: 100},
|
|
User: &User{ID: 601},
|
|
Account: &Account{ID: 701},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, cheaperCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, cheaperCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
require.True(t, usageRepo.lastLog.ActualCost > 0, "cost must not be zero")
|
|
// 审计链完整保留：请求/发送模型不因计费切换被改写，响应模型与 mismatch 记录在案。
|
|
require.Equal(t, pricier, usageRepo.lastLog.Model)
|
|
require.Equal(t, pricier, usageRepo.lastLog.RequestedModel)
|
|
require.NotNil(t, usageRepo.lastLog.UpstreamResponseModel)
|
|
require.Equal(t, cheaper, *usageRepo.lastLog.UpstreamResponseModel)
|
|
require.NotNil(t, usageRepo.lastLog.UpstreamModelMismatch)
|
|
require.True(t, *usageRepo.lastLog.UpstreamModelMismatch)
|
|
}
|
|
|
|
func TestGatewayServiceRecordUsage_ResponseModelRejectsPricierResponseModel(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newGatewayRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{})
|
|
tokens := UsageTokens{InputTokens: 100, OutputTokens: 50}
|
|
cheaper, pricier, cheaperCost, _ := orderedResponseBillingModels(t, svc.billingService, tokens, anthropicCheapFixtureModel, anthropicPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &RecordUsageInput{
|
|
Result: &ForwardResult{
|
|
RequestID: "gateway_response_model_forged_upgrade",
|
|
Usage: ClaudeUsage{InputTokens: 100, OutputTokens: 50},
|
|
Model: cheaper,
|
|
UpstreamResponseModel: pricier, // forged/upgraded declaration must not raise cost
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 501, Quota: 100},
|
|
User: &User{ID: 601},
|
|
Account: &Account{ID: 701},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: cheaper,
|
|
ChannelMappedModel: cheaper,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, cheaperCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, cheaperCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
func TestGatewayServiceRecordUsage_ResponseModelSafeFallbacks(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
responseModel func(cheaper string) string
|
|
conflict bool
|
|
source string
|
|
}{
|
|
{
|
|
name: "in_stream_conflict_falls_back_to_baseline",
|
|
responseModel: func(cheaper string) string { return cheaper },
|
|
conflict: true,
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "empty_response_model_falls_back_to_baseline",
|
|
responseModel: func(string) string { return "" },
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "unpriced_response_model_falls_back_to_baseline",
|
|
responseModel: func(string) string { return "zz-unpriced-response-model" },
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "default_channel_mapped_mode_ignores_response_model",
|
|
responseModel: func(cheaper string) string { return cheaper },
|
|
source: BillingModelSourceChannelMapped,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newGatewayRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{})
|
|
tokens := UsageTokens{InputTokens: 100, OutputTokens: 50}
|
|
cheaper, pricier, _, pricierCost := orderedResponseBillingModels(t, svc.billingService, tokens, anthropicCheapFixtureModel, anthropicPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &RecordUsageInput{
|
|
Result: &ForwardResult{
|
|
RequestID: "gateway_response_model_fallback_" + tt.name,
|
|
Usage: ClaudeUsage{InputTokens: 100, OutputTokens: 50},
|
|
Model: pricier,
|
|
UpstreamResponseModel: tt.responseModel(cheaper),
|
|
UpstreamResponseModelConflict: tt.conflict,
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 501, Quota: 100},
|
|
User: &User{ID: 601},
|
|
Account: &Account{ID: 701},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: tt.source,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, pricierCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, pricierCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
})
|
|
}
|
|
}
|
|
|
|
// --- OpenAI gateway (OpenAIGatewayService.RecordUsage) ---
|
|
|
|
func TestOpenAIGatewayServiceRecordUsage_ResponseModelBillsCheaperResponseModel(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newOpenAIRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{}, nil)
|
|
tokens := UsageTokens{InputTokens: 20, OutputTokens: 10}
|
|
cheaper, pricier, cheaperCost, _ := orderedResponseBillingModels(t, svc.billingService, tokens, openAICheapFixtureModel, openAIPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &OpenAIRecordUsageInput{
|
|
Result: &OpenAIForwardResult{
|
|
RequestID: "openai_response_model_downgrade",
|
|
Model: pricier,
|
|
UpstreamModel: pricier,
|
|
UpstreamResponseModel: cheaper,
|
|
Usage: OpenAIUsage{InputTokens: 20, OutputTokens: 10},
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 10},
|
|
User: &User{ID: 20},
|
|
Account: &Account{ID: 30},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, cheaperCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, cheaperCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
require.True(t, usageRepo.lastLog.ActualCost > 0, "cost must not be zero")
|
|
// 审计链完整保留。
|
|
require.Equal(t, pricier, usageRepo.lastLog.Model)
|
|
require.NotNil(t, usageRepo.lastLog.UpstreamResponseModel)
|
|
require.Equal(t, cheaper, *usageRepo.lastLog.UpstreamResponseModel)
|
|
require.NotNil(t, usageRepo.lastLog.UpstreamModelMismatch)
|
|
require.True(t, *usageRepo.lastLog.UpstreamModelMismatch)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceRecordUsage_ResponseModelRejectsPricierResponseModel(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newOpenAIRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{}, nil)
|
|
tokens := UsageTokens{InputTokens: 20, OutputTokens: 10}
|
|
cheaper, pricier, cheaperCost, _ := orderedResponseBillingModels(t, svc.billingService, tokens, openAICheapFixtureModel, openAIPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &OpenAIRecordUsageInput{
|
|
Result: &OpenAIForwardResult{
|
|
RequestID: "openai_response_model_forged_upgrade",
|
|
Model: cheaper,
|
|
UpstreamModel: cheaper,
|
|
UpstreamResponseModel: pricier,
|
|
Usage: OpenAIUsage{InputTokens: 20, OutputTokens: 10},
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 10},
|
|
User: &User{ID: 20},
|
|
Account: &Account{ID: 30},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: cheaper,
|
|
ChannelMappedModel: cheaper,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, cheaperCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, cheaperCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceRecordUsage_ResponseModelSafeFallbacks(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
responseModel func(cheaper string) string
|
|
conflict bool
|
|
source string
|
|
}{
|
|
{
|
|
name: "in_stream_conflict_falls_back_to_baseline",
|
|
responseModel: func(cheaper string) string { return cheaper },
|
|
conflict: true,
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "empty_response_model_falls_back_to_baseline",
|
|
responseModel: func(string) string { return "" },
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "unpriced_response_model_falls_back_to_baseline",
|
|
responseModel: func(string) string { return "zz-unpriced-response-model" },
|
|
source: BillingModelSourceResponse,
|
|
},
|
|
{
|
|
name: "default_channel_mapped_mode_ignores_response_model",
|
|
responseModel: func(cheaper string) string { return cheaper },
|
|
source: BillingModelSourceChannelMapped,
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newOpenAIRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{}, nil)
|
|
tokens := UsageTokens{InputTokens: 20, OutputTokens: 10}
|
|
cheaper, pricier, _, pricierCost := orderedResponseBillingModels(t, svc.billingService, tokens, openAICheapFixtureModel, openAIPriceyFixtureModel)
|
|
|
|
err := svc.RecordUsage(context.Background(), &OpenAIRecordUsageInput{
|
|
Result: &OpenAIForwardResult{
|
|
RequestID: "openai_response_model_fallback_" + tt.name,
|
|
Model: pricier,
|
|
UpstreamModel: pricier,
|
|
UpstreamResponseModel: tt.responseModel(cheaper),
|
|
UpstreamResponseModelConflict: tt.conflict,
|
|
Usage: OpenAIUsage{InputTokens: 20, OutputTokens: 10},
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 10},
|
|
User: &User{ID: 20},
|
|
Account: &Account{ID: 30},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: tt.source,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, pricierCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, pricierCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
})
|
|
}
|
|
}
|
|
|
|
// --- 准入规则本身 ---
|
|
|
|
func TestResponseModelBillingDeclaration(t *testing.T) {
|
|
t.Parallel()
|
|
tests := []struct {
|
|
name string
|
|
source string
|
|
model string
|
|
conflict bool
|
|
mediaBilled bool
|
|
want string
|
|
}{
|
|
{name: "opted_in_and_clean", source: BillingModelSourceResponse, model: " claude-sonnet-4 ", want: "claude-sonnet-4"},
|
|
{name: "other_source_never_looks_at_response", source: BillingModelSourceChannelMapped, model: "claude-sonnet-4"},
|
|
{name: "empty_source_never_looks_at_response", source: "", model: "claude-sonnet-4"},
|
|
{name: "upstream_source_never_looks_at_response", source: BillingModelSourceUpstream, model: "claude-sonnet-4"},
|
|
{name: "in_stream_conflict_rejected", source: BillingModelSourceResponse, model: "claude-sonnet-4", conflict: true},
|
|
{name: "media_billed_request_rejected", source: BillingModelSourceResponse, model: "claude-sonnet-4", mediaBilled: true},
|
|
{name: "blank_declaration_rejected", source: BillingModelSourceResponse, model: " "},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, tt.want, responseModelBillingDeclaration(tt.source, tt.model, tt.conflict, tt.mediaBilled))
|
|
})
|
|
}
|
|
}
|
|
|
|
// 上游自报的模型名是外部输入。GetModelPricing 的系列兜底会给任意含 "haiku" 的名字
|
|
// 返回最便宜的系列价，因此计费准入必须走"确定性识别"，否则上游随手编一个名字就能
|
|
// 把账单压到地板价。本用例把这个差异钉死。
|
|
func TestBillingServiceHasIdentifiedTokenPricing_RejectsFamilyGuesses(t *testing.T) {
|
|
t.Parallel()
|
|
billing := newGatewayRecordUsageServiceForTest(
|
|
&openAIRecordUsageLogRepoStub{}, &openAIRecordUsageUserRepoStub{}, &openAIRecordUsageSubRepoStub{},
|
|
).billingService
|
|
|
|
require.True(t, billing.HasIdentifiedTokenPricing("claude-sonnet-4"))
|
|
require.True(t, billing.HasIdentifiedTokenPricing(" CLAUDE-SONNET-4 "), "识别应当忽略大小写与空白")
|
|
require.True(t, billing.HasIdentifiedTokenPricing("gpt-5.4-nano"))
|
|
|
|
const forged = "totally-made-up-haiku-v9"
|
|
if _, err := billing.GetModelPricing(forged); err == nil {
|
|
// 这正是本函数存在的理由：宽松查价对编造的名字也会成功。
|
|
require.False(t, billing.HasIdentifiedTokenPricing(forged),
|
|
"family-guessed pricing must not qualify a model as a billing basis")
|
|
}
|
|
require.False(t, billing.HasIdentifiedTokenPricing(""))
|
|
require.False(t, billing.HasIdentifiedTokenPricing("zz-unpriced-response-model"))
|
|
// Versioned media ids may inherit a text card via GetModelPricing; the
|
|
// identified-token gate must still reject them so response-model billing
|
|
// cannot adopt grok-4.5 rates for image/audio/video ids.
|
|
require.False(t, billing.HasIdentifiedTokenPricing("grok-2-image-1212"))
|
|
require.False(t, billing.HasIdentifiedTokenPricing("grok-2-audio"))
|
|
require.False(t, billing.HasIdentifiedTokenPricing("grok-5-video"))
|
|
}
|
|
|
|
func TestGatewayServiceRecordUsage_ResponseModelRejectsUnidentifiedFamilyName(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newGatewayRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{})
|
|
tokens := UsageTokens{InputTokens: 100, OutputTokens: 50}
|
|
const forged = "totally-made-up-haiku-v9"
|
|
|
|
baselineCost, err := svc.billingService.CalculateCost(anthropicPriceyFixtureModel, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
// 前提：这个编造的名字确实能被宽松查价算出更低的费用——正是必须被拒绝的那条路径。
|
|
forgedCost, err := svc.billingService.CalculateCost(forged, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
require.Less(t, forgedCost.TotalCost, baselineCost.TotalCost)
|
|
|
|
err = svc.RecordUsage(context.Background(), &RecordUsageInput{
|
|
Result: &ForwardResult{
|
|
RequestID: "gateway_response_model_forged_family_name",
|
|
Usage: ClaudeUsage{InputTokens: 100, OutputTokens: 50},
|
|
Model: anthropicPriceyFixtureModel,
|
|
UpstreamResponseModel: forged,
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 501, Quota: 100},
|
|
User: &User{ID: 601},
|
|
Account: &Account{ID: 701},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: anthropicPriceyFixtureModel,
|
|
ChannelMappedModel: anthropicPriceyFixtureModel,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, baselineCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, baselineCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceRecordUsage_ResponseModelRejectsUnidentifiedFamilyName(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newOpenAIRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{}, nil)
|
|
tokens := UsageTokens{InputTokens: 20, OutputTokens: 10}
|
|
const forged = "totally-made-up-haiku-v9"
|
|
|
|
baselineCost, err := svc.billingService.CalculateCost(openAIPriceyFixtureModel, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
forgedCost, err := svc.billingService.CalculateCost(forged, tokens, 1.1)
|
|
require.NoError(t, err)
|
|
require.Less(t, forgedCost.TotalCost, baselineCost.TotalCost)
|
|
|
|
err = svc.RecordUsage(context.Background(), &OpenAIRecordUsageInput{
|
|
Result: &OpenAIForwardResult{
|
|
RequestID: "openai_response_model_forged_family_name",
|
|
Model: openAIPriceyFixtureModel,
|
|
UpstreamModel: openAIPriceyFixtureModel,
|
|
UpstreamResponseModel: forged,
|
|
Usage: OpenAIUsage{InputTokens: 20, OutputTokens: 10},
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 10},
|
|
User: &User{ID: 20},
|
|
Account: &Account{ID: 30},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: openAIPriceyFixtureModel,
|
|
ChannelMappedModel: openAIPriceyFixtureModel,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
require.InDelta(t, baselineCost.ActualCost, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, baselineCost.ActualCost, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
// --- 成本准入的三条不变式 ---
|
|
|
|
func TestResponseModelBillingAdoptable(t *testing.T) {
|
|
t.Parallel()
|
|
cost := func(total float64) *CostBreakdown {
|
|
return &CostBreakdown{TotalCost: total, ActualCost: total}
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
baseline *CostBreakdown
|
|
response *CostBreakdown
|
|
baselineChannelPriced bool
|
|
responseChannelPriced bool
|
|
want bool
|
|
}{
|
|
// 1. 不得更贵
|
|
{name: "cheaper_adopted", baseline: cost(1), response: cost(0.5), want: true},
|
|
{name: "equal_adopted", baseline: cost(1), response: cost(1), want: true},
|
|
{name: "float_noise_within_epsilon_adopted", baseline: cost(1), response: cost(1 + 1e-13), want: true},
|
|
{name: "pricier_rejected", baseline: cost(1), response: cost(1.0001)},
|
|
|
|
// 2. 不得把一笔本应计费的请求归零（价格表里有显式写 0 的条目，能通过确定性识别）
|
|
{name: "zeroing_a_billable_request_rejected", baseline: cost(1), response: cost(0)},
|
|
{name: "negative_cost_rejected_as_zeroing", baseline: cost(1), response: cost(-1)},
|
|
{name: "already_zero_baseline_unaffected", baseline: cost(0), response: cost(0), want: true},
|
|
|
|
// 3. 不得从渠道定价跨到全局价格表（否则渠道加价被带日期的自报模型名绕过）
|
|
{name: "channel_priced_baseline_to_global_rejected", baseline: cost(1), response: cost(0.5), baselineChannelPriced: true},
|
|
{name: "channel_priced_on_both_sides_adopted", baseline: cost(1), response: cost(0.5), baselineChannelPriced: true, responseChannelPriced: true, want: true},
|
|
{name: "global_baseline_to_channel_priced_adopted", baseline: cost(1), response: cost(0.5), responseChannelPriced: true, want: true},
|
|
|
|
{name: "nil_baseline_rejected", response: cost(0.5)},
|
|
{name: "nil_response_rejected", baseline: cost(1)},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
require.Equal(t, tt.want, responseModelBillingAdoptable(
|
|
tt.baseline, tt.response, tt.baselineChannelPriced, tt.responseChannelPriced,
|
|
))
|
|
})
|
|
}
|
|
}
|
|
|
|
// --- 按次/按量计费请求一律不采纳（门的调用点接线） ---
|
|
//
|
|
// 搜索附加费是叠加在 token 成本之上的，所以"采纳与否"会体现在最终金额上，本用例因此
|
|
// 能真正区分两条分支。语音（AudioUsage）走的是与模型无关的按量单价，采纳与否金额相同，
|
|
// 无法用金额断言区分，故只由 TestResponseModelBillingDeclaration 覆盖门本身。
|
|
|
|
func TestGatewayServiceRecordUsage_ResponseModelSkippedForSearchSurchargedRequest(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newGatewayRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{})
|
|
tokens := UsageTokens{InputTokens: 100, OutputTokens: 50}
|
|
cheaper, pricier, _, pricierCost := orderedResponseBillingModels(t, svc.billingService, tokens, anthropicCheapFixtureModel, anthropicPriceyFixtureModel)
|
|
|
|
const searchCalls = 2
|
|
searchCost := svc.billingService.CalculateSearchCost(searchCalls, nil, 1.1)
|
|
require.NotNil(t, searchCost)
|
|
require.Greater(t, searchCost.ActualCost, 0.0, "夹具附加费必须非零，否则断言分不出两条分支")
|
|
|
|
err := svc.RecordUsage(context.Background(), &RecordUsageInput{
|
|
Result: &ForwardResult{
|
|
RequestID: "gateway_response_model_search_surcharge",
|
|
Usage: ClaudeUsage{InputTokens: 100, OutputTokens: 50},
|
|
Model: pricier,
|
|
UpstreamResponseModel: cheaper,
|
|
SearchCount: searchCalls,
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 501, Quota: 100},
|
|
User: &User{ID: 601},
|
|
Account: &Account{ID: 701},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
want := pricierCost.ActualCost + searchCost.ActualCost
|
|
require.InDelta(t, want, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, want, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
func TestOpenAIGatewayServiceRecordUsage_ResponseModelSkippedForSearchSurchargedRequest(t *testing.T) {
|
|
usageRepo := &openAIRecordUsageLogRepoStub{inserted: true}
|
|
userRepo := &openAIRecordUsageUserRepoStub{}
|
|
svc := newOpenAIRecordUsageServiceForTest(usageRepo, userRepo, &openAIRecordUsageSubRepoStub{}, nil)
|
|
tokens := UsageTokens{InputTokens: 20, OutputTokens: 10}
|
|
cheaper, pricier, _, pricierCost := orderedResponseBillingModels(t, svc.billingService, tokens, openAICheapFixtureModel, openAIPriceyFixtureModel)
|
|
|
|
const searchCalls = 3
|
|
searchCost := svc.billingService.CalculateSearchCost(searchCalls, nil, 1.1)
|
|
require.NotNil(t, searchCost)
|
|
require.Greater(t, searchCost.ActualCost, 0.0, "夹具附加费必须非零，否则断言分不出两条分支")
|
|
|
|
err := svc.RecordUsage(context.Background(), &OpenAIRecordUsageInput{
|
|
Result: &OpenAIForwardResult{
|
|
RequestID: "openai_response_model_search_surcharge",
|
|
Model: pricier,
|
|
UpstreamModel: pricier,
|
|
UpstreamResponseModel: cheaper,
|
|
SearchCount: searchCalls,
|
|
Usage: OpenAIUsage{InputTokens: 20, OutputTokens: 10},
|
|
Duration: time.Second,
|
|
},
|
|
APIKey: &APIKey{ID: 10},
|
|
User: &User{ID: 20},
|
|
Account: &Account{ID: 30},
|
|
ChannelUsageFields: ChannelUsageFields{
|
|
ChannelID: 9,
|
|
OriginalModel: pricier,
|
|
ChannelMappedModel: pricier,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
},
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
require.NotNil(t, usageRepo.lastLog)
|
|
want := pricierCost.ActualCost + searchCost.ActualCost
|
|
require.InDelta(t, want, usageRepo.lastLog.ActualCost, 1e-12)
|
|
require.InDelta(t, want, userRepo.lastAmount, 1e-12)
|
|
}
|
|
|
|
// --- 渠道配置透传 ---
|
|
|
|
func TestToUsageFields_ResponseModelSourcePassesThrough(t *testing.T) {
|
|
r := ChannelMappingResult{
|
|
MappedModel: "claude-fable-5",
|
|
ChannelID: 4,
|
|
Mapped: false,
|
|
BillingModelSource: BillingModelSourceResponse,
|
|
}
|
|
fields := r.ToUsageFields("claude-fable-5", "claude-fable-5")
|
|
require.Equal(t, int64(4), fields.ChannelID)
|
|
require.Equal(t, BillingModelSourceResponse, fields.BillingModelSource)
|
|
}
|