mirror of
https://gitee.com/wanwujie/sub2api
synced 2026-04-16 12:54:45 +08:00
a1e2ffd586
- Replace heavy RefreshAccountToken with lightweight tryFillProjectID (loadCodeAssist → onboardUser → fallback), consistent with Antigravity-Manager's behavior - Add sync.Map cooldown/dedup (60s) to prevent repeated fill attempts - Add fallback project_id "bamboo-precept-lgxtn" matching AM - Extract mergeCredentials helper to eliminate duplication - Use slog structured logging instead of log.Printf - Fix time.Sleep in OnboardUser to context-aware select - Fix strings.NewReader(string(bodyBytes)) → bytes.NewReader(bodyBytes) - Remove redundant tc := tc in test (Go 1.22+) - Add nil guard in persistProjectID for test safety
64 lines
1.2 KiB
Go
64 lines
1.2 KiB
Go
package service
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestResolveDefaultTierID(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
tests := []struct {
|
|
name string
|
|
loadRaw map[string]any
|
|
want string
|
|
}{
|
|
{
|
|
name: "missing allowedTiers",
|
|
loadRaw: map[string]any{
|
|
"paidTier": map[string]any{"id": "g1-pro-tier"},
|
|
},
|
|
want: "",
|
|
},
|
|
{
|
|
name: "allowedTiers but no default",
|
|
loadRaw: map[string]any{
|
|
"allowedTiers": []any{
|
|
map[string]any{"id": "free-tier", "isDefault": false},
|
|
map[string]any{"id": "standard-tier", "isDefault": false},
|
|
},
|
|
},
|
|
want: "",
|
|
},
|
|
{
|
|
name: "default tier found",
|
|
loadRaw: map[string]any{
|
|
"allowedTiers": []any{
|
|
map[string]any{"id": "free-tier", "isDefault": true},
|
|
map[string]any{"id": "standard-tier", "isDefault": false},
|
|
},
|
|
},
|
|
want: "free-tier",
|
|
},
|
|
{
|
|
name: "default tier id with spaces",
|
|
loadRaw: map[string]any{
|
|
"allowedTiers": []any{
|
|
map[string]any{"id": " standard-tier ", "isDefault": true},
|
|
},
|
|
},
|
|
want: "standard-tier",
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
got := resolveDefaultTierID(tc.loadRaw)
|
|
if got != tc.want {
|
|
t.Fatalf("resolveDefaultTierID() = %q, want %q", got, tc.want)
|
|
}
|
|
})
|
|
}
|
|
}
|