junhong_cmp_fiber/tests/unit/commission_withdrawal_service_test.go

package unit

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"

	"github.com/break/junhong_cmp_fiber/internal/model/dto"
	"github.com/break/junhong_cmp_fiber/internal/service/commission_withdrawal"
	"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 createWithdrawalTestContext(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 TestCommissionWithdrawalService_ListWithdrawalRequests(t *testing.T) {
	tx := testutils.NewTestTransaction(t)
	rdb := testutils.GetTestRedis(t)
	testutils.CleanTestRedisKeys(t, rdb)

	shopStore := postgres.NewShopStore(tx, rdb)
	accountStore := postgres.NewAccountStore(tx, rdb)
	walletStore := postgres.NewWalletStore(tx, rdb)
	walletTransactionStore := postgres.NewWalletTransactionStore(tx, rdb)
	commissionWithdrawalRequestStore := postgres.NewCommissionWithdrawalRequestStore(tx, rdb)

	service := commission_withdrawal.New(tx, shopStore, accountStore, walletStore, walletTransactionStore, commissionWithdrawalRequestStore)

	t.Run("查询提现申请列表-空结果", func(t *testing.T) {
		ctx := createWithdrawalTestContext(1)

		req := &dto.WithdrawalRequestListReq{
			Page:     1,
			PageSize: 20,
		}

		result, err := service.ListWithdrawalRequests(ctx, req)
		require.NoError(t, err)
		assert.NotNil(t, result)
		assert.GreaterOrEqual(t, result.Total, int64(0))
	})

	t.Run("按状态筛选提现申请", func(t *testing.T) {
		ctx := createWithdrawalTestContext(1)

		status := 1
		req := &dto.WithdrawalRequestListReq{
			Page:     1,
			PageSize: 20,
			Status:   &status,
		}

		result, err := service.ListWithdrawalRequests(ctx, req)
		require.NoError(t, err)
		assert.NotNil(t, result)
	})

	t.Run("按时间范围筛选提现申请", func(t *testing.T) {
		ctx := createWithdrawalTestContext(1)

		req := &dto.WithdrawalRequestListReq{
			Page:      1,
			PageSize:  20,
			StartTime: "2025-01-01 00:00:00",
			EndTime:   "2025-12-31 23:59:59",
		}

		result, err := service.ListWithdrawalRequests(ctx, req)
		require.NoError(t, err)
		assert.NotNil(t, result)
	})
}

func TestCommissionWithdrawalService_Approve(t *testing.T) {
	tx := testutils.NewTestTransaction(t)
	rdb := testutils.GetTestRedis(t)
	testutils.CleanTestRedisKeys(t, rdb)

	shopStore := postgres.NewShopStore(tx, rdb)
	accountStore := postgres.NewAccountStore(tx, rdb)
	walletStore := postgres.NewWalletStore(tx, rdb)
	walletTransactionStore := postgres.NewWalletTransactionStore(tx, rdb)
	commissionWithdrawalRequestStore := postgres.NewCommissionWithdrawalRequestStore(tx, rdb)

	service := commission_withdrawal.New(tx, shopStore, accountStore, walletStore, walletTransactionStore, commissionWithdrawalRequestStore)

	t.Run("审批不存在的提现申请应失败", func(t *testing.T) {
		ctx := createWithdrawalTestContext(1)

		req := &dto.ApproveWithdrawalReq{
			PaymentType: "manual",
		}

		_, err := service.Approve(ctx, 99999, req)
		assert.Error(t, err)
	})
}

func TestCommissionWithdrawalService_Reject(t *testing.T) {
	tx := testutils.NewTestTransaction(t)
	rdb := testutils.GetTestRedis(t)
	testutils.CleanTestRedisKeys(t, rdb)

	shopStore := postgres.NewShopStore(tx, rdb)
	accountStore := postgres.NewAccountStore(tx, rdb)
	walletStore := postgres.NewWalletStore(tx, rdb)
	walletTransactionStore := postgres.NewWalletTransactionStore(tx, rdb)
	commissionWithdrawalRequestStore := postgres.NewCommissionWithdrawalRequestStore(tx, rdb)

	service := commission_withdrawal.New(tx, shopStore, accountStore, walletStore, walletTransactionStore, commissionWithdrawalRequestStore)

	t.Run("拒绝不存在的提现申请应失败", func(t *testing.T) {
		ctx := createWithdrawalTestContext(1)

		req := &dto.RejectWithdrawalReq{
			Remark: "测试拒绝原因",
		}

		_, err := service.Reject(ctx, 99999, req)
		assert.Error(t, err)
	})
}

func TestCommissionWithdrawalService_ConcurrentApproval(t *testing.T) {
	t.Run("并发审批测试-状态检查", func(t *testing.T) {
		assert.True(t, true)
	})
}

func TestCommissionWithdrawalService_InsufficientBalance(t *testing.T) {
	t.Run("余额不足测试", func(t *testing.T) {
		assert.True(t, true)
	})
}