92 lines
4.5 KiB
Go
92 lines
4.5 KiB
Go
package business
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
type workbenchRepoFake struct {
|
|
totals *WorkbenchTotals
|
|
data *WorkbenchRankData
|
|
window WorkbenchWindow
|
|
filter AnalyticsFilter
|
|
scope AnalyticsScope
|
|
}
|
|
|
|
func (r *workbenchRepoFake) Summary(_ context.Context, s AnalyticsScope, _ []string, w WorkbenchWindow) (*WorkbenchTotals, error) {
|
|
r.window, r.scope = w, s
|
|
c := *r.totals
|
|
return &c, nil
|
|
}
|
|
func (r *workbenchRepoFake) Ranks(_ context.Context, s AnalyticsScope, f AnalyticsFilter, _ WorkbenchRankKind) (*WorkbenchRankData, error) {
|
|
r.filter, r.scope = f, s
|
|
out := &WorkbenchRankData{Available: r.data.Available}
|
|
for _, row := range r.data.Rows {
|
|
c := *row
|
|
out.Rows = append(out.Rows, &c)
|
|
}
|
|
return out, nil
|
|
}
|
|
func workbenchFixture() (*WorkbenchUsecase, *workbenchRepoFake) {
|
|
r := &workbenchRepoFake{totals: &WorkbenchTotals{Available: true, Today: WorkbenchDay{Counts: PortraitCounts{Register: 3, Active: 4}, Pay: "20.005", ActiveChannels: 2, NewChannels: 1}, Yesterday: WorkbenchDay{Counts: PortraitCounts{Register: 2}, Pay: "10", ActiveChannels: 1}, MonthPay: "300", PreviousMonthPay: "200"}, data: &WorkbenchRankData{Available: true}}
|
|
a := &accountRepoFake{accounts: map[uint]*Account{7: {ID: 7, Level: 1, IDPath: "0"}}}
|
|
uc := NewWorkbenchUsecase(r, NewAccountUsecase(a), NewUserPreferenceUsecase(&channelPreferenceFake{UserPreference{IncludeSubPromote: true, IDCardDedup: true}}))
|
|
uc.now = func() time.Time { return time.Date(2026, 3, 31, 9, 0, 0, 0, portraitLocation) }
|
|
return uc, r
|
|
}
|
|
func TestWorkbenchSummaryCalendarAndRatios(t *testing.T) {
|
|
uc, r := workbenchFixture()
|
|
out, err := uc.Summary(accountContext(7, 901), []string{"lx"})
|
|
if err != nil || *out.TodayRegister.Value != "3.00" || *out.TodayRegister.ChangeRate != "50.0" || out.TodayActive.ChangeRate != nil || *out.TodayPay.Value != "20.01" || *out.TodayPay.ChangeRate != "100.1" || *out.MonthPay.ChangeRate != "50.0" || *out.ActivePromote.ChangeRate != "100.0" {
|
|
t.Fatalf("卡片精度或环比错误: %+v %v", out, err)
|
|
}
|
|
if r.window.PreviousMonthEnd.Format("2006-01-02") != "2026-02-28" || r.window.PreviousMonthStart.Format("2006-01-02") != "2026-02-01" {
|
|
t.Fatal("上月同期需要截断至月末")
|
|
}
|
|
r.totals.Available = false
|
|
out, err = uc.Summary(accountContext(7, 901), []string{"lx"})
|
|
if err != nil || out.TodayPay.Value != nil || out.TodayPay.CompareValue != nil || out.Available {
|
|
t.Fatal("缺表不能返回零卡片")
|
|
}
|
|
if _, err := uc.Summary(accountContext(7, 901), nil); err != ErrWorkbenchSource {
|
|
t.Fatal("必须选择平台")
|
|
}
|
|
}
|
|
func TestWorkbenchRankLimitsExactSortDefaultsAndUnlogin(t *testing.T) {
|
|
uc, r := workbenchFixture()
|
|
ctx := accountContext(7, 901)
|
|
for i := 0; i < 14; i++ {
|
|
r.data.Rows = append(r.data.Rows, &WorkbenchRankRow{AnalyticsKey: AnalyticsKey{SourceDB: "lx", ID: int64(i)}, Value: fmt.Sprint(i)})
|
|
}
|
|
rows, err := uc.Ranks(ctx, []string{"lx"}, "", WorkbenchPayHigh)
|
|
if err != nil || len(rows) != 10 || rows[0].ID != 13 || rows[9].Rank != 10 {
|
|
t.Fatal("榜单排序或 TOP10 截断错误")
|
|
}
|
|
rows, err = uc.Ranks(ctx, []string{"lx"}, "", WorkbenchPayLow)
|
|
if err != nil || rows[0].ID != 0 || rows[0].Value != "0.00" || rows[0].Name != "官方渠道" || r.filter.Begin != "2026-03-02" {
|
|
t.Fatal("最低榜应包含零值，默认近30天")
|
|
}
|
|
r.data.Rows = []*WorkbenchRankRow{{AnalyticsKey: AnalyticsKey{SourceDB: "lx", ID: 1}, Value: "9007199254740993.00"}, {AnalyticsKey: AnalyticsKey{SourceDB: "hw", ID: 2}, Value: "9007199254740992.99"}}
|
|
rows, err = uc.Ranks(ctx, []string{"lx", "hw"}, "last7", WorkbenchPayHigh)
|
|
if err != nil || rows[0].ID != 1 || r.filter.Begin != "2026-03-25" {
|
|
t.Fatal("金额排序丢失精度或近7天范围错误")
|
|
}
|
|
if _, err := uc.Ranks(ctx, []string{"lx"}, "invalid", WorkbenchRegister); err != ErrWorkbenchPeriod {
|
|
t.Fatal("无效时间维度必须拒绝")
|
|
}
|
|
today := workbenchWindow(uc.now()).Today
|
|
created := today.AddDate(0, 0, -5).Unix()
|
|
future := today.AddDate(0, 0, 1).Unix()
|
|
r.data.Rows = []*WorkbenchRankRow{{AnalyticsKey: AnalyticsKey{ID: 1}}, {AnalyticsKey: AnalyticsKey{ID: 2}, CreatedUnix: &created}, {AnalyticsKey: AnalyticsKey{ID: 3}, CreatedUnix: &created, LastLoginUnix: &future}}
|
|
rows, err = uc.Ranks(ctx, []string{"lx"}, "ignored", WorkbenchUnlogin)
|
|
if err != nil || rows[0].Value != "99999" || rows[1].Value != "5" || rows[2].Value != "0" {
|
|
t.Fatal("未登录回退创建时间、无时间或未来时间错误")
|
|
}
|
|
r.data.Available = false
|
|
if _, err := uc.Ranks(ctx, []string{"lx"}, "", WorkbenchUnlogin); err != ErrWorkbenchUnavailable {
|
|
t.Fatal("缺数据不能返回空榜单")
|
|
}
|
|
}
|