李建琦
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
43 lines
1.5 KiB
Go
43 lines
1.5 KiB
Go
package admin
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/Wei-Shaw/sub2api/internal/handler/dto"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestEnrichShadowParentInfo(t *testing.T) {
|
|
pid := int64(100)
|
|
parent := &service.Account{
|
|
ID: 100,
|
|
Credentials: map[string]any{
|
|
"email": "owner@example.com",
|
|
"plan_type": "pro",
|
|
"subscription_expires_at": "2026-12-31T00:00:00Z",
|
|
"chatgpt_account_id": "acct_123",
|
|
},
|
|
Extra: map[string]any{"privacy_mode": "training_off"},
|
|
}
|
|
parents := map[int64]*service.Account{100: parent}
|
|
|
|
shadow := AccountWithConcurrency{Account: &dto.Account{ID: 200, ParentAccountID: &pid}}
|
|
normal := AccountWithConcurrency{Account: &dto.Account{ID: 1}}
|
|
orphan := AccountWithConcurrency{Account: &dto.Account{ID: 201, ParentAccountID: ptrInt64(999)}}
|
|
items := []AccountWithConcurrency{shadow, normal, orphan}
|
|
|
|
enrichShadowParentInfo(items, parents)
|
|
|
|
require.Equal(t, "owner@example.com", items[0].ParentEmail, "影子回填母账号邮箱")
|
|
require.Equal(t, "pro", items[0].ParentPlanType)
|
|
require.Equal(t, "training_off", items[0].ParentPrivacyMode)
|
|
require.Equal(t, "2026-12-31T00:00:00Z", items[0].ParentSubscriptionExpiresAt)
|
|
require.Equal(t, "acct_123", items[0].ParentChatGPTAccountID)
|
|
|
|
require.Empty(t, items[1].ParentEmail, "非影子不回填")
|
|
require.Empty(t, items[2].ParentEmail, "母账号缺失时优雅留空")
|
|
}
|
|
|
|
func ptrInt64(v int64) *int64 { return &v }
|