wangjia
126 lines
4.3 KiB
Go
126 lines
4.3 KiB
Go
package store_test
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/wangjia/pay/internal/model"
|
|
"github.com/wangjia/pay/internal/store"
|
|
)
|
|
|
|
func TestGetOrderAndAttemptByRef(t *testing.T) {
|
|
s := store.NewOrderStore(model.OpenTestDB(t))
|
|
seedOrder(t, s, "PAY-Q1") // 复用 order_test.go 的 seedOrder(建 order + attempt provider_ref="P-PAY-Q1")
|
|
|
|
o, err := s.GetOrder("PAY-Q1")
|
|
if err != nil || o.OutTradeNo != "PAY-Q1" {
|
|
t.Fatalf("GetOrder = %+v, %v", o, err)
|
|
}
|
|
if _, err := s.GetOrder("NOPE"); !errors.Is(err, store.ErrOrderNotFound) {
|
|
t.Fatalf("缺单应 ErrOrderNotFound, got %v", err)
|
|
}
|
|
|
|
att, err := s.AttemptByProviderRef("P-PAY-Q1")
|
|
if err != nil || att.OutTradeNo != "PAY-Q1" || att.Channel != "crypto" {
|
|
t.Fatalf("AttemptByProviderRef = %+v, %v", att, err)
|
|
}
|
|
if _, err := s.AttemptByProviderRef("P-UNKNOWN"); !errors.Is(err, store.ErrAttemptNotFound) {
|
|
t.Fatalf("缺尝试应 ErrAttemptNotFound, got %v", err)
|
|
}
|
|
}
|
|
|
|
func TestListPendingAndExpire(t *testing.T) {
|
|
s := store.NewOrderStore(model.OpenTestDB(t))
|
|
seedOrder(t, s, "PAY-Q2")
|
|
|
|
pending, err := s.ListAttemptsByStatus(model.AttemptPending, 10)
|
|
if err != nil || len(pending) != 1 {
|
|
t.Fatalf("ListAttemptsByStatus pending = %d, %v", len(pending), err)
|
|
}
|
|
|
|
n, err := s.ExpirePendingAttempts("PAY-Q2")
|
|
if err != nil || n != 1 {
|
|
t.Fatalf("ExpirePendingAttempts = %d, %v", n, err)
|
|
}
|
|
pending2, _ := s.ListAttemptsByStatus(model.AttemptPending, 10)
|
|
if len(pending2) != 0 {
|
|
t.Fatalf("弃过期后应无 pending, got %d", len(pending2))
|
|
}
|
|
_ = time.Now
|
|
}
|
|
|
|
func TestExpireStaleOrders(t *testing.T) {
|
|
db := model.OpenTestDB(t)
|
|
s := store.NewOrderStore(db)
|
|
now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)
|
|
|
|
// 三张单:旧 pending(应过期)、新 pending(未到点)、旧 paid(不动)。
|
|
mk := func(no string, st model.OrderStatusV2, createdAgo time.Duration) {
|
|
o := &model.OrderV2{OutTradeNo: no, AmountMinor: 100, Currency: "USDT", Status: st}
|
|
if err := s.CreateOrder(o); err != nil {
|
|
t.Fatalf("create %s: %v", no, err)
|
|
}
|
|
// 直接回填 created_at 制造账龄(GORM 默认写 now)。
|
|
if err := db.Model(&model.OrderV2{}).Where("out_trade_no = ?", no).
|
|
Update("created_at", now.Add(-createdAgo)).Error; err != nil {
|
|
t.Fatalf("backdate %s: %v", no, err)
|
|
}
|
|
}
|
|
mk("STALE", model.OrderPendingV2, 2*time.Hour) // 旧 pending
|
|
mk("FRESH", model.OrderPendingV2, 5*time.Minute) // 新 pending
|
|
mk("PAID", model.OrderPaidV2, 2*time.Hour) // 旧 paid
|
|
|
|
cutoff := now.Add(-time.Hour) // TTL=1h
|
|
n, err := s.ExpireStaleOrders(cutoff, 500)
|
|
if err != nil {
|
|
t.Fatalf("expire: %v", err)
|
|
}
|
|
if n != 1 {
|
|
t.Fatalf("应关闭 1 张(仅 STALE), got %d", n)
|
|
}
|
|
assertStatus := func(no string, want model.OrderStatusV2) {
|
|
o, _ := s.GetOrder(no)
|
|
if o.Status != want {
|
|
t.Fatalf("%s status = %v, want %v", no, o.Status, want)
|
|
}
|
|
}
|
|
assertStatus("STALE", model.OrderExpiredV2)
|
|
assertStatus("FRESH", model.OrderPendingV2)
|
|
assertStatus("PAID", model.OrderPaidV2)
|
|
|
|
// 幂等:再跑一次不再关闭(STALE 已非 pending)。
|
|
if n2, _ := s.ExpireStaleOrders(cutoff, 500); n2 != 0 {
|
|
t.Fatalf("重跑应 0, got %d", n2)
|
|
}
|
|
}
|
|
|
|
func TestSumPaidAttemptMinorByAccountSince(t *testing.T) {
|
|
db := model.OpenTestDB(t)
|
|
s := store.NewOrderStore(db)
|
|
now := time.Date(2026, 7, 10, 12, 0, 0, 0, time.UTC)
|
|
dayStart := time.Date(2026, 7, 10, 0, 0, 0, 0, time.UTC)
|
|
|
|
mk := func(no, acct string, minor int64, st model.AttemptStatus, paidAgo time.Duration) {
|
|
paid := now.Add(-paidAgo)
|
|
a := &model.Attempt{OutTradeNo: no, Channel: "alipay", AccountID: acct, ProviderRef: "R-" + no,
|
|
AmountMinor: minor, Currency: "CNY", Status: st, PaidAt: &paid}
|
|
if err := s.CreateAttempt(a); err != nil {
|
|
t.Fatalf("attempt %s: %v", no, err)
|
|
}
|
|
}
|
|
mk("A", "acct-1", 10000, model.AttemptPaid, 1*time.Hour) // 今日,计入
|
|
mk("B", "acct-1", 5000, model.AttemptPaid, 2*time.Hour) // 今日,计入 → acct-1=15000
|
|
mk("C", "acct-2", 7000, model.AttemptPaid, 30*time.Minute) // 今日 acct-2=7000
|
|
mk("D", "acct-1", 9999, model.AttemptPending, 10*time.Minute) // 未付,不计
|
|
mk("E", "acct-1", 8888, model.AttemptPaid, 20*time.Hour) // 昨天(paid_at < dayStart),不计
|
|
|
|
got, err := s.SumPaidAttemptMinorByAccountSince(dayStart)
|
|
if err != nil {
|
|
t.Fatalf("sum: %v", err)
|
|
}
|
|
if got["acct-1"] != 15000 || got["acct-2"] != 7000 {
|
|
t.Fatalf("聚合 = %+v, want acct-1=15000 acct-2=7000", got)
|
|
}
|
|
}
|