activity_discount_repo_test.go 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369
  1. // ActivityDiscountRepo 单测(AC-06):可用额口径 / dump 3 参汇总 / 原子增减 / 批量单语句 CASE。
  2. package repository
  3. import (
  4. "context"
  5. "errors"
  6. "strings"
  7. "testing"
  8. "time"
  9. "gorm.io/gorm"
  10. "yuansiwei/unified-backend/internal/academic/model"
  11. "yuansiwei/unified-backend/internal/academic/money"
  12. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  13. appscope "yuansiwei/unified-backend/internal/infra/scope"
  14. )
  15. // seedPool 插入一条赠送金额池(显式 ID)。
  16. func seedPool(t *testing.T, db *gorm.DB, id, inst, campus int64, userID, discount, used string) {
  17. t.Helper()
  18. row := &model.CoursesActivityUserDiscount{
  19. ID: id,
  20. ActivityID: 1,
  21. InstitutionID: inst,
  22. CampusID: campus,
  23. UserID: userID,
  24. Income: "0.00",
  25. DiscountAmount: discount,
  26. DiscountAmountUsed: used,
  27. CreateTime: time.Now(),
  28. }
  29. if err := db.Create(row).Error; err != nil {
  30. t.Fatalf("seed pool(%d) 失败: %v", id, err)
  31. }
  32. }
  33. // amountOf 读取池的「已使用金额」(money 归类,规避 sqlite NUMERIC 回读差异)。
  34. func amountOf(t *testing.T, db *gorm.DB, id int64) money.Amount {
  35. t.Helper()
  36. var row model.CoursesActivityUserDiscount
  37. if err := db.Where("id = ?", id).First(&row).Error; err != nil {
  38. t.Fatalf("读池(%d) 失败: %v", id, err)
  39. }
  40. amt, err := money.Parse(row.DiscountAmountUsed)
  41. if err != nil {
  42. t.Fatalf("解析 used(%d) 失败: %v", id, err)
  43. }
  44. return amt
  45. }
  46. // TestListUsableFilterAndOrder 可用口径(discount > used)+ 活动维度 + 游标方向。
  47. func TestListUsableFilterAndOrder(t *testing.T) {
  48. db := newActivityRepoDB(t)
  49. user := "01HUSER000000000000000011"
  50. seedPool(t, db, 1, 1, 2, user, "10.00", "0.00") // 可用
  51. seedPool(t, db, 2, 1, 2, user, "20.00", "20.00") // 用尽:不可用
  52. seedPool(t, db, 3, 1, 2, user, "-5.00", "0.00") // 负赠送:不可用
  53. seedPool(t, db, 4, 1, 2, user, "7.50", "2.50") // 可用
  54. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  55. ctx := context.Background()
  56. asc, err := repo.ListUsable(ctx, user, 0, OrderAsc)
  57. if err != nil {
  58. t.Fatalf("ListUsable(asc) 失败: %v", err)
  59. }
  60. if len(asc) != 2 || asc[0].ID != 1 || asc[1].ID != 4 {
  61. t.Fatalf("可用行应为 {1,4}(升序),实得 %+v", asc)
  62. }
  63. desc, err := repo.ListUsable(ctx, user, 0, OrderDesc)
  64. if err != nil {
  65. t.Fatalf("ListUsable(desc) 失败: %v", err)
  66. }
  67. if len(desc) != 2 || desc[0].ID != 4 || desc[1].ID != 1 {
  68. t.Fatalf("可用行应为 {4,1}(降序),实得 %+v", desc)
  69. }
  70. if _, err := repo.ListUsable(ctx, "", 0, OrderAsc); !errors.Is(err, infraerrors.ErrBadParam) {
  71. t.Fatalf("空 user_id 应 ErrBadParam,实得 %v", err)
  72. }
  73. if _, err := repo.ListUsable(ctx, user, 0, "desc,1;--"); !errors.Is(err, infraerrors.ErrBadParam) {
  74. t.Fatalf("非法方向应 ErrBadParam,实得 %v", err)
  75. }
  76. if _, err := repo.ListUsable(ctx, user, -1, OrderAsc); !errors.Is(err, infraerrors.ErrBadParam) {
  77. t.Fatalf("activity_id<0 应 ErrBadParam,实得 %v", err)
  78. }
  79. // activityID>0 只取该活动(seedPool 里全是活动 1)
  80. onlyOne, err := repo.ListUsable(ctx, user, 1, OrderAsc)
  81. if err != nil {
  82. t.Fatalf("ListUsable(activity=1) 失败: %v", err)
  83. }
  84. if len(onlyOne) != 2 {
  85. t.Fatalf("活动 1 可用行应为 2,实得 %d", len(onlyOne))
  86. }
  87. none, err := repo.ListUsable(ctx, user, 2, OrderAsc)
  88. if err != nil {
  89. t.Fatalf("ListUsable(activity=2) 失败: %v", err)
  90. }
  91. if len(none) != 0 {
  92. t.Fatalf("活动 2 应无可用行(活动维度隔离),实得 %d", len(none))
  93. }
  94. }
  95. // TestListUsableActivityIsolation A 活动的扣减不影响 B 活动的池(audit A8,dump 763216)。
  96. //
  97. // 若 ListUsable 漏了 `activity_id = ?`(或不传 activityID),扣 A 活动时会按 id 升序先吃光 B 活动的池
  98. // → 用户"用 A 的赠送额度却扣了 B 的池",且退费时退回对象错位。本用例是该缺陷的回归门禁。
  99. func TestListUsableActivityIsolation(t *testing.T) {
  100. db := newActivityRepoDB(t)
  101. user := "01HUSER000000000000000021"
  102. // 活动 1 的池 id=1(后建),活动 2 的池 id=2(id 更大)——故意让 id 序与活动序相反
  103. if err := db.Create(&model.CoursesActivityUserDiscount{
  104. ID: 1, ActivityID: 1, InstitutionID: 1, CampusID: 2, UserID: user,
  105. Income: "0.00", DiscountAmount: "10.00", DiscountAmountUsed: "0.00",
  106. }).Error; err != nil {
  107. t.Fatalf("seed 池 1 失败: %v", err)
  108. }
  109. if err := db.Create(&model.CoursesActivityUserDiscount{
  110. ID: 2, ActivityID: 2, InstitutionID: 1, CampusID: 2, UserID: user,
  111. Income: "0.00", DiscountAmount: "20.00", DiscountAmountUsed: "0.00",
  112. }).Error; err != nil {
  113. t.Fatalf("seed 池 2 失败: %v", err)
  114. }
  115. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  116. ctx := context.Background()
  117. rows, err := repo.ListUsable(ctx, user, 2, OrderAsc)
  118. if err != nil {
  119. t.Fatalf("ListUsable(activity=2) 失败: %v", err)
  120. }
  121. if len(rows) != 1 || rows[0].ID != 2 {
  122. t.Fatalf("活动 2 只应命中池 2,实得 %+v", rows)
  123. }
  124. if _, err := repo.IncUsed(ctx, db, 2, money.FromCents(2000)); err != nil {
  125. t.Fatalf("扣减活动 2 的池失败: %v", err)
  126. }
  127. if got := amountOf(t, db, 1); got != 0 {
  128. t.Fatalf("活动 1 的池不得被活动 2 的扣减影响,实得 %s", got.String())
  129. }
  130. }
  131. // TestSumUsableThreeParamScoping dump usable 3 参**等值**口径(audit A7,dump 763182-763183)。
  132. //
  133. // dump:`WHERE schoolid=? AND campusid=? AND userid=?`,两个作用域参数传 0 视为**不筛**(平台/机构级汇总)。
  134. // 关键差异(相对旧实现的 `IN (campus,0)` 超集):机构级池(campus=0)**不并入**指定校区可用额。
  135. //
  136. // 池清单(用户 user,可用 = discount - used):
  137. //
  138. // pool1 inst1 campus2 可用 10 pool2 inst1 campus0 可用 15(机构级池)
  139. // pool3 inst1 campus3 可用 100 pool4 inst2 campus2 可用 50
  140. // pool5 inst1 campus2 可用 0(用尽) pool6 他人 —— 用户过滤永远排除
  141. func TestSumUsableThreeParamScoping(t *testing.T) {
  142. db := newActivityRepoDB(t)
  143. user := "01HUSER000000000000000012"
  144. seedPool(t, db, 1, 1, 2, user, "10.00", "0.00")
  145. seedPool(t, db, 2, 1, 0, user, "20.00", "5.00")
  146. seedPool(t, db, 3, 1, 3, user, "100.00", "0.00")
  147. seedPool(t, db, 4, 2, 2, user, "50.00", "0.00")
  148. seedPool(t, db, 5, 1, 2, user, "5.00", "5.00")
  149. seedPool(t, db, 6, 1, 2, "01HOTHER00000000000001", "999.00", "0.00")
  150. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  151. ctx := context.Background()
  152. // ① 本校区分池:inst=1 AND campus=2 → pool1(10) + pool5(0) = 10.00(**不含**机构级池 15)
  153. if got, err := repo.SumUsable(ctx, user, 1, 2); err != nil || got.Cents() != 1000 {
  154. t.Fatalf("SumUsable(1,2) 应为 10.00(等值口径,不含 campus=0 池),实得 %s err=%v", got.String(), err)
  155. }
  156. // ② 不限定校区:inst=1 → 10 + 15 + 100 = 125.00
  157. if got, err := repo.SumUsable(ctx, user, 1, 0); err != nil || got.Cents() != 12500 {
  158. t.Fatalf("SumUsable(1,0) 应为 125.00,实得 %s err=%v", got.String(), err)
  159. }
  160. // ③ 平台视角:inst=0 且 campus=0 → 不过滤机构/校区 → 10+15+100+50 = 175.00
  161. if got, err := repo.SumUsable(ctx, user, 0, 0); err != nil || got.Cents() != 17500 {
  162. t.Fatalf("SumUsable(0,0) 平台视角应为 175.00,实得 %s err=%v", got.String(), err)
  163. }
  164. // ④ 只按校区分池(跨机构同名校区):campus=2 → pool1(10) + pool4(50) = 60.00
  165. if got, err := repo.SumUsable(ctx, user, 0, 2); err != nil || got.Cents() != 6000 {
  166. t.Fatalf("SumUsable(0,2) 应为 60.00,实得 %s err=%v", got.String(), err)
  167. }
  168. // ⑤ 他机构等值:inst=2 → 50.00(越权防线已按裁定移到 Handler 角色白名单,仓储不再拦)
  169. if got, err := repo.SumUsable(ctx, user, 2, 2); err != nil || got.Cents() != 5000 {
  170. t.Fatalf("SumUsable(2,2) 应为 50.00,实得 %s err=%v", got.String(), err)
  171. }
  172. // 参数防御:负数非法(0 才是"不筛")
  173. if _, err := repo.SumUsable(ctx, user, -1, 2); !errors.Is(err, infraerrors.ErrBadParam) {
  174. t.Fatalf("institutionID<0 应 ErrBadParam,实得 %v", err)
  175. }
  176. if _, err := repo.SumUsable(ctx, "", 1, 2); !errors.Is(err, infraerrors.ErrBadParam) {
  177. t.Fatalf("空 user_id 应 ErrBadParam,实得 %v", err)
  178. }
  179. // 无可用行 → 0(不是错误)
  180. if got, err := repo.SumUsable(ctx, "01HNOBODY00000000000001", 1, 2); err != nil || !got.IsZero() {
  181. t.Fatalf("无可用行应返回 0,实得 %s err=%v", got.String(), err)
  182. }
  183. }
  184. // TestIncUsedAndBatch 原子增减 + 批量单语句 CASE(正负增量都精确到分)。
  185. func TestIncUsedAndBatch(t *testing.T) {
  186. db := newActivityRepoDB(t)
  187. user := "01HUSER000000000000000013"
  188. seedPool(t, db, 1, 1, 2, user, "100.00", "0.00")
  189. seedPool(t, db, 2, 1, 2, user, "100.00", "0.00")
  190. seedPool(t, db, 3, 1, 2, user, "100.00", "10.00")
  191. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  192. runner := NewTxRunner(db)
  193. ctx := context.Background()
  194. // 单账户:消费 12.34
  195. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  196. n, e := repo.IncUsed(ctx, tx, 1, money.FromCents(1234))
  197. if e != nil {
  198. return e
  199. }
  200. if n != 1 {
  201. t.Fatalf("IncUsed 应影响 1 行,实得 %d", n)
  202. }
  203. return nil
  204. }); err != nil {
  205. t.Fatalf("IncUsed 失败: %v", err)
  206. }
  207. if got := amountOf(t, db, 1); got.Cents() != 1234 {
  208. t.Fatalf("id=1 used 应为 12.34,实得 %s", got.String())
  209. }
  210. // 批量:id1 +1.00 / id2 +2.50 / id3 -0.01(退回冲正)
  211. sqls := captureSQL(t, db)
  212. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  213. _, e := repo.IncUsedBatch(ctx, tx, []UsedDelta{
  214. {DiscountID: 1, Delta: money.FromCents(100)},
  215. {DiscountID: 2, Delta: money.FromCents(250)},
  216. {DiscountID: 3, Delta: money.FromCents(-1)},
  217. })
  218. return e
  219. }); err != nil {
  220. t.Fatalf("IncUsedBatch 失败: %v", err)
  221. }
  222. // 立即断言(后续读库也会被回调捕获,故此处先判再读)
  223. if len(*sqls) != 1 {
  224. t.Fatalf("批量增减必须 1 条语句(禁止逐行 UPDATE),实得 %d 条: %v", len(*sqls), *sqls)
  225. }
  226. if !strings.Contains((*sqls)[0], "CASE id") {
  227. t.Fatalf("批量增减应为 CASE WHEN 单语句,SQL=%q", (*sqls)[0])
  228. }
  229. if got := amountOf(t, db, 1); got.Cents() != 1334 {
  230. t.Fatalf("id=1 used 应为 13.34,实得 %s", got.String())
  231. }
  232. if got := amountOf(t, db, 2); got.Cents() != 250 {
  233. t.Fatalf("id=2 used 应为 2.50,实得 %s", got.String())
  234. }
  235. if got := amountOf(t, db, 3); got.Cents() != 999 {
  236. t.Fatalf("id=3 used 应为 9.99(10.00-0.01),实得 %s", got.String())
  237. }
  238. }
  239. // TestIncUsedBatchEdgeCases 空集合/零增量/非法 id 不发 SQL;同 id 重复出现以后者为准。
  240. func TestIncUsedBatchEdgeCases(t *testing.T) {
  241. db := newActivityRepoDB(t)
  242. user := "01HUSER000000000000000014"
  243. seedPool(t, db, 1, 1, 2, user, "100.00", "0.00")
  244. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  245. runner := NewTxRunner(db)
  246. ctx := context.Background()
  247. sqls := captureSQL(t, db)
  248. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  249. n, e := repo.IncUsedBatch(ctx, tx, nil)
  250. if e != nil {
  251. return e
  252. }
  253. if n != 0 {
  254. t.Fatalf("空集合应 0 行,实得 %d", n)
  255. }
  256. _, e = repo.IncUsedBatch(ctx, tx, []UsedDelta{{DiscountID: 0, Delta: money.FromCents(1)}, {DiscountID: 1, Delta: 0}})
  257. return e
  258. }); err != nil {
  259. t.Fatalf("IncUsedBatch 边界用例失败: %v", err)
  260. }
  261. if len(*sqls) != 0 {
  262. t.Fatalf("无有效条目不应发 SQL,实得 %v", *sqls)
  263. }
  264. // 同 id 两次:后者覆盖前者
  265. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  266. _, e := repo.IncUsedBatch(ctx, tx, []UsedDelta{
  267. {DiscountID: 1, Delta: money.FromCents(500)},
  268. {DiscountID: 1, Delta: money.FromCents(700)},
  269. })
  270. return e
  271. }); err != nil {
  272. t.Fatalf("IncUsedBatch(重复 id) 失败: %v", err)
  273. }
  274. if got := amountOf(t, db, 1); got.Cents() != 700 {
  275. t.Fatalf("重复 id 应以后者为准(7.00),实得 %s", got.String())
  276. }
  277. // tx 缺失 → ErrInternal
  278. if _, err := repo.IncUsed(nil, nil, 1, money.FromCents(1)); !errors.Is(err, infraerrors.ErrInternal) {
  279. t.Fatalf("tx 为 nil 应 ErrInternal,实得 %v", err)
  280. }
  281. if _, err := repo.IncUsedBatch(ctx, nil, []UsedDelta{{DiscountID: 1, Delta: 1}}); !errors.Is(err, infraerrors.ErrInternal) {
  282. t.Fatalf("tx 为 nil 应 ErrInternal,实得 %v", err)
  283. }
  284. if _, err := repo.IncUsed(ctx, repo.DB, 0, money.FromCents(1)); !errors.Is(err, infraerrors.ErrBadParam) {
  285. t.Fatalf("id<=0 应 ErrBadParam,实得 %v", err)
  286. }
  287. }
  288. // TestInsertDiscountAndUsedLog 两条流水写入(参数校验 + 落库形态)。
  289. func TestInsertDiscountAndUsedLog(t *testing.T) {
  290. db := newActivityRepoDB(t)
  291. user := "01HUSER000000000000000015"
  292. seedPool(t, db, 1, 1, 2, user, "100.00", "0.00")
  293. repo := NewActivityDiscountRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  294. runner := NewTxRunner(db)
  295. ctx := context.Background()
  296. stuCourse := int64(901)
  297. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  298. return repo.InsertDiscountLog(ctx, tx, &model.CoursesActivityUserDiscountLog{
  299. ActivityUserDiscountID: 1,
  300. CoursesStudentID: &stuCourse,
  301. DiscountAmountUsed: "-10.00",
  302. CreateTime: time.Now(),
  303. })
  304. }); err != nil {
  305. t.Fatalf("InsertDiscountLog 失败: %v", err)
  306. }
  307. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  308. return repo.InsertUsedLog(ctx, tx, &model.CoursesActivityUsedLog{
  309. ActivityID: 1,
  310. UserID: user,
  311. OrderNo: ptrString("901"),
  312. ActivityMoneyUsed: "-10.00",
  313. CreateTime: time.Now(),
  314. })
  315. }); err != nil {
  316. t.Fatalf("InsertUsedLog 失败: %v", err)
  317. }
  318. var dlogCount, ulogCount int64
  319. if err := db.Model(&model.CoursesActivityUserDiscountLog{}).Count(&dlogCount).Error; err != nil {
  320. t.Fatalf("count discount log 失败: %v", err)
  321. }
  322. if err := db.Model(&model.CoursesActivityUsedLog{}).Count(&ulogCount).Error; err != nil {
  323. t.Fatalf("count used log 失败: %v", err)
  324. }
  325. if dlogCount != 1 || ulogCount != 1 {
  326. t.Fatalf("两条流水应各 1 行,实得 discount=%d used=%d", dlogCount, ulogCount)
  327. }
  328. // 参数/事务防御
  329. if err := repo.InsertDiscountLog(ctx, nil, &model.CoursesActivityUserDiscountLog{ActivityUserDiscountID: 1}); !errors.Is(err, infraerrors.ErrInternal) {
  330. t.Fatalf("tx nil 应 ErrInternal,实得 %v", err)
  331. }
  332. if err := repo.InsertDiscountLog(ctx, repo.DB, &model.CoursesActivityUserDiscountLog{ActivityUserDiscountID: 0}); !errors.Is(err, infraerrors.ErrBadParam) {
  333. t.Fatalf("缺 discount id 应 ErrBadParam,实得 %v", err)
  334. }
  335. if err := repo.InsertUsedLog(ctx, repo.DB, &model.CoursesActivityUsedLog{ActivityID: 0, UserID: user}); !errors.Is(err, infraerrors.ErrBadParam) {
  336. t.Fatalf("缺 activity_id 应 ErrBadParam,实得 %v", err)
  337. }
  338. if err := repo.InsertUsedLog(ctx, repo.DB, nil); !errors.Is(err, infraerrors.ErrBadParam) {
  339. t.Fatalf("nil 行应 ErrBadParam,实得 %v", err)
  340. }
  341. }
  342. // ptrString 字符串指针构造(测试用)。
  343. func ptrString(s string) *string { return &s }