wangjia
45 lines
1.4 KiB
Go
45 lines
1.4 KiB
Go
package store_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/wangjia/pay/internal/model"
|
|
"github.com/wangjia/pay/internal/store"
|
|
)
|
|
|
|
func TestWebhookOutboxEnqueueIdempotent(t *testing.T) {
|
|
ws := store.NewWebhookStore(model.OpenTestDB(t))
|
|
|
|
if err := ws.EnqueueDelivery("PAY-1", "pangolin", "payment.succeeded", `{"a":1}`); err != nil {
|
|
t.Fatalf("enqueue#1: %v", err)
|
|
}
|
|
// 幂等:同 (out_trade_no,event_type) 再入队不新增行、不报错。
|
|
if err := ws.EnqueueDelivery("PAY-1", "pangolin", "payment.succeeded", `{"a":1}`); err != nil {
|
|
t.Fatalf("enqueue#2: %v", err)
|
|
}
|
|
list, _ := ws.ListUndelivered(10)
|
|
if len(list) != 1 {
|
|
t.Fatalf("应恰 1 行待投递, got %d", len(list))
|
|
}
|
|
|
|
if err := ws.MarkDelivered(list[0].ID); err != nil {
|
|
t.Fatalf("markDelivered: %v", err)
|
|
}
|
|
if again, _ := ws.ListUndelivered(10); len(again) != 0 {
|
|
t.Fatalf("投递后应 0 待投递, got %d", len(again))
|
|
}
|
|
}
|
|
|
|
func TestWebhookMarkFailed(t *testing.T) {
|
|
ws := store.NewWebhookStore(model.OpenTestDB(t))
|
|
_ = ws.EnqueueDelivery("PAY-2", "jiu", "payment.succeeded", `{}`)
|
|
list, _ := ws.ListUndelivered(10)
|
|
if err := ws.MarkFailed(list[0].ID, "boom"); err != nil {
|
|
t.Fatalf("markFailed: %v", err)
|
|
}
|
|
again, _ := ws.ListUndelivered(10)
|
|
if len(again) != 1 || again[0].Attempts != 1 || again[0].LastError != "boom" {
|
|
t.Fatalf("失败后应仍待投递且 attempts=1, got %+v", again)
|
|
}
|
|
}
|