package gateway_test import ( "context" "errors" "testing" "github.com/wangjia/pay/internal/gateway" "github.com/wangjia/pay/internal/model" "github.com/wangjia/pay/internal/provider" ) // 把 fake 订单推到 paid(复用 P2/P3 的 SyncPendingAttempts + SetQueryResult 路径)。 func createAndPay(t *testing.T, g *gateway.Gateway, fp interface { SetQueryResult(string, provider.PaidEvent) }, orders interface { ListAttemptsByStatus(model.AttemptStatus, int) ([]model.Attempt, error) }) string { t.Helper() res, err := g.CreateOrder(context.Background(), gateway.CreateOrderInput{ SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1", }) if err != nil { t.Fatalf("create: %v", err) } atts, _ := orders.ListAttemptsByStatus(model.AttemptPending, 10) fp.SetQueryResult(atts[0].ProviderRef, provider.PaidEvent{ ProviderRef: atts[0].ProviderRef, Status: provider.PaidSucceeded, PaidAmountMinor: 29990000, PaidCurrency: "USDT", }) if _, err := g.SyncPendingAttempts(context.Background(), 10); err != nil { t.Fatalf("sync: %v", err) } return res.OrderNo } func TestRefundPartialThenFull(t *testing.T) { g, fp, spy, orders := newGateway(t) fp.EnableRefund("fake-refund-ref", provider.PaidSucceeded, nil) // fake 变可退渠道,同步成功 no := createAndPay(t, g, fp, orders) // 部分退 1/3(总价 29990000) r1, err := g.Refund(context.Background(), gateway.RefundInput{ OutTradeNo: no, AmountMinor: 9990000, Reason: "test", BizSystem: "pangolin", }) if err != nil || r1.Status != string(model.RefundSucceeded) { t.Fatalf("refund1 = %+v, %v", r1, err) } if o, _ := orders.GetOrder(no); o.Status != model.OrderPartRefundedV2 { t.Fatalf("after partial: order = %s want partially_refunded", o.Status) } // 退款事件已入队(refund.succeeded,带 refund_id) last := spy.calls[len(spy.calls)-1] if last["event_type"] != "refund.succeeded" || last["refund_id"] != r1.RefundID { t.Fatalf("refund webhook = %+v", last) } // 超退守卫:再退 25000000 > 剩余 20000000 → 拒 if _, err := g.Refund(context.Background(), gateway.RefundInput{ OutTradeNo: no, AmountMinor: 25000000, BizSystem: "pangolin", }); !errors.Is(err, gateway.ErrRefundAmountInvalid) { t.Fatalf("over-refund err = %v want ErrRefundAmountInvalid", err) } // 退剩余 → refunded r2, err := g.Refund(context.Background(), gateway.RefundInput{ OutTradeNo: no, AmountMinor: 20000000, BizSystem: "pangolin", }) if err != nil || r2.Status != string(model.RefundSucceeded) { t.Fatalf("refund2 = %+v, %v", r2, err) } if o, _ := orders.GetOrder(no); o.Status != model.OrderRefundedV2 { t.Fatalf("after full: order = %s want refunded", o.Status) } } func TestRefundNotRefundableWhenPending(t *testing.T) { g, _, _, _ := newGateway(t) res, _ := g.CreateOrder(context.Background(), gateway.CreateOrderInput{SKU: "pro_year", Method: "fake"}) if _, err := g.Refund(context.Background(), gateway.RefundInput{OutTradeNo: res.OrderNo, AmountMinor: 1}); !errors.Is(err, gateway.ErrOrderNotRefundable) { t.Fatalf("err = %v want ErrOrderNotRefundable", err) } } func TestRefundManualForCryptoLikeChannel(t *testing.T) { g, fp, spy, orders := newGateway(t) // fp 不 EnableRefund → SupportsRefund=false → 建 manual_pending,不动 order、不发事件 no := createAndPay(t, g, fp, orders) nCalls := len(spy.calls) r, err := g.Refund(context.Background(), gateway.RefundInput{OutTradeNo: no, AmountMinor: 9990000, BizSystem: "pangolin"}) if err != nil || r.Status != string(model.RefundManualPending) { t.Fatalf("manual refund = %+v, %v", r, err) } if o, _ := orders.GetOrder(no); o.Status != model.OrderPaidV2 { t.Fatalf("manual pending 不应动 order,得 %s", o.Status) } if len(spy.calls) != nCalls { t.Fatal("manual pending 不应发退款事件") } // 待办可捞 list, _ := g.ListManualPendingRefunds(10) if len(list) != 1 || list[0].RefundID != r.RefundID { t.Fatalf("manual list = %+v", list) } // 运营回填完成 → 成功落地 + 事件 + 态机 done, err := g.CompleteManualRefund(context.Background(), r.RefundID, "tron-tx-hash") if err != nil || done.Status != string(model.RefundSucceeded) { t.Fatalf("complete = %+v, %v", done, err) } if o, _ := orders.GetOrder(no); o.Status != model.OrderPartRefundedV2 { t.Fatalf("after complete: order = %s want partially_refunded", o.Status) } last := spy.calls[len(spy.calls)-1] if last["event_type"] != "refund.succeeded" || last["provider_refund_ref"] != "tron-tx-hash" { t.Fatalf("complete webhook = %+v", last) } }