package services import ( "context" "database/sql" "encoding/json" "errors" "fmt" "testing" "time" "quyun/v2/app/commands/testx" "quyun/v2/app/errorx" "quyun/v2/app/http/tenant/dto" "quyun/v2/app/requests" "quyun/v2/database" "quyun/v2/database/fields" "quyun/v2/database/models" "quyun/v2/pkg/consts" "github.com/samber/lo" . "github.com/smartystreets/goconvey/convey" "github.com/stretchr/testify/suite" _ "go.ipao.vip/atom" "go.ipao.vip/atom/contracts" "go.ipao.vip/gen/types" "go.uber.org/dig" ) func newLegacyOrderSnapshot() types.JSONType[fields.OrdersSnapshot] { return types.NewJSONType(fields.OrdersSnapshot{ Kind: "legacy", Data: json.RawMessage([]byte("{}")), }) } type OrderTestSuiteInjectParams struct { dig.In DB *sql.DB Initials []contracts.Initial `group:"initials"` // nolint:structcheck } type OrderTestSuite struct { suite.Suite OrderTestSuiteInjectParams } func Test_Order(t *testing.T) { providers := testx.Default().With(Provide) testx.Serve(providers, t, func(p OrderTestSuiteInjectParams) { suite.Run(t, &OrderTestSuite{OrderTestSuiteInjectParams: p}) }) } func (s *OrderTestSuite) truncate(ctx context.Context, tableNames ...string) { database.Truncate(ctx, s.DB, tableNames...) } func (s *OrderTestSuite) seedTenantUser(ctx context.Context, tenantID, userID, balance, frozen int64) { now := time.Now().UTC() _, err := s.DB.ExecContext(ctx, ` INSERT INTO users (id, username, password, roles, status, metas, created_at, updated_at, balance, balance_frozen) VALUES ($1, $2, 'x', ARRAY['user'], $3, '{}'::jsonb, $4, $4, $5, $6) ON CONFLICT (id) DO UPDATE SET balance = EXCLUDED.balance, balance_frozen = EXCLUDED.balance_frozen, updated_at = EXCLUDED.updated_at `, userID, fmt.Sprintf("u%d", userID), consts.UserStatusVerified, now, balance, frozen) So(err, ShouldBeNil) _, err = s.DB.ExecContext(ctx, ` INSERT INTO tenant_users (tenant_id, user_id, role, status, created_at, updated_at) VALUES ($1, $2, ARRAY['member'], $3, $4, $4) ON CONFLICT (tenant_id, user_id) DO UPDATE SET role = EXCLUDED.role, status = EXCLUDED.status, updated_at = EXCLUDED.updated_at `, tenantID, userID, consts.UserStatusVerified, now) So(err, ShouldBeNil) } func (s *OrderTestSuite) seedPublishedContent(ctx context.Context, tenantID, ownerUserID int64) *models.Content { m := &models.Content{ TenantID: tenantID, UserID: ownerUserID, Title: "标题", Description: "描述", Status: consts.ContentStatusPublished, Visibility: consts.ContentVisibilityTenantOnly, PreviewSeconds: consts.DefaultContentPreviewSeconds, PreviewDownloadable: false, PublishedAt: time.Now().UTC(), } So(m.Create(ctx), ShouldBeNil) return m } func (s *OrderTestSuite) seedContentPrice(ctx context.Context, tenantID, contentID, priceAmount int64) { p := &models.ContentPrice{ TenantID: tenantID, UserID: 1, ContentID: contentID, Currency: consts.CurrencyCNY, PriceAmount: priceAmount, DiscountType: consts.DiscountTypeNone, DiscountValue: 0, DiscountStartAt: time.Time{}, DiscountEndAt: time.Time{}, } So(p.Create(ctx), ShouldBeNil) } func (s *OrderTestSuite) Test_MyOrderPage() { Convey("Order.MyOrderPage", s.T(), func() { ctx := s.T().Context() now := time.Now().UTC() tenantID := int64(1) userID := int64(2) s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) Convey("参数非法应返回错误", func() { _, err := Order.MyOrderPage(ctx, 0, userID, &dto.MyOrderListFilter{}) So(err, ShouldNotBeNil) }) Convey("空数据应返回 total=0", func() { pager, err := Order.MyOrderPage(ctx, tenantID, userID, &dto.MyOrderListFilter{}) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 0) }) Convey("按 content_id 过滤", func() { o1 := &models.Order{ TenantID: tenantID, UserID: userID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o1.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: userID, OrderID: o1.ID, ContentID: 111, ContentUserID: 1, AmountPaid: 100, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: userID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 200, Snapshot: newLegacyOrderSnapshot(), PaidAt: now.Add(time.Minute), CreatedAt: now.Add(time.Minute), UpdatedAt: now.Add(time.Minute), } So(o2.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: userID, OrderID: o2.ID, ContentID: 222, ContentUserID: 1, AmountPaid: 200, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now.Add(time.Minute), UpdatedAt: now.Add(time.Minute), }).Create(ctx), ShouldBeNil) pager, err := Order.MyOrderPage(ctx, tenantID, userID, &dto.MyOrderListFilter{ ContentID: lo.ToPtr(int64(111)), }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) }) } func (s *OrderTestSuite) Test_MyOrderDetail() { Convey("Order.MyOrderDetail", s.T(), func() { ctx := s.T().Context() tenantID := int64(1) userID := int64(2) s.truncate(ctx, models.TableNameOrder, models.TableNameOrderItem) Convey("参数非法应返回错误", func() { _, err := Order.MyOrderDetail(ctx, 0, userID, 1) So(err, ShouldNotBeNil) }) Convey("订单不存在应返回错误", func() { _, err := Order.MyOrderDetail(ctx, tenantID, userID, 999) So(err, ShouldNotBeNil) }) }) } func (s *OrderTestSuite) Test_AdminOrderPage() { Convey("Order.AdminOrderPage", s.T(), func() { ctx := s.T().Context() now := time.Now().UTC() tenantID := int64(1) s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) Convey("参数非法应返回错误", func() { _, err := Order.AdminOrderPage(ctx, 0, &dto.AdminOrderListFilter{}) So(err, ShouldNotBeNil) }) Convey("空数据应返回 total=0", func() { pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{}) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 0) }) Convey("按 paid_at 时间窗过滤", func() { o1 := &models.Order{ TenantID: tenantID, UserID: 2, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now.Add(-time.Hour), CreatedAt: now.Add(-time.Hour), UpdatedAt: now.Add(-time.Hour), } So(o1.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 2, OrderID: o1.ID, ContentID: 333, ContentUserID: 1, AmountPaid: 100, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now.Add(-time.Hour), UpdatedAt: now.Add(-time.Hour), }).Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: 3, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 200, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o2.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 3, OrderID: o2.ID, ContentID: 444, ContentUserID: 1, AmountPaid: 200, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) from := now.Add(-10 * time.Minute) to := now.Add(10 * time.Minute) pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ PaidAtFrom: &from, PaidAtTo: &to, }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) Convey("按 content_id 过滤", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) o1 := &models.Order{ TenantID: tenantID, UserID: 2, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o1.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 2, OrderID: o1.ID, ContentID: 555, ContentUserID: 1, AmountPaid: 100, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ ContentID: lo.ToPtr(int64(555)), }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) Convey("按 username 关键字过滤", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder, models.TableNameUser) u1 := &models.User{ Username: "alice", Password: "x", Roles: types.NewArray([]consts.Role{consts.RoleUser}), Status: consts.UserStatusVerified, Metas: types.JSON([]byte("{}")), CreatedAt: now, UpdatedAt: now, } So(u1.Create(ctx), ShouldBeNil) u2 := &models.User{ Username: "bob", Password: "x", Roles: types.NewArray([]consts.Role{consts.RoleUser}), Status: consts.UserStatusVerified, Metas: types.JSON([]byte("{}")), CreatedAt: now, UpdatedAt: now, } So(u2.Create(ctx), ShouldBeNil) o1 := &models.Order{ TenantID: tenantID, UserID: u1.ID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o1.Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: u2.ID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o2.Create(ctx), ShouldBeNil) username := "ali" pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ Username: &username, }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) items, ok := pager.Items.([]*models.Order) So(ok, ShouldBeTrue) So(items[0].UserID, ShouldEqual, u1.ID) }) Convey("按 content_title 关键字过滤", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder, models.TableNameContent) c1 := &models.Content{ TenantID: tenantID, UserID: 1, Title: "Go 教程", Description: "desc", Status: consts.ContentStatusPublished, Visibility: consts.ContentVisibilityTenantOnly, PreviewSeconds: consts.DefaultContentPreviewSeconds, PreviewDownloadable: false, PublishedAt: now, CreatedAt: now, UpdatedAt: now, } So(c1.Create(ctx), ShouldBeNil) c2 := &models.Content{ TenantID: tenantID, UserID: 1, Title: "Rust 教程", Description: "desc", Status: consts.ContentStatusPublished, Visibility: consts.ContentVisibilityTenantOnly, PreviewSeconds: consts.DefaultContentPreviewSeconds, PreviewDownloadable: false, PublishedAt: now, CreatedAt: now, UpdatedAt: now, } So(c2.Create(ctx), ShouldBeNil) o1 := &models.Order{ TenantID: tenantID, UserID: 2, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o1.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 2, OrderID: o1.ID, ContentID: c1.ID, ContentUserID: 1, AmountPaid: 100, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: 3, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o2.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 3, OrderID: o2.ID, ContentID: c2.ID, ContentUserID: 1, AmountPaid: 100, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) title := "Go" pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ ContentTitle: &title, }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) Convey("按 created_at 时间窗过滤", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) o1 := &models.Order{ TenantID: tenantID, UserID: 2, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now.Add(-time.Hour), UpdatedAt: now.Add(-time.Hour), } So(o1.Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: 3, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 200, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o2.Create(ctx), ShouldBeNil) from := now.Add(-10 * time.Minute) to := now.Add(10 * time.Minute) pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ CreatedAtFrom: &from, CreatedAtTo: &to, }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) Convey("按排序字段(asc/desc)排序(白名单)", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) o1 := &models.Order{ TenantID: tenantID, UserID: 2, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 500, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now.Add(-time.Hour), UpdatedAt: now.Add(-time.Hour), } So(o1.Create(ctx), ShouldBeNil) o2 := &models.Order{ TenantID: tenantID, UserID: 3, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o2.Create(ctx), ShouldBeNil) asc := "amount_paid" pagerAsc, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ SortQueryFilter: requests.SortQueryFilter{Asc: &asc}, }) So(err, ShouldBeNil) So(pagerAsc.Total, ShouldEqual, 2) itemsAsc, ok := pagerAsc.Items.([]*models.Order) So(ok, ShouldBeTrue) So(itemsAsc[0].AmountPaid, ShouldEqual, 100) desc := "created_at" pagerDesc, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ SortQueryFilter: requests.SortQueryFilter{Desc: &desc}, }) So(err, ShouldBeNil) So(pagerDesc.Total, ShouldEqual, 2) itemsDesc, ok := pagerDesc.Items.([]*models.Order) So(ok, ShouldBeTrue) So(itemsDesc[0].CreatedAt.After(itemsDesc[1].CreatedAt), ShouldBeTrue) }) Convey("组合筛选:user_id + status + amount_paid 区间 + content_id", func() { s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder) statusPaid := consts.OrderStatusPaid userID := int64(7) contentID := int64(777) // 命中:user_id=7, status=paid, amount_paid=500, content_id=777 oHit := &models.Order{ TenantID: tenantID, UserID: userID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 500, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(oHit.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: userID, OrderID: oHit.ID, ContentID: contentID, ContentUserID: 1, AmountPaid: 500, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) // 不命中:amount_paid 不在区间 oNoAmount := &models.Order{ TenantID: tenantID, UserID: userID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 50, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(oNoAmount.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: userID, OrderID: oNoAmount.ID, ContentID: contentID, ContentUserID: 1, AmountPaid: 50, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) // 不命中:status 不同 oNoStatus := &models.Order{ TenantID: tenantID, UserID: userID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusCreated, Currency: consts.CurrencyCNY, AmountPaid: 500, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(oNoStatus.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: userID, OrderID: oNoStatus.ID, ContentID: contentID, ContentUserID: 1, AmountPaid: 500, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) // 不命中:user_id 不同 oNoUser := &models.Order{ TenantID: tenantID, UserID: 8, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 500, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(oNoUser.Create(ctx), ShouldBeNil) So((&models.OrderItem{ TenantID: tenantID, UserID: 8, OrderID: oNoUser.ID, ContentID: contentID, ContentUserID: 1, AmountPaid: 500, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, }).Create(ctx), ShouldBeNil) min := int64(100) max := int64(900) pager, err := Order.AdminOrderPage(ctx, tenantID, &dto.AdminOrderListFilter{ UserID: &userID, Status: &statusPaid, ContentID: &contentID, AmountPaidMin: &min, AmountPaidMax: &max, }) So(err, ShouldBeNil) So(pager.Total, ShouldEqual, 1) }) }) } func (s *OrderTestSuite) Test_AdminOrderDetail() { Convey("Order.AdminOrderDetail", s.T(), func() { ctx := s.T().Context() tenantID := int64(1) s.truncate(ctx, models.TableNameOrder, models.TableNameOrderItem) Convey("参数非法应返回错误", func() { _, err := Order.AdminOrderDetail(ctx, 0, 1) So(err, ShouldNotBeNil) }) Convey("订单不存在应返回错误", func() { _, err := Order.AdminOrderDetail(ctx, tenantID, 999) So(err, ShouldNotBeNil) }) }) } func (s *OrderTestSuite) Test_AdminOrderExportCSV() { Convey("Order.AdminOrderExportCSV", s.T(), func() { ctx := s.T().Context() now := time.Now().UTC() tenantID := int64(1) s.truncate(ctx, models.TableNameOrderItem, models.TableNameOrder, models.TableNameUser, models.TableNameContent) Convey("参数非法应返回错误", func() { _, err := Order.AdminOrderExportCSV(ctx, 0, &dto.AdminOrderListFilter{}) So(err, ShouldNotBeNil) }) Convey("导出应返回 CSV 且包含表头", func() { u := &models.User{ Username: "alice", Password: "x", Roles: types.NewArray([]consts.Role{consts.RoleUser}), Status: consts.UserStatusVerified, Metas: types.JSON([]byte("{}")), CreatedAt: now, UpdatedAt: now, } So(u.Create(ctx), ShouldBeNil) o := &models.Order{ TenantID: tenantID, UserID: u.ID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountPaid: 123, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(o.Create(ctx), ShouldBeNil) resp, err := Order.AdminOrderExportCSV(ctx, tenantID, &dto.AdminOrderListFilter{}) So(err, ShouldBeNil) So(resp, ShouldNotBeNil) So(resp.ContentType, ShouldEqual, "text/csv") So(resp.Filename, ShouldContainSubstring, "tenant_1_orders_") So(resp.CSV, ShouldContainSubstring, "id,tenant_id,user_id,type,status,amount_paid,paid_at,created_at") So(resp.CSV, ShouldContainSubstring, "content_purchase") }) }) } func (s *OrderTestSuite) Test_AdminRefundOrder() { Convey("Order.AdminRefundOrder", s.T(), func() { ctx := s.T().Context() now := time.Now().UTC() tenantID := int64(1) operatorUserID := int64(10) buyerUserID := int64(20) s.truncate( ctx, models.TableNameTenantLedger, models.TableNameContentAccess, models.TableNameOrderItem, models.TableNameOrder, models.TableNameTenantUser, models.TableNameUser, ) Convey("参数非法应返回错误", func() { _, err := Order.AdminRefundOrder(ctx, 0, operatorUserID, 1, false, "", "", now) So(err, ShouldNotBeNil) }) Convey("订单非已支付状态应返回状态冲突", func() { s.seedTenantUser(ctx, tenantID, buyerUserID, 0, 0) orderModel := &models.Order{ TenantID: tenantID, UserID: buyerUserID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusCreated, Currency: consts.CurrencyCNY, AmountOriginal: 100, AmountDiscount: 0, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(orderModel.Create(ctx), ShouldBeNil) _, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因", "", now) So(err, ShouldNotBeNil) var appErr *errorx.AppError So(errors.As(err, &appErr), ShouldBeTrue) So(appErr.Code, ShouldEqual, errorx.CodeStatusConflict) }) Convey("已超过默认退款时间窗且非强制应失败", func() { s.seedTenantUser(ctx, tenantID, buyerUserID, 0, 0) orderModel := &models.Order{ TenantID: tenantID, UserID: buyerUserID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountOriginal: 100, AmountDiscount: 0, AmountPaid: 100, Snapshot: newLegacyOrderSnapshot(), PaidAt: now.Add(-consts.DefaultOrderRefundWindow).Add(-time.Second), CreatedAt: now, UpdatedAt: now, } So(orderModel.Create(ctx), ShouldBeNil) _, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因", "", now) So(err, ShouldNotBeNil) }) Convey("成功退款应回收权益并入账", func() { s.seedTenantUser(ctx, tenantID, buyerUserID, 0, 0) contentID := int64(123) orderModel := &models.Order{ TenantID: tenantID, UserID: buyerUserID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountOriginal: 300, AmountDiscount: 0, AmountPaid: 300, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(orderModel.Create(ctx), ShouldBeNil) item := &models.OrderItem{ TenantID: tenantID, UserID: buyerUserID, OrderID: orderModel.ID, ContentID: contentID, ContentUserID: 999, AmountPaid: 300, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, } So(item.Create(ctx), ShouldBeNil) access := &models.ContentAccess{ TenantID: tenantID, UserID: buyerUserID, ContentID: contentID, OrderID: orderModel.ID, Status: consts.ContentAccessStatusActive, CreatedAt: now, UpdatedAt: now, RevokedAt: time.Time{}, } So(access.Create(ctx), ShouldBeNil) refunding, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因", "", now.Add(time.Minute)) So(err, ShouldBeNil) So(refunding, ShouldNotBeNil) So(refunding.Status, ShouldEqual, consts.OrderStatusRefunding) // refunding 期间重复请求应幂等返回 refunding(并允许重复触发入队,不影响最终结果)。 refunding2, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因2", "", now.Add(90*time.Second)) So(err, ShouldBeNil) So(refunding2, ShouldNotBeNil) So(refunding2.Status, ShouldEqual, consts.OrderStatusRefunding) refunded, err := Order.ProcessRefundingOrder(ctx, &ProcessRefundingOrderParams{ TenantID: tenantID, OrderID: orderModel.ID, OperatorUserID: operatorUserID, Force: false, Reason: "原因", Now: now.Add(2 * time.Minute), }) So(err, ShouldBeNil) So(refunded, ShouldNotBeNil) So(refunded.Status, ShouldEqual, consts.OrderStatusRefunded) // worker 重试/重复执行应幂等:不重复入账、不重复回收权益。 refundedRetry, err := Order.ProcessRefundingOrder(ctx, &ProcessRefundingOrderParams{ TenantID: tenantID, OrderID: orderModel.ID, OperatorUserID: operatorUserID, Force: false, Reason: "原因", Now: now.Add(5 * time.Minute), }) So(err, ShouldBeNil) So(refundedRetry, ShouldNotBeNil) So(refundedRetry.Status, ShouldEqual, consts.OrderStatusRefunded) var u models.User So(_db.WithContext(ctx).Where("id = ?", buyerUserID).First(&u).Error, ShouldBeNil) So(u.Balance, ShouldEqual, 300) var access2 models.ContentAccess So(_db.WithContext(ctx).Where("tenant_id = ? AND user_id = ? AND content_id = ?", tenantID, buyerUserID, contentID).First(&access2).Error, ShouldBeNil) So(access2.Status, ShouldEqual, consts.ContentAccessStatusRevoked) So(access2.RevokedAt.IsZero(), ShouldBeFalse) refunded2, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因2", "", now.Add(3*time.Minute)) So(err, ShouldBeNil) So(refunded2.Status, ShouldEqual, consts.OrderStatusRefunded) var u2 models.User So(_db.WithContext(ctx).Where("id = ?", buyerUserID).First(&u2).Error, ShouldBeNil) So(u2.Balance, ShouldEqual, 300) var ledgers []*models.TenantLedger So(_db.WithContext(ctx). Where("tenant_id = ? AND user_id = ? AND idempotency_key = ?", tenantID, buyerUserID, fmt.Sprintf("refund:%d", orderModel.ID)). Find(&ledgers).Error, ShouldBeNil) So(len(ledgers), ShouldEqual, 1) }) Convey("failed 状态允许重新发起退款(paid/failed -> refunding)", func() { s.truncate( ctx, models.TableNameTenantLedger, models.TableNameContentAccess, models.TableNameOrderItem, models.TableNameOrder, models.TableNameTenantUser, models.TableNameUser, ) s.seedTenantUser(ctx, tenantID, buyerUserID, 0, 0) contentID := int64(123) orderModel := &models.Order{ TenantID: tenantID, UserID: buyerUserID, Type: consts.OrderTypeContentPurchase, Status: consts.OrderStatusPaid, Currency: consts.CurrencyCNY, AmountOriginal: 300, AmountDiscount: 0, AmountPaid: 300, Snapshot: newLegacyOrderSnapshot(), PaidAt: now, CreatedAt: now, UpdatedAt: now, } So(orderModel.Create(ctx), ShouldBeNil) item := &models.OrderItem{ TenantID: tenantID, UserID: buyerUserID, OrderID: orderModel.ID, ContentID: contentID, ContentUserID: 999, AmountPaid: 300, Snapshot: types.NewJSONType(fields.OrderItemsSnapshot{}), CreatedAt: now, UpdatedAt: now, } So(item.Create(ctx), ShouldBeNil) access := &models.ContentAccess{ TenantID: tenantID, UserID: buyerUserID, ContentID: contentID, OrderID: orderModel.ID, Status: consts.ContentAccessStatusActive, CreatedAt: now, UpdatedAt: now, RevokedAt: time.Time{}, } So(access.Create(ctx), ShouldBeNil) // 先发起一次退款进入 refunding,再模拟异步失败进入 failed。 refunding, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因", "", now.Add(time.Minute)) So(err, ShouldBeNil) So(refunding.Status, ShouldEqual, consts.OrderStatusRefunding) So(Order.MarkRefundFailed(ctx, tenantID, orderModel.ID, now.Add(2*time.Minute)), ShouldBeNil) var failed models.Order So(_db.WithContext(ctx).Where("tenant_id = ? AND id = ?", tenantID, orderModel.ID).First(&failed).Error, ShouldBeNil) So(failed.Status, ShouldEqual, consts.OrderStatusFailed) // failed -> refunding 允许重新发起,并再次入队(幂等)。 refunding2, err := Order.AdminRefundOrder(ctx, tenantID, operatorUserID, orderModel.ID, false, "原因2", "", now.Add(3*time.Minute)) So(err, ShouldBeNil) So(refunding2.Status, ShouldEqual, consts.OrderStatusRefunding) }) Convey("不可重试错误分类应稳定", func() { So(IsRefundJobNonRetryableError(nil), ShouldBeFalse) So(IsRefundJobNonRetryableError(errors.New("x")), ShouldBeFalse) So(IsRefundJobNonRetryableError(errorx.ErrInvalidParameter), ShouldBeTrue) So(IsRefundJobNonRetryableError(errorx.ErrRecordNotFound), ShouldBeTrue) So(IsRefundJobNonRetryableError(errorx.ErrStatusConflict), ShouldBeTrue) So(IsRefundJobNonRetryableError(errorx.ErrPreconditionFailed), ShouldBeTrue) So(IsRefundJobNonRetryableError(errorx.ErrPermissionDenied), ShouldBeTrue) So(IsRefundJobNonRetryableError(errorx.ErrInternalError), ShouldBeFalse) }) }) } func (s *OrderTestSuite) Test_PurchaseContent() { Convey("Order.PurchaseContent", s.T(), func() { ctx := s.T().Context() now := time.Now().UTC() tenantID := int64(1) ownerUserID := int64(100) buyerUserID := int64(200) s.truncate( ctx, models.TableNameTenantLedger, models.TableNameContentAccess, models.TableNameOrderItem, models.TableNameOrder, models.TableNameContentPrice, models.TableNameContent, models.TableNameTenantUser, ) Convey("参数非法应返回错误", func() { _, err := Order.PurchaseContent(ctx, nil) So(err, ShouldNotBeNil) _, err = Order.PurchaseContent(ctx, &PurchaseContentParams{TenantID: 0, UserID: 1, ContentID: 1}) So(err, ShouldNotBeNil) }) Convey("内容未发布应返回前置条件失败", func() { s.seedTenantUser(ctx, tenantID, buyerUserID, 1000, 0) content := &models.Content{ TenantID: tenantID, UserID: ownerUserID, Title: "标题", Description: "描述", Status: consts.ContentStatusDraft, Visibility: consts.ContentVisibilityTenantOnly, PreviewSeconds: consts.DefaultContentPreviewSeconds, PreviewDownloadable: false, } So(content.Create(ctx), ShouldBeNil) _, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_not_published", Now: now, }) So(err, ShouldNotBeNil) var appErr *errorx.AppError So(errors.As(err, &appErr), ShouldBeTrue) So(appErr.Code, ShouldEqual, errorx.CodePreconditionFailed) }) Convey("免费内容购买应创建订单并授予权益(幂等)", func() { s.seedTenantUser(ctx, tenantID, buyerUserID, 1000, 0) content := s.seedPublishedContent(ctx, tenantID, ownerUserID) res1, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_free_1", Now: now, }) So(err, ShouldBeNil) So(res1, ShouldNotBeNil) So(res1.AmountPaid, ShouldEqual, 0) So(res1.Order, ShouldNotBeNil) So(res1.Order.Status, ShouldEqual, consts.OrderStatusPaid) So(res1.Access, ShouldNotBeNil) So(res1.Access.Status, ShouldEqual, consts.ContentAccessStatusActive) snap := res1.Order.Snapshot.Data() So(snap.Kind, ShouldEqual, string(consts.OrderTypeContentPurchase)) var snapData fields.OrdersContentPurchaseSnapshot So(json.Unmarshal(snap.Data, &snapData), ShouldBeNil) So(snapData.ContentID, ShouldEqual, content.ID) So(snapData.ContentTitle, ShouldEqual, content.Title) So(snapData.AmountPaid, ShouldEqual, int64(0)) res2, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_free_1", Now: now.Add(time.Second), }) So(err, ShouldBeNil) So(res2.Order.ID, ShouldEqual, res1.Order.ID) }) Convey("付费内容购买应冻结+扣款并授予权益(幂等)", func() { s.truncate( ctx, models.TableNameTenantLedger, models.TableNameContentAccess, models.TableNameOrderItem, models.TableNameOrder, models.TableNameContentPrice, models.TableNameContent, models.TableNameTenantUser, models.TableNameUser, ) s.seedTenantUser(ctx, tenantID, buyerUserID, 1000, 0) content := s.seedPublishedContent(ctx, tenantID, ownerUserID) s.seedContentPrice(ctx, tenantID, content.ID, 300) res1, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_paid_1", Now: now, }) So(err, ShouldBeNil) So(res1, ShouldNotBeNil) So(res1.AmountPaid, ShouldEqual, 300) So(res1.Order, ShouldNotBeNil) So(res1.Order.Status, ShouldEqual, consts.OrderStatusPaid) So(res1.Access, ShouldNotBeNil) So(res1.Access.Status, ShouldEqual, consts.ContentAccessStatusActive) snap := res1.Order.Snapshot.Data() So(snap.Kind, ShouldEqual, string(consts.OrderTypeContentPurchase)) var snapData fields.OrdersContentPurchaseSnapshot So(json.Unmarshal(snap.Data, &snapData), ShouldBeNil) So(snapData.ContentID, ShouldEqual, content.ID) So(snapData.AmountPaid, ShouldEqual, int64(300)) So(snapData.AmountOriginal, ShouldEqual, int64(300)) itemSnap := res1.OrderItem.Snapshot.Data() So(itemSnap.ContentID, ShouldEqual, content.ID) So(itemSnap.AmountPaid, ShouldEqual, int64(300)) var u models.User So(_db.WithContext(ctx).Where("id = ?", buyerUserID).First(&u).Error, ShouldBeNil) So(u.Balance, ShouldEqual, 700) So(u.BalanceFrozen, ShouldEqual, 0) res2, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_paid_1", Now: now.Add(2 * time.Second), }) So(err, ShouldBeNil) So(res2.Order.ID, ShouldEqual, res1.Order.ID) var u2 models.User So(_db.WithContext(ctx).Where("id = ?", buyerUserID).First(&u2).Error, ShouldBeNil) So(u2.Balance, ShouldEqual, 700) So(u2.BalanceFrozen, ShouldEqual, 0) }) Convey("存在回滚标记时应稳定返回“失败+已回滚”", func() { s.truncate( ctx, models.TableNameTenantLedger, models.TableNameContentAccess, models.TableNameOrderItem, models.TableNameOrder, models.TableNameContentPrice, models.TableNameContent, models.TableNameTenantUser, models.TableNameUser, ) s.seedTenantUser(ctx, tenantID, buyerUserID, 1000, 0) content := s.seedPublishedContent(ctx, tenantID, ownerUserID) s.seedContentPrice(ctx, tenantID, content.ID, 300) rollbackKey := "idem_rollback_1:rollback" ledger := &models.TenantLedger{ TenantID: tenantID, UserID: buyerUserID, OrderID: 0, Type: consts.TenantLedgerTypeUnfreeze, Amount: 1, BalanceBefore: 0, BalanceAfter: 0, FrozenBefore: 0, FrozenAfter: 0, IdempotencyKey: rollbackKey, Remark: "rollback marker", CreatedAt: now, UpdatedAt: now, } So(ledger.Create(ctx), ShouldBeNil) _, err := Order.PurchaseContent(ctx, &PurchaseContentParams{ TenantID: tenantID, UserID: buyerUserID, ContentID: content.ID, IdempotencyKey: "idem_rollback_1", Now: now.Add(time.Second), }) So(err, ShouldNotBeNil) So(err.Error(), ShouldContainSubstring, "失败+已回滚") }) }) }