完善租客等级免押与积分管理

2026-07-26 18:07:10 +08:00
parent 379536e5c5
commit c594b543d1
24 changed files with 1296 additions and 178 deletions
@@ -1,12 +1,99 @@
 package rentergrowth

 import (
+	"encoding/json"
 	"testing"

 	"hfb_sys/backend/internal/database"
 	"hfb_sys/backend/internal/model"
 )

+func TestDefaultConfigIncludesDepositFreeQuota(t *testing.T) {
+	cfg := DefaultConfig()
+	wants := map[string]int64{
+		DefaultLevelCode: 0,
+		"platinum":       DefaultPlatinumQuotaCent,
+		"diamond":        DefaultDiamondQuotaCent,
+		"peak":           DefaultPeakQuotaCent,
+	}
+	for _, level := range cfg.Levels {
+		if level.DepositFreeQuotaCent != wants[level.Code] {
+			t.Fatalf("等级 %s 免押额度 = %d, want %d", level.Code, level.DepositFreeQuotaCent, wants[level.Code])
+		}
+	}
+}
+
+func TestLegacyConfigBackfillsQuotaButKeepsExplicitZero(t *testing.T) {
+	legacy := `{"enabled":true,"points_per_yuan":1,"levels":[{"code":"normal","name":"普通","min_points":0,"discount_bps":10000},{"code":"platinum","name":"铂金","min_points":300,"discount_bps":9900},{"code":"diamond","name":"钻石","min_points":1000,"discount_bps":9800},{"code":"peak","name":"巅峰","min_points":5000,"discount_bps":9500}]}`
+	cfg, err := ParseConfigValue(legacy)
+	if err != nil {
+		t.Fatalf("解析旧配置失败：%v", err)
+	}
+	if LevelForPoints(cfg, 300).DepositFreeQuotaCent != DefaultPlatinumQuotaCent {
+		t.Fatalf("旧配置未补齐铂金免押额度")
+	}
+
+	var raw map[string]any
+	if err := json.Unmarshal([]byte(legacy), &raw); err != nil {
+		t.Fatalf("解析测试 JSON 失败：%v", err)
+	}
+	levels := raw["levels"].([]any)
+	levels[1].(map[string]any)["deposit_free_quota_cent"] = float64(0)
+	explicitZero, err := json.Marshal(raw)
+	if err != nil {
+		t.Fatalf("生成测试配置失败：%v", err)
+	}
+	cfg, err = ParseConfigValue(string(explicitZero))
+	if err != nil {
+		t.Fatalf("解析显式零额度配置失败：%v", err)
+	}
+	if LevelForPoints(cfg, 300).DepositFreeQuotaCent != 0 {
+		t.Fatalf("显式配置的零额度被默认值覆盖")
+	}
+}
+
+func TestSnapshotCombinesLevelAndManualDepositQuota(t *testing.T) {
+	db := database.NewTestDB()
+	if err := db.AutoMigrate(&model.User{}, &model.SystemConfig{}); err != nil {
+		t.Fatalf("migrate failed: %v", err)
+	}
+	user := model.User{Phone: "13800009999", RenterGrowthPoints: 1000, DepositFreeQuotaCent: 30_000}
+	if err := db.Create(&user).Error; err != nil {
+		t.Fatalf("create user failed: %v", err)
+	}
+	snapshot, err := SnapshotForUser(db, user.ID)
+	if err != nil {
+		t.Fatalf("SnapshotForUser() error = %v", err)
+	}
+	if snapshot.LevelDepositFreeQuotaCent != DefaultDiamondQuotaCent || snapshot.ManualDepositFreeQuotaCent != 30_000 || snapshot.EffectiveDepositFreeQuotaCent != 130_000 {
+		t.Fatalf("免押额度快照不正确：%+v", snapshot)
+	}
+}
+
+func TestSyncAllUserLevelsUsesUpdatedThresholds(t *testing.T) {
+	db := database.NewTestDB()
+	if err := db.AutoMigrate(&model.User{}); err != nil {
+		t.Fatalf("migrate failed: %v", err)
+	}
+	users := []model.User{
+		{Phone: "13800009001", RenterGrowthPoints: 2000, RenterGrowthLevel: DefaultLevelCode},
+		{Phone: "13800009002", RenterGrowthPoints: 6000, RenterGrowthLevel: DefaultLevelCode},
+	}
+	if err := db.Create(&users).Error; err != nil {
+		t.Fatalf("create users failed: %v", err)
+	}
+	if err := SyncAllUserLevels(db, DefaultConfig()); err != nil {
+		t.Fatalf("SyncAllUserLevels() error = %v", err)
+	}
+	var saved []model.User
+	if err := db.Order("id ASC").Find(&saved).Error; err != nil {
+		t.Fatalf("find users failed: %v", err)
+	}
+	if saved[0].RenterGrowthLevel != "diamond" || saved[1].RenterGrowthLevel != "peak" {
+		t.Fatalf("同步后等级不正确：%s/%s", saved[0].RenterGrowthLevel, saved[1].RenterGrowthLevel)
+	}
+}
+
 func TestLevelForPointsStartsFromNormal(t *testing.T) {
 	cfg := DefaultConfig()
 	cases := []struct {