975e6b1563
- gofmt: 修复 admin_service/antigravity_oauth_service/token_refresh_service 格式 - staticcheck S1009: 移除 SetUserSettingsResponse.IsSuccess 中冗余的 nil 检查 - unused: 将仅测试使用的 applyAntigravitySubscriptionResult 移至测试文件 Made-with: Cursor
68 lines
1.8 KiB
Go
68 lines
1.8 KiB
Go
//go:build unit
|
|
|
|
package service
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func applyAntigravitySubscriptionResult(account *Account, result AntigravitySubscriptionResult) (map[string]any, map[string]any) {
|
|
credentials := make(map[string]any)
|
|
for k, v := range account.Credentials {
|
|
credentials[k] = v
|
|
}
|
|
credentials["plan_type"] = result.PlanType
|
|
|
|
extra := make(map[string]any)
|
|
for k, v := range account.Extra {
|
|
extra[k] = v
|
|
}
|
|
if result.SubscriptionStatus != "" {
|
|
extra["subscription_status"] = result.SubscriptionStatus
|
|
} else {
|
|
delete(extra, "subscription_status")
|
|
}
|
|
if result.SubscriptionError != "" {
|
|
extra["subscription_error"] = result.SubscriptionError
|
|
} else {
|
|
delete(extra, "subscription_error")
|
|
}
|
|
return credentials, extra
|
|
}
|
|
|
|
func TestApplyAntigravityPrivacyMode_SetsInMemoryExtra(t *testing.T) {
|
|
account := &Account{}
|
|
|
|
applyAntigravityPrivacyMode(account, AntigravityPrivacySet)
|
|
|
|
if account.Extra == nil {
|
|
t.Fatal("expected account.Extra to be initialized")
|
|
}
|
|
if got := account.Extra["privacy_mode"]; got != AntigravityPrivacySet {
|
|
t.Fatalf("expected privacy_mode %q, got %v", AntigravityPrivacySet, got)
|
|
}
|
|
}
|
|
|
|
func TestApplyAntigravityPrivacyMode_PreservedBySubscriptionResult(t *testing.T) {
|
|
account := &Account{
|
|
Credentials: map[string]any{
|
|
"access_token": "token",
|
|
},
|
|
Extra: map[string]any{
|
|
"existing": "value",
|
|
},
|
|
}
|
|
applyAntigravityPrivacyMode(account, AntigravityPrivacySet)
|
|
|
|
_, extra := applyAntigravitySubscriptionResult(account, AntigravitySubscriptionResult{
|
|
PlanType: "Pro",
|
|
})
|
|
|
|
if got := extra["privacy_mode"]; got != AntigravityPrivacySet {
|
|
t.Fatalf("expected subscription writeback to keep privacy_mode %q, got %v", AntigravityPrivacySet, got)
|
|
}
|
|
if got := extra["existing"]; got != "value" {
|
|
t.Fatalf("expected existing extra fields to be preserved, got %v", got)
|
|
}
|
|
}
|