wangjia
8051b0fb16
1. CreateOrderInput/RetryOrder 加 Metadata 通道,handler 白名单过滤(is_mobile/render)
后原样传给 provider.CreateRequest;alipay adapter 据 is_mobile 选 wap/page。
2. alipay adapter 移植 v1 当面付(TradePreCreate):Metadata["render"]=="qr" → 二维码
render_type,payload={qr_content,display_amount,currency},默认 2 小时窗口。
3. NewOutTradeNo 随机部分 8→16 hex 防生日碰撞(仍 <=64 字符,合规 DB 列/支付宝上限);
v2 改状态端点(下单/重试/取消)加 per-IP 内存令牌桶限流,默认开 30/min,
config.rate_limit.disabled 可关;callback/GET 查询不限。
顺带修:internal/reconcile/sync_test.go 的 gateway.New 调用漏传 refunds store,
预先存在的编译期回归(与本次改动无关,但挡住 go test ./... 全绿,一并修掉)。
Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_013nMthbVEmQquxBRKb9Fj8u
102 lines
3.5 KiB
Go
102 lines
3.5 KiB
Go
package handler_test
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
|
|
"github.com/wangjia/pay/config"
|
|
"github.com/wangjia/pay/internal/accounts"
|
|
"github.com/wangjia/pay/internal/gateway"
|
|
"github.com/wangjia/pay/internal/model"
|
|
"github.com/wangjia/pay/internal/provider"
|
|
"github.com/wangjia/pay/internal/provider/fake"
|
|
"github.com/wangjia/pay/internal/router"
|
|
"github.com/wangjia/pay/internal/store"
|
|
)
|
|
|
|
// jiu 反馈波 item 3:v2 改状态端点(POST /orders 等)超过 config.RateLimit 设定的
|
|
// per-IP 速率应回 429。config.C 用完整重赋值 + defer 恢复(与 refund_test.go
|
|
// buildRefundEngine 同一套约定),避免污染同包其它测试的全局配置状态。
|
|
func TestV2CreateOrderRateLimited(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
prev := config.C
|
|
config.C = config.Config{RateLimit: config.RateLimitConfig{RequestsPerMin: 2}}
|
|
defer func() { config.C = prev }()
|
|
|
|
db := model.OpenTestDB(t)
|
|
orders := store.NewOrderStore(db)
|
|
refunds := store.NewRefundStore(db)
|
|
preg := provider.NewRegistry()
|
|
preg.Register(fake.New())
|
|
areg := accounts.New([]config.AccountConfig{
|
|
{AccountID: "fake-a1", Channel: "fake", Region: "global", Enabled: true, Weight: 1},
|
|
})
|
|
picker := accounts.NewRouter(areg, nil, nil)
|
|
g := gateway.New(orders, refunds, preg, picker, oneResolver{}, nopEnqueuer{}, "global")
|
|
r := gin.New()
|
|
router.SetupV2(r, g)
|
|
|
|
body := map[string]any{"sku": "pro_year", "method": "fake"}
|
|
var codes []int
|
|
for i := 0; i < 3; i++ {
|
|
w, _ := do(t, r, http.MethodPost, "/api/v2/orders", body)
|
|
codes = append(codes, w.Code)
|
|
}
|
|
if codes[0] != http.StatusOK || codes[1] != http.StatusOK {
|
|
t.Fatalf("前 2 次(= burst=RequestsPerMin)应放行, got codes=%v", codes)
|
|
}
|
|
if codes[2] != http.StatusTooManyRequests {
|
|
t.Fatalf("第 3 次(超 burst)应 429, got codes=%v", codes)
|
|
}
|
|
}
|
|
|
|
// callback 端点(渠道来源,不是终端用户)不挂限流:哪怕下单类速率上限压得很低,
|
|
// 回调也不该被误伤,否则会把渠道异步通知的正常重投当成攻击拦掉。
|
|
func TestV2CallbackNotRateLimited(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
prev := config.C
|
|
config.C = config.Config{RateLimit: config.RateLimitConfig{RequestsPerMin: 1}}
|
|
defer func() { config.C = prev }()
|
|
|
|
r, _ := buildEngineWithStore(t)
|
|
for i := 0; i < 5; i++ {
|
|
w, _ := do(t, r, http.MethodPost, "/api/v2/callback/fake", map[string]any{
|
|
"provider_ref": "GHOST", "status": "succeeded", "amount_minor": 1, "currency": "USDT",
|
|
})
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("callback 第 %d 次不应被限流, code=%d", i+1, w.Code)
|
|
}
|
|
}
|
|
}
|
|
|
|
// disabled:true 应完全放行改状态端点,不管速率配多低。
|
|
func TestV2RateLimitDisabled(t *testing.T) {
|
|
gin.SetMode(gin.TestMode)
|
|
prev := config.C
|
|
config.C = config.Config{RateLimit: config.RateLimitConfig{Disabled: true, RequestsPerMin: 1}}
|
|
defer func() { config.C = prev }()
|
|
|
|
db := model.OpenTestDB(t)
|
|
orders := store.NewOrderStore(db)
|
|
refunds := store.NewRefundStore(db)
|
|
preg := provider.NewRegistry()
|
|
preg.Register(fake.New())
|
|
areg := accounts.New([]config.AccountConfig{
|
|
{AccountID: "fake-a1", Channel: "fake", Region: "global", Enabled: true, Weight: 1},
|
|
})
|
|
picker := accounts.NewRouter(areg, nil, nil)
|
|
g := gateway.New(orders, refunds, preg, picker, oneResolver{}, nopEnqueuer{}, "global")
|
|
r := gin.New()
|
|
router.SetupV2(r, g)
|
|
|
|
body := map[string]any{"sku": "pro_year", "method": "fake"}
|
|
for i := 0; i < 5; i++ {
|
|
w, _ := do(t, r, http.MethodPost, "/api/v2/orders", body)
|
|
if w.Code != http.StatusOK {
|
|
t.Fatalf("disabled=true 时第 %d 次不应被限流, code=%d", i+1, w.Code)
|
|
}
|
|
}
|
|
}
|