huang
e2e6a64ba4
- 创建 internal/service/package/service_test.go 文件 - 测试 Create 方法：创建成功、编码重复失败、系列不存在失败 - 测试 UpdateStatus 方法：禁用时自动强制下架、启用时保持原上架状态 - 测试 UpdateShelfStatus 方法：启用状态可上架、禁用状态不能上架、下架成功 - 测试 Get 方法：获取成功、不存在返回错误 - 测试 List 方法：列表查询、按类型过滤、按状态过滤 - 使用 testutils.NewTestTransaction 创建测试事务 - 使用 middleware.SetUserContext 设置用户上下文 - 使用唯一的 PackageCode（基于时间戳） - 重点覆盖双状态逻辑的测试
373 lines
10 KiB
Go
373 lines
10 KiB
Go
package packagepkg
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/break/junhong_cmp_fiber/internal/model/dto"
|
|
"github.com/break/junhong_cmp_fiber/internal/store/postgres"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/break/junhong_cmp_fiber/pkg/errors"
|
|
"github.com/break/junhong_cmp_fiber/pkg/middleware"
|
|
"github.com/break/junhong_cmp_fiber/tests/testutils"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func generateUniquePackageCode(prefix string) string {
|
|
return fmt.Sprintf("%s_%d", prefix, time.Now().UnixNano())
|
|
}
|
|
|
|
func TestPackageService_Create(t *testing.T) {
|
|
tx := testutils.NewTestTransaction(t)
|
|
packageStore := postgres.NewPackageStore(tx)
|
|
packageSeriesStore := postgres.NewPackageSeriesStore(tx)
|
|
svc := New(packageStore, packageSeriesStore)
|
|
|
|
ctx := middleware.SetUserContext(context.Background(), &middleware.UserContextInfo{
|
|
UserID: 1,
|
|
UserType: constants.UserTypePlatform,
|
|
})
|
|
|
|
t.Run("创建成功", func(t *testing.T) {
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_CREATE"),
|
|
PackageName: "创建测试套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
|
|
resp, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.NotZero(t, resp.ID)
|
|
assert.Equal(t, req.PackageCode, resp.PackageCode)
|
|
assert.Equal(t, req.PackageName, resp.PackageName)
|
|
assert.Equal(t, constants.StatusEnabled, resp.Status)
|
|
assert.Equal(t, 2, resp.ShelfStatus)
|
|
})
|
|
|
|
t.Run("编码重复失败", func(t *testing.T) {
|
|
code := generateUniquePackageCode("PKG_DUP")
|
|
req1 := &dto.CreatePackageRequest{
|
|
PackageCode: code,
|
|
PackageName: "第一个套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
_, err := svc.Create(ctx, req1)
|
|
require.NoError(t, err)
|
|
|
|
req2 := &dto.CreatePackageRequest{
|
|
PackageCode: code,
|
|
PackageName: "第二个套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
_, err = svc.Create(ctx, req2)
|
|
require.Error(t, err)
|
|
appErr, ok := err.(*errors.AppError)
|
|
require.True(t, ok)
|
|
assert.Equal(t, errors.CodeConflict, appErr.Code)
|
|
})
|
|
|
|
t.Run("系列不存在失败", func(t *testing.T) {
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_SERIES"),
|
|
PackageName: "系列测试套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
SeriesID: func() *uint { id := uint(99999); return &id }(),
|
|
}
|
|
|
|
_, err := svc.Create(ctx, req)
|
|
require.Error(t, err)
|
|
appErr, ok := err.(*errors.AppError)
|
|
require.True(t, ok)
|
|
assert.Equal(t, errors.CodeNotFound, appErr.Code)
|
|
})
|
|
}
|
|
|
|
func TestPackageService_UpdateStatus(t *testing.T) {
|
|
tx := testutils.NewTestTransaction(t)
|
|
packageStore := postgres.NewPackageStore(tx)
|
|
packageSeriesStore := postgres.NewPackageSeriesStore(tx)
|
|
svc := New(packageStore, packageSeriesStore)
|
|
|
|
ctx := middleware.SetUserContext(context.Background(), &middleware.UserContextInfo{
|
|
UserID: 1,
|
|
UserType: constants.UserTypePlatform,
|
|
})
|
|
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_STATUS"),
|
|
PackageName: "状态测试套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("禁用套餐时自动强制下架", func(t *testing.T) {
|
|
err := svc.UpdateShelfStatus(ctx, created.ID, 1)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err := svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, pkg.ShelfStatus)
|
|
|
|
err = svc.UpdateStatus(ctx, created.ID, 2)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err = svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 2, pkg.Status)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
})
|
|
|
|
t.Run("启用套餐时保持原上架状态", func(t *testing.T) {
|
|
req2 := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_ENABLE"),
|
|
PackageName: "启用测试套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created2, err := svc.Create(ctx, req2)
|
|
require.NoError(t, err)
|
|
|
|
err = svc.UpdateShelfStatus(ctx, created2.ID, 1)
|
|
require.NoError(t, err)
|
|
|
|
err = svc.UpdateStatus(ctx, created2.ID, 2)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err := svc.Get(ctx, created2.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 2, pkg.Status)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
|
|
err = svc.UpdateStatus(ctx, created2.ID, 1)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err = svc.Get(ctx, created2.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, pkg.Status)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
})
|
|
}
|
|
|
|
func TestPackageService_UpdateShelfStatus(t *testing.T) {
|
|
tx := testutils.NewTestTransaction(t)
|
|
packageStore := postgres.NewPackageStore(tx)
|
|
packageSeriesStore := postgres.NewPackageSeriesStore(tx)
|
|
svc := New(packageStore, packageSeriesStore)
|
|
|
|
ctx := middleware.SetUserContext(context.Background(), &middleware.UserContextInfo{
|
|
UserID: 1,
|
|
UserType: constants.UserTypePlatform,
|
|
})
|
|
|
|
t.Run("启用状态的套餐可以上架", func(t *testing.T) {
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_SHELF_ENABLE"),
|
|
PackageName: "上架测试-启用",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err := svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, pkg.Status)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
|
|
err = svc.UpdateShelfStatus(ctx, created.ID, 1)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err = svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, pkg.ShelfStatus)
|
|
})
|
|
|
|
t.Run("禁用状态的套餐不能上架", func(t *testing.T) {
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_SHELF_DISABLE"),
|
|
PackageName: "上架测试-禁用",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
|
|
err = svc.UpdateStatus(ctx, created.ID, 2)
|
|
require.NoError(t, err)
|
|
|
|
err = svc.UpdateShelfStatus(ctx, created.ID, 1)
|
|
require.Error(t, err)
|
|
appErr, ok := err.(*errors.AppError)
|
|
require.True(t, ok)
|
|
assert.Equal(t, errors.CodeInvalidStatus, appErr.Code)
|
|
|
|
pkg, err := svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
})
|
|
|
|
t.Run("下架成功", func(t *testing.T) {
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_SHELF_OFF"),
|
|
PackageName: "下架测试",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
|
|
err = svc.UpdateShelfStatus(ctx, created.ID, 1)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err := svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 1, pkg.ShelfStatus)
|
|
|
|
err = svc.UpdateShelfStatus(ctx, created.ID, 2)
|
|
require.NoError(t, err)
|
|
|
|
pkg, err = svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, 2, pkg.ShelfStatus)
|
|
})
|
|
}
|
|
|
|
func TestPackageService_Get(t *testing.T) {
|
|
tx := testutils.NewTestTransaction(t)
|
|
packageStore := postgres.NewPackageStore(tx)
|
|
packageSeriesStore := postgres.NewPackageSeriesStore(tx)
|
|
svc := New(packageStore, packageSeriesStore)
|
|
|
|
ctx := middleware.SetUserContext(context.Background(), &middleware.UserContextInfo{
|
|
UserID: 1,
|
|
UserType: constants.UserTypePlatform,
|
|
})
|
|
|
|
req := &dto.CreatePackageRequest{
|
|
PackageCode: generateUniquePackageCode("PKG_GET"),
|
|
PackageName: "查询测试套餐",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
}
|
|
created, err := svc.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
|
|
t.Run("获取成功", func(t *testing.T) {
|
|
resp, err := svc.Get(ctx, created.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, created.PackageCode, resp.PackageCode)
|
|
assert.Equal(t, created.PackageName, resp.PackageName)
|
|
assert.Equal(t, created.ID, resp.ID)
|
|
})
|
|
|
|
t.Run("不存在返回错误", func(t *testing.T) {
|
|
_, err := svc.Get(ctx, 99999)
|
|
require.Error(t, err)
|
|
appErr, ok := err.(*errors.AppError)
|
|
require.True(t, ok)
|
|
assert.Equal(t, errors.CodeNotFound, appErr.Code)
|
|
})
|
|
}
|
|
|
|
func TestPackageService_List(t *testing.T) {
|
|
tx := testutils.NewTestTransaction(t)
|
|
packageStore := postgres.NewPackageStore(tx)
|
|
packageSeriesStore := postgres.NewPackageSeriesStore(tx)
|
|
svc := New(packageStore, packageSeriesStore)
|
|
|
|
ctx := middleware.SetUserContext(context.Background(), &middleware.UserContextInfo{
|
|
UserID: 1,
|
|
UserType: constants.UserTypePlatform,
|
|
})
|
|
|
|
packages := []dto.CreatePackageRequest{
|
|
{
|
|
PackageCode: generateUniquePackageCode("PKG_LIST_001"),
|
|
PackageName: "列表测试套餐1",
|
|
PackageType: "formal",
|
|
DurationMonths: 1,
|
|
Price: 9900,
|
|
},
|
|
{
|
|
PackageCode: generateUniquePackageCode("PKG_LIST_002"),
|
|
PackageName: "列表测试套餐2",
|
|
PackageType: "addon",
|
|
DurationMonths: 1,
|
|
Price: 4900,
|
|
},
|
|
{
|
|
PackageCode: generateUniquePackageCode("PKG_LIST_003"),
|
|
PackageName: "列表测试套餐3",
|
|
PackageType: "formal",
|
|
DurationMonths: 12,
|
|
Price: 99900,
|
|
},
|
|
}
|
|
|
|
for _, p := range packages {
|
|
_, err := svc.Create(ctx, &p)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
t.Run("列表查询", func(t *testing.T) {
|
|
req := &dto.PackageListRequest{
|
|
Page: 1,
|
|
PageSize: 10,
|
|
}
|
|
|
|
resp, total, err := svc.List(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.Greater(t, total, int64(0))
|
|
assert.Greater(t, len(resp), 0)
|
|
})
|
|
|
|
t.Run("按套餐类型过滤", func(t *testing.T) {
|
|
packageType := "formal"
|
|
req := &dto.PackageListRequest{
|
|
Page: 1,
|
|
PageSize: 10,
|
|
PackageType: &packageType,
|
|
}
|
|
|
|
resp, _, err := svc.List(ctx, req)
|
|
require.NoError(t, err)
|
|
for _, p := range resp {
|
|
assert.Equal(t, packageType, p.PackageType)
|
|
}
|
|
})
|
|
|
|
t.Run("按状态过滤", func(t *testing.T) {
|
|
status := constants.StatusEnabled
|
|
req := &dto.PackageListRequest{
|
|
Page: 1,
|
|
PageSize: 10,
|
|
Status: &status,
|
|
}
|
|
|
|
resp, _, err := svc.List(ctx, req)
|
|
require.NoError(t, err)
|
|
for _, p := range resp {
|
|
assert.Equal(t, status, p.Status)
|
|
}
|
|
})
|
|
}
|