用户管理增加余额调账功能

支持后台加减可用余额并写入资金流水与审计，优化操作列按钮样式。
2026-07-13 21:30:38 +08:00
parent a2a9ade68e
commit b96d6141f5
10 changed files with 642 additions and 57 deletions
@@ -164,6 +164,92 @@ func TestManualRealnameRejectsVerifiedUser(t *testing.T) {
 	}
 }

+func TestAdjustWalletDebitAndAudit(t *testing.T) {
+	db := setupAdminUserTestDB(t)
+	user := model.User{
+		Phone:          "13900000003",
+		Nickname:       "有余额用户",
+		RealnameStatus: "verified",
+		Status:         "active",
+	}
+	if err := db.Create(&user).Error; err != nil {
+		t.Fatalf("创建用户失败：%v", err)
+	}
+	if err := db.Create(&model.WalletAccount{
+		UserID:               user.ID,
+		AvailableBalanceCent: 15000,
+		FrozenBalanceCent:    3000,
+		Status:               "active",
+	}).Error; err != nil {
+		t.Fatalf("创建钱包失败：%v", err)
+	}
+
+	repo := NewRepository(db)
+	got, err := repo.AdjustWallet(t.Context(), 88, user.ID, WalletAdjustRequest{
+		Direction:   "out",
+		AmountCent:  5000,
+		Reason:      "线下已转账核销",
+		ReferenceNo: "OFFLINE-001",
+	}, AuditMeta{IP: "127.0.0.1", UserAgent: "test", RequestID: "req-wallet-1"})
+	if err != nil {
+		t.Fatalf("AdjustWallet() error = %v", err)
+	}
+	if got.AvailableBalanceCent != 10000 {
+		t.Fatalf("可用余额 = %d, want 10000", got.AvailableBalanceCent)
+	}
+	if got.FrozenBalanceCent != 3000 {
+		t.Fatalf("冻结余额不应变化，got %d", got.FrozenBalanceCent)
+	}
+
+	var ledger model.WalletLedger
+	if err := db.Where("user_id = ? AND biz_type = ?", user.ID, "admin_debit").First(&ledger).Error; err != nil {
+		t.Fatalf("查询资金流水失败：%v", err)
+	}
+	if ledger.Direction != "out" || ledger.AmountCent != 5000 || ledger.BalanceType != "available" {
+		t.Fatalf("流水不正确：%+v", ledger)
+	}
+
+	var audit model.AuditLog
+	if err := db.Where("action = ?", "admin_user.wallet_adjust").First(&audit).Error; err != nil {
+		t.Fatalf("查询审计日志失败：%v", err)
+	}
+	if audit.ActorID != 88 || audit.BizID == nil || *audit.BizID != user.ID {
+		t.Fatalf("审计主体不正确：actor=%d biz=%v", audit.ActorID, audit.BizID)
+	}
+	detail := string(audit.Detail)
+	if !strings.Contains(detail, "线下已转账核销") || !strings.Contains(detail, "OFFLINE-001") {
+		t.Fatalf("审计详情缺少原因或外部单号：%s", detail)
+	}
+}
+
+func TestAdjustWalletInsufficientBalance(t *testing.T) {
+	db := setupAdminUserTestDB(t)
+	user := model.User{
+		Phone:  "13900000004",
+		Status: "active",
+	}
+	if err := db.Create(&user).Error; err != nil {
+		t.Fatalf("创建用户失败：%v", err)
+	}
+	if err := db.Create(&model.WalletAccount{
+		UserID:               user.ID,
+		AvailableBalanceCent: 100,
+		Status:               "active",
+	}).Error; err != nil {
+		t.Fatalf("创建钱包失败：%v", err)
+	}
+
+	repo := NewRepository(db)
+	_, err := repo.AdjustWallet(t.Context(), 1, user.ID, WalletAdjustRequest{
+		Direction:  "out",
+		AmountCent: 200,
+		Reason:     "超额扣款",
+	}, AuditMeta{})
+	if !errors.Is(err, ErrInsufficientBalance) {
+		t.Fatalf("AdjustWallet() error = %v, want ErrInsufficientBalance", err)
+	}
+}
+
 func setupAdminUserTestDB(t *testing.T) *gorm.DB {
 	t.Helper()
 	db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{})
@@ -177,6 +263,8 @@ func setupAdminUserTestDB(t *testing.T) *gorm.DB {
 		&model.RentalOrder{},
 		&model.RentalListing{},
 		&model.Dispute{},
+		&model.WalletAccount{},
+		&model.WalletLedger{},
 	); err != nil {
 		t.Fatalf("数据库迁移失败：%v", err)
 	}