233 lines
11 KiB
Go
233 lines
11 KiB
Go
package business
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"gorm.io/gorm"
|
|
"kra/internal/biz/business"
|
|
)
|
|
|
|
func prepareAnalyticsData(t *testing.T) (*gorm.DB, business.AnalyticsRepo, business.AnalyticsScope, business.AnalyticsFilter) {
|
|
t.Helper()
|
|
db, _ := prepareChannels(t)
|
|
for _, sql := range []string{
|
|
"INSERT INTO tab_promote_apply VALUES('hw',100,1,1)",
|
|
"ALTER TABLE tab_sw_report_day ADD COLUMN game_name TEXT",
|
|
"ALTER TABLE tab_sw_report_day_key ADD COLUMN game_name TEXT",
|
|
"UPDATE tab_sw_report_day SET game_name=CASE WHEN game_id=1 THEN '历史游戏甲' ELSE '游戏乙' END",
|
|
"UPDATE tab_sw_report_day_key SET game_name=CASE WHEN game_id=1 THEN '历史游戏甲' ELSE '游戏乙' END",
|
|
"UPDATE tab_sw_report_day_key SET identity_key='A' WHERE (source_db='hw' OR promote_id=103) AND metric=1",
|
|
"INSERT INTO tab_sw_report_day(source_db,promote_id,game_id,dedup_mode,stat_date,total_pay,game_name) VALUES('lx',0,1,0,'2026-11-01',5,'历史游戏甲'),('lx',999,1,0,'2026-11-01',7,'历史游戏甲')",
|
|
"INSERT INTO tab_sw_report_day_key(source_db,promote_id,game_id,dedup_mode,stat_date,metric,identity_key,game_name) VALUES('lx',0,1,1,'2026-11-01',1,'official','历史游戏甲'),('lx',999,1,1,'2026-11-01',1,'orphan','历史游戏甲'),('lx',100,1,1,'2026-11-01',2,'A','历史游戏甲'),('lx',103,1,1,'2026-11-02',2,'A','历史游戏甲'),('hw',100,1,1,'2026-11-01',2,'A','历史游戏甲'),('lx',100,1,1,'2026-11-01',3,'A','历史游戏甲'),('lx',103,1,1,'2026-11-02',3,'A','历史游戏甲'),('hw',100,1,1,'2026-11-01',3,'A','历史游戏甲')",
|
|
} {
|
|
if err := db.Exec(sql).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
scope := business.AnalyticsScope{Viewer: channelTestScope(), IncludeSub: true, IDCardDedup: true}
|
|
filter := business.AnalyticsFilter{Sources: []string{"lx", "hw"}, Begin: "2026-11-01", End: "2026-11-03"}
|
|
return db, NewAnalyticsRepo(testData{db}), scope, filter
|
|
}
|
|
func TestAnalyticsChannelTreeAndAdminOfficialRows(t *testing.T) {
|
|
_, repo, scope, filter := prepareAnalyticsData(t)
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repo.FillNames(context.Background(), business.ChannelAnalytics, scope, filter, data.Rows); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(data.Rows) != 3 || data.GlobalCounts != nil {
|
|
t.Fatalf("普通商务统计范围错误: %+v", data)
|
|
}
|
|
var lx *business.AnalyticsRow
|
|
for _, row := range data.Rows {
|
|
if row.SourceDB == "lx" && row.ID == 100 {
|
|
lx = row
|
|
}
|
|
if row.ID == 0 || row.ID == 999 {
|
|
t.Fatal("受约束商务看到官方/未知渠道")
|
|
}
|
|
}
|
|
if lx == nil || lx.Counts.Register != 2 || portraitTestMoney(lx.Pay) != "60.60" || lx.Name != "root-channel" || lx.BusinessName() != "lx-business" {
|
|
t.Fatalf("渠道树去重、充值或商务展示错误: %+v", lx)
|
|
}
|
|
scope.Viewer = business.UserScope{All: true}
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 5 || data.GlobalCounts.Register != 4 || data.GlobalCounts.Active != 2 || data.GlobalCounts.Pay != 1 || portraitTestMoney(data.GlobalPay) != "2070.60" {
|
|
t.Fatalf("超管全局去重或官方/历史渠道遗漏: %+v %v", data, err)
|
|
}
|
|
}
|
|
|
|
func TestAnalyticsFiltersIncludeIdleChannelsAndRecalculateTotals(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
if err := db.Exec("INSERT INTO tab_promote(id,source_db,account,promote_level,parent_id,top_promote_id,busier_id) VALUES(200,'lx','idle',1,0,0,14)").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
yes, no := true, false
|
|
filter.HasRegister, filter.HasPay = &no, &no
|
|
for _, viewer := range []business.UserScope{scope.Viewer, {All: true}} {
|
|
scope.Viewer = viewer
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
found := false
|
|
for _, row := range data.Rows {
|
|
if row.Counts.Register != 0 || portraitTestMoney(row.Pay) != "0.00" {
|
|
t.Fatalf("筛选后的每行必须同时满足两个条件: %+v", row)
|
|
}
|
|
found = found || row.SourceDB == "lx" && row.ID == 200
|
|
}
|
|
if !found {
|
|
t.Fatal("整个期间没有事件的可见渠道不能遗漏")
|
|
}
|
|
if viewer.All && (data.GlobalCounts.Register != 0 || portraitTestMoney(data.GlobalPay) != "0.00") {
|
|
t.Fatalf("管理员合计未跟随过滤: %+v", data)
|
|
}
|
|
}
|
|
filter.HasRegister, filter.HasPay = &yes, &yes
|
|
filter.Sources, filter.PromoteName = []string{"lx"}, "root-channel"
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 1 || portraitTestMoney(data.GlobalPay) != "60.60" || data.GlobalCounts.Register != data.Rows[0].Counts.Register {
|
|
t.Fatalf("平台、名称及指标筛选须共同约束列表与全量合计: %+v %v", data, err)
|
|
}
|
|
}
|
|
|
|
func TestAnalyticsEffectiveCountIncludesQualifiedDescendants(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
qualified := time.Date(2026, 11, 2, 0, 0, 0, 0, time.FixedZone("Asia/Shanghai", 8*3600)).Unix()
|
|
if err := db.Exec("UPDATE sys_business_effectiveness_state SET rule_key='audit' WHERE id=1").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, key := range []business.ChannelKey{{SourceDB: "lx", ID: 100}, {SourceDB: "lx", ID: 101}, {SourceDB: "yq", ID: 100}} {
|
|
if err := db.Create(&effectivenessPO{RuleKey: "audit", SourceDB: key.SourceDB, PromoteID: key.ID, QualifiedAt: &qualified, WindowPay: "3000"}).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
filter.PromoteName = "root-channel"
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 1 || data.EffectiveCount != 2 {
|
|
t.Fatalf("合并渠道树展示仍须计入下级达标，且不能混入权限外同号渠道: %+v %v", data, err)
|
|
}
|
|
scope.IncludeSub = false
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || data.EffectiveCount != 1 {
|
|
t.Fatalf("关闭包含下级后只统计自身: %+v %v", data, err)
|
|
}
|
|
filter.Begin = "2026-11-03"
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || data.EffectiveCount != 0 {
|
|
t.Fatalf("有效端必须按达标时间归属统计区间: %+v %v", data, err)
|
|
}
|
|
}
|
|
func TestAnalyticsFiltersKeepHierarchyAndSnapshotNames(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
filter.PromoteName = "child"
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 1 || data.Rows[0].ID != 100 || data.Rows[0].Counts.Register != 2 || portraitTestMoney(data.Rows[0].Pay) != "60.60" {
|
|
t.Fatalf("模糊渠道筛选应扩到同树可见渠道: %+v %v", data, err)
|
|
}
|
|
if err := db.Exec("UPDATE tab_promote SET register_type=1 WHERE source_db='lx' AND id=101").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
filter.PromoteName = ""
|
|
kind := 1
|
|
filter.PromoteType = &kind
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := repo.FillNames(context.Background(), business.ChannelAnalytics, scope, filter, data.Rows); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err != nil || len(data.Rows) != 1 || data.Rows[0].ID != 100 || portraitTestMoney(data.Rows[0].Pay) != "20.20" || data.Rows[0].PromoteType == nil || *data.Rows[0].PromoteType != 0 {
|
|
t.Fatalf("类型筛选应限制参与渠道，同时显示归根后的当前资料: %+v %v", data, err)
|
|
}
|
|
filter.PromoteType = nil
|
|
filter.GameNames = []string{"历史游戏甲"}
|
|
data, err = repo.Aggregate(context.Background(), business.GameAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 2 {
|
|
t.Fatalf("按游戏快照名筛选错误: %+v %v", data, err)
|
|
}
|
|
if err := repo.FillNames(context.Background(), business.GameAnalytics, scope, filter, data.Rows); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
for _, row := range data.Rows {
|
|
if row.SourceDB == "lx" && (row.ID != 1 || row.Name != "游戏甲" || row.Counts.Register != 1) {
|
|
t.Fatalf("应按报表快照筛选、用当前游戏名展示: %+v", row)
|
|
}
|
|
}
|
|
filter.GameNames = []string{"游戏甲"}
|
|
data, err = repo.Aggregate(context.Background(), business.GameAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 0 {
|
|
t.Fatal("不能用当前游戏表名替代报表快照过滤")
|
|
}
|
|
}
|
|
|
|
func TestAnalyticsChannelSearchMatchesNicknameAndKeepsScope(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
if err := db.Exec("UPDATE tab_promote SET nickname='同名推广' WHERE id=100").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
scope.Viewer = business.UserScope{SelfID: 9, OnlySelf: true}
|
|
filter.PromoteName = "推广"
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 1 || data.Rows[0].SourceDB != "lx" || data.Rows[0].ID != 100 {
|
|
t.Fatalf("nickname search must remain scoped: %+v %v", data, err)
|
|
}
|
|
filter.PromoteName = "%"
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 0 {
|
|
t.Fatal("wildcard keyword must not broaden the query")
|
|
}
|
|
}
|
|
|
|
func TestGameAuthorizationFiltersAggregatedFactsByPlatform(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
scope.Viewer = business.UserScope{SelfID: 9, OnlySelf: true}
|
|
for _, sql := range []string{
|
|
"INSERT INTO tab_game(source_db,id,game_name) VALUES('lx',77,'未申请游戏'),('hw',77,'其他平台游戏')",
|
|
"INSERT INTO tab_promote_apply(source_db,promote_id,game_id,status) VALUES('hw',100,77,1)",
|
|
"INSERT INTO tab_sw_report_day(source_db,promote_id,game_id,dedup_mode,stat_date,total_pay,game_name) VALUES('lx',100,77,0,'2026-11-01',9999999,'未申请游戏')",
|
|
} {
|
|
if err := db.Exec(sql).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
data, err := repo.Aggregate(context.Background(), business.GameAnalytics, scope, filter)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if len(data.Rows) == 0 {
|
|
t.Fatal("approved games must remain visible")
|
|
}
|
|
for _, row := range data.Rows {
|
|
if row.ID == 77 || row.SourceDB != "lx" {
|
|
t.Fatal("unapproved game facts or another platform's approval escaped authorization")
|
|
}
|
|
}
|
|
}
|
|
func TestAnalyticsMissingScopeAndTablesDoNotBroaden(t *testing.T) {
|
|
db, repo, scope, filter := prepareAnalyticsData(t)
|
|
scope.IncludeSub = false
|
|
data, err := repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || len(data.Rows) != 2 {
|
|
t.Fatal("关闭下级仅保留一级渠道统计")
|
|
}
|
|
if err := db.Exec("DROP TABLE tab_sw_report_day_key").Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || data.Available {
|
|
t.Fatal("缺身份键表不能返回零汇总")
|
|
}
|
|
scope.Viewer = business.UserScope{SelfID: 500, OnlySelf: true}
|
|
data, err = repo.Aggregate(context.Background(), business.ChannelAnalytics, scope, filter)
|
|
if err != nil || !data.Available || len(data.Rows) != 0 {
|
|
t.Fatal("无绑定应为确定空范围")
|
|
}
|
|
}
|