118 lines
4.8 KiB
Go
118 lines
4.8 KiB
Go
package adminfinance
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"hfb_sys/backend/internal/auditlog"
|
|
"hfb_sys/backend/internal/timeutil"
|
|
|
|
"gorm.io/driver/sqlite"
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/logger"
|
|
)
|
|
|
|
func TestOperatingExpenseCreateListAndVoid(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
|
|
if err != nil {
|
|
t.Fatalf("打开测试数据库失败: %v", err)
|
|
}
|
|
for _, statement := range []string{
|
|
`CREATE TABLE operating_expenses (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT, expense_no TEXT NOT NULL UNIQUE, category TEXT NOT NULL,
|
|
payee_name TEXT NOT NULL, amount_cent INTEGER NOT NULL, occurred_at DATETIME NOT NULL,
|
|
remark TEXT NOT NULL, voucher_url TEXT NOT NULL, status TEXT NOT NULL, created_by INTEGER NOT NULL,
|
|
voided_by INTEGER, voided_at DATETIME, void_reason TEXT NOT NULL DEFAULT '', created_at DATETIME, updated_at DATETIME
|
|
)`,
|
|
`CREATE TABLE admin_users (id INTEGER PRIMARY KEY, nickname TEXT, username TEXT)`,
|
|
`CREATE TABLE audit_logs (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT, actor_type TEXT, actor_id INTEGER, action TEXT, biz_type TEXT,
|
|
biz_id INTEGER, ip TEXT, user_agent TEXT, detail BLOB, created_at DATETIME
|
|
)`,
|
|
} {
|
|
if err := db.Exec(statement).Error; err != nil {
|
|
t.Fatalf("创建测试表失败: %v", err)
|
|
}
|
|
}
|
|
if err := db.Exec(`INSERT INTO admin_users (id, nickname, username) VALUES (7, '运营甲', 'ops_a'), (8, '运营乙', 'ops_b')`).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
repo := NewRepository(db)
|
|
occurredAt := time.Date(2026, 8, 20, 10, 30, 0, 0, timeutil.ShanghaiLocation())
|
|
created, err := repo.CreateOperatingExpense(t.Context(), CreateOperatingExpenseRequest{
|
|
Category: "推广投放",
|
|
PayeeName: "测试媒体",
|
|
AmountCent: 12800,
|
|
OccurredAt: occurredAt,
|
|
Remark: "八月推广费用",
|
|
VoucherURL: "/api/files/object?key=operating-expense%2Fvoucher.webp",
|
|
}, 7, auditlog.Meta{RequestID: "req-create"})
|
|
if err != nil {
|
|
t.Fatalf("创建运营开支失败: %v", err)
|
|
}
|
|
if created.ID == 0 || created.ExpenseNo == "" || created.Status != "paid" || created.CreatedByName != "运营甲" {
|
|
t.Fatalf("创建结果不正确: %+v", created)
|
|
}
|
|
|
|
list, err := repo.OperatingExpenses(t.Context(), OperatingExpenseQuery{
|
|
DateType: "occurred", StartDate: occurredAt.Add(-time.Hour), EndDate: occurredAt.Add(time.Hour), Page: 1, PageSize: 20,
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("查询运营开支失败: %v", err)
|
|
}
|
|
if list.Total != 1 || list.Summary.PaidAmountCent != 12800 || list.Items[0].Category != "推广投放" {
|
|
t.Fatalf("运营开支列表不正确: %+v", list)
|
|
}
|
|
|
|
voided, err := repo.VoidOperatingExpense(t.Context(), created.ID, "重复录入", 8, auditlog.Meta{RequestID: "req-void"})
|
|
if err != nil {
|
|
t.Fatalf("作废运营开支失败: %v", err)
|
|
}
|
|
if voided.Status != "voided" || voided.VoidedByName != "运营乙" || voided.VoidReason != "重复录入" {
|
|
t.Fatalf("作废结果不正确: %+v", voided)
|
|
}
|
|
if _, err := repo.VoidOperatingExpense(t.Context(), created.ID, "重复作废", 8, auditlog.Meta{}); !errors.Is(err, ErrOperatingExpenseNotPaid) {
|
|
t.Fatalf("重复作废错误 = %v, want ErrOperatingExpenseNotPaid", err)
|
|
}
|
|
}
|
|
|
|
func TestOperatingExpenseDashboardSummaryExcludesVoided(t *testing.T) {
|
|
db, err := gorm.Open(sqlite.Open(":memory:"), &gorm.Config{Logger: logger.Default.LogMode(logger.Silent)})
|
|
if err != nil {
|
|
t.Fatalf("打开测试数据库失败: %v", err)
|
|
}
|
|
if err := db.Exec(`CREATE TABLE operating_expenses (
|
|
id INTEGER PRIMARY KEY, amount_cent INTEGER NOT NULL, occurred_at DATETIME NOT NULL, status TEXT NOT NULL
|
|
)`).Error; err != nil {
|
|
t.Fatalf("创建运营开支表失败: %v", err)
|
|
}
|
|
loc := timeutil.ShanghaiLocation()
|
|
inRange := time.Date(2026, 8, 20, 10, 0, 0, 0, loc)
|
|
if err := db.Exec(`INSERT INTO operating_expenses (id, amount_cent, occurred_at, status) VALUES
|
|
(1, 12000, ?, 'paid'), (2, 8000, ?, 'voided'), (3, 5000, ?, 'paid')`, inRange, inRange, inRange.AddDate(0, 0, -1)).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
summary, err := NewRepository(db).operatingExpenseSummary(t.Context(), DashboardQuery{
|
|
StartDate: time.Date(2026, 8, 20, 0, 0, 0, 0, loc),
|
|
EndDate: time.Date(2026, 8, 20, 23, 59, 59, 0, loc),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("运营开支仪表盘统计失败: %v", err)
|
|
}
|
|
if summary.AmountCent != 12000 || summary.Count != 1 {
|
|
t.Fatalf("运营开支统计 = %+v, want 12000/1", summary)
|
|
}
|
|
daily, err := NewRepository(db).dailyOperatingExpenses(t.Context(), DashboardQuery{
|
|
StartDate: time.Date(2026, 8, 20, 0, 0, 0, 0, loc),
|
|
EndDate: time.Date(2026, 8, 20, 23, 59, 59, 0, loc),
|
|
})
|
|
if err != nil {
|
|
t.Fatalf("每日运营开支统计失败: %v", err)
|
|
}
|
|
if len(daily) != 1 || daily[0].Date != "2026-08-20" || daily[0].AmountCent != 12000 || daily[0].Count != 1 {
|
|
t.Fatalf("每日运营开支统计 = %+v, want 2026-08-20/12000/1", daily)
|
|
}
|
|
}
|