feat: 接入商务用户资料和全局统计偏好

internal/data/business/preference_test.go

@@ -0,0 +1,72 @@
+package business
+
+import (
+	"context"
+	"github.com/glebarez/sqlite"
+	"gorm.io/gorm"
+	"kra/internal/biz/business"
+	"path/filepath"
+	"testing"
+)
+
+type testData struct{ db *gorm.DB }
+
+func (d testData) DB() *gorm.DB { return d.db }
+
+func openTestDB(t *testing.T) *gorm.DB {
+	t.Helper()
+	db, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "business.db")), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	sqlDB, err := db.DB()
+	if err != nil {
+		t.Fatal(err)
+	}
+	t.Cleanup(func() { _ = sqlDB.Close() })
+	for _, step := range Migrations() {
+		if err := step.Migrate(db); err != nil {
+			t.Fatal(err)
+		}
+	}
+	return db
+}
+
+func TestPreferencePreservesLegacyValuesAndUpserts(t *testing.T) {
+	db := openTestDB(t)
+	r := NewUserPreferenceRepo(testData{db})
+	ctx := context.Background()
+	if pref, err := r.Find(ctx, 5); err != nil || pref != nil {
+		t.Fatalf("缺省记录错误: %+v %v", pref, err)
+	}
+	if err := db.Exec("INSERT INTO sys_user_preference(user_id,include_sub_promote,idcard_dedup) VALUES (5,NULL,2),(6,1,1)").Error; err != nil {
+		t.Fatal(err)
+	}
+	pref, err := r.Find(ctx, 5)
+	if err != nil || !pref.IncludeSubPromote || !pref.IDCardDedup {
+		t.Fatalf("旧 NULL 和非零值应为开启: %+v %v", pref, err)
+	}
+	for i := 0; i < 2; i++ {
+		if err := r.Save(ctx, 5, business.UserPreference{}); err != nil {
+			t.Fatal(err)
+		}
+	}
+	for _, step := range Migrations() {
+		if err := step.Migrate(db); err != nil {
+			t.Fatal(err)
+		}
+	}
+	pref, err = r.Find(ctx, 5)
+	if err != nil || pref.IncludeSubPromote || pref.IDCardDedup {
+		t.Fatalf("重复迁移不能重置偏好: %+v %v", pref, err)
+	}
+	other, err := r.Find(ctx, 6)
+	if err != nil || !other.IncludeSubPromote || !other.IDCardDedup {
+		t.Fatalf("影响其他用户: %+v %v", other, err)
+	}
+	var count int64
+	db.Model(&preferencePO{}).Where("user_id = ?", 5).Count(&count)
+	if count != 1 {
+		t.Fatalf("重复写入产生多条记录: %d", count)
+	}
+}