108 lines
4.0 KiB
Go
108 lines
4.0 KiB
Go
|
package logic
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"testing"
|
|||
|
|
"time"
|
|||
|
|
|
|||
|
|
"github.com/gogf/gf/v2/os/gctx"
|
|||
|
|
"github.com/gogf/gf/v2/os/gtime"
|
|||
|
|
|
|||
|
|
"tool-api/internal/consts"
|
|||
|
|
"tool-api/internal/model/entity"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// TestEvaluateNoteQuota 覆盖额度边界：
|
|||
|
|
// - 免费未满(49) → 允许新建
|
|||
|
|
// - 免费刚好满(50) → 禁止新建（对应「第 51 条返回 4003」的前置条件）
|
|||
|
|
// - 会员过期回落且活跃数超限(51) → 仅禁止新建，不删数据
|
|||
|
|
// - 会员(不限量) → 允许新建
|
|||
|
|
//
|
|||
|
|
// 该函数是 SaveNote/RestoreNote 真实调用的额度判定，非复述逻辑。
|
|||
|
|
func TestEvaluateNoteQuota(t *testing.T) {
|
|||
|
|
cases := []struct {
|
|||
|
|
name string
|
|||
|
|
activeCount int
|
|||
|
|
isMember bool
|
|||
|
|
wantLimit int
|
|||
|
|
wantUnlimit bool
|
|||
|
|
wantCreate bool
|
|||
|
|
}{
|
|||
|
|
{"免费_未满", 49, false, consts.NoteFreeLimit, false, true},
|
|||
|
|
{"免费_刚好满", 50, false, consts.NoteFreeLimit, false, false},
|
|||
|
|
{"会员过期回落_超限仅禁新建", 51, false, consts.NoteFreeLimit, false, false},
|
|||
|
|
{"会员_超限仍不限", 51, true, 0, true, true},
|
|||
|
|
{"会员_零条", 0, true, 0, true, true},
|
|||
|
|
}
|
|||
|
|
for _, c := range cases {
|
|||
|
|
limit, unlimited, canCreate := evaluateNoteQuota(c.activeCount, c.isMember)
|
|||
|
|
if limit != c.wantLimit || unlimited != c.wantUnlimit || canCreate != c.wantCreate {
|
|||
|
|
t.Fatalf("%s: got(limit=%d, unlimited=%v, canCreate=%v) want(%d, %v, %v)",
|
|||
|
|
c.name, limit, unlimited, canCreate, c.wantLimit, c.wantUnlimit, c.wantCreate)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestEffectiveLevelKeyMemberFallback 证明会员判定确实走 EffectiveLevelKey，
|
|||
|
|
// 且会员到期后回落默认等级（保证「会员过期即回落免费额度」）。
|
|||
|
|
func TestEffectiveLevelKeyMemberFallback(t *testing.T) {
|
|||
|
|
if got := EffectiveLevelKey(nil); got != consts.DefaultLevelKey {
|
|||
|
|
t.Fatalf("nil user: got %s want %s", got, consts.DefaultLevelKey)
|
|||
|
|
}
|
|||
|
|
// 永久会员（level_expire_at 为空）
|
|||
|
|
if got := EffectiveLevelKey(&entity.Users{LevelKey: "v2"}); got != "v2" {
|
|||
|
|
t.Fatalf("永久会员: got %s want v2", got)
|
|||
|
|
}
|
|||
|
|
// 未过期会员
|
|||
|
|
if got := EffectiveLevelKey(&entity.Users{LevelKey: "v2", LevelExpireAt: gtime.Now().Add(time.Hour)}); got != "v2" {
|
|||
|
|
t.Fatalf("未过期会员: got %s want v2", got)
|
|||
|
|
}
|
|||
|
|
// 已过期会员 → 回落默认（免费）等级
|
|||
|
|
if got := EffectiveLevelKey(&entity.Users{LevelKey: "v2", LevelExpireAt: gtime.Now().Add(-time.Hour)}); got != consts.DefaultLevelKey {
|
|||
|
|
t.Fatalf("已过期会员: got %s want %s", got, consts.DefaultLevelKey)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestSendNoteRemindWithEmptyTemplateDegrades 覆盖「模板未配置时接口正常、无 panic」：
|
|||
|
|
// 直接调用真实发送函数并传入空模板 ID，断言不 panic 且返回 nil（降级）。
|
|||
|
|
func TestSendNoteRemindWithEmptyTemplateDegrades(t *testing.T) {
|
|||
|
|
defer func() {
|
|||
|
|
if r := recover(); r != nil {
|
|||
|
|
t.Fatalf("模板未配置时不应 panic，但发生了 panic: %v", r)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
ctx := gctx.New()
|
|||
|
|
err := sendNoteRemindWithTemplate(ctx, NoteRemindSendReq{
|
|||
|
|
NoteId: 1,
|
|||
|
|
Openid: "o_test_openid",
|
|||
|
|
Title: "会议要点",
|
|||
|
|
Content: "待办清单：a、b、c",
|
|||
|
|
Time: "2026-09-20 09:00:00",
|
|||
|
|
}, "")
|
|||
|
|
if err != nil {
|
|||
|
|
t.Fatalf("模板未配置时应降级返回 nil，但返回 err=%v", err)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestShouldClaimRemind 证明「同一提醒只发一次」的条件更新语义：
|
|||
|
|
// 只有当抢占了 remind_status=1→2 的那一次（rowsAffected==1）才继续发送。
|
|||
|
|
func TestShouldClaimRemind(t *testing.T) {
|
|||
|
|
if !shouldClaimRemind(1) {
|
|||
|
|
t.Fatalf("rowsAffected=1 时应继续发送")
|
|||
|
|
}
|
|||
|
|
if shouldClaimRemind(0) {
|
|||
|
|
t.Fatalf("rowsAffected=0 时应跳过（已被其它实例/轮次抢占）")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// TestJoinAndSplitTags 覆盖标签规整（去空白/去空项/去重）与回拆分。
|
|||
|
|
func TestJoinAndSplitTags(t *testing.T) {
|
|||
|
|
csv := joinNoteTags([]string{" 工作 ", "", "待办", "工作"})
|
|||
|
|
if csv != "工作,待办" {
|
|||
|
|
t.Fatalf("joinNoteTags got %q want %q", csv, "工作,待办")
|
|||
|
|
}
|
|||
|
|
tags := splitNoteTags(csv)
|
|||
|
|
if len(tags) != 2 || tags[0] != "工作" || tags[1] != "待办" {
|
|||
|
|
t.Fatalf("splitNoteTags got %v want [工作 待办]", tags)
|
|||
|
|
}
|
|||
|
|
}
|