hfb_sys/backend/internal/modules/withdrawal/repository_integration_test.go

package withdrawal

import (
	"context"
	"errors"
	"testing"

	"hfb_sys/backend/internal/database"
	"hfb_sys/backend/internal/model"
	"hfb_sys/backend/internal/modules/wallet"
	"hfb_sys/backend/pkg/crypto"

	"gorm.io/gorm"
)

type withdrawalFixture struct {
	db        *gorm.DB
	repo      *Repository
	userID    uint64
	adminID   uint64
	accountID uint64
}

func newWithdrawalFixture(t *testing.T) withdrawalFixture {
	t.Helper()
	db := database.NewTestDB()
	if err := database.AutoMigrateIfMissing(db,
		&model.User{},
		&model.AdminUser{},
		&model.UserPaymentAccount{},
		&model.WithdrawalRequest{},
		&model.WalletAccount{},
		&model.WalletLedger{},
	); err != nil {
		t.Fatalf("migrate fixture: %v", err)
	}

	user := model.User{Phone: "13800000001", Nickname: "test-user", Status: "active"}
	admin := model.AdminUser{Username: "reviewer", PasswordHash: "test", Nickname: "reviewer", Status: "active"}
	if err := db.Create(&user).Error; err != nil {
		t.Fatalf("create user: %v", err)
	}
	if err := db.Create(&admin).Error; err != nil {
		t.Fatalf("create admin: %v", err)
	}
	account := model.UserPaymentAccount{
		UserID: user.ID, AccountType: "bank", AccountName: "Test User",
		AccountNo: "6222021234567890", BankName: "Test Bank", Status: "active",
	}
	if err := db.Create(&account).Error; err != nil {
		t.Fatalf("create payment account: %v", err)
	}
	if err := db.Create(&model.WalletAccount{
		UserID: user.ID, AvailableBalanceCent: 100_000, Status: "active",
	}).Error; err != nil {
		t.Fatalf("create wallet: %v", err)
	}

	return withdrawalFixture{
		db: db, repo: NewRepository(db, wallet.NewRepository(db), &crypto.MockEncryptor{}),
		userID: user.ID, adminID: admin.ID, accountID: account.ID,
	}
}

func (f withdrawalFixture) create(t *testing.T, amountCent int64) *WithdrawalDTO {
	t.Helper()
	dto, err := f.repo.Create(context.Background(), f.userID, CreateWithdrawalRequest{
		PaymentAccountID: f.accountID,
		AmountCent:       amountCent,
	})
	if err != nil {
		t.Fatalf("Create: %v", err)
	}
	return dto
}

func (f withdrawalFixture) assertBalance(t *testing.T, available, frozen int64) {
	t.Helper()
	var account model.WalletAccount
	if err := f.db.Where("user_id = ?", f.userID).First(&account).Error; err != nil {
		t.Fatalf("load wallet: %v", err)
	}
	if account.AvailableBalanceCent != available || account.FrozenBalanceCent != frozen {
		t.Fatalf("balance = available:%d frozen:%d, want available:%d frozen:%d",
			account.AvailableBalanceCent, account.FrozenBalanceCent, available, frozen)
	}
}

func TestWithdrawalCreateAndCancelPreservesBalance(t *testing.T) {
	f := newWithdrawalFixture(t)
	dto := f.create(t, 20_000)
	if dto.Status != "pending" || dto.FeeCent != 0 || dto.ActualAmountCent != 20_000 {
		t.Fatalf("created withdrawal = %+v", dto)
	}
	f.assertBalance(t, 80_000, 20_000)

	if err := f.repo.Cancel(context.Background(), f.userID, dto.ID); err != nil {
		t.Fatalf("Cancel: %v", err)
	}
	f.assertBalance(t, 100_000, 0)
	if err := f.repo.Cancel(context.Background(), f.userID, dto.ID); !errors.Is(err, ErrWithdrawalLocked) {
		t.Fatalf("second Cancel error = %v, want ErrWithdrawalLocked", err)
	}
	f.assertBalance(t, 100_000, 0)
}

func TestWithdrawalRejectedReturnsFrozenBalanceOnce(t *testing.T) {
	f := newWithdrawalFixture(t)
	dto := f.create(t, 30_000)
	approved := false
	result, err := f.repo.Review(context.Background(), f.adminID, dto.ID, ReviewWithdrawalRequest{
		Approved: &approved,
		Remark:   "account mismatch",
	})
	if err != nil {
		t.Fatalf("Review rejected: %v", err)
	}
	if result.Status != "rejected" || result.ReviewedByName != "reviewer" {
		t.Fatalf("review result = %+v", result)
	}
	f.assertBalance(t, 100_000, 0)
	if _, err := f.repo.Review(context.Background(), f.adminID, dto.ID, ReviewWithdrawalRequest{Approved: &approved}); !errors.Is(err, ErrWithdrawalLocked) {
		t.Fatalf("second Review error = %v, want ErrWithdrawalLocked", err)
	}
}

func TestWithdrawalApprovedAndPaidConsumesFrozenBalanceOnce(t *testing.T) {
	f := newWithdrawalFixture(t)
	dto := f.create(t, 40_000)
	approved := true
	if _, err := f.repo.Review(context.Background(), f.adminID, dto.ID, ReviewWithdrawalRequest{Approved: &approved}); err != nil {
		t.Fatalf("Review approved: %v", err)
	}
	f.assertBalance(t, 60_000, 40_000)

	result, err := f.repo.ConfirmPayment(context.Background(), f.adminID, dto.ID, ConfirmPaymentRequest{
		PaymentProofURL: "proof/test.png",
		Remark:          "paid",
	})
	if err != nil {
		t.Fatalf("ConfirmPayment: %v", err)
	}
	if result.Status != "completed" || result.PaidByName != "reviewer" {
		t.Fatalf("payment result = %+v", result)
	}
	f.assertBalance(t, 60_000, 0)
	if _, err := f.repo.ConfirmPayment(context.Background(), f.adminID, dto.ID, ConfirmPaymentRequest{}); !errors.Is(err, ErrWithdrawalLocked) {
		t.Fatalf("second ConfirmPayment error = %v, want ErrWithdrawalLocked", err)
	}
	f.assertBalance(t, 60_000, 0)
}

func TestWithdrawalCreateRollsBackWhenBalanceIsInsufficient(t *testing.T) {
	f := newWithdrawalFixture(t)
	_, err := f.repo.Create(context.Background(), f.userID, CreateWithdrawalRequest{
		PaymentAccountID: f.accountID,
		AmountCent:       100_001,
	})
	if !errors.Is(err, wallet.ErrInsufficientBalance) {
		t.Fatalf("Create error = %v, want ErrInsufficientBalance", err)
	}
	var count int64
	if err := f.db.Model(&model.WithdrawalRequest{}).Count(&count).Error; err != nil {
		t.Fatalf("count withdrawals: %v", err)
	}
	if count != 0 {
		t.Fatalf("withdrawal count = %d, want 0", count)
	}
	f.assertBalance(t, 100_000, 0)
}

func TestWithdrawalDetailIncludesDistributionSiteInfo(t *testing.T) {
	f := newWithdrawalFixture(t)
	if err := f.db.AutoMigrate(&model.DistributionSite{}); err != nil {
		t.Fatal(err)
	}
	site := model.DistributionSite{SiteCode: "sub_site", SiteName: "分销子站", Domain: "sub.example.com", Status: "active"}
	if err := f.db.Create(&site).Error; err != nil {
		t.Fatal(err)
	}
	wReq := model.WithdrawalRequest{
		WithdrawNo:         "WD2026TESTSITE",
		UserID:             f.userID,
		DistributionSiteID: site.ID,
		AmountCent:         5000,
		ActualAmountCent:   5000,
		Status:             "pending",
	}
	if err := f.db.Create(&wReq).Error; err != nil {
		t.Fatal(err)
	}

	detail, err := f.repo.AdminFindByID(context.Background(), wReq.ID)
	if err != nil {
		t.Fatalf("AdminFindByID failed: %v", err)
	}
	if detail.DistributionSiteID != site.ID || detail.DistributionSiteName != "分销子站" || detail.DistributionSiteCode != "sub_site" {
		t.Fatalf("unexpected detail site info: %+v", detail)
	}
}