Files
Scoreboard/server/main_test.go
T
AskaEth 2d207fd654 feat: 赛事项目切换 + 记住上次打开的赛事(修「导入赛事包后没数据」)
根因:服务端启动只认 -project(缺省 default),Console 从不传;导入只是把包解到
event_projs/<名字>,没有任何切换入口 → 导入的赛事永远打不开,页面看到的是空的 default。

- 服务端 -project 缺省改为空串;解析顺序:显式 flag(并写回记忆)> <dataDir>/active_project.txt
  > default;backup/passwd 口径一致;启动日志打印生效项目与来源
- GET /api/projects(admin)列出各赛事(是否当前/是否有 event.db/大小/更新时间)
- POST /api/projects/activate(admin):校验后原子写记忆文件 → 审计 → 返回
  {ok,name,restarting:true} → 触发优雅关闭;Console 对非用户发起的退出会自动拉起,
  重启后即打开新赛事(无需改 Console)。本来就是当前赛事则 restarting:false 不重启
- 管理端「导出」页新增赛事项目列表与「切换为当前赛事」;导入成功后提示「已导入为 X,
  但还不是当前赛事」并给「立即切换并重启服务端」;切换后轮询 /api/health 等它回来再刷新
- 文档:使用手册(导入≠切换、生效四步)、常见问题(导入后看不到数据)

自检:服务端 12 包全绿;前端 vue-tsc + build、node --test 66/66;console 14 文件通过。
2026-09-10 09:47:06 +08:00

256 lines
9.5 KiB
Go

package main
import (
"os"
"path/filepath"
"strings"
"testing"
"git.yeij.top/AskaEth/Scoreboard/server/internal/httpapi"
)
// CLI 备份必须用 event 前缀(与备份服务一致),否则 PruneBackups 按 <prefix>-* 清理不到,
// 备份目录会只增不减。
func TestRunBackupUsesEventPrefix(t *testing.T) {
root := t.TempDir()
dataDir := filepath.Join(root, "data")
writeEventDB(t, dataDir)
cfgPath := writeConfig(t, root, dataDir)
if err := runBackup([]string{"-config", cfgPath, "-project", "default", "-reason", "manual"}); err != nil {
t.Fatalf("runBackup: %v", err)
}
entries, err := os.ReadDir(filepath.Join(dataDir, "backups"))
if err != nil {
t.Fatalf("读取备份目录: %v", err)
}
eventBackup := false
for _, e := range entries {
if strings.HasPrefix(e.Name(), "default-") {
t.Errorf("CLI 备份不应使用项目名前缀: %s", e.Name())
}
if strings.HasPrefix(e.Name(), "event-manual-") {
eventBackup = true
}
}
if !eventBackup {
t.Errorf("未生成 event- 前缀的赛事库备份,实际文件: %v", entries)
}
}
// 数据目录搬到安装目录之外后,CLI 备份必须跟 -data 走,否则备份的是旧目录(甚至是空目录)。
func TestRunBackupDataFlagOverridesConfig(t *testing.T) {
root := t.TempDir()
cfgDataDir := filepath.Join(root, "cfg-data")
flagDataDir := filepath.Join(root, "flag-data")
writeEventDB(t, cfgDataDir)
writeEventDB(t, flagDataDir)
cfgPath := writeConfig(t, root, cfgDataDir)
if err := runBackup([]string{"-config", cfgPath, "-project", "default", "-reason", "manual", "-data", flagDataDir}); err != nil {
t.Fatalf("runBackup: %v", err)
}
entries, err := os.ReadDir(filepath.Join(flagDataDir, "backups"))
if err != nil {
t.Fatalf("读取 -data 指向的备份目录: %v", err)
}
if len(entries) == 0 {
t.Fatalf("-data 指向的目录里没有备份文件")
}
if _, err := os.Stat(filepath.Join(cfgDataDir, "backups")); !os.IsNotExist(err) {
t.Fatalf("配置里的数据目录不应被写入: %v", err)
}
}
// -data 传空串等于不传:仍然用配置文件里的数据目录,便携与手工运行的行为不变。
func TestRunBackupEmptyDataFlagKeepsConfig(t *testing.T) {
root := t.TempDir()
cfgDataDir := filepath.Join(root, "cfg-data")
writeEventDB(t, cfgDataDir)
cfgPath := writeConfig(t, root, cfgDataDir)
if err := runBackup([]string{"-config", cfgPath, "-project", "default", "-reason", "manual", "-data", ""}); err != nil {
t.Fatalf("runBackup: %v", err)
}
if _, err := os.Stat(filepath.Join(cfgDataDir, "backups")); err != nil {
t.Fatalf("空 -data 时应写配置里的数据目录: %v", err)
}
}
// writeEventDB 造一个空的赛事库;runBackup 只在 event.db 存在时备份赛事库,空库也走同一条 VACUUM INTO 路径。
func writeEventDB(t *testing.T, dataDir string) {
t.Helper()
writeProjectEventDB(t, dataDir, defaultProjectName)
}
// writeProjectEventDB 在指定项目下造一个空的赛事库。
func writeProjectEventDB(t *testing.T, dataDir, project string) {
t.Helper()
projDir := filepath.Join(dataDir, "event_projs", project)
if err := os.MkdirAll(projDir, 0o755); err != nil {
t.Fatalf("创建赛事目录: %v", err)
}
if err := os.WriteFile(filepath.Join(projDir, "event.db"), nil, 0o644); err != nil {
t.Fatalf("创建赛事库: %v", err)
}
}
// writeActiveProjectFile 写记忆文件,模拟「上次打开过的赛事」。
func writeActiveProjectFile(t *testing.T, dataDir, name string) {
t.Helper()
writeActiveProjectRaw(t, dataDir, name+"\n")
}
// writeActiveProjectRaw 直接写记忆文件原文,用于构造被改坏的记忆文件。
func writeActiveProjectRaw(t *testing.T, dataDir, content string) {
t.Helper()
if err := os.MkdirAll(dataDir, 0o755); err != nil {
t.Fatalf("创建数据目录: %v", err)
}
if err := os.WriteFile(filepath.Join(dataDir, httpapi.ActiveProjectFile), []byte(content), 0o644); err != nil {
t.Fatalf("写记忆文件: %v", err)
}
}
// 启动解析顺序:显式 -project > 记忆文件里的上一次赛事 > default。
// 这正是「导入赛事包后管理面板还是空的」的修复点:Console 不传 -project,只能靠记忆文件。
func TestResolveProjectNamePriority(t *testing.T) {
dataDir := t.TempDir()
name, source, err := resolveProjectName(dataDir, "")
if err != nil || name != defaultProjectName || source != projectSourceDefault {
t.Fatalf("没有记忆文件时应回落 default,得到 %q/%q/%v", name, source, err)
}
name, source, err = resolveProjectName(dataDir, " ")
if err != nil || name != defaultProjectName || source != projectSourceDefault {
t.Fatalf("空白 -project 等于没传,得到 %q/%q/%v", name, source, err)
}
writeActiveProjectFile(t, dataDir, "秋季赛")
name, source, err = resolveProjectName(dataDir, "")
if err != nil || name != "秋季赛" || source != projectSourceFile {
t.Fatalf("有记忆文件时应用它,得到 %q/%q/%v", name, source, err)
}
name, source, err = resolveProjectName(dataDir, "冬季赛")
if err != nil || name != "冬季赛" || source != projectSourceFlag {
t.Fatalf("显式 -project 应压过记忆文件,得到 %q/%q/%v", name, source, err)
}
// 显式传了非法名字必须报错退出,不能静默回落到别的赛事——那会让现场打开错的库。
if _, _, err := resolveProjectName(dataDir, "../逃逸"); err == nil {
t.Fatal("非法 -project 应报错")
}
if _, _, err := resolveProjectName(dataDir, "a/b"); err == nil {
t.Fatal("含分隔符的 -project 应报错")
}
}
// 记忆文件被手工改坏时回落 default,而不是打开一个非法或半截的目录。
func TestResolveProjectNameInvalidMemoryFile(t *testing.T) {
for _, tc := range []struct{ name, content string }{
{"路径穿越", "../逃逸\n"},
{"多行", "好名字\n坏名字\n"},
{"只有空白", " \n"},
{"末尾点号", "赛事.\n"},
} {
dataDir := t.TempDir()
writeActiveProjectRaw(t, dataDir, tc.content)
name, source, err := resolveProjectName(dataDir, "")
if err != nil || name != defaultProjectName || source != projectSourceDefault {
t.Errorf("%s:非法记忆文件应回落 default,得到 %q/%q/%v", tc.name, name, source, err)
}
}
// 数据目录还不存在(首次启动)同样回落 default,不能报错拦启动。
name, source, err := resolveProjectName(filepath.Join(t.TempDir(), "还没有"), "")
if err != nil || name != defaultProjectName || source != projectSourceDefault {
t.Fatalf("数据目录不存在时应回落 default,得到 %q/%q/%v", name, source, err)
}
}
// CLI 备份必须跟服务端打开同一个赛事库:项目名同样从记忆文件解析。
func TestRunBackupUsesRememberedProject(t *testing.T) {
root := t.TempDir()
dataDir := filepath.Join(root, "data")
writeProjectEventDB(t, dataDir, "秋季赛")
writeActiveProjectFile(t, dataDir, "秋季赛")
cfgPath := writeConfig(t, root, dataDir)
if err := runBackup([]string{"-config", cfgPath, "-reason", "manual"}); err != nil {
t.Fatalf("runBackup: %v", err)
}
entries, err := os.ReadDir(filepath.Join(dataDir, "backups"))
if err != nil {
t.Fatalf("读取备份目录: %v", err)
}
eventBackup := false
for _, e := range entries {
if strings.HasPrefix(e.Name(), "event-manual-") {
eventBackup = true
}
}
if !eventBackup {
t.Fatalf("应按记忆文件里的 秋季赛 备份赛事库,实际文件: %v", entries)
}
// 备份是只读操作,绝不能顺手改掉「下次打开哪个赛事」。
raw, err := os.ReadFile(filepath.Join(dataDir, httpapi.ActiveProjectFile))
if err != nil || strings.TrimSpace(string(raw)) != "秋季赛" {
t.Fatalf("记忆文件不该被备份改写,得到 %q / %v", raw, err)
}
}
// 显式 -project 优先于记忆文件:指到没有 event.db 的项目时就不备份赛事库。
func TestRunBackupExplicitProjectBeatsMemory(t *testing.T) {
root := t.TempDir()
dataDir := filepath.Join(root, "data")
writeProjectEventDB(t, dataDir, "秋季赛")
writeActiveProjectFile(t, dataDir, "秋季赛")
cfgPath := writeConfig(t, root, dataDir)
if err := runBackup([]string{"-config", cfgPath, "-project", "没导入过", "-reason", "manual"}); err != nil {
t.Fatalf("runBackup: %v", err)
}
entries, err := os.ReadDir(filepath.Join(dataDir, "backups"))
if err != nil {
t.Fatalf("读取备份目录: %v", err)
}
for _, e := range entries {
if strings.HasPrefix(e.Name(), "event-") {
t.Fatalf("显式 -project 指到空项目时不该备份赛事库,实际文件: %s", e.Name())
}
}
}
// writeConfig 写一份只指定数据目录的最小配置,返回配置路径。
func writeConfig(t *testing.T, dir, dataDir string) string {
t.Helper()
path := filepath.Join(dir, "config.yaml")
raw := "dataDir: " + dataDir + "\nserver:\n port: 19540\n"
if err := os.WriteFile(path, []byte(raw), 0o644); err != nil {
t.Fatalf("写配置: %v", err)
}
return path
}
// 现场按文档用 findstr 搜纯 ASCII 锚找初始密码;中文匹配在 chcp 936/65001 下不可靠。
func TestInitialAdminBannerAsciiAnchor(t *testing.T) {
const anchor = "=== INITIAL ADMIN CREATED (password above) ==="
banner := initialAdminBanner("admin", "s3cret-pass")
if !strings.Contains(banner, anchor) {
t.Fatalf("banner 缺少可搜索的 ASCII 锚 %q:\n%s", anchor, banner)
}
if !strings.Contains(banner, "s3cret-pass") {
t.Fatalf("banner 应包含初始密码:\n%s", banner)
}
// 锚在密码之后,文案 password above 才成立。
if strings.Index(banner, "s3cret-pass") > strings.Index(banner, anchor) {
t.Fatalf("ASCII 锚应在密码行之后:\n%s", banner)
}
}