Files
hi-server/internal/model/lottery/registry_test.go
T
shanshanzhong147 a46fb83054 新功能(#3): 抽奖 Stage 1 handler 真实业务对接 + 邀请钩子
Closes HIF-3 (阶段 PR B)

PR B:handler 真实业务对接 + 邀请钩子(迭代含 R1 修复)

- 迁移 02157_lottery_grant_ledger:external_ref UNIQUE 作为发奖幂等键
- log.CommissionTypeLottery=339(架构师批准的新常量)
- DispatchRequest.IdempotencyKey(架构师 review 建议第 2 条)
- GrantLedger + LedgerService.Reserve:INSERT ON CONFLICT DO NOTHING 幂等 upsert
- VPNDurationHandler:ResolveEffectiveUser 归位家庭 owner + UpdateSubscribe,ExpireTime 三分支对齐 grantGiftDays
- CommissionHandler:UpdateCommission + WriteCommissionLog(339),发给中奖者本人,不做家庭组归位
- Handler 从 model 层迁到 logic 层(避免 model → logic 反向依赖);noop 保留在 model 层
- InviteHook:fire-and-forget 独立 goroutine + 10s timeout,扫 running 活动的 invite_success 源
- ServiceContext 新增 LotteryChance / LotteryLedger / LotteryInviteHook
- activateOrderLogic.handleCommission 两条分支通过 invokeInviteHookIfEligible 助手触发,助手内统一 gate IsNew(架构师 R1 打回后的修复)

架构师 R1 打回:branch B 未按"首次付款激活"gate,续费也会给 referer 发抽奖机会 → 已通过助手函数集中收拢,避免 branch A/B 判断漂移。

测试覆盖率:model 76.5% / handler 73.2% / hook 91.7%;补 4 个回归测试覆盖 IsNew 门槛(含关键的 DoesNotFireOnRenewal)。

线上仍零可见变更:无对外路由,钩子仅在活动 status=running 时生效,Stage 1 全流程无活动记录时 loadRunningActivities 返回空。
2026-07-08 21:19:28 -07:00

47 lines
1.1 KiB
Go

package lottery
import (
"context"
"errors"
"testing"
"gorm.io/gorm"
)
func TestRegistry_GetMissing(t *testing.T) {
r := NewRegistry()
if _, ok := r.Get("nope"); ok {
t.Fatalf("empty registry should not return handler")
}
if _, err := r.MustGet("nope"); !errors.Is(err, ErrHandlerNotRegistered) {
t.Fatalf("MustGet on missing type should return ErrHandlerNotRegistered, got %v", err)
}
}
func TestRegistry_RegisterAndGet(t *testing.T) {
r := NewRegistry()
r.Register(NewNoopHandler())
h, err := r.MustGet(PrizeTypeNone)
if err != nil {
t.Fatalf("MustGet(%q): %v", PrizeTypeNone, err)
}
if h.Type() != PrizeTypeNone {
t.Fatalf("Type mismatch: want %q got %q", PrizeTypeNone, h.Type())
}
}
func TestNoopHandler_AlwaysAutoClaimed(t *testing.T) {
h := NewNoopHandler()
if !h.IsAuto() {
t.Fatal("noop must be auto")
}
res, err := h.Dispatch(context.Background(), (*gorm.DB)(nil), DispatchRequest{})
if err != nil {
t.Fatalf("noop dispatch err: %v", err)
}
if res.State != DispatchStateAutoClaimed {
t.Fatalf("noop should dispatch as auto_claimed, got %q", res.State)
}
}