71f61bbc47
P0-1: Credits degraded response retry + fail-open - Add isAntigravityDegradedResponse() to detect transient API failures - Retry up to 3 times with exponential backoff (500ms/1s/2s) - Invalidate singleflight cache between retries - Fail-open after exhausting retries instead of 5h circuit break P1-1: Fix channel restriction pre-check timing conflict - Swap checkClaudeCodeRestriction before checkChannelPricingRestriction - Ensures channel restriction is checked against final fallback groupID P1-2: Add interval pricing validation (frontend + backend) - Backend: ValidateIntervals() with boundary, price, overlap checks - Frontend: validateIntervals() with Chinese error messages - Rules: MinTokens>=0, MaxTokens>MinTokens, prices>=0, no overlap P2: Fix cross-platform same-model pricing/mapping override - Store cache keys using original platform instead of group platform - Lookup across matching platforms (antigravity→anthropic→gemini) - Prevents anthropic/gemini same-name models from overwriting each other
131 lines
3.5 KiB
Go
131 lines
3.5 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/pkg/ctxkey"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestSelectAccountForModelWithExclusions_UsesFallbackGroupForChannelRestriction(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
groupID := int64(10)
|
|
fallbackID := int64(11)
|
|
ch := Channel{
|
|
ID: 1,
|
|
Status: StatusActive,
|
|
GroupIDs: []int64{fallbackID},
|
|
RestrictModels: true,
|
|
ModelPricing: []ChannelModelPricing{
|
|
{Platform: PlatformAnthropic, Models: []string{"claude-sonnet-4-6"}},
|
|
},
|
|
}
|
|
channelSvc := newTestChannelService(makeStandardRepo(ch, map[int64]string{
|
|
fallbackID: PlatformAnthropic,
|
|
}))
|
|
accountRepo := &mockAccountRepoForPlatform{
|
|
accounts: []Account{
|
|
{ID: 1, Platform: PlatformAnthropic, Priority: 1, Status: StatusActive, Schedulable: true},
|
|
},
|
|
accountsByID: map[int64]*Account{},
|
|
}
|
|
for i := range accountRepo.accounts {
|
|
accountRepo.accountsByID[accountRepo.accounts[i].ID] = &accountRepo.accounts[i]
|
|
}
|
|
groupRepo := &mockGroupRepoForGateway{
|
|
groups: map[int64]*Group{
|
|
groupID: {
|
|
ID: groupID,
|
|
Platform: PlatformAnthropic,
|
|
Status: StatusActive,
|
|
ClaudeCodeOnly: true,
|
|
FallbackGroupID: &fallbackID,
|
|
Hydrated: true,
|
|
},
|
|
fallbackID: {
|
|
ID: fallbackID,
|
|
Platform: PlatformAnthropic,
|
|
Status: StatusActive,
|
|
Hydrated: true,
|
|
},
|
|
},
|
|
}
|
|
|
|
svc := &GatewayService{
|
|
accountRepo: accountRepo,
|
|
groupRepo: groupRepo,
|
|
channelService: channelSvc,
|
|
cfg: testConfig(),
|
|
}
|
|
|
|
ctx := context.WithValue(context.Background(), ctxkey.Group, groupRepo.groups[groupID])
|
|
account, err := svc.SelectAccountForModelWithExclusions(ctx, &groupID, "", "claude-sonnet-4-6", nil)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, account)
|
|
require.Equal(t, int64(1), account.ID)
|
|
}
|
|
|
|
func TestSelectAccountWithLoadAwareness_UsesFallbackGroupForChannelRestriction(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
groupID := int64(10)
|
|
fallbackID := int64(11)
|
|
ch := Channel{
|
|
ID: 1,
|
|
Status: StatusActive,
|
|
GroupIDs: []int64{fallbackID},
|
|
RestrictModels: true,
|
|
ModelPricing: []ChannelModelPricing{
|
|
{Platform: PlatformAnthropic, Models: []string{"claude-sonnet-4-6"}},
|
|
},
|
|
}
|
|
channelSvc := newTestChannelService(makeStandardRepo(ch, map[int64]string{
|
|
fallbackID: PlatformAnthropic,
|
|
}))
|
|
accountRepo := &mockAccountRepoForPlatform{
|
|
accounts: []Account{
|
|
{ID: 1, Platform: PlatformAnthropic, Priority: 1, Status: StatusActive, Schedulable: true},
|
|
},
|
|
accountsByID: map[int64]*Account{},
|
|
}
|
|
for i := range accountRepo.accounts {
|
|
accountRepo.accountsByID[accountRepo.accounts[i].ID] = &accountRepo.accounts[i]
|
|
}
|
|
groupRepo := &mockGroupRepoForGateway{
|
|
groups: map[int64]*Group{
|
|
groupID: {
|
|
ID: groupID,
|
|
Platform: PlatformAnthropic,
|
|
Status: StatusActive,
|
|
ClaudeCodeOnly: true,
|
|
FallbackGroupID: &fallbackID,
|
|
Hydrated: true,
|
|
},
|
|
fallbackID: {
|
|
ID: fallbackID,
|
|
Platform: PlatformAnthropic,
|
|
Status: StatusActive,
|
|
Hydrated: true,
|
|
},
|
|
},
|
|
}
|
|
|
|
svc := &GatewayService{
|
|
accountRepo: accountRepo,
|
|
groupRepo: groupRepo,
|
|
channelService: channelSvc,
|
|
cfg: testConfig(),
|
|
}
|
|
|
|
ctx := context.WithValue(context.Background(), ctxkey.Group, groupRepo.groups[groupID])
|
|
result, err := svc.SelectAccountWithLoadAwareness(ctx, &groupID, "", "claude-sonnet-4-6", nil, "", 0)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, result)
|
|
require.NotNil(t, result.Account)
|
|
require.Equal(t, int64(1), result.Account.ID)
|
|
}
|