Blizzard
bd829cfecb
owner/admin 在桌面端生成一张微信带参二维码发给团队，成员用微信扫码关注 即自动加入租户，并收到「✅ 已加入团队【X】」被动回复。与登录二维码同一微信机制， scene 加 inv_ 前缀分流；扫码入组走被动回复，不需要 access_token、不碰 IP 白名单。 - store: TenantInvite(可复用码=有效期+人数上限+可撤销三道闸)；RedeemInvite 幂等入组、 同一人重复扫不重复消耗名额、复活已移除者；带单测钉死过期/撤销/满员/幂等。 - handler: 建码/列表/撤销端点(RequireTenantRole admin)；WxMPEvent inv_ 分支。 - 顺手修潜在生产 bug：微信用户此前都建成空邮箱，User.Email 整列唯一索引下第二个微信 用户就撞唯一约束建号失败(登录"全通"只因当前仅一个微信用户)。改为按 openid 合成占位 邮箱 wx-<openid>@wx.local，绕开冲突且不动索引。邀请功能会批量建微信用户，非修不可。 - desktop: 顶栏「邀请成员」入口(租户 admin/owner 可见) + 二维码弹窗(生成/大图展示/撤销)。 Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
75 lines
3.4 KiB
Go
75 lines
3.4 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/glebarez/sqlite" // 纯 Go sqlite（无 CGO）：DB 背书的单测在 CI ubuntu 无 Postgres 服务时也能跑
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
"gorm.io/gorm/schema"
|
|
)
|
|
|
|
// newTestStore 起一个内存 sqlite，迁移同款模型 + 建那道支付幂等兜底的部分唯一索引 +
|
|
// 挂租户作用域回调，尽量贴近生产 Postgres 的行为（核心事务/CAS/OnConflict 在两者一致）。
|
|
// 返回的 *Postgres 直接复用生产的 store 方法——测的是真逻辑，不是替身。
|
|
func newTestStore(t *testing.T) *Postgres {
|
|
t.Helper()
|
|
// 静音 gorm 日志：计费路径故意查 pricing/setting 取不到时回退默认，属预期空查询，别刷屏。
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{
|
|
Logger: logger.Default.LogMode(logger.Silent),
|
|
// 必须与 OpenPostgres 用同一套命名策略：多数模型有显式 TableName() 碰巧对得上，
|
|
// 但 Task 这类没有的会退化成 "tasks"，于是写裸 SQL 的查询在测试里查无此表。
|
|
NamingStrategy: schema.NamingStrategy{TablePrefix: "sundynix_", SingularTable: true},
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("打开内存 sqlite 失败: %v", err)
|
|
}
|
|
sqlDB, err := db.DB()
|
|
if err != nil {
|
|
t.Fatalf("取 *sql.DB 失败: %v", err)
|
|
}
|
|
sqlDB.SetMaxOpenConns(1) // :memory: 每连接一个库，锁死单连接才共享同一份数据
|
|
|
|
if err := db.AutoMigrate(
|
|
&User{}, &Tenant{}, &TenantMember{}, &TenantInvite{}, &CreditLedger{}, &PaymentOrder{},
|
|
&RedeemCode{}, &CreditPack{}, &UsageEvent{}, &UsageRollup{}, &Setting{}, &Pricing{}, &LLMModel{},
|
|
&AuditLog{}, &Task{}, &Eval{}, &SubscriptionPlan{}, &Subscription{},
|
|
&KB{}, // 租户作用域模型，验证隔离插件
|
|
); err != nil {
|
|
t.Fatalf("AutoMigrate 失败: %v", err)
|
|
}
|
|
// 支付入账幂等兜底闸：与 pgsql.go 生产建的同一道部分唯一索引（sqlite 同样支持）。
|
|
if err := db.Exec(`CREATE UNIQUE INDEX idx_ledger_grant_ref ON sundynix_credit_ledger (kind, ref) WHERE kind = 'grant' AND ref <> ''`).Error; err != nil {
|
|
t.Fatalf("建幂等索引失败: %v", err)
|
|
}
|
|
if err := db.Exec(`CREATE UNIQUE INDEX idx_ledger_refund_ref ON sundynix_credit_ledger (kind, ref) WHERE kind = 'adjust' AND ref <> ''`).Error; err != nil {
|
|
t.Fatalf("建退款幂等索引失败: %v", err)
|
|
}
|
|
registerTenantScope(db)
|
|
return &Postgres{db: db}
|
|
}
|
|
|
|
// seedTenant 建一个租户行（GrantCredits/MarkOrderPaid 靠 UpdateColumn 更新它的物化余额，
|
|
// 无租户行则余额更新落空 → 破坏「余额 == SUM(ledger)」不变量，故测试必须先建）。
|
|
func seedTenant(t *testing.T, p *Postgres, id string) {
|
|
t.Helper()
|
|
if err := p.db.Create(&Tenant{BaseModel: BaseModel{ID: id}, Name: "T-" + id, Slug: "slug-" + id, Status: "active"}).Error; err != nil {
|
|
t.Fatalf("建租户失败: %v", err)
|
|
}
|
|
}
|
|
|
|
// balanceEqualsSumLedger 是计费系统的核心不变量：物化余额 == 账本分录之和。
|
|
// 任何入账/扣费路径违反它都是对账事故。
|
|
func assertBalanceInvariant(t *testing.T, p *Postgres, tenantID string) {
|
|
t.Helper()
|
|
ctx := WithoutTenant(context.Background())
|
|
var sum int64
|
|
p.db.WithContext(ctx).Model(&CreditLedger{}).Where("tenant_id = ?", tenantID).
|
|
Select("coalesce(sum(credits_micro),0)").Scan(&sum)
|
|
bal := p.TenantBalance(ctx, tenantID)
|
|
if bal != sum {
|
|
t.Fatalf("不变量破坏：物化余额=%d 但 SUM(ledger)=%d", bal, sum)
|
|
}
|
|
}
|