AskaEth
bc8b1ee6c0
- M3 认证/权限/审计/限流 + 全部 REST 接口 - M4 管理面板（MIUIx + 成绩录入/积分榜下钻/大屏控制） - M5a 大屏页面（六页 + 明细覆盖层 + 缩放适配） - M6 WebSocket 实时同步中心 - M7 xlsx 导出 + 赛事包 zip + 备份 - 规则包 YAML 导入导出（差异预览） - 字体 vendor 子集化（4 套 x2 字重） - 集成：realtime/export/rules 接入 main+httpapi；backup 子命令 - 修复：中间件包装 ResponseWriter 丢失 Hijacker，导致 WS 握手 501
124 lines
3.4 KiB
Go
124 lines
3.4 KiB
Go
package export
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestBackupAndRetention(t *testing.T) {
|
|
db := openEventDB(t)
|
|
seedEvent(t, db)
|
|
|
|
dir := t.TempDir()
|
|
base := time.Date(2026, 9, 9, 10, 0, 0, 0, time.Local)
|
|
|
|
info, err := Backup(db, BackupOptions{Dir: dir, Prefix: "event", Reason: "publish", Keep: 3, Now: base})
|
|
if err != nil {
|
|
t.Fatalf("备份: %v", err)
|
|
}
|
|
if info.Size == 0 || info.SHA256 == "" {
|
|
t.Errorf("备份信息不完整: %+v", info)
|
|
}
|
|
if info.Reason != "publish" {
|
|
t.Errorf("备份原因 = %q，期望 publish", info.Reason)
|
|
}
|
|
if _, err := os.Stat(info.Path); err != nil {
|
|
t.Fatalf("备份文件不存在: %v", err)
|
|
}
|
|
|
|
// 快照必须可打开，且数据与源库一致。
|
|
bdb := openSQLite(t, info.Path)
|
|
if got := countRows(t, bdb, "teams"); got != 2 {
|
|
t.Errorf("备份内 teams = %d，期望 2", got)
|
|
}
|
|
if got := countRows(t, bdb, "score_ledger"); got != 2 {
|
|
t.Errorf("备份内 score_ledger = %d，期望 2", got)
|
|
}
|
|
_ = bdb.Close()
|
|
|
|
// 再生成 5 份，Keep=3 应只留最近 3 份。
|
|
for i := 1; i <= 5; i++ {
|
|
if _, err := Backup(db, BackupOptions{
|
|
Dir: dir, Prefix: "event", Reason: "auto", Keep: 3,
|
|
Now: base.Add(time.Duration(i) * time.Minute),
|
|
}); err != nil {
|
|
t.Fatalf("第 %d 次备份: %v", i, err)
|
|
}
|
|
}
|
|
list, err := ListBackups(dir, "event")
|
|
if err != nil {
|
|
t.Fatalf("列出备份: %v", err)
|
|
}
|
|
if len(list) != 3 {
|
|
t.Fatalf("保留份数 = %d，期望 3", len(list))
|
|
}
|
|
if list[0].Reason != "auto" {
|
|
t.Errorf("最新备份原因 = %q，期望 auto", list[0].Reason)
|
|
}
|
|
|
|
latest, ok, err := LatestBackup(dir, "event")
|
|
if err != nil || !ok {
|
|
t.Fatalf("最近备份: ok=%v err=%v", ok, err)
|
|
}
|
|
if !latest.CreatedAt.Equal(base.Add(5 * time.Minute)) {
|
|
t.Errorf("最近备份时间 = %s，期望 %s", latest.CreatedAt, base.Add(5*time.Minute))
|
|
}
|
|
|
|
if _, err := PruneBackups(dir, "event", 1); err != nil {
|
|
t.Fatalf("手动清理: %v", err)
|
|
}
|
|
list, err = ListBackups(dir, "event")
|
|
if err != nil {
|
|
t.Fatalf("清理后列出备份: %v", err)
|
|
}
|
|
if len(list) != 1 {
|
|
t.Errorf("清理后份数 = %d，期望 1", len(list))
|
|
}
|
|
}
|
|
|
|
func TestBackupBeforePublishAndUniqueNames(t *testing.T) {
|
|
db := openEventDB(t)
|
|
seedEvent(t, db)
|
|
|
|
dir := t.TempDir()
|
|
base := time.Date(2026, 9, 9, 11, 0, 0, 0, time.Local)
|
|
|
|
a, err := Backup(db, BackupOptions{Dir: dir, Prefix: "event", Now: base, SkipPrune: true})
|
|
if err != nil {
|
|
t.Fatalf("备份 A: %v", err)
|
|
}
|
|
b, err := Backup(db, BackupOptions{Dir: dir, Prefix: "event", Now: base, SkipPrune: true})
|
|
if err != nil {
|
|
t.Fatalf("备份 B: %v", err)
|
|
}
|
|
if a.Path == b.Path {
|
|
t.Errorf("同一秒的两份备份重名: %s", a.Path)
|
|
}
|
|
|
|
info, err := BackupBeforePublish(db, BackupOptions{Dir: dir, Prefix: "event", Keep: 20, Now: base.Add(time.Hour)})
|
|
if err != nil {
|
|
t.Fatalf("发布前备份: %v", err)
|
|
}
|
|
if info.Reason != "publish" {
|
|
t.Errorf("发布前备份原因 = %q，期望 publish", info.Reason)
|
|
}
|
|
if filepath.Dir(info.Path) != dir {
|
|
t.Errorf("备份目录 = %q，期望 %q", filepath.Dir(info.Path), dir)
|
|
}
|
|
}
|
|
|
|
func TestListBackupsMissingDir(t *testing.T) {
|
|
list, err := ListBackups(filepath.Join(t.TempDir(), "不存在"), "event")
|
|
if err != nil {
|
|
t.Fatalf("目录不存在不应报错: %v", err)
|
|
}
|
|
if len(list) != 0 {
|
|
t.Errorf("备份数 = %d，期望 0", len(list))
|
|
}
|
|
if _, ok, err := LatestBackup(filepath.Join(t.TempDir(), "不存在"), "event"); err != nil || ok {
|
|
t.Errorf("空目录的最近备份: ok=%v err=%v", ok, err)
|
|
}
|
|
}
|