order_pay_repo_test.go 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307
  1. // OrderPayRepo 单测(AC-03):三通道可见性(含 NULL 行裁定 §2.5)/ CAS 幂等 / 分页 / 幂等单号。
  2. //
  3. // 测试栈:glebarez/sqlite 文件库(**只迁 t_order_pay 单表**——sqlite 索引名库级全局,
  4. // `uk_biz_no` 与工单/支付日志表重名,混迁会撞名,原因见 expense_repo_test.go 注释)。
  5. package repository
  6. import (
  7. "context"
  8. "errors"
  9. "path/filepath"
  10. "testing"
  11. "time"
  12. gormsqlite "github.com/glebarez/sqlite"
  13. "gorm.io/gorm"
  14. "yuansiwei/unified-backend/internal/academic/model"
  15. infraerrors "yuansiwei/unified-backend/internal/infra/errors"
  16. appscope "yuansiwei/unified-backend/internal/infra/scope"
  17. )
  18. // newOrderPayRepoDB 支付单表文件库。
  19. func newOrderPayRepoDB(t *testing.T) *gorm.DB {
  20. t.Helper()
  21. dsn := filepath.Join(t.TempDir(), "academic_order_pay_repo_test.db")
  22. db, err := gorm.Open(gormsqlite.Open(dsn), &gorm.Config{})
  23. if err != nil {
  24. t.Fatalf("gorm.Open(支付单文件库) 失败: %v", err)
  25. }
  26. t.Cleanup(func() {
  27. if sqlDB, e := db.DB(); e == nil {
  28. _ = sqlDB.Close()
  29. }
  30. })
  31. if err := db.AutoMigrate(&model.OrderPay{}); err != nil {
  32. t.Fatalf("AutoMigrate(t_order_pay) 失败: %v", err)
  33. }
  34. return db
  35. }
  36. // seedOrderPay 插入一条支付单(机构/校区可为 nil,用于覆盖 NULL 行裁定)。
  37. //
  38. // createTime 传零值即由 DB 默认值兜底(model 侧 `default:CURRENT_TIMESTAMP`)。
  39. func seedOrderPay(t *testing.T, db *gorm.DB, bizNo string, inst, campus *int64, userID *string, createTime time.Time) {
  40. t.Helper()
  41. row := &model.OrderPay{
  42. BizNo: bizNo, InstitutionID: inst, CampusID: campus, UserID: userID,
  43. PayMoney: "100.00", Status: model.OrderPayStatusUnpaid, CreateTime: createTime,
  44. }
  45. if err := db.Create(row).Error; err != nil {
  46. t.Fatalf("seed 支付单 %s 失败: %v", bizNo, err)
  47. }
  48. }
  49. // ptrI64 int64 指针助手(NULL 与 0 语义不同,显式构造便于阅读)。
  50. func ptrI64(v int64) *int64 { return &v }
  51. // ptrStr 字符串指针助手。
  52. func ptrStr(v string) *string { return &v }
  53. // TestOrderPayListByUserIncludesNullScope 通道①:付款人主口径——本人付的 NULL 机构行**必须**可见。
  54. func TestOrderPayListByUserIncludesNullScope(t *testing.T) {
  55. db := newOrderPayRepoDB(t)
  56. const owner = "01HUSER000000000000000301"
  57. const otherUser = "01HUSER000000000000000302"
  58. base := time.Date(2026, 3, 1, 0, 0, 0, 0, time.UTC)
  59. seedOrderPay(t, db, "PAY-A", ptrI64(1), ptrI64(2), ptrStr(owner), base)
  60. seedOrderPay(t, db, "PAY-B", nil, nil, ptrStr(owner), base.Add(time.Minute)) // 家长代付/历史行:NULL 机构
  61. seedOrderPay(t, db, "PAY-C", ptrI64(2), ptrI64(5), ptrStr(otherUser), base)
  62. repo := NewOrderPayRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  63. rows, err := repo.ListByUser(context.Background(), owner, PageQuery{PageSize: 10})
  64. if err != nil {
  65. t.Fatalf("ListByUser 失败: %v", err)
  66. }
  67. if len(rows) != 2 {
  68. t.Fatalf("本人订单数 = %d,期望 2(含 NULL 机构行,且不含他人订单)", len(rows))
  69. }
  70. if rows[0].BizNo != "PAY-B" || rows[1].BizNo != "PAY-A" {
  71. t.Fatalf("排序应为 create_time DESC:%s / %s", rows[0].BizNo, rows[1].BizNo)
  72. }
  73. nullSeen := false
  74. for _, r := range rows {
  75. if r.InstitutionID == nil {
  76. nullSeen = true
  77. }
  78. }
  79. if !nullSeen {
  80. t.Fatal("付款人通道必须返回 NULL 机构行(§2.5:钱是本人付的)")
  81. }
  82. }
  83. // TestOrderPayListByScopeExcludesNullRows 通道②:显式机构参数——NULL 行**不得**出现;非超管越权拒绝。
  84. func TestOrderPayListByScopeExcludesNullRows(t *testing.T) {
  85. db := newOrderPayRepoDB(t)
  86. const user = "01HUSER000000000000000303"
  87. seedOrderPay(t, db, "PAY-D", ptrI64(1), ptrI64(2), ptrStr(user), time.Time{})
  88. seedOrderPay(t, db, "PAY-E", nil, nil, ptrStr(user), time.Time{}) // NULL 行:institution/campus 都空
  89. seedOrderPay(t, db, "PAY-F", ptrI64(2), ptrI64(5), ptrStr(user), time.Time{})
  90. ctx := context.Background()
  91. repo := NewOrderPayRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  92. all, err := repo.ListByScope(ctx, 1, nil, PageQuery{PageSize: 10})
  93. if err != nil {
  94. t.Fatalf("ListByScope(inst=1) 失败: %v", err)
  95. }
  96. if len(all) != 1 || all[0].BizNo != "PAY-D" {
  97. t.Fatalf("机构 1 只应有 1 条非 NULL 行,实得 %d:%v", len(all), all)
  98. }
  99. withCampus, err := repo.ListByScope(ctx, 1, ptrI64(2), PageQuery{PageSize: 10})
  100. if err != nil {
  101. t.Fatalf("ListByScope(inst=1,campus=2) 失败: %v", err)
  102. }
  103. if len(withCampus) != 1 || withCampus[0].BizNo != "PAY-D" {
  104. t.Fatalf("机构 1 校区 2 只应有 1 条,实得 %d", len(withCampus))
  105. }
  106. emptyCampus, err := repo.ListByScope(ctx, 1, ptrI64(9), PageQuery{PageSize: 10})
  107. if err != nil {
  108. t.Fatalf("ListByScope(inst=1,campus=9) 失败: %v", err)
  109. }
  110. if emptyCampus == nil || len(emptyCampus) != 0 {
  111. t.Fatalf("空结果必须是非 nil 空切片:%v", emptyCampus)
  112. }
  113. // 非超管借参数跨机构 → ErrUnscoped
  114. if _, err := repo.ListByScope(ctx, 2, nil, PageQuery{}); !errors.Is(err, infraerrors.ErrUnscoped) {
  115. t.Fatalf("跨机构参数错误 = %v,期望 ErrUnscoped", err)
  116. }
  117. // 机构参数非法
  118. if _, err := repo.ListByScope(ctx, 0, nil, PageQuery{}); !errors.Is(err, infraerrors.ErrBadParam) {
  119. t.Fatalf("institution_id=0 错误 = %v,期望 ErrBadParam", err)
  120. }
  121. // nil 作用域(系统任务)不得走参数通道
  122. sys := NewOrderPayRepo(db, nil)
  123. if _, err := sys.ListByScope(ctx, 1, nil, PageQuery{}); !errors.Is(err, infraerrors.ErrUnscoped) {
  124. t.Fatalf("nil 作用域错误 = %v,期望 ErrUnscoped", err)
  125. }
  126. // 超管可看任意机构
  127. admin := NewOrderPayRepo(db, &appscope.Scope{SuperAdmin: true, InstitutionID: 9})
  128. adminRows, err := admin.ListByScope(ctx, 2, nil, PageQuery{PageSize: 10})
  129. if err != nil {
  130. t.Fatalf("超管查询失败: %v", err)
  131. }
  132. if len(adminRows) != 1 || adminRows[0].BizNo != "PAY-F" {
  133. t.Fatalf("超管应看到机构 2 的 1 条,实得 %d", len(adminRows))
  134. }
  135. }
  136. // TestOrderPayGetByOrderNo 通道③:系统通道按 biz_no 命中;pay_order 兜底;不存在 → ErrNotFound。
  137. func TestOrderPayGetByOrderNo(t *testing.T) {
  138. db := newOrderPayRepoDB(t)
  139. const user = "01HUSER000000000000000304"
  140. seedOrderPay(t, db, "PAY-G", nil, nil, ptrStr(user), time.Time{})
  141. if err := db.Model(&model.OrderPay{}).Where("biz_no = ?", "PAY-G").
  142. UpdateColumn("pay_order", "4200001234").Error; err != nil {
  143. t.Fatalf("回填 pay_order 失败: %v", err)
  144. }
  145. sys := NewOrderPayRepo(db, nil) // 回调路径:无请求作用域
  146. ctx := context.Background()
  147. byBiz, err := sys.GetByOrderNo(ctx, "PAY-G")
  148. if err != nil {
  149. t.Fatalf("按 biz_no 查询失败: %v", err)
  150. }
  151. if byBiz.BizNo != "PAY-G" || byBiz.InstitutionID != nil {
  152. t.Fatalf("命中结果不符:%s / %v", byBiz.BizNo, byBiz.InstitutionID)
  153. }
  154. byPayOrder, err := sys.GetByOrderNo(ctx, "4200001234")
  155. if err != nil {
  156. t.Fatalf("按第三方流水号兜底查询失败: %v", err)
  157. }
  158. if byPayOrder.BizNo != "PAY-G" {
  159. t.Fatalf("兜底命中结果不符:%s", byPayOrder.BizNo)
  160. }
  161. if _, err := sys.GetByOrderNo(ctx, "PAY-NOT-EXIST"); !errors.Is(err, infraerrors.ErrNotFound) {
  162. t.Fatalf("不存在单号错误 = %v,期望 ErrNotFound", err)
  163. }
  164. if _, err := sys.GetByOrderNo(ctx, " "); !errors.Is(err, infraerrors.ErrBadParam) {
  165. t.Fatalf("空单号错误 = %v,期望 ErrBadParam", err)
  166. }
  167. }
  168. // TestOrderPayUpdateStatusCASIdempotent CAS:首次成功、重复调用 affected=0、非法状态拒绝。
  169. func TestOrderPayUpdateStatusCASIdempotent(t *testing.T) {
  170. db := newOrderPayRepoDB(t)
  171. repo := NewOrderPayRepo(db, nil)
  172. runner := NewTxRunner(db)
  173. ctx := context.Background()
  174. const user = "01HUSER000000000000000305"
  175. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  176. return repo.Create(ctx, tx, &model.OrderPay{
  177. BizNo: "PAY-H", UserID: ptrStr(user), InstitutionID: ptrI64(1), CampusID: ptrI64(2),
  178. PayMoney: "88.00", Status: model.OrderPayStatusUnpaid,
  179. })
  180. }); err != nil {
  181. t.Fatalf("创建支付单失败: %v", err)
  182. }
  183. // 首次回调:0 → 1
  184. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  185. affected, e := repo.UpdateStatusCAS(ctx, tx, "PAY-H",
  186. model.OrderPayStatusUnpaid, model.OrderPayStatusSucceeded)
  187. if e != nil {
  188. return e
  189. }
  190. if affected != 1 {
  191. t.Fatalf("首次 CAS affected = %d,期望 1", affected)
  192. }
  193. return nil
  194. }); err != nil {
  195. t.Fatalf("首次 CAS 失败: %v", err)
  196. }
  197. // 重复回调:affected=0(幂等,禁止重复入账)
  198. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  199. affected, e := repo.UpdateStatusCAS(ctx, tx, "PAY-H",
  200. model.OrderPayStatusUnpaid, model.OrderPayStatusSucceeded)
  201. if e != nil {
  202. return e
  203. }
  204. if affected != 0 {
  205. t.Fatalf("重复 CAS affected = %d,期望 0", affected)
  206. }
  207. return nil
  208. }); err != nil {
  209. t.Fatalf("重复 CAS 失败: %v", err)
  210. }
  211. // 不存在的单号:affected=0(不报错,由 Service 判 404/幂等)
  212. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  213. affected, e := repo.UpdateStatusCAS(ctx, tx, "PAY-NONE",
  214. model.OrderPayStatusUnpaid, model.OrderPayStatusSucceeded)
  215. if e != nil {
  216. return e
  217. }
  218. if affected != 0 {
  219. t.Fatalf("不存在单号 affected = %d,期望 0", affected)
  220. }
  221. return nil
  222. }); err != nil {
  223. t.Fatalf("不存在单号用例失败: %v", err)
  224. }
  225. // 非法状态值 → ErrBadParam
  226. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error {
  227. _, e := repo.UpdateStatusCAS(ctx, tx, "PAY-H", model.OrderPayStatusSucceeded, 9)
  228. if !errors.Is(e, infraerrors.ErrBadParam) {
  229. t.Fatalf("非法状态错误 = %v,期望 ErrBadParam", e)
  230. }
  231. return nil
  232. }); err != nil {
  233. t.Fatalf("非法状态用例失败: %v", err)
  234. }
  235. }
  236. // TestOrderPayCreateDuplicateBizNo 下单幂等:biz_no 撞键 → ErrDuplicateBizNo(3003)。
  237. func TestOrderPayCreateDuplicateBizNo(t *testing.T) {
  238. db := newOrderPayRepoDB(t)
  239. repo := NewOrderPayRepo(db, nil)
  240. runner := NewTxRunner(db)
  241. ctx := context.Background()
  242. order := &model.OrderPay{BizNo: "PAY-I", PayMoney: "1.00", Status: model.OrderPayStatusUnpaid}
  243. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error { return repo.Create(ctx, tx, order) }); err != nil {
  244. t.Fatalf("创建支付单失败: %v", err)
  245. }
  246. dup := &model.OrderPay{BizNo: "PAY-I", PayMoney: "1.00", Status: model.OrderPayStatusUnpaid}
  247. err := runner.WithinTx(ctx, func(tx *gorm.DB) error { return repo.Create(ctx, tx, dup) })
  248. if !errors.Is(err, ErrDuplicateBizNo) {
  249. t.Fatalf("重复单号错误 = %v,期望 ErrDuplicateBizNo", err)
  250. }
  251. // 缺 biz_no → ErrBadParam
  252. bad := &model.OrderPay{PayMoney: "1.00"}
  253. if err := runner.WithinTx(ctx, func(tx *gorm.DB) error { return repo.Create(ctx, tx, bad) }); !errors.Is(err, infraerrors.ErrBadParam) {
  254. t.Fatalf("缺 biz_no 错误 = %v,期望 ErrBadParam", err)
  255. }
  256. }
  257. // TestOrderPayListPagingEmpty 分页:空结果非 nil;pageSize 钳制 200。
  258. func TestOrderPayListPagingEmpty(t *testing.T) {
  259. db := newOrderPayRepoDB(t)
  260. repo := NewOrderPayRepo(db, &appscope.Scope{InstitutionID: 1, CampusID: 2})
  261. ctx := context.Background()
  262. empty, err := repo.ListByUser(ctx, "01HUSER000000000000000399", PageQuery{})
  263. if err != nil {
  264. t.Fatalf("空结果查询失败: %v", err)
  265. }
  266. if empty == nil || len(empty) != 0 {
  267. t.Fatalf("空结果必须是非 nil 空切片:%v", empty)
  268. }
  269. if got := (PageQuery{PageSize: 1000}).Limit(); got != MaxPageSize {
  270. t.Fatalf("pageSize 钳制 = %d,期望 %d", got, MaxPageSize)
  271. }
  272. // 空 user_id → ErrBadParam
  273. if _, err := repo.ListByUser(ctx, " ", PageQuery{}); !errors.Is(err, infraerrors.ErrBadParam) {
  274. t.Fatalf("空 user_id 错误 = %v,期望 ErrBadParam", err)
  275. }
  276. }