huang
91c9bbfeb8
All checks were successful
构建并部署到测试环境（无 SSH） / build-and-deploy (push) Successful in 4m35s
新增功能： - 店铺佣金查询：店铺佣金统计、店铺佣金记录列表、店铺提现记录 - 佣金提现审批：提现申请列表、审批通过、审批拒绝 - 提现配置管理：配置列表、新增配置、获取当前生效配置 - 企业管理：企业列表、创建、更新、删除、获取详情 - 企业卡授权：授权列表、批量授权、批量取消授权、统计 - 客户账号管理：账号列表、创建、更新状态、重置密码 - 我的佣金：佣金统计、佣金记录、提现申请、提现记录 数据库变更： - 扩展 tb_commission_withdrawal_request 新增提现单号等字段 - 扩展 tb_account 新增 is_primary 字段 - 扩展 tb_commission_record 新增 shop_id、balance_after - 扩展 tb_commission_withdrawal_setting 新增每日提现次数限制 - 扩展 tb_iot_card、tb_device 新增 shop_id 冗余字段 - 新建 tb_enterprise_card_authorization 企业卡授权表 - 新建 tb_asset_allocation_record 资产分配记录表 - 数据迁移：owner_type 枚举值 agent 统一为 shop 测试： - 新增 7 个单元测试文件覆盖各服务 - 修复集成测试 Redis 依赖问题
358 lines
11 KiB
Go
358 lines
11 KiB
Go
package unit
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/break/junhong_cmp_fiber/internal/model"
|
|
"github.com/break/junhong_cmp_fiber/internal/service/enterprise"
|
|
"github.com/break/junhong_cmp_fiber/internal/store/postgres"
|
|
"github.com/break/junhong_cmp_fiber/pkg/constants"
|
|
"github.com/break/junhong_cmp_fiber/tests/testutils"
|
|
)
|
|
|
|
func createEnterpriseTestContext(userID uint) context.Context {
|
|
ctx := context.Background()
|
|
ctx = context.WithValue(ctx, constants.ContextKeyUserID, userID)
|
|
ctx = context.WithValue(ctx, constants.ContextKeyUserType, constants.UserTypePlatform)
|
|
return ctx
|
|
}
|
|
|
|
func TestEnterpriseService_Create(t *testing.T) {
|
|
db, redisClient := testutils.SetupTestDB(t)
|
|
defer testutils.TeardownTestDB(t, db, redisClient)
|
|
|
|
enterpriseStore := postgres.NewEnterpriseStore(db, redisClient)
|
|
shopStore := postgres.NewShopStore(db, redisClient)
|
|
accountStore := postgres.NewAccountStore(db, redisClient)
|
|
|
|
service := enterprise.New(db, enterpriseStore, shopStore, accountStore)
|
|
|
|
t.Run("创建企业-含账号创建", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
req := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "测试企业",
|
|
EnterpriseCode: "ENT_TEST_001",
|
|
ContactName: "测试联系人",
|
|
ContactPhone: "13800000001",
|
|
LoginPhone: "13900000001",
|
|
Password: "Test123456",
|
|
}
|
|
|
|
result, err := service.Create(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.Equal(t, "测试企业", result.Enterprise.EnterpriseName)
|
|
assert.Equal(t, "ENT_TEST_001", result.Enterprise.EnterpriseCode)
|
|
assert.Equal(t, constants.StatusEnabled, result.Enterprise.Status)
|
|
assert.Greater(t, result.AccountID, uint(0))
|
|
})
|
|
|
|
t.Run("创建企业-企业编号已存在应失败", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
req1 := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "企业一",
|
|
EnterpriseCode: "ENT_DUP_001",
|
|
ContactName: "联系人一",
|
|
ContactPhone: "13800000010",
|
|
LoginPhone: "13900000010",
|
|
Password: "Test123456",
|
|
}
|
|
_, err := service.Create(ctx, req1)
|
|
require.NoError(t, err)
|
|
|
|
req2 := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "企业二",
|
|
EnterpriseCode: "ENT_DUP_001",
|
|
ContactName: "联系人二",
|
|
ContactPhone: "13800000011",
|
|
LoginPhone: "13900000011",
|
|
Password: "Test123456",
|
|
}
|
|
_, err = service.Create(ctx, req2)
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("创建企业-手机号已存在应失败", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
req1 := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "企业三",
|
|
EnterpriseCode: "ENT_PHONE_001",
|
|
ContactName: "联系人三",
|
|
ContactPhone: "13800000020",
|
|
LoginPhone: "13900000020",
|
|
Password: "Test123456",
|
|
}
|
|
_, err := service.Create(ctx, req1)
|
|
require.NoError(t, err)
|
|
|
|
req2 := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "企业四",
|
|
EnterpriseCode: "ENT_PHONE_002",
|
|
ContactName: "联系人四",
|
|
ContactPhone: "13800000021",
|
|
LoginPhone: "13900000020",
|
|
Password: "Test123456",
|
|
}
|
|
_, err = service.Create(ctx, req2)
|
|
assert.Error(t, err)
|
|
})
|
|
|
|
t.Run("创建企业-未授权用户应失败", func(t *testing.T) {
|
|
ctx := context.Background()
|
|
|
|
req := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "未授权企业",
|
|
EnterpriseCode: "ENT_UNAUTH_001",
|
|
ContactName: "联系人",
|
|
ContactPhone: "13800000030",
|
|
LoginPhone: "13900000030",
|
|
Password: "Test123456",
|
|
}
|
|
|
|
_, err := service.Create(ctx, req)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestEnterpriseService_Update(t *testing.T) {
|
|
db, redisClient := testutils.SetupTestDB(t)
|
|
defer testutils.TeardownTestDB(t, db, redisClient)
|
|
|
|
enterpriseStore := postgres.NewEnterpriseStore(db, redisClient)
|
|
shopStore := postgres.NewShopStore(db, redisClient)
|
|
accountStore := postgres.NewAccountStore(db, redisClient)
|
|
|
|
service := enterprise.New(db, enterpriseStore, shopStore, accountStore)
|
|
|
|
t.Run("编辑企业", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
createReq := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "待编辑企业",
|
|
EnterpriseCode: "ENT_EDIT_001",
|
|
ContactName: "原联系人",
|
|
ContactPhone: "13800000040",
|
|
LoginPhone: "13900000040",
|
|
Password: "Test123456",
|
|
}
|
|
createResult, err := service.Create(ctx, createReq)
|
|
require.NoError(t, err)
|
|
|
|
newName := "编辑后企业"
|
|
newContact := "新联系人"
|
|
updateReq := &model.UpdateEnterpriseRequest{
|
|
EnterpriseName: &newName,
|
|
ContactName: &newContact,
|
|
}
|
|
|
|
updated, err := service.Update(ctx, createResult.Enterprise.ID, updateReq)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "编辑后企业", updated.EnterpriseName)
|
|
assert.Equal(t, "新联系人", updated.ContactName)
|
|
})
|
|
|
|
t.Run("编辑不存在的企业应失败", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
newName := "不存在企业"
|
|
updateReq := &model.UpdateEnterpriseRequest{
|
|
EnterpriseName: &newName,
|
|
}
|
|
|
|
_, err := service.Update(ctx, 99999, updateReq)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestEnterpriseService_UpdateStatus(t *testing.T) {
|
|
db, redisClient := testutils.SetupTestDB(t)
|
|
defer testutils.TeardownTestDB(t, db, redisClient)
|
|
|
|
enterpriseStore := postgres.NewEnterpriseStore(db, redisClient)
|
|
shopStore := postgres.NewShopStore(db, redisClient)
|
|
accountStore := postgres.NewAccountStore(db, redisClient)
|
|
|
|
service := enterprise.New(db, enterpriseStore, shopStore, accountStore)
|
|
|
|
t.Run("禁用企业-账号同步禁用", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
createReq := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "待禁用企业",
|
|
EnterpriseCode: "ENT_STATUS_001",
|
|
ContactName: "联系人",
|
|
ContactPhone: "13800000050",
|
|
LoginPhone: "13900000050",
|
|
Password: "Test123456",
|
|
}
|
|
createResult, err := service.Create(ctx, createReq)
|
|
require.NoError(t, err)
|
|
|
|
err = service.UpdateStatus(ctx, createResult.Enterprise.ID, constants.StatusDisabled)
|
|
require.NoError(t, err)
|
|
|
|
ent, err := service.GetByID(ctx, createResult.Enterprise.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, constants.StatusDisabled, ent.Status)
|
|
|
|
var account model.Account
|
|
err = db.Where("enterprise_id = ?", createResult.Enterprise.ID).First(&account).Error
|
|
require.NoError(t, err)
|
|
assert.Equal(t, constants.StatusDisabled, account.Status)
|
|
})
|
|
|
|
t.Run("启用企业-账号同步启用", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
createReq := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "待启用企业",
|
|
EnterpriseCode: "ENT_STATUS_002",
|
|
ContactName: "联系人",
|
|
ContactPhone: "13800000051",
|
|
LoginPhone: "13900000051",
|
|
Password: "Test123456",
|
|
}
|
|
createResult, err := service.Create(ctx, createReq)
|
|
require.NoError(t, err)
|
|
|
|
err = service.UpdateStatus(ctx, createResult.Enterprise.ID, constants.StatusDisabled)
|
|
require.NoError(t, err)
|
|
|
|
err = service.UpdateStatus(ctx, createResult.Enterprise.ID, constants.StatusEnabled)
|
|
require.NoError(t, err)
|
|
|
|
ent, err := service.GetByID(ctx, createResult.Enterprise.ID)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, constants.StatusEnabled, ent.Status)
|
|
|
|
var account model.Account
|
|
err = db.Where("enterprise_id = ?", createResult.Enterprise.ID).First(&account).Error
|
|
require.NoError(t, err)
|
|
assert.Equal(t, constants.StatusEnabled, account.Status)
|
|
})
|
|
|
|
t.Run("更新不存在企业状态应失败", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
err := service.UpdateStatus(ctx, 99999, constants.StatusDisabled)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestEnterpriseService_UpdatePassword(t *testing.T) {
|
|
db, redisClient := testutils.SetupTestDB(t)
|
|
defer testutils.TeardownTestDB(t, db, redisClient)
|
|
|
|
enterpriseStore := postgres.NewEnterpriseStore(db, redisClient)
|
|
shopStore := postgres.NewShopStore(db, redisClient)
|
|
accountStore := postgres.NewAccountStore(db, redisClient)
|
|
|
|
service := enterprise.New(db, enterpriseStore, shopStore, accountStore)
|
|
|
|
t.Run("修改企业账号密码", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
createReq := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "密码测试企业",
|
|
EnterpriseCode: "ENT_PWD_001",
|
|
ContactName: "联系人",
|
|
ContactPhone: "13800000060",
|
|
LoginPhone: "13900000060",
|
|
Password: "OldPass123",
|
|
}
|
|
createResult, err := service.Create(ctx, createReq)
|
|
require.NoError(t, err)
|
|
|
|
err = service.UpdatePassword(ctx, createResult.Enterprise.ID, "NewPass456")
|
|
require.NoError(t, err)
|
|
|
|
var account model.Account
|
|
err = db.Where("enterprise_id = ?", createResult.Enterprise.ID).First(&account).Error
|
|
require.NoError(t, err)
|
|
assert.NotEqual(t, "OldPass123", account.Password)
|
|
assert.NotEqual(t, "NewPass456", account.Password)
|
|
})
|
|
|
|
t.Run("修改不存在企业密码应失败", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
err := service.UpdatePassword(ctx, 99999, "NewPass789")
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestEnterpriseService_List(t *testing.T) {
|
|
db, redisClient := testutils.SetupTestDB(t)
|
|
defer testutils.TeardownTestDB(t, db, redisClient)
|
|
|
|
enterpriseStore := postgres.NewEnterpriseStore(db, redisClient)
|
|
shopStore := postgres.NewShopStore(db, redisClient)
|
|
accountStore := postgres.NewAccountStore(db, redisClient)
|
|
|
|
service := enterprise.New(db, enterpriseStore, shopStore, accountStore)
|
|
|
|
t.Run("查询企业列表-空结果", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
req := &model.EnterpriseListReq{
|
|
Page: 1,
|
|
PageSize: 20,
|
|
}
|
|
|
|
result, err := service.List(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.GreaterOrEqual(t, result.Total, int64(0))
|
|
})
|
|
|
|
t.Run("查询企业列表-按名称筛选", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
for i := 0; i < 3; i++ {
|
|
createReq := &model.CreateEnterpriseReq{
|
|
EnterpriseName: "列表测试企业",
|
|
EnterpriseCode: "ENT_LIST_" + string(rune('A'+i)),
|
|
ContactName: "联系人",
|
|
ContactPhone: "1380000007" + string(rune('0'+i)),
|
|
LoginPhone: "1390000007" + string(rune('0'+i)),
|
|
Password: "Test123456",
|
|
}
|
|
_, err := service.Create(ctx, createReq)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
req := &model.EnterpriseListReq{
|
|
Page: 1,
|
|
PageSize: 20,
|
|
EnterpriseName: "列表测试",
|
|
}
|
|
|
|
result, err := service.List(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
assert.GreaterOrEqual(t, result.Total, int64(3))
|
|
})
|
|
|
|
t.Run("查询企业列表-按状态筛选", func(t *testing.T) {
|
|
ctx := createEnterpriseTestContext(1)
|
|
|
|
status := constants.StatusEnabled
|
|
req := &model.EnterpriseListReq{
|
|
Page: 1,
|
|
PageSize: 20,
|
|
Status: &status,
|
|
}
|
|
|
|
result, err := service.List(ctx, req)
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, result)
|
|
})
|
|
}
|