subtrack
1f7d0f0cc4
build / build-and-test (push) Successful in 3m19s
1) 工作台速录 - 手机尾号输入框下方新增 3×4 触摸小键盘（1-9、0、⌫、✓），点 ✓ 直接提交；实体键盘仍可输入 - 抽屉顶部新增秒级时钟（HH:MM:SS），位于日期标题上方 2) 日历格子 - 移除排单缩略信息（客户名/套餐色签/“x/y 单”小字） - 保留左侧打手颜色竖条，中间改为大字「待完成 / 总数」，例如 1/2 = 还剩 1 单、共 2 单 - 大屏格子按此前约定保持原样 3) 客户停用 / 恢复 / 彻底删除 - customers 新增 archived 列（0/1）；停用后默认从客户列表、日历、待办、大屏、到期提醒、徽标中隐藏，订阅与打卡记录原样保留 - 恢复后立即回到日历；列表右上角可切换「显示已停用」，已停用行带标记 - 彻底删除为二次确认操作，依赖外键级联清理订阅、打卡、暂停、追加记录与加急标记 - 停用/恢复/删除都会写入操作日志 4) 按钮收纳 - 任务行只露出「完成」，跳过 / 加急（取消加急）/ 追加收进「⋯」菜单；已完成行只露「撤销打卡」等，点空白处自动收起 - 加急、跳过等原有能力不变，仅入口折叠 5) 顶栏 - 在「今日到期」「3 天内到期」右侧新增「7 天内到期」徽标，固定统计未来 7 天 6) 视觉 - 移除管理端背景的横线装饰，改为纯底色；大屏网格背景保留 数据库兼容性： - 只新增 customers.archived 一列，老库启动时自动 ALTER，可重复迁移；原有表结构与接口不变 测试： - 新增单测 TestCustomerArchiveFilters：停用后日历/到期/徽标/搜索/列表全部过滤、恢复回日历、彻底删除级联清理订阅与打卡 - 功能验证 20 项（键盘、格子计数、折叠菜单、7 天徽标、时钟、停用/恢复/删除、背景）全部通过 - 原有 119 项验收回归全绿；go vet / go test ./... 通过；版本号 1.5.0
1109 lines
32 KiB
Go
1109 lines
32 KiB
Go
package store
|
||
|
||
import (
|
||
"database/sql"
|
||
"errors"
|
||
"os"
|
||
"path/filepath"
|
||
"strconv"
|
||
"strings"
|
||
"testing"
|
||
"time"
|
||
|
||
"daili/internal/domain"
|
||
)
|
||
|
||
func newTestDB(t *testing.T) *DB {
|
||
t.Helper()
|
||
db, err := Open(filepath.Join(t.TempDir(), "test.db"))
|
||
if err != nil {
|
||
t.Fatalf("打开数据库失败: %v", err)
|
||
}
|
||
t.Cleanup(func() { _ = db.Close() })
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatalf("迁移失败: %v", err)
|
||
}
|
||
return db
|
||
}
|
||
|
||
func mustPackage(t *testing.T, db *DB, name, unit string, amount int) *Package {
|
||
t.Helper()
|
||
id, err := db.CreatePackage(&Package{Name: name, Unit: unit, DefaultAmount: amount, Color: "#2F5D50"})
|
||
if err != nil {
|
||
t.Fatalf("创建套餐失败: %v", err)
|
||
}
|
||
p, err := db.Package(id)
|
||
if err != nil {
|
||
t.Fatalf("读取套餐失败: %v", err)
|
||
}
|
||
return p
|
||
}
|
||
|
||
func mustCustomer(t *testing.T, db *DB, tail, name string) int64 {
|
||
t.Helper()
|
||
id, err := db.CreateCustomer(&Customer{Tail: tail, Name: name})
|
||
if err != nil {
|
||
t.Fatalf("创建客户失败: %v", err)
|
||
}
|
||
return id
|
||
}
|
||
|
||
func TestCreateSubAndDuplicate(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
})
|
||
if err != nil {
|
||
t.Fatalf("创建订阅失败: %v", err)
|
||
}
|
||
if sub.EndDate != domain.Date(domain.AddDays(today, 29)) {
|
||
t.Fatalf("结束日=%s", sub.EndDate)
|
||
}
|
||
if sub.Unit != domain.UnitDay || sub.Status != domain.StatusActive {
|
||
t.Fatalf("unit=%s status=%s", sub.Unit, sub.Status)
|
||
}
|
||
|
||
if _, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
}); err == nil {
|
||
t.Fatal("同一客户同一套餐应禁止重复订阅")
|
||
}
|
||
|
||
// 初始记录
|
||
adj, err := db.AdjustmentsBySub(sub.ID)
|
||
if err != nil {
|
||
t.Fatalf("读取追加记录失败: %v", err)
|
||
}
|
||
if len(adj) != 1 || adj[0].Kind != "initial" || adj[0].DaysAdded != 30 {
|
||
t.Fatalf("初始追加记录不正确: %+v", adj)
|
||
}
|
||
}
|
||
|
||
func TestExtendDaysFromTodayWhenExpired(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.AddDays(today, -40)), Days: 30,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if sub.EndDate != domain.Date(domain.AddDays(today, -11)) {
|
||
t.Fatalf("end=%s", sub.EndDate)
|
||
}
|
||
|
||
newEnd := domain.AddDays(today, 29)
|
||
plan, err := db.ExtendDays(sub.ID, domain.Date(newEnd), "续费", 0)
|
||
if err != nil {
|
||
t.Fatalf("追加失败: %v", err)
|
||
}
|
||
if domain.Date(plan.Base) != domain.Date(today) {
|
||
t.Fatalf("已到期订阅应从今天起算，实际 %s", domain.Date(plan.Base))
|
||
}
|
||
if plan.Days != 30 {
|
||
t.Fatalf("追加天数=%d want 30", plan.Days)
|
||
}
|
||
got, _ := db.Sub(sub.ID)
|
||
if got.EndDate != domain.Date(newEnd) {
|
||
t.Fatalf("新结束日=%s", got.EndDate)
|
||
}
|
||
if got.Status != domain.StatusActive {
|
||
t.Fatalf("状态=%s", got.Status)
|
||
}
|
||
|
||
adj, _ := db.AdjustmentsBySub(sub.ID)
|
||
if len(adj) != 2 || adj[0].Kind != "extend_days" || adj[0].DaysAdded != 30 {
|
||
t.Fatalf("追加记录不正确: %+v", adj[0])
|
||
}
|
||
if adj[0].BaseDate != domain.Date(today) {
|
||
t.Fatalf("记录起点=%s", adj[0].BaseDate)
|
||
}
|
||
}
|
||
|
||
func TestExtendDaysFromOldEndWhenActive(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
|
||
sub, _ := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
})
|
||
newEnd := domain.AddDays(today, 39)
|
||
plan, err := db.ExtendDays(sub.ID, domain.Date(newEnd), "", 0)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if domain.Date(plan.Base) != domain.Date(domain.AddDays(today, 30)) {
|
||
t.Fatalf("未到期应从原结束日次日起算，实际 %s", domain.Date(plan.Base))
|
||
}
|
||
if plan.Days != 10 {
|
||
t.Fatalf("追加天数=%d want 10", plan.Days)
|
||
}
|
||
}
|
||
|
||
func TestExtendDaysRejectsBackwards(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
|
||
sub, _ := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
})
|
||
if _, err := db.ExtendDays(sub.ID, domain.Date(domain.AddDays(today, 10)), "", 0); err == nil {
|
||
t.Fatal("早于计算起点的日期应被拒绝")
|
||
}
|
||
}
|
||
|
||
func TestAddCountAndCheckinConsumption(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "深渊十次", domain.UnitCount, 10)
|
||
cid := mustCustomer(t, db, "5678", "大鹏")
|
||
today := domain.Today()
|
||
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), TotalCount: 10,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if sub.TotalCount != 10 || sub.UsedCount != 0 {
|
||
t.Fatalf("total=%d used=%d", sub.TotalCount, sub.UsedCount)
|
||
}
|
||
|
||
// 打卡完成 → 扣 1 次
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(today), "done", "", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ := db.Sub(sub.ID)
|
||
if got.UsedCount != 1 {
|
||
t.Fatalf("used=%d want 1", got.UsedCount)
|
||
}
|
||
|
||
// 同一天改成「跳过」→ 返还
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(today), "skipped", "", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ = db.Sub(sub.ID)
|
||
if got.UsedCount != 0 {
|
||
t.Fatalf("used=%d want 0", got.UsedCount)
|
||
}
|
||
|
||
// 再改回完成 → 再扣
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(today), "done", "", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ = db.Sub(sub.ID)
|
||
if got.UsedCount != 1 {
|
||
t.Fatalf("used=%d want 1", got.UsedCount)
|
||
}
|
||
|
||
// 撤销打卡 → 返还
|
||
ci, err := db.CheckinFor(sub.ID, domain.Date(today))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := db.DeleteCheckin(ci.ID, 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ = db.Sub(sub.ID)
|
||
if got.UsedCount != 0 {
|
||
t.Fatalf("used=%d want 0", got.UsedCount)
|
||
}
|
||
|
||
// 次数用尽后不能再打卡
|
||
_, _ = db.sql.Exec(`UPDATE subscriptions SET used_count=total_count WHERE id=?`, sub.ID)
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(domain.AddDays(today, -1)), "done", "", 0); err == nil {
|
||
t.Fatal("次数用尽时打卡应被拒绝")
|
||
}
|
||
|
||
// 追加次数
|
||
if err := db.AddCount(sub.ID, 5, "", "加购", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ = db.Sub(sub.ID)
|
||
if got.TotalCount != 15 {
|
||
t.Fatalf("total=%d want 15", got.TotalCount)
|
||
}
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(domain.AddDays(today, -1)), "done", "", 0); err != nil {
|
||
t.Fatalf("追加后应可打卡: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestPauseResumeShiftsEndDate(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
|
||
sub, _ := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
})
|
||
if err := db.PauseSub(sub.ID, 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ := db.Sub(sub.ID)
|
||
if got.Status != domain.StatusPaused {
|
||
t.Fatalf("status=%s", got.Status)
|
||
}
|
||
|
||
// 把暂停开始时间改成 5 天前，模拟暂停了 5 天
|
||
oldStart := domain.Date(domain.AddDays(today, -5)) + " 09:00:00"
|
||
if _, err := db.sql.Exec(`UPDATE sub_pauses SET start_at=? WHERE sub_id=? AND end_at IS NULL`, oldStart, sub.ID); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
shift, err := db.ResumeSub(sub.ID, 0)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if shift != 5 {
|
||
t.Fatalf("顺延天数=%d want 5", shift)
|
||
}
|
||
got, _ = db.Sub(sub.ID)
|
||
if got.EndDate != domain.Date(domain.AddDays(today, 34)) {
|
||
t.Fatalf("end=%s want %s", got.EndDate, domain.Date(domain.AddDays(today, 34)))
|
||
}
|
||
if got.Status != domain.StatusActive {
|
||
t.Fatalf("status=%s", got.Status)
|
||
}
|
||
}
|
||
|
||
func TestCalendarSubsExcludesOpenEndedCount(t *testing.T) {
|
||
db := newTestDB(t)
|
||
dayPkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cntPkg := mustPackage(t, db, "深渊十次", domain.UnitCount, 10)
|
||
today := domain.Today()
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
|
||
daySub, _ := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: dayPkg.ID, StartDate: domain.Date(today), Days: 30,
|
||
})
|
||
cid2 := mustCustomer(t, db, "5678", "大鹏")
|
||
if _, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid2, PackageID: cntPkg.ID, StartDate: domain.Date(today), TotalCount: 10,
|
||
}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
subs, err := db.CalendarSubs(domain.Date(today), domain.Date(today))
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(subs) != 1 || subs[0].ID != daySub.ID {
|
||
t.Fatalf("无有效期的按次订阅不应出现在日历: %+v", subs)
|
||
}
|
||
}
|
||
|
||
func TestBackup(t *testing.T) {
|
||
db := newTestDB(t)
|
||
mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
path, err := db.BackupNow(14)
|
||
if err != nil {
|
||
t.Fatalf("备份失败: %v", err)
|
||
}
|
||
list, err := db.ListBackups()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(list) != 1 {
|
||
t.Fatalf("备份数量=%d want 1 (%s)", len(list), path)
|
||
}
|
||
// 同日再备份：文件名按秒生成，允许重复
|
||
if _, err := db.BackupNow(1); err != nil {
|
||
// 同一秒内会重名，算作已存在，不视为失败
|
||
t.Logf("第二次备份返回: %v", err)
|
||
}
|
||
list, _ = db.ListBackups()
|
||
if len(list) > 1 {
|
||
t.Fatalf("保留份数未生效: %d", len(list))
|
||
}
|
||
}
|
||
|
||
func TestSettingsAndBadges(t *testing.T) {
|
||
db := newTestDB(t)
|
||
_ = db.SetSetting("warn_days", "5")
|
||
if got := db.Setting("warn_days", "3"); got != "5" {
|
||
t.Fatalf("setting=%s", got)
|
||
}
|
||
if got := db.Setting("missing", "def"); got != "def" {
|
||
t.Fatalf("默认值=%s", got)
|
||
}
|
||
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
today := domain.Today()
|
||
_, _ = db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.AddDays(today, -30)), Days: 31,
|
||
})
|
||
due, due3, _, err := db.Badges(today)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if due != 1 || due3 != 1 {
|
||
t.Fatalf("due=%d due3=%d want 1/1", due, due3)
|
||
}
|
||
}
|
||
|
||
func TestSearchByTail(t *testing.T) {
|
||
db := newTestDB(t)
|
||
mustCustomer(t, db, "8899", "阿飞")
|
||
mustCustomer(t, db, "8890", "阿飞二号")
|
||
today := domain.Today()
|
||
list, total, err := db.ListCustomers("8899", 20, 0, domain.Date(today), 3, false)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if total != 1 || len(list) != 1 || list[0].Name != "阿飞" {
|
||
t.Fatalf("搜索命中 %d 条", total)
|
||
}
|
||
list, total, _ = db.ListCustomers("889", 20, 0, domain.Date(today), 3, false)
|
||
if total != 2 {
|
||
t.Fatalf("前缀搜索应命中 2 条，实际 %d", total)
|
||
}
|
||
}
|
||
|
||
func TestUniqueCustomer(t *testing.T) {
|
||
db := newTestDB(t)
|
||
mustCustomer(t, db, "1234", "老张")
|
||
_, err := db.CreateCustomer(&Customer{Tail: "1234", Name: "老张"})
|
||
if err == nil {
|
||
t.Fatal("重复的尾号+姓名应被拒绝")
|
||
}
|
||
if _, err := db.CreateCustomer(&Customer{Tail: "1234", Name: "另一位"}); err != nil {
|
||
t.Fatalf("同尾号不同姓名应允许: %v", err)
|
||
}
|
||
}
|
||
|
||
var _ = time.Now
|
||
|
||
func TestScreenTopLimitedToTen(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
today := domain.Today()
|
||
for i := 0; i < 12; i++ {
|
||
cid, err := db.CreateCustomer(&Customer{Tail: "T" + strconv.Itoa(i), Name: "客户" + strconv.Itoa(i)})
|
||
if err != nil {
|
||
t.Fatalf("建客户失败: %v", err)
|
||
}
|
||
if _, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(today), Days: 30 - i,
|
||
}); err != nil {
|
||
t.Fatalf("建订阅失败: %v", err)
|
||
}
|
||
}
|
||
st, err := db.ScreenStats(today, 3)
|
||
if err != nil {
|
||
t.Fatalf("统计失败: %v", err)
|
||
}
|
||
if len(st.Top) != 10 {
|
||
t.Fatalf("TOP 榜应为 10 条，实际 %d 条", len(st.Top))
|
||
}
|
||
}
|
||
|
||
func TestBackupKeepPrune(t *testing.T) {
|
||
db := newTestDB(t)
|
||
mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
dir := db.BackupDir()
|
||
names := []string{
|
||
"daili-20260101-000001.db",
|
||
"daili-20260102-000001.db",
|
||
"daili-20260103-000001.db",
|
||
}
|
||
for _, n := range names {
|
||
if _, err := db.BackupTo(filepath.Join(dir, n), 0); err != nil {
|
||
t.Fatalf("备份失败: %v", err)
|
||
}
|
||
}
|
||
if err := db.pruneBackups(2); err != nil {
|
||
t.Fatalf("清理失败: %v", err)
|
||
}
|
||
list, err := db.ListBackups()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if len(list) != 2 {
|
||
t.Fatalf("保留 2 份，实际 %d 份: %v", len(list), list)
|
||
}
|
||
if list[0] != names[2] || list[1] != names[1] {
|
||
t.Fatalf("保留的不是最新两份: %v", list)
|
||
}
|
||
}
|
||
|
||
func TestBackupDirOverride(t *testing.T) {
|
||
db := newTestDB(t)
|
||
custom := filepath.Join(t.TempDir(), "自定义备份")
|
||
db.SetBackupDir(custom)
|
||
if db.BackupDir() != custom {
|
||
t.Fatalf("备份目录未生效: %s", db.BackupDir())
|
||
}
|
||
if _, err := db.BackupNow(5); err != nil {
|
||
t.Fatalf("备份失败: %v", err)
|
||
}
|
||
if _, err := os.Stat(custom); err != nil {
|
||
t.Fatalf("备份目录未创建: %v", err)
|
||
}
|
||
}
|
||
|
||
// copyFile 用于模拟“用户只拷贝数据库文件”。
|
||
func copyFile(t *testing.T, src, dst string) {
|
||
t.Helper()
|
||
data, err := os.ReadFile(src)
|
||
if err != nil {
|
||
t.Fatalf("读取 %s 失败: %v", src, err)
|
||
}
|
||
if err := os.WriteFile(dst, data, 0o644); err != nil {
|
||
t.Fatalf("写入 %s 失败: %v", dst, err)
|
||
}
|
||
}
|
||
|
||
func TestSingleFileDatabase(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "daili.db")
|
||
db, err := Open(path)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if mode, err := db.JournalMode(); err != nil || !strings.EqualFold(mode, "delete") {
|
||
t.Fatalf("journal_mode=%s err=%v", mode, err)
|
||
}
|
||
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
if _, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 30,
|
||
}); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for _, suffix := range []string{"-wal", "-shm"} {
|
||
if _, err := os.Stat(path + suffix); err == nil {
|
||
t.Fatalf("运行中不应出现 %s", suffix)
|
||
}
|
||
}
|
||
if err := db.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for _, suffix := range []string{"-wal", "-shm", "-journal"} {
|
||
if _, err := os.Stat(path + suffix); err == nil {
|
||
t.Fatalf("关闭后不应残留 %s", suffix)
|
||
}
|
||
}
|
||
if fi, err := os.Stat(path); err != nil || fi.Size() < 8192 {
|
||
t.Fatalf("主库大小异常: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestCopyDatabaseIsLossless(t *testing.T) {
|
||
srcDir := t.TempDir()
|
||
srcPath := filepath.Join(srcDir, "daili.db")
|
||
db, err := Open(srcPath)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "1234", "老张")
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 30,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(domain.Today()), "done", "迁移前打卡", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 只拷贝 daili.db 一个文件
|
||
dstDir := t.TempDir()
|
||
dstPath := filepath.Join(dstDir, "daili.db")
|
||
copyFile(t, srcPath, dstPath)
|
||
|
||
cp, err := Open(dstPath)
|
||
if err != nil {
|
||
t.Fatalf("打开副本失败: %v", err)
|
||
}
|
||
defer cp.Close()
|
||
if n, err := cp.CustomerCount(); err != nil || n != 1 {
|
||
t.Fatalf("客户数=%d err=%v", n, err)
|
||
}
|
||
got, err := cp.Sub(sub.ID)
|
||
if err != nil || got.CustomerTail != "1234" || got.EndDate != sub.EndDate {
|
||
t.Fatalf("订阅丢失: %+v err=%v", got, err)
|
||
}
|
||
ci, err := cp.CheckinFor(sub.ID, domain.Date(domain.Today()))
|
||
if err != nil || ci.Note != "迁移前打卡" {
|
||
t.Fatalf("打卡记录丢失: %+v err=%v", ci, err)
|
||
}
|
||
for _, suffix := range []string{"-wal", "-shm"} {
|
||
if _, err := os.Stat(dstPath + suffix); err == nil {
|
||
t.Fatalf("副本目录出现了 %s", suffix)
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestDeleteDatabaseResets(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "daili.db")
|
||
db, err := Open(path)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
mustCustomer(t, db, "1234", "老张")
|
||
if err := db.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for _, p := range []string{path, path + "-wal", path + "-shm", path + "-journal"} {
|
||
_ = os.Remove(p)
|
||
}
|
||
|
||
fresh, err := Open(path)
|
||
if err != nil {
|
||
t.Fatalf("删除后重建失败: %v", err)
|
||
}
|
||
defer fresh.Close()
|
||
if err := fresh.Migrate(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if n, _ := fresh.CustomerCount(); n != 0 {
|
||
t.Fatalf("删库后应无客户，实际 %d", n)
|
||
}
|
||
if n, _ := fresh.CountUsers(); n != 0 {
|
||
t.Fatalf("删库后应无管理员，实际 %d", n)
|
||
}
|
||
}
|
||
|
||
func TestStaleSidecarsCleanup(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "daili.db")
|
||
// 模拟“删了主库但留下旧版本 WAL”的情况
|
||
if err := os.WriteFile(path+"-wal", []byte("stale"), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := os.WriteFile(path+"-shm", []byte("stale"), 0o644); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
db, err := Open(path)
|
||
if err != nil {
|
||
t.Fatalf("打开失败: %v", err)
|
||
}
|
||
defer db.Close()
|
||
for _, suffix := range []string{"-wal", "-shm"} {
|
||
if _, err := os.Stat(path + suffix); err == nil {
|
||
t.Fatalf("残留的 %s 应被清理", suffix)
|
||
}
|
||
}
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
|
||
func TestOpenLegacyWALDatabaseKeepsData(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "legacy.db")
|
||
|
||
// 模拟 v1.1 的 WAL 库
|
||
dsn := strings.ReplaceAll(path, "\\", "/") + "?_pragma=journal_mode(WAL)"
|
||
old, err := sql.Open("sqlite", dsn)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := old.Exec(`CREATE TABLE legacy(x INTEGER)`); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := old.Exec(`INSERT INTO legacy VALUES(42)`); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := old.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
db, err := Open(path)
|
||
if err != nil {
|
||
t.Fatalf("打开旧库失败: %v", err)
|
||
}
|
||
defer db.Close()
|
||
var x int
|
||
if err := db.sql.QueryRow(`SELECT x FROM legacy`).Scan(&x); err != nil || x != 42 {
|
||
t.Fatalf("旧库数据丢失: x=%d err=%v", x, err)
|
||
}
|
||
if mode, _ := db.JournalMode(); !strings.EqualFold(mode, "delete") {
|
||
t.Fatalf("旧库未切到单文件模式: %s", mode)
|
||
}
|
||
}
|
||
|
||
func TestPackageCalendarSeparate(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "每日任务", domain.UnitDay, 30)
|
||
if pkg.CalendarSeparate {
|
||
t.Fatal("新建套餐默认不应勾选独立日历")
|
||
}
|
||
pkg.CalendarSeparate = true
|
||
if err := db.UpdatePackage(pkg); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, err := db.Package(pkg.ID)
|
||
if err != nil || !got.CalendarSeparate {
|
||
t.Fatalf("独立日历标记未保存: %+v err=%v", got, err)
|
||
}
|
||
list, err := db.ListPackages(true)
|
||
if err != nil || len(list) != 1 || !list[0].CalendarSeparate {
|
||
t.Fatalf("列表未带独立日历标记: %+v err=%v", list, err)
|
||
}
|
||
got.CalendarSeparate = false
|
||
if err := db.UpdatePackage(got); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if again, _ := db.Package(pkg.ID); again.CalendarSeparate {
|
||
t.Fatal("取消勾选失败")
|
||
}
|
||
}
|
||
|
||
func TestTaskUrgentFlags(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
cid := mustCustomer(t, db, "4321", "老王")
|
||
sub1, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 30,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
pkg2 := mustPackage(t, db, "按次", domain.UnitCount, 10)
|
||
sub2, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg2.ID, StartDate: domain.Date(domain.Today()),
|
||
TotalCount: 10, EndDate: domain.Date(domain.AddDays(domain.Today(), 30)),
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
d1 := domain.Date(domain.Today())
|
||
d2 := domain.Date(domain.AddDays(domain.Today(), 1))
|
||
|
||
if err := db.SetUrgent(sub1.ID, d1, true); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.SetUrgent(sub2.ID, d2, true); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
flags, err := db.UrgentFlags(d1, d2)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if !flags[sub1.ID][d1] || !flags[sub2.ID][d2] {
|
||
t.Fatalf("加急标记缺失: %+v", flags)
|
||
}
|
||
if ok, _ := db.IsUrgent(sub1.ID, d1); !ok {
|
||
t.Fatal("IsUrgent 应为 true")
|
||
}
|
||
if n, _ := db.CountUrgent(d1, d2); n != 2 {
|
||
t.Fatalf("加急计数=%d 期望 2", n)
|
||
}
|
||
// 重复设置不报错（UPSERT）
|
||
if err := db.SetUrgent(sub1.ID, d1, true); err != nil {
|
||
t.Fatalf("重复标记加急失败: %v", err)
|
||
}
|
||
// 取消
|
||
if err := db.SetUrgent(sub1.ID, d1, false); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if ok, _ := db.IsUrgent(sub1.ID, d1); ok {
|
||
t.Fatal("取消加急后仍为 true")
|
||
}
|
||
if n, _ := db.CountUrgent(d1, d2); n != 1 {
|
||
t.Fatalf("取消后计数=%d 期望 1", n)
|
||
}
|
||
// 非法日期
|
||
if err := db.SetUrgent(sub1.ID, "2026-13-40", true); err == nil {
|
||
t.Fatal("非法日期应报错")
|
||
}
|
||
// 不存在的订阅
|
||
if err := db.SetUrgent(99999, d1, true); err == nil {
|
||
t.Fatal("不存在的订阅应报错")
|
||
}
|
||
}
|
||
|
||
// TestMigrateOldDatabaseAddsColumn 模拟 v1.1 的旧库：packages 表没有 calendar_separate。
|
||
func TestMigrateOldDatabaseAddsColumn(t *testing.T) {
|
||
dir := t.TempDir()
|
||
path := filepath.Join(dir, "old.db")
|
||
dsn := strings.ReplaceAll(path, "\\", "/")
|
||
old, err := sql.Open("sqlite", dsn)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
stmts := []string{
|
||
`CREATE TABLE packages (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
name TEXT NOT NULL,
|
||
unit TEXT NOT NULL,
|
||
default_amount INTEGER NOT NULL DEFAULT 30,
|
||
price_cents INTEGER,
|
||
color TEXT NOT NULL DEFAULT '#2F5D50',
|
||
note TEXT NOT NULL DEFAULT '',
|
||
archived INTEGER NOT NULL DEFAULT 0,
|
||
sort_order INTEGER NOT NULL DEFAULT 0,
|
||
created_at TEXT NOT NULL,
|
||
updated_at TEXT NOT NULL
|
||
)`,
|
||
`INSERT INTO packages(name,unit,default_amount,color,note,archived,sort_order,created_at,updated_at)
|
||
VALUES('老套餐','day',30,'#2F5D50','',0,1,'2026-01-01 00:00:00','2026-01-01 00:00:00')`,
|
||
`CREATE TABLE users (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
username TEXT NOT NULL UNIQUE,
|
||
display_name TEXT NOT NULL DEFAULT '',
|
||
password_hash TEXT NOT NULL,
|
||
is_active INTEGER NOT NULL DEFAULT 1,
|
||
created_at TEXT NOT NULL,
|
||
last_login_at TEXT
|
||
)`,
|
||
`INSERT INTO users(username,display_name,password_hash,is_active,created_at)
|
||
VALUES('admin','老板','x',1,'2026-01-01 00:00:00')`,
|
||
`CREATE TABLE customers (
|
||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||
tail TEXT NOT NULL,
|
||
name TEXT NOT NULL,
|
||
game_name TEXT NOT NULL DEFAULT '',
|
||
game_server TEXT NOT NULL DEFAULT '',
|
||
contact TEXT NOT NULL DEFAULT '',
|
||
tags TEXT NOT NULL DEFAULT '',
|
||
source TEXT NOT NULL DEFAULT '',
|
||
note TEXT NOT NULL DEFAULT '',
|
||
created_at TEXT NOT NULL,
|
||
updated_at TEXT NOT NULL,
|
||
UNIQUE (tail, name)
|
||
)`,
|
||
`INSERT INTO customers(tail,name,created_at,updated_at)
|
||
VALUES('1001','老客户','2026-01-01 00:00:00','2026-01-01 00:00:00')`,
|
||
}
|
||
for _, s := range stmts {
|
||
if _, err := old.Exec(s); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
}
|
||
if err := old.Close(); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
db, err := Open(path)
|
||
if err != nil {
|
||
t.Fatalf("打开旧库失败: %v", err)
|
||
}
|
||
defer db.Close()
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatalf("迁移失败: %v", err)
|
||
}
|
||
pkg, err := db.Package(1)
|
||
if err != nil {
|
||
t.Fatalf("读取老套餐失败: %v", err)
|
||
}
|
||
if pkg.Name != "老套餐" || pkg.CalendarSeparate {
|
||
t.Fatalf("老数据/默认值不对: %+v", pkg)
|
||
}
|
||
// 新列可写
|
||
pkg.CalendarSeparate = true
|
||
if err := db.UpdatePackage(pkg); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if got, _ := db.Package(1); !got.CalendarSeparate {
|
||
t.Fatal("迁移后新列不可写")
|
||
}
|
||
// 老库的 users / customers 也补上了新列
|
||
var userColor string
|
||
if err := db.sql.QueryRow(`SELECT IFNULL(color,'') FROM users WHERE id=1`).Scan(&userColor); err != nil {
|
||
t.Fatalf("users.color 未补上: %v", err)
|
||
}
|
||
if err := db.SetUserColor(1, "#123456"); err != nil {
|
||
t.Fatalf("老库 users.color 不可写: %v", err)
|
||
}
|
||
c, err := db.Customer(1)
|
||
if err != nil {
|
||
t.Fatalf("老客户读取失败: %v", err)
|
||
}
|
||
if c.AssigneeID != 0 {
|
||
t.Fatalf("老客户不应有打手: %+v", c)
|
||
}
|
||
if err := db.SetCustomerAssignee(1, 1); err != nil {
|
||
t.Fatalf("老库 customers.assignee_id 不可写: %v", err)
|
||
}
|
||
if c2, _ := db.Customer(1); c2.AssigneeID != 1 {
|
||
t.Fatalf("指派未生效: %+v", c2)
|
||
}
|
||
|
||
// 新表 task_flags 已建立
|
||
var tbl string
|
||
if err := db.sql.QueryRow(`SELECT name FROM sqlite_master WHERE type='table' AND name='task_flags'`).Scan(&tbl); err != nil {
|
||
t.Fatalf("task_flags 表未建立: %v", err)
|
||
}
|
||
// 迁移可重复执行
|
||
if err := db.Migrate(); err != nil {
|
||
t.Fatalf("重复迁移失败: %v", err)
|
||
}
|
||
}
|
||
|
||
func TestUserColorAndAssignee(t *testing.T) {
|
||
db := newTestDB(t)
|
||
|
||
// 新建管理员自动分配调色板颜色
|
||
u1, err := db.CreateUser("dashou1", "阿强", "daili123")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
u2, err := db.CreateUser("dashou2", "小美", "daili123")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if u1.Color == "" || u2.Color == "" || u1.Color == u2.Color {
|
||
t.Fatalf("自动颜色不合理: %q %q", u1.Color, u2.Color)
|
||
}
|
||
|
||
// 改色（小写转大写）
|
||
if err := db.SetUserColor(u2.ID, "#c0392b"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, err := db.UserByID(u2.ID)
|
||
if err != nil || got.Color != "#C0392B" {
|
||
t.Fatalf("颜色未保存或未规范化: %+v err=%v", got, err)
|
||
}
|
||
// 非法颜色
|
||
if err := db.SetUserColor(u2.ID, "red"); err == nil {
|
||
t.Fatal("非法颜色应报错")
|
||
}
|
||
if err := db.SetUserColor(u2.ID, "#12345"); err == nil {
|
||
t.Fatal("长度不对应报错")
|
||
}
|
||
// 不存在的账户
|
||
if err := db.SetUserColor(9999, "#123456"); err == nil {
|
||
t.Fatal("不存在的账户应报错")
|
||
}
|
||
|
||
// 客户指派
|
||
cid := mustCustomer(t, db, "3001", "客户甲")
|
||
if err := db.SetCustomerAssignee(cid, u1.ID); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
c, err := db.Customer(cid)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if c.AssigneeID != u1.ID || c.AssigneeName != "阿强" || c.AssigneeColor != "#2F5D50" {
|
||
t.Fatalf("客户指派信息不对: %+v", c)
|
||
}
|
||
// 取消指派
|
||
if err := db.SetCustomerAssignee(cid, 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if c2, _ := db.Customer(cid); c2.AssigneeID != 0 || c2.AssigneeName != "" {
|
||
t.Fatalf("取消指派失败: %+v", c2)
|
||
}
|
||
// 指派给不存在的打手
|
||
if err := db.SetCustomerAssignee(cid, 9999); err == nil {
|
||
t.Fatal("不存在的打手应报错")
|
||
}
|
||
|
||
// 客户创建时直接带指派
|
||
cid2, err := db.CreateCustomer(&Customer{Tail: "3002", Name: "客户乙", AssigneeID: u2.ID})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
c3, _ := db.Customer(cid2)
|
||
if c3.AssigneeName != "小美" || c3.AssigneeColor != "#C0392B" {
|
||
t.Fatalf("新建客户的指派未生效: %+v", c3)
|
||
}
|
||
|
||
// 订阅联表能带出打手
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid2, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 30,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
gotSub, err := db.Sub(sub.ID)
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if gotSub.AssigneeName != "小美" || gotSub.AssigneeColor != "#C0392B" {
|
||
t.Fatalf("订阅未带出打手: %+v", gotSub)
|
||
}
|
||
|
||
// 只有启用中的账户可作为打手候选
|
||
if err := db.SetUserActive(u1.ID, false); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
users, err := db.ActiveUsers()
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
for _, u := range users {
|
||
if u.ID == u1.ID {
|
||
t.Fatal("停用账户不应出现在打手候选里")
|
||
}
|
||
}
|
||
}
|
||
|
||
func TestUpdateUserByAdmin(t *testing.T) {
|
||
db := newTestDB(t)
|
||
|
||
admin, err := db.CreateUser("admin", "老板", "daili123")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
u, err := db.CreateUser("dashou", "阿强", "dashou123")
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
// 建一个会话，稍后用密码重置把它踢掉
|
||
token := "tok-" + domain.Date(domain.Today())
|
||
if err := db.CreateSession(token, u.ID, time.Hour, "127.0.0.1"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := db.SessionUser(token); err != nil {
|
||
t.Fatalf("会话应有效: %v", err)
|
||
}
|
||
|
||
// 只改昵称：会话保留
|
||
if err := db.UpdateUserByAdmin(u.ID, "dashou", "强哥", ""); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
got, _ := db.UserByID(u.ID)
|
||
if got.DisplayName != "强哥" || got.Username != "dashou" {
|
||
t.Fatalf("昵称未更新: %+v", got)
|
||
}
|
||
if _, err := db.SessionUser(token); err != nil {
|
||
t.Fatalf("未改密码不应踢下线: %v", err)
|
||
}
|
||
|
||
// 改用户名 + 重置密码：会话失效，新密码可登录
|
||
if err := db.UpdateUserByAdmin(u.ID, "laoban", "强哥", "newpass123"); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := db.SessionUser(token); err == nil {
|
||
t.Fatal("改密后旧会话应失效")
|
||
}
|
||
if _, err := db.Authenticate("laoban", "newpass123"); err != nil {
|
||
t.Fatalf("新密码应可登录: %v", err)
|
||
}
|
||
if _, err := db.Authenticate("laoban", "dashou123"); err == nil {
|
||
t.Fatal("旧密码不应还能登录")
|
||
}
|
||
|
||
// 用户名冲突
|
||
if err := db.UpdateUserByAdmin(u.ID, "admin", "", ""); !errors.Is(err, ErrDuplicate) {
|
||
t.Fatalf("重名应返回 ErrDuplicate, got %v", err)
|
||
}
|
||
// 空用户名
|
||
if err := db.UpdateUserByAdmin(u.ID, " ", "", ""); err == nil {
|
||
t.Fatal("空用户名应报错")
|
||
}
|
||
// 超管自己也在表里，但接口层才限制“只能改别人”
|
||
if admin.ID != 1 {
|
||
t.Fatalf("首个账户 id 应为 1，实际 %d", admin.ID)
|
||
}
|
||
}
|
||
|
||
func TestCustomerArchiveFilters(t *testing.T) {
|
||
db := newTestDB(t)
|
||
pkg := mustPackage(t, db, "包月日常", domain.UnitDay, 30)
|
||
|
||
// 一个正常客户 + 一个按次客户
|
||
cid := mustCustomer(t, db, "7001", "小张")
|
||
sub, err := db.CreateSub(CreateSubInput{
|
||
CustomerID: cid, PackageID: pkg.ID, StartDate: domain.Date(domain.Today()), Days: 10,
|
||
})
|
||
if err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
// 打卡 + 加急，稍后验证级联删除
|
||
if err := db.SaveCheckin(sub.ID, domain.Date(domain.Today()), "done", "", 0); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if err := db.SetUrgent(sub.ID, domain.Date(domain.Today()), true); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
|
||
today := domain.Date(domain.Today())
|
||
subs, err := db.CalendarSubs(today, today)
|
||
if err != nil || len(subs) != 1 {
|
||
t.Fatalf("停用前日历应有 1 条: %v err=%v", len(subs), err)
|
||
}
|
||
if n, _ := db.CustomerCount(); n != 1 {
|
||
t.Fatalf("停用前客户数=1, got %d", n)
|
||
}
|
||
|
||
// 停用
|
||
if err := db.SetCustomerArchived(cid, true); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if subs, _ := db.CalendarSubs(today, today); len(subs) != 0 {
|
||
t.Fatalf("停用后不应上日历: %d", len(subs))
|
||
}
|
||
if due, _ := db.DueOn(domain.Date(domain.AddDays(domain.Today(), 9))); len(due) != 0 {
|
||
t.Fatalf("停用后不应提醒到期: %d", len(due))
|
||
}
|
||
if n, _ := db.CustomerCount(); n != 0 {
|
||
t.Fatalf("停用后客户数应为 0, got %d", n)
|
||
}
|
||
if list, total, _ := db.ListCustomers("", 50, 0, today, 3, false); total != 0 || len(list) != 0 {
|
||
t.Fatalf("默认列表不应包含停用客户: total=%d", total)
|
||
}
|
||
if list, total, _ := db.ListCustomers("", 50, 0, today, 3, true); total != 1 || len(list) != 1 || !list[0].Archived {
|
||
t.Fatalf("已停用视图应包含: total=%d list=%+v", total, list)
|
||
}
|
||
if c, _ := db.Customer(cid); !c.Archived {
|
||
t.Fatal("Customer() 应返回 Archived=true")
|
||
}
|
||
if due, _, _, _ := db.Badges(domain.Today()); due != 0 {
|
||
t.Fatalf("停用后徽标应为 0, got %d", due)
|
||
}
|
||
if all, _ := db.CustomersByTail("7001"); len(all) != 0 {
|
||
t.Fatalf("停用客户不应被尾号搜到: %d", len(all))
|
||
}
|
||
|
||
// 恢复
|
||
if err := db.SetCustomerArchived(cid, false); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if subs, _ := db.CalendarSubs(today, today); len(subs) != 1 {
|
||
t.Fatalf("恢复后应回到日历: %d", len(subs))
|
||
}
|
||
|
||
// 彻底删除：订阅/打卡/加急一起消失
|
||
if err := db.DeleteCustomer(cid); err != nil {
|
||
t.Fatal(err)
|
||
}
|
||
if _, err := db.Customer(cid); err == nil {
|
||
t.Fatal("删除后不应还能查到客户")
|
||
}
|
||
if subs, _ := db.CalendarSubs(today, today); len(subs) != 0 {
|
||
t.Fatalf("删除后订阅应级联消失: %d", len(subs))
|
||
}
|
||
var flags, checkins int
|
||
_ = db.sql.QueryRow(`SELECT COUNT(*) FROM task_flags`).Scan(&flags)
|
||
_ = db.sql.QueryRow(`SELECT COUNT(*) FROM checkins`).Scan(&checkins)
|
||
if flags != 0 || checkins != 0 {
|
||
t.Fatalf("删除后打卡/加急应级联删除: flags=%d checkins=%d", flags, checkins)
|
||
}
|
||
if err := db.DeleteCustomer(9999); err == nil {
|
||
t.Fatal("删除不存在的客户应报错")
|
||
}
|
||
}
|