feat: 迁移渠道列表新增和游戏累计分布

internal/data/business/channel_write_test.go

@@ -0,0 +1,87 @@
+package business
+
+import (
+	"context"
+	"errors"
+	"fmt"
+	"path/filepath"
+	"testing"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/gorm"
+	"kra/internal/biz/business"
+)
+
+type channelDatabases struct{ platform *gorm.DB }
+
+func (d channelDatabases) Database(name string) (*gorm.DB, error) {
+	if name == "lx" {
+		return d.platform, nil
+	}
+	return nil, errors.New("数据源未配置")
+}
+func TestChannelWriteIsAtomicInNamedPlatformDatabase(t *testing.T) {
+	db, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "platform.db")), &gorm.Config{})
+	if err != nil {
+		t.Fatal(err)
+	}
+	pool, _ := db.DB()
+	t.Cleanup(func() { _ = pool.Close() })
+	if err := db.AutoMigrate(&platformChannelPO{}, &channelSettlementPO{}); err != nil {
+		t.Fatal(err)
+	}
+	for _, sql := range []string{"CREATE UNIQUE INDEX channel_account ON tab_promote(account)", "CREATE TABLE tab_promote_business(id INTEGER PRIMARY KEY,promote_ids TEXT)", "INSERT INTO tab_promote_business VALUES(14,'5,6')"} {
+		if err := db.Exec(sql).Error; err != nil {
+			t.Fatal(err)
+		}
+	}
+	repo := NewChannelWriteRepo(channelDatabases{db})
+	input := &business.ChannelWrite{ChannelCreate: business.ChannelCreate{SourceDB: "lx", Account: "中文渠道", Password: "encoded-password", BusierID: 14, RegisterType: 1}, Status: -3, Level: 1, SettlementDays: 7, CreatedUnix: 1000, NextCountUnix: 605800}
+	id, err := repo.Create(context.Background(), input)
+	if err != nil {
+		t.Fatal(err)
+	}
+	var channel platformChannelPO
+	if err := db.First(&channel, id).Error; err != nil {
+		t.Fatal(err)
+	}
+	if channel.Status != -3 || channel.PromoteLevel != 1 || channel.ParentID != 0 || channel.TopPromoteID != 0 || channel.Password != "encoded-password" || channel.CreateTime != 1000 || channel.RegisterType != 1 {
+		t.Fatalf("源平台渠道默认字段错误: %+v", channel)
+	}
+	var settlement channelSettlementPO
+	if err := db.Where("promote_id = ?", id).First(&settlement).Error; err != nil {
+		t.Fatal(err)
+	}
+	if settlement.TimeType != 0 || settlement.DayPeriod != 7 || settlement.NextCountTime != 605800 || settlement.PromoteAccount != "中文渠道" {
+		t.Fatalf("结算周期字段错误: %+v", settlement)
+	}
+	var related string
+	db.Table("tab_promote_business").Select("promote_ids").Where("id=14").Scan(&related)
+	if related != fmt.Sprintf("5,6,%d", id) {
+		t.Fatal("追加关系覆盖了已有渠道")
+	}
+	if _, err := repo.Create(context.Background(), input); err != business.ErrChannelExists {
+		t.Fatal("重复账号没有拒绝")
+	}
+	if err := db.Exec("DROP TABLE tab_promote_settlement_time").Error; err != nil {
+		t.Fatal(err)
+	}
+	input.Account = "rollback"
+	if _, err := repo.Create(context.Background(), input); err == nil {
+		t.Fatal("缺结算表应回滚")
+	}
+	var count int64
+	db.Model(&platformChannelPO{}).Count(&count)
+	if count != 1 {
+		t.Fatal("创建失败留下了渠道")
+	}
+	var after string
+	db.Table("tab_promote_business").Select("promote_ids").Where("id=14").Scan(&after)
+	if after != related {
+		t.Fatal("创建失败仍改动关联")
+	}
+	input.SourceDB = "hw"
+	if _, err := repo.Create(context.Background(), input); err != business.ErrPlatformUnavailable {
+		t.Fatal("未配置平台不应回退到其他数据源")
+	}
+}