wangjia
74 lines
2.2 KiB
Go
74 lines
2.2 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
|
|
}
|
|
|
|
// TestMarkDisputed 覆盖 P8 Task6 打标:条件 UPDATE 只在 disputed=false 时翻转,重投同一
|
|
// dispute(第二次调用)幂等 no-op(rows_affected=0,不报错)。
|
|
func TestMarkDisputed(t *testing.T) {
|
|
s := store.NewOrderStore(model.OpenTestDB(t))
|
|
seedOrder(t, s, "PAY-Q3")
|
|
|
|
flipped, err := s.MarkDisputed("PAY-Q3")
|
|
if err != nil || !flipped {
|
|
t.Fatalf("first MarkDisputed: flipped=%v err=%v", flipped, err)
|
|
}
|
|
o, err := s.GetOrder("PAY-Q3")
|
|
if err != nil || !o.Disputed {
|
|
t.Fatalf("order after MarkDisputed = %+v, %v", o, err)
|
|
}
|
|
|
|
// 重投同一 dispute → 幂等 no-op,不报错。
|
|
again, err := s.MarkDisputed("PAY-Q3")
|
|
if err != nil || again {
|
|
t.Fatalf("dup MarkDisputed: again=%v err=%v", again, err)
|
|
}
|
|
}
|