9da079a5ee
Some checks failed
Security Scan / backend-security (push) Failing after 3s
Security Scan / frontend-security (push) Failing after 5s
CI / test (push) Failing after 3s
CI / frontend (push) Failing after 3s
CI / golangci-lint (push) Failing after 3s
CI / windsurf-platform (macos-latest) (push) Has been cancelled
CI / windsurf-platform (windows-latest) (push) Has been cancelled
110 lines
3.2 KiB
Go
110 lines
3.2 KiB
Go
package service
|
||
|
||
import (
|
||
"testing"
|
||
"time"
|
||
|
||
"github.com/Wei-Shaw/sub2api/internal/pkg/antigravity"
|
||
)
|
||
|
||
func TestAccountHasUsableCredits(t *testing.T) {
|
||
cases := []struct {
|
||
name string
|
||
acct *Account
|
||
want bool
|
||
}{
|
||
{name: "nil 账号_false", acct: nil, want: false},
|
||
{name: "Extra 为空_false（保守策略）", acct: &Account{}, want: false},
|
||
{
|
||
name: "余额 0_false",
|
||
acct: &Account{Extra: map[string]any{
|
||
extraKeyCreditsBalance: 0.0,
|
||
extraKeyCreditsCheckedAt: time.Now().UTC().Format(time.RFC3339),
|
||
}},
|
||
want: false,
|
||
},
|
||
{
|
||
name: "余额 > 0_无耗尽标记_true",
|
||
acct: &Account{Extra: map[string]any{
|
||
extraKeyCreditsBalance: 1.5,
|
||
extraKeyCreditsCheckedAt: time.Now().UTC().Format(time.RFC3339),
|
||
}},
|
||
want: true,
|
||
},
|
||
{
|
||
name: "余额 > 0_刚耗尽_false",
|
||
acct: &Account{Extra: map[string]any{
|
||
extraKeyCreditsBalance: 5.0,
|
||
extraKeyCreditsCheckedAt: time.Now().UTC().Format(time.RFC3339),
|
||
extraKeyCreditsExhaustedAt: time.Now().UTC().Format(time.RFC3339),
|
||
}},
|
||
want: false,
|
||
},
|
||
{
|
||
name: "余额 > 0_耗尽标记已过期_true",
|
||
acct: &Account{Extra: map[string]any{
|
||
extraKeyCreditsBalance: 5.0,
|
||
extraKeyCreditsCheckedAt: time.Now().UTC().Format(time.RFC3339),
|
||
extraKeyCreditsExhaustedAt: time.Now().Add(-2 * time.Hour).UTC().Format(time.RFC3339),
|
||
}},
|
||
want: true,
|
||
},
|
||
{
|
||
name: "余额来自字符串_仍可识别",
|
||
acct: &Account{Extra: map[string]any{
|
||
extraKeyCreditsBalance: "10.5",
|
||
}},
|
||
want: true,
|
||
},
|
||
}
|
||
|
||
for _, tc := range cases {
|
||
t.Run(tc.name, func(t *testing.T) {
|
||
if got := AccountHasUsableCredits(tc.acct); got != tc.want {
|
||
t.Errorf("AccountHasUsableCredits = %v, want %v", got, tc.want)
|
||
}
|
||
})
|
||
}
|
||
}
|
||
|
||
func TestRefreshAccountCreditsFromLoadCodeAssist_累加_GOOGLE_ONE_AI(t *testing.T) {
|
||
acct := &Account{Extra: map[string]any{
|
||
// 设置一个旧的耗尽标记，验证刷新后会被清除
|
||
extraKeyCreditsExhaustedAt: time.Now().Add(-1 * time.Minute).UTC().Format(time.RFC3339),
|
||
}}
|
||
|
||
resp := &antigravity.LoadCodeAssistResponse{
|
||
PaidTier: &antigravity.PaidTierInfo{
|
||
AvailableCredits: []antigravity.AvailableCredit{
|
||
{CreditType: antigravity.CreditTypeGoogleOneAI, CreditAmount: "8.5"},
|
||
{CreditType: "OTHER_TYPE", CreditAmount: "100.0"}, // 应被忽略
|
||
{CreditType: "", CreditAmount: "1.5"}, // 空 type 视为 GOOGLE_ONE_AI 兼容
|
||
},
|
||
},
|
||
}
|
||
|
||
refreshAccountCreditsFromLoadCodeAssist(acct, resp)
|
||
|
||
if got := readFloat(acct.Extra[extraKeyCreditsBalance]); got != 10.0 {
|
||
t.Errorf("余额累加错误: got %v, want 10.0", got)
|
||
}
|
||
if _, present := acct.Extra[extraKeyCreditsExhaustedAt]; present {
|
||
t.Errorf("刷新后耗尽标记应被清除")
|
||
}
|
||
if !AccountHasUsableCredits(acct) {
|
||
t.Errorf("刷新后账号应可用 credits")
|
||
}
|
||
}
|
||
|
||
func TestRefreshAccountCreditsFromLoadCodeAssist_无_paidTier_余额_0(t *testing.T) {
|
||
acct := &Account{}
|
||
refreshAccountCreditsFromLoadCodeAssist(acct, &antigravity.LoadCodeAssistResponse{})
|
||
|
||
if got := readFloat(acct.Extra[extraKeyCreditsBalance]); got != 0 {
|
||
t.Errorf("无 paidTier 时余额应为 0: got %v", got)
|
||
}
|
||
if AccountHasUsableCredits(acct) {
|
||
t.Errorf("零余额账号不应被视为可用")
|
||
}
|
||
}
|