feat(pay-v2): P8 Task5 订阅取消(API+入站)+ past_due 状态

This commit is contained in:
wangjia
2026-07-10 18:18:55 +08:00
parent b8bdb2c193
commit 62abdc1d65
8 changed files with 669 additions and 13 deletions
+25
View File
@@ -105,6 +105,31 @@ func (h *GatewayHandler) GetStatus(c *gin.Context) {
util.RespondSuccess(c, v)
}
// GetSubscription GET /api/v2/subscriptions/:sub_id —— 查订阅状态 + 续费锚点。
func (h *GatewayHandler) GetSubscription(c *gin.Context) {
v, err := h.g.GetSubscription(c.Param("sub_id"))
if err != nil {
util.RespondError(c, http.StatusNotFound, "subscription_not_found", "订阅不存在")
return
}
util.RespondSuccess(c, v)
}
// CancelSubscription POST /api/v2/subscriptions/:sub_id/cancel —— 主动取消:查订阅 → 渠道
// 取消 → 本地翻 canceled + 入队 subscription.canceled(幂等,重复调用 no-op)。
func (h *GatewayHandler) CancelSubscription(c *gin.Context) {
if err := h.g.CancelSubscription(c.Request.Context(), c.Param("sub_id")); err != nil {
if errors.Is(err, store.ErrSubNotFound) {
util.RespondError(c, http.StatusNotFound, "subscription_not_found", "订阅不存在")
return
}
log.Printf("[v2 subscription] 取消失败 sub_id=%s: %v", c.Param("sub_id"), err)
util.RespondError(c, http.StatusInternalServerError, "cancel_failed", "取消失败,请稍后重试")
return
}
util.RespondSuccess(c, gin.H{"canceled": true})
}
type retryRequest struct {
Method string `json:"method"`
}
+143
View File
@@ -248,3 +248,146 @@ func (p *eurFakeProvider) VerifyCallback(_ context.Context, in provider.Callback
func (p *eurFakeProvider) Query(_ context.Context, req provider.QueryRequest) (*provider.PaidEvent, error) {
return nil, errors.New("not implemented")
}
// subFakeProvider 实现 provider.SubscriptionProvider,供 handler 层 CancelSubscription/
// GetSubscription 端到端测试用(独立于 internal/gateway 包的 fakeSubProvider,同构但不能跨
// 测试包复用未导出类型)。VerifyCallback 原样透传测试构造的 provider.PaidEvent JSON。
type subFakeProvider struct {
sessionRef string
}
func (p *subFakeProvider) Method() string { return "subfake" }
func (p *subFakeProvider) Capabilities() provider.Capabilities {
return provider.Capabilities{
RenderTypes: []provider.RenderType{provider.RenderRedirect},
SupportsRecurring: true,
RecurringKind: provider.RecurringKindGatewayScheduled,
SettleCurrencies: []string{"USD"},
Regions: []string{"global"},
}
}
func (p *subFakeProvider) Create(_ context.Context, _ provider.CreateRequest) (*provider.Session, error) {
return nil, errors.New("subFakeProvider: one-time Create not used")
}
func (p *subFakeProvider) CreateSubscriptionCheckout(_ context.Context, req provider.CreateRequest) (*provider.Session, error) {
return &provider.Session{
ProviderRef: p.sessionRef, RenderType: provider.RenderRedirect,
Payload: map[string]any{"url": "https://checkout.example/" + p.sessionRef},
}, nil
}
func (p *subFakeProvider) CancelSubscription(_ context.Context, _ string) error { return nil }
func (p *subFakeProvider) VerifyCallback(_ context.Context, in provider.CallbackInput) (*provider.PaidEvent, error) {
var ev provider.PaidEvent
if err := json.Unmarshal(in.Raw, &ev); err != nil {
return nil, err
}
return &ev, nil
}
func (p *subFakeProvider) Query(_ context.Context, req provider.QueryRequest) (*provider.PaidEvent, error) {
return &provider.PaidEvent{ProviderRef: req.ProviderRef, Status: provider.PaidPending}, nil
}
type subResolver struct{}
func (subResolver) Resolve(sku, currency string) (int64, string, string, error) {
return 2999, "Pro 月付", "pro_monthly", nil
}
// buildSubEngine 装配一套支持订阅的路由(subfake 渠道),供取消/查询端点测试用。
func buildSubEngine(t *testing.T) (*gin.Engine, *subFakeProvider) {
t.Helper()
gin.SetMode(gin.TestMode)
db := model.OpenTestDB(t)
orders := store.NewOrderStore(db)
refunds := store.NewRefundStore(db)
subs := store.NewSubscriptionStore(db)
preg := provider.NewRegistry()
fp := &subFakeProvider{sessionRef: "cs_sub_1"}
preg.Register(fp)
areg := accounts.New([]config.AccountConfig{
{AccountID: "subfake-a1", Channel: "subfake", Region: "global", Enabled: true, Weight: 1},
})
picker := accounts.NewRouter(areg, nil, nil)
g := gateway.New(orders, refunds, preg, picker, subResolver{}, nopEnqueuer{}, "global", subs)
r := gin.New()
router.SetupV2(r, g)
return r, fp
}
// TestV2SubscriptionNotFound404 —— 未知 sub_id 的取消/查询都应 404,而不是 500。
func TestV2SubscriptionNotFound404(t *testing.T) {
r, _ := buildSubEngine(t)
wGet, _ := do(t, r, http.MethodGet, "/api/v2/subscriptions/SUB-GHOST", nil)
if wGet.Code != http.StatusNotFound {
t.Fatalf("get unknown sub code=%d, want 404", wGet.Code)
}
wCancel, _ := do(t, r, http.MethodPost, "/api/v2/subscriptions/SUB-GHOST/cancel", nil)
if wCancel.Code != http.StatusNotFound {
t.Fatalf("cancel unknown sub code=%d, want 404", wCancel.Code)
}
}
// TestV2SubscriptionCancelLifecycle 端到端:建订阅 → 首期支付回调激活 → 查询 active →
// 取消 → 查询 canceled;取消端点重复调用幂等(仍 200,canceled=true)。
func TestV2SubscriptionCancelLifecycle(t *testing.T) {
r, fp := buildSubEngine(t)
w, out := do(t, r, http.MethodPost, "/api/v2/subscriptions", map[string]any{
"sku": "pro_monthly", "method": "subfake", "biz_system": "", "biz_ref": "u-1",
})
if w.Code != http.StatusOK {
t.Fatalf("create subscription code=%d body=%v", w.Code, out)
}
data := out["data"].(map[string]any)
subID := data["sub_id"].(string)
// 激活前:GetSubscription 404(Subscription 行在首期支付回调时才诞生)。
wPre, _ := do(t, r, http.MethodGet, "/api/v2/subscriptions/"+subID, nil)
if wPre.Code != http.StatusNotFound {
t.Fatalf("get before activation code=%d, want 404", wPre.Code)
}
// 首期支付回调激活。
raw, err := json.Marshal(provider.PaidEvent{
Kind: provider.EventPayment, ProviderRef: fp.sessionRef, Status: provider.PaidSucceeded,
PaidAmountMinor: 2999, PaidCurrency: "USD", SubscriptionRef: "sub_h1",
})
if err != nil {
t.Fatalf("marshal event: %v", err)
}
wc, _ := do(t, r, http.MethodPost, "/api/v2/callback/subfake", json.RawMessage(raw))
if wc.Code != http.StatusOK {
t.Fatalf("activate callback code=%d", wc.Code)
}
wActive, outActive := do(t, r, http.MethodGet, "/api/v2/subscriptions/"+subID, nil)
if wActive.Code != http.StatusOK {
t.Fatalf("get after activation code=%d body=%v", wActive.Code, outActive)
}
if outActive["data"].(map[string]any)["status"] != "active" {
t.Fatalf("status = %v, want active", outActive["data"])
}
// 取消。
wCancel, outCancel := do(t, r, http.MethodPost, "/api/v2/subscriptions/"+subID+"/cancel", nil)
if wCancel.Code != http.StatusOK || outCancel["data"].(map[string]any)["canceled"] != true {
t.Fatalf("cancel = %d %v", wCancel.Code, outCancel)
}
wAfter, outAfter := do(t, r, http.MethodGet, "/api/v2/subscriptions/"+subID, nil)
if wAfter.Code != http.StatusOK || outAfter["data"].(map[string]any)["status"] != "canceled" {
t.Fatalf("status after cancel = %d %v", wAfter.Code, outAfter)
}
// 重复取消:幂等 200,不报错。
wCancel2, outCancel2 := do(t, r, http.MethodPost, "/api/v2/subscriptions/"+subID+"/cancel", nil)
if wCancel2.Code != http.StatusOK || outCancel2["data"].(map[string]any)["canceled"] != true {
t.Fatalf("repeat cancel = %d %v", wCancel2.Code, outCancel2)
}
}