116 lines
5.0 KiB
Go
116 lines
5.0 KiB
Go
package business
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"math/big"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/glebarez/sqlite"
|
|
"gorm.io/gorm"
|
|
"kra/internal/biz/business"
|
|
)
|
|
|
|
type activationDatabases struct{ oa *gorm.DB }
|
|
|
|
func (d activationDatabases) Database(name string) (*gorm.DB, error) {
|
|
if name != "oa" || d.oa == nil {
|
|
return nil, errors.New("未配置")
|
|
}
|
|
return d.oa, nil
|
|
}
|
|
func prepareActivations(t *testing.T) (*gorm.DB, *gorm.DB, business.ActivationRepo) {
|
|
t.Helper()
|
|
db := prepareAccounts(t)
|
|
oa, err := gorm.Open(sqlite.Open(filepath.Join(t.TempDir(), "oa.db")), &gorm.Config{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
pool, _ := oa.DB()
|
|
t.Cleanup(func() { _ = pool.Close() })
|
|
if err := oa.AutoMigrate(&activationPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
start := time.Unix(1000, 0).UTC()
|
|
end := time.Unix(2000, 0).UTC()
|
|
channel := uint64(100)
|
|
amount := "30.30"
|
|
pos := []activationPO{{ID: 1, PromoteID: &channel, PlatformName: "lx", PromoteAccount: "渠道一", BusierAccount: "Own", BindAt: &start, ReachAt: &end, Status: "reached", AccumAmount: &amount}, {ID: 2, PlatformName: "hw", BusierAccount: "own", Status: "pending"}, {ID: 3, PlatformName: "lx", BusierAccount: "Other", Status: "in_progress"}, {ID: 4, PlatformName: "lx", BusierAccount: "Own", Status: "rejected", DeletedAt: &end}}
|
|
if err := oa.Create(&pos).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, sql := range []string{
|
|
"INSERT INTO sys_user_bind_platform(user_id,source_db,platform_busier_id,platform_busier_account,status) VALUES(7,'lx',1,'Own','0'),(8,'lx',2,'Child','0'),(7,'hw',3,'Pending','3')",
|
|
"ALTER TABLE tab_promote ADD COLUMN top_promote_id INTEGER",
|
|
"ALTER TABLE tab_promote ADD COLUMN parent_id INTEGER",
|
|
"INSERT INTO tab_promote(id,source_db,top_promote_id,parent_id) VALUES(100,'lx',0,0),(101,'lx',0,100),(101,'lx',0,100),(102,'lx',100,101),(103,'lx',200,200),(100,'hw',0,0)",
|
|
"CREATE TABLE tab_spend(source_db TEXT,promote_id INTEGER,pay_status INTEGER,pay_time INTEGER,game_id INTEGER,game_name TEXT,pay_amount DECIMAL(20,2))",
|
|
"INSERT INTO tab_spend VALUES('lx',100,1,1000,1,'A',10.10),('lx',101,1,2000,1,'Z',20.20),('lx',102,1,1500,2,'B',3.30),('lx',103,1,1500,1,'X',900),('hw',100,1,1500,1,'X',999),('lx',100,1,999,1,'X',800),('lx',100,1,2001,1,'X',700),('lx',100,0,1500,1,'X',600)",
|
|
} {
|
|
if err := db.Exec(sql).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
return db, oa, NewActivationRepo(testData{db}, activationDatabases{oa})
|
|
}
|
|
func TestActivationListsReadOAAndRestrictAccounts(t *testing.T) {
|
|
db, _, repo := prepareActivations(t)
|
|
ctx := context.Background()
|
|
accounts, err := repo.OwnAccounts(ctx, 7)
|
|
if err != nil || len(accounts) != 1 || accounts[0] != (business.ActivationAccount{SourceDB: "lx", Account: "Own"}) {
|
|
t.Fatalf("本人已通过账号范围错误: %+v %v", accounts, err)
|
|
}
|
|
page, err := repo.List(ctx, business.ActivationScope{Accounts: []business.ActivationAccount{{SourceDB: "lx", Account: "OWN"}}}, business.ActivationFilter{})
|
|
if err != nil || page.Total != 1 || page.Rows[0].ID != 1 {
|
|
t.Fatalf("OA 列表或排序错误: %+v %v", page, err)
|
|
}
|
|
page, err = repo.List(ctx, business.ActivationScope{All: true}, business.ActivationFilter{PlatformNames: []string{"lx"}, Statuses: []string{"reached"}})
|
|
if err != nil || page.Total != 1 || page.Rows[0].AccumAmount != "30.30" {
|
|
t.Fatalf("平台或状态筛选错误: %+v %v", page, err)
|
|
}
|
|
if _, err := repo.Find(ctx, 4); err != business.ErrActivationNotFound {
|
|
t.Fatal("软删除 OA 记录仍可访问")
|
|
}
|
|
// 即使主库有同名表，也不能在 OA 未配置时回退到主库。
|
|
if err := db.AutoMigrate(&activationPO{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
missing := NewActivationRepo(testData{db}, activationDatabases{})
|
|
page, err = missing.List(ctx, business.ActivationScope{All: true}, business.ActivationFilter{})
|
|
if err != nil || page.Available {
|
|
t.Fatal("OA 缺配置必须明确不可用")
|
|
}
|
|
if _, err := missing.Find(ctx, 1); err != business.ErrActivationSource {
|
|
t.Fatal("OA 缺配置不应读取主库")
|
|
}
|
|
}
|
|
func TestActivationSpendTreePlatformAndInclusiveBounds(t *testing.T) {
|
|
db, _, repo := prepareActivations(t)
|
|
end := int64(2000)
|
|
query := business.ActivationSpendQuery{PlatformName: "lx", PromoteID: 100, Start: 1000, End: &end}
|
|
items, err := repo.Spend(context.Background(), query)
|
|
amount := func(value string) string {
|
|
n, ok := new(big.Rat).SetString(value)
|
|
if !ok {
|
|
return "无效金额"
|
|
}
|
|
return n.FloatString(2)
|
|
}
|
|
if err != nil || len(items) != 2 || items[0].GameName != "Z" || amount(items[0].Amount) != "30.30" || amount(items[1].Amount) != "3.30" {
|
|
t.Fatalf("渠道树、同号平台、成功支付、起止边界或重复渠道导致金额错误: %+v %v", items, err)
|
|
}
|
|
if err := db.Exec("DROP TABLE tab_spend").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := repo.Spend(context.Background(), query); err != business.ErrActivationMetrics {
|
|
t.Fatal("缺流水表未报告不可用")
|
|
}
|
|
query.PromoteID = 404
|
|
items, err = repo.Spend(context.Background(), query)
|
|
if err != nil || len(items) != 0 {
|
|
t.Fatal("确定没有渠道时应返回真实空流水")
|
|
}
|
|
}
|