hd4k-downloader/internal/core/core_test.go

package core

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"testing"
	"time"

	"hd4k-downloader/internal/hd4k"
)

// ---------------------------------------------------------------- 测试辅助

func newTestStore(t *testing.T) *Store {
	t.Helper()
	dir := t.TempDir()
	s := NewStore(filepath.Join(dir, "state.json"))
	t.Cleanup(func() {
		s.Stop()
		s.Flush()
	})
	return s
}

func sampleGallery() *hd4k.Gallery {
	return &hd4k.Gallery{
		ID:    "migmg.ssuu.uk/content/09/myua-sweetbox-vol52.html",
		URL:   "https://migmg.ssuu.uk/content/09/myua-sweetbox-vol52.html",
		Host:  "migmg.ssuu.uk",
		Path:  "/content/09/myua-sweetbox-vol52.html",
		Slug:  "myua-sweetbox-vol52.html",
		Title: "Myu_a SWEETBOX Vol.52 [190MB-53photos]",
		Total: 3,
		Images: []hd4k.ImageInfo{
			{Index: 1, URL: "https://i0.wp.com/pic.4khd.com/a/1.webp?w=1300", Filename: "tbox-001.webp"},
			{Index: 2, URL: "https://i0.wp.com/pic.4khd.com/a/2.webp?w=1300", Filename: "tbox-002.webp"},
			{Index: 3, URL: "https://i0.wp.com/pic.4khd.com/a/3.webp?w=1300", Filename: "tbox-003.webp"},
		},
		Meta: hd4k.GalleryMeta{Category: "09", Slug: "myua-sweetbox-vol52.html", Host: "migmg.ssuu.uk"},
	}
}

// ---------------------------------------------------------------- 目录模板

func TestBuildDirTemplates(t *testing.T) {
	g := sampleGallery()

	cases := []struct{ tpl, want string }{
		{"{title}", "Myu_a SWEETBOX Vol.52 [190MB-53photos]"},
		{"{name}", "myua-sweetbox-vol52"},
		{"{slug}", "myua-sweetbox-vol52.html"},
		{"{host}/{title}", "migmg.ssuu.uk/Myu_a SWEETBOX Vol.52 [190MB-53photos]"},
		{"{category}/{name}", "09/myua-sweetbox-vol52"},
		{"{unknown}", "{unknown}"}, // 未知占位符原样保留
	}

	for _, c := range cases {
		got := BuildDir(c.tpl, g)
		if got != filepath.FromSlash(c.want) {
			t.Errorf("BuildDir(%q) = %q, 想要 %q", c.tpl, got, filepath.FromSlash(c.want))
		}
	}
}

// 占位符替换出来的值里的斜杠，绝不能变成真的目录层级。
// 站点标题里可能带 / : * ? " < > | 这些字符，直接落盘就是非法文件名。
func TestBuildDirNoInjectedHierarchy(t *testing.T) {
	g := sampleGallery()
	g.Title = `Bad/Title: With*Weird|Chars?And"Quotes"<Here>`

	got := BuildDir("{title}", g)
	segs := strings.Split(got, string(filepath.Separator))
	if len(segs) != 1 {
		t.Fatalf("标题里的非法字符造出了 %d 层目录: %q", len(segs), got)
	}
	if !validSegmentName(segs[0]) {
		t.Errorf("段名仍不合法: %q", segs[0])
	}
}

// 模板里**写出来的**分隔符才产生分层
func TestBuildDirTemplateSeparators(t *testing.T) {
	g := sampleGallery()
	got := BuildDir("{category}/{name}/{title}", g)
	if len(strings.Split(got, string(filepath.Separator))) != 3 {
		t.Errorf("模板里两个斜杠应产生 3 层: %q", got)
	}
}

func TestBuildDirWindowsReservedName(t *testing.T) {
	g := sampleGallery()
	g.Title = "CON"
	got := BuildDir("{title}", g)
	if strings.EqualFold(got, "CON") {
		t.Errorf("Windows 保留名没被转义: %q", got)
	}
}

func TestBuildDirEmptyTemplateDefaultsToTitle(t *testing.T) {
	g := sampleGallery()
	if got := BuildDir("", g); got != SanitizeSegment(g.Title, g.Slug) {
		t.Errorf("空模板应回落到标题: %q", got)
	}
}

func TestListPlaceholders(t *testing.T) {
	ph := ListPlaceholders()
	want := map[string]bool{
		"{title}": false, "{name}": false, "{slug}": false,
		"{host}": false, "{category}": false, "{id}": false,
	}
	for _, p := range ph {
		if _, ok := want[p]; ok {
			want[p] = true
		}
	}
	for p, seen := range want {
		if !seen {
			t.Errorf("占位符列表缺了 %s", p)
		}
	}
}

// ---------------------------------------------------------------- Store

func TestStoreAddTasksDedup(t *testing.T) {
	s := newTestStore(t)
	s.UpdateSettings(map[string]interface{}{"downloadRoot": t.TempDir()})

	ref := hd4k.GalleryRef{
		ID:   "migmg.ssuu.uk/content/09/x.html",
		URL:  "https://migmg.ssuu.uk/content/09/x.html",
		Host: "migmg.ssuu.uk",
		Path: "/content/09/x.html",
		Slug: "x.html",
	}
	if n := s.AddTasks([]hd4k.GalleryRef{ref, ref}); n != 1 {
		t.Errorf("重复添加应只算 1 个, 实际 %d", n)
	}
	if got := len(s.Tasks()); got != 1 {
		t.Errorf("任务数 = %d, 想要 1", got)
	}
}

// 回归测试：UpsertTask / FindTask 必须深拷贝 Pages。
// 浅拷贝的话，图片级 worker 继续写同一个 map，而后台 Flush 正在
// json.Marshal 整个 state → fatal error: concurrent map write（进程闪退）。
func TestStoreDoesNotAliasPageMap(t *testing.T) {
	s := newTestStore(t)

	stored := s.UpsertTask(Task{
		ID:    "h/content/09/x.html",
		Title: "t",
		Pages: map[string]string{"1": "done"},
	})

	// 拿返回的副本改 Pages，不应影响内部
	stored.Pages["1"] = "TAMPERED"
	if got := s.FindTask("h/content/09/x.html").Pages["1"]; got != "done" {
		t.Errorf("UpsertTask 返回值与内部共享了 Pages map: %q", got)
	}

	// FindTask 返回的也不应共享
	found := s.FindTask("h/content/09/x.html")
	found.Pages["2"] = "NEW"
	if _, leak := s.FindTask("h/content/09/x.html").Pages["2"]; leak {
		t.Error("FindTask 返回的 map 与内部共享了")
	}
}

// 回归测试：UpdateSettings 曾经在持写锁时调 Settings() 取返回值，
// 而 Go 的 RWMutex 不可重入 → 永久死锁。
// 症状是「点保存设置后整个程序卡死」，且不报任何错，只有这里能抓到。
func TestUpdateSettingsDoesNotDeadlock(t *testing.T) {
	s := newTestStore(t)

	done := make(chan struct{})
	go func() {
		defer close(done)
		for i := 0; i < 50; i++ {
			s.UpdateSettings(map[string]interface{}{"timeout": 1000 + i})
		}
	}()

	select {
	case <-done:
	case <-time.After(5 * time.Second):
		t.Fatal("UpdateSettings 死锁了（RWMutex 不可重入）")
	}
}

// 返回值不能和内部共享 Sites 切片 —— 调用方改它就绕过了锁
func TestUpdateSettingsDoesNotLeakSiteSlice(t *testing.T) {
	s := newTestStore(t)
	out := s.UpdateSettings(map[string]interface{}{})
	out.Sites = append(out.Sites, Site{Host: "leaked.example", Enabled: true})

	for _, site := range s.Settings().Sites {
		if site.Host == "leaked.example" {
			t.Fatal("UpdateSettings 的返回值泄漏了内部 Sites 切片")
		}
	}
}

func TestStoreClearKeepsFiles(t *testing.T) {
	s := newTestStore(t)
	s.AddTasks([]hd4k.GalleryRef{{ID: "a"}, {ID: "b"}})

	s.UpsertTask(Task{ID: "a", Status: "done"})
	s.UpsertTask(Task{ID: "b", Status: "queued"})

	if n := s.ClearFinished(); n != 1 {
		t.Errorf("清完成应删 1 个, 实际 %d", n)
	}
	tasks := s.Tasks()
	if len(tasks) != 1 || tasks[0].ID != "b" {
		t.Errorf("清理后应只剩 b: %+v", tasks)
	}
}

func TestStoreRequeueFailed(t *testing.T) {
	s := newTestStore(t)
	for id, st := range map[string]string{
		"a": "failed", "b": "partial", "c": "done", "d": "queued", "e": "running",
	} {
		s.UpsertTask(Task{ID: id, Status: st})
	}
	n := s.RequeueFailed()
	// failed + partial + running = 3
	if n != 3 {
		t.Errorf("重排队数 = %d, 想要 3", n)
	}
	if s.FindTask("c").Status != "done" {
		t.Error("已完成的任务不该被重排队")
	}
}

func TestStoreRecoverOnStart(t *testing.T) {
	s := newTestStore(t)
	s.UpsertTask(Task{ID: "a", Status: "downloading"})
	s.UpsertTask(Task{ID: "b", Status: "parsing"})
	s.UpsertTask(Task{ID: "c", Status: "running"})
	if n := s.RecoverOnStart(); n != 3 {
		t.Errorf("归位数 = %d, 想要 3", n)
	}
	for _, id := range []string{"a", "b", "c"} {
		if got := s.FindTask(id).Status; got != "queued" {
			t.Errorf("%s 状态 = %q, 想要 queued", id, got)
		}
	}
}

func TestStoreSelectAll(t *testing.T) {
	s := newTestStore(t)
	s.AddTasks([]hd4k.GalleryRef{{ID: "a"}, {ID: "b"}, {ID: "c"}})
	out := s.SelectAll([]string{"a", "c"})
	sel := map[string]bool{}
	for _, t := range out {
		sel[t.ID] = t.Selected
	}
	if !sel["a"] || sel["b"] || !sel["c"] {
		t.Errorf("勾选状态不对: %v", sel)
	}
}

func TestStoreSettingsMergeAndSites(t *testing.T) {
	s := newTestStore(t)
	s.UpdateSettings(map[string]interface{}{"timeout": 12345})

	if got := s.Settings().Timeout; got != 12345 {
		t.Errorf("Timeout = %d, 想要 12345", got)
	}

	// 站点列表是结构体，走 map 补丁容易被反序列化坏 —— 必须能正确写入
	s.UpdateSettings(map[string]interface{}{
		"sites": []hd4k.Site{{Host: "example.com", Label: "EX", Enabled: true}},
	})
	sites := s.Settings().Sites
	if len(sites) != 1 || sites[0].Host != "example.com" {
		t.Fatalf("站点列表写入失败: %#v", sites)
	}

	// 改别的不该把站点清掉
	s.UpdateSettings(map[string]interface{}{"timeout": 999})
	if len(s.Settings().Sites) != 1 {
		t.Errorf("改其它设置时站点列表被清空了: %#v", s.Settings().Sites)
	}
}

func TestStoreFlushReload(t *testing.T) {
	dir := t.TempDir()
	p := filepath.Join(dir, "state.json")

	s := NewStore(p)
	s.UpdateSettings(map[string]interface{}{"timeout": 4321})
	s.AddTasks([]hd4k.GalleryRef{{ID: "x", Slug: "x.html", Title: "标题"}})
	s.Stop()
	s.Flush()

	s2 := NewStore(p)
	if got := s2.Settings().Timeout; got != 4321 {
		t.Errorf("重载后 Timeout = %d, 想要 4321", got)
	}
	if got := s2.FindTask("x"); got == nil || got.Title != "标题" {
		t.Errorf("重载后任务丢失: %#v", got)
	}
}

// ---------------------------------------------------------------- 清单

func TestManifestRoundTrip(t *testing.T) {
	dir := t.TempDir()
	g := sampleGallery()

	pageMap := map[string]ManifestPage{}
	for _, im := range g.Images {
		pageMap[fmt.Sprint(im.Index)] = ManifestPage{
			Page: im.Index, File: im.Filename, URL: im.URL, Finished: true,
		}
	}
	meta := g.Meta
	in := Manifest{
		ID: g.ID, URL: g.URL, Host: g.Host, Path: g.Path, Slug: g.Slug,
		Title: g.Title, TotalPages: 3, Total: 3,
		DonePages: []int{3, 1, 2},
		PageMap:   pageMap, Meta: &meta,
	}
	if err := WriteManifest(dir, in); err != nil {
		t.Fatal(err)
	}

	out, err := ReadManifest(dir)
	if err != nil {
		t.Fatal(err)
	}
	if out == nil {
		t.Fatal("清单读不出来")
	}
	if out.Title != g.Title || out.Total != 3 {
		t.Errorf("清单内容不对: %#v", out)
	}
	if !sortedInts(out.DonePages) {
		t.Errorf("DonePages 未排序: %v", out.DonePages)
	}

	// 关键能力：清单里存的是稳定直链，所以能 0 请求重建画廊
	back, ok := out.GalleryFromManifest()
	if !ok {
		t.Fatal("应能从清单重建画廊")
	}
	if len(back.Images) != 3 {
		t.Fatalf("重建后图片数 = %d, 想要 3", len(back.Images))
	}
	for i, im := range back.Images {
		if im.Index != i+1 {
			t.Errorf("第 %d 张序号 = %d", i, im.Index)
		}
		if im.URL == "" {
			t.Error("重建的图片丢了直链")
		}
		if !strings.Contains(im.URL, "w=1300") {
			t.Errorf("直链丢了 w=1300（会 429）: %s", im.URL)
		}
	}
}

func TestManifestIncompleteFallsBack(t *testing.T) {
	dir := t.TempDir()
	// 有 PageMap 但缺 URL —— 不能用，宁可重新解析
	m := Manifest{
		ID:      "x",
		Version: 2,
		PageMap: map[string]ManifestPage{"1": {Page: 1, File: "a.webp"}},
	}
	if err := WriteManifest(dir, m); err != nil {
		t.Fatal(err)
	}
	out, _ := ReadManifest(dir)
	if _, ok := out.GalleryFromManifest(); ok {
		t.Error("信息不全的清单不该被当成可用")
	}
}

func TestReadManifestMissingIsNil(t *testing.T) {
	m, err := ReadManifest(t.TempDir())
	if err != nil {
		t.Errorf("文件不存在不该报错: %v", err)
	}
	if m != nil {
		t.Error("文件不存在应返回 nil")
	}
}

func TestFindManifestByID(t *testing.T) {
	root := t.TempDir()
	id := "migmg.ssuu.uk/content/09/x.html"
	if err := WriteManifest(filepath.Join(root, "x"), Manifest{ID: id}); err != nil {
		t.Fatal(err)
	}
	if got := FindManifestByID(root, id); got == nil {
		t.Fatal("按 ID 找不到清单")
	}
	if FindManifestByID(root, "nope") != nil {
		t.Error("不该匹配到不存在的 ID")
	}
}

func sortedInts(s []int) bool {
	for i := 1; i < len(s); i++ {
		if s[i] < s[i-1] {
			return false
		}
	}
	return true
}

// ---------------------------------------------------------------- 桥口

func newTestBridge(t *testing.T) *Bridge {
	t.Helper()
	s := newTestStore(t)
	s.UpdateSettings(map[string]interface{}{
		"bridgePort":   0,
		"downloadRoot": t.TempDir(),
	})
	b := NewBridge(s, func(level, msg string) {}, func(t Task, added bool, batch bool) {})
	t.Cleanup(b.Stop)
	return b
}

// 起一个真实的桥口（127.0.0.1 上的随机端口）
func startBridge(t *testing.T) (*Bridge, string, *Store) {
	t.Helper()
	s := newTestStore(t)
	s.UpdateSettings(map[string]interface{}{
		"downloadRoot": t.TempDir(),
	})
	b := NewBridge(s, func(level, msg string) {}, func(t Task, added bool, batch bool) {})
	t.Cleanup(b.Stop)

	port := 0
	for p := 38000; p < 38100; p++ {
		ln, err := net.Listen("tcp", fmt.Sprintf("127.0.0.1:%d", p))
		if err == nil {
			ln.Close()
			port = p
			break
		}
	}
	if port == 0 {
		t.Skip("找不到空闲端口，跳过桥口测试")
	}
	b.Start(port)
	return b, fmt.Sprintf("http://127.0.0.1:%d", port), s
}

func TestBridgePingAndInfo(t *testing.T) {
	_, base, _ := startBridge(t)

	resp, err := http.Get(base + "/ping")
	if err != nil {
		t.Fatal(err)
	}
	resp.Body.Close()
	if resp.StatusCode != 200 {
		t.Errorf("/ping 状态 = %d", resp.StatusCode)
	}

	resp, err = http.Get(base + "/info")
	if err != nil {
		t.Fatal(err)
	}
	defer resp.Body.Close()
	var info map[string]interface{}
	if err := json.NewDecoder(resp.Body).Decode(&info); err != nil {
		t.Fatal(err)
	}
	if info["token"] != BridgeToken {
		t.Errorf("/info token = %v, 想要 %v", info["token"], BridgeToken)
	}
	// 站点域名（含别名）返回给油猴，用于生成 @match
	hosts, _ := info["hosts"].([]interface{})
	if len(hosts) == 0 {
		t.Error("/info 没返回站点域名")
	}
}

func TestBridgeAddRequiresToken(t *testing.T) {
	_, base, _ := startBridge(t)

	req, _ := http.NewRequest("POST", base+"/add", strings.NewReader(`{"url":"https://migmg.ssuu.uk/content/09/x.html"}`))
	req.Header.Set("Content-Type", "application/json")
	// 故意不带 token
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		t.Fatal(err)
	}
	resp.Body.Close()
	if resp.StatusCode != http.StatusForbidden {
		t.Errorf("无 token 应返回 403, 实际 %d", resp.StatusCode)
	}
}

// 推的是已 404 的 www.4khd.com —— 应归一到主域名 migmg.ssuu.uk。
// 实测 migmg.ssuu.uk 首页里 90 个 /content/ 链接全都指向 www.4khd.com，
// 不归一的话后端会去请求死域名。
func TestBridgeAddNormalizesAliasDomain(t *testing.T) {
	_, base, s := startBridge(t)

	postJSON(t, base+"/add", map[string]interface{}{
		"url":   "https://www.4khd.com/content/09/myua-sweetbox-vol52.html",
		"title": "Myu_a SWEETBOX Vol.52",
	})

	task := s.FindTask("migmg.ssuu.uk/content/09/myua-sweetbox-vol52.html")
	if task == nil {
		t.Fatalf("没找到归一化后的任务，现有: %+v", s.Tasks())
	}
	if task.Title != "Myu_a SWEETBOX Vol.52" {
		t.Errorf("标题没带过来: %q", task.Title)
	}
}

func TestBridgeAddBatchDedupAndInvalid(t *testing.T) {
	_, base, s := startBridge(t)

	postJSON(t, base+"/add-batch", map[string]interface{}{
		"items": []map[string]interface{}{
			{"url": "https://migmg.ssuu.uk/content/09/a.html", "title": "A"},
			{"url": "https://www.4khd.com/content/09/a.html", "title": "A dup"}, // 别名 → 同一个
			{"url": "https://migmg.ssuu.uk/content/09/b.html", "title": "B"},
			{"url": "not-a-url"}, // 无效
		},
	})

	if got := len(s.Tasks()); got != 2 {
		t.Errorf("任务数 = %d, 想要 2（别名应去重、无效应忽略）: %+v", got, s.Tasks())
	}
}

func TestBridgeAddRejectsNonGallery(t *testing.T) {
	_, base, s := startBridge(t)

	resp := postJSONRaw(t, base+"/add", map[string]interface{}{
		"url": "https://migmg.ssuu.uk/pages/popular",
	})
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusBadRequest {
		t.Errorf("非画廊链接应返回 400, 实际 %d", resp.StatusCode)
	}
	if len(s.Tasks()) != 0 {
		t.Error("非画廊链接不该入库")
	}
}

// CORS 预检：Origin 必须在站点白名单里
func TestBridgeCORSOnlyForKnownOrigins(t *testing.T) {
	_, base, _ := startBridge(t)

	req, _ := http.NewRequest("OPTIONS", base+"/add", nil)
	req.Header.Set("Origin", "https://migmg.ssuu.uk")
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		t.Fatal(err)
	}
	resp.Body.Close()
	if got := resp.Header.Get("Access-Control-Allow-Origin"); got != "https://migmg.ssuu.uk" {
		t.Errorf("已知站点的 Origin 应被放行, 实际 %q", got)
	}

	req2, _ := http.NewRequest("OPTIONS", base+"/add", nil)
	req2.Header.Set("Origin", "https://evil.example")
	resp2, err := http.DefaultClient.Do(req2)
	if err != nil {
		t.Fatal(err)
	}
	resp2.Body.Close()
	if got := resp2.Header.Get("Access-Control-Allow-Origin"); got != "" {
		t.Errorf("陌生 Origin 不应被放行, 实际 %q", got)
	}
}

// 桥口只绑回环地址，不能暴露到局域网
func TestBridgeBindsLoopbackOnly(t *testing.T) {
	b, _, _ := startBridge(t)
	if b.Port() <= 0 {
		t.Fatal("端口没起来")
	}

	addrs, err := net.InterfaceAddrs()
	if err != nil {
		return
	}
	client := &http.Client{Timeout: 2 * time.Second}
	for _, addr := range addrs {
		ipnet, ok := addr.(*net.IPNet)
		if !ok || ipnet.IP.IsLoopback() || ipnet.IP.To4() == nil {
			continue
		}
		target := fmt.Sprintf("http://%s:%d/ping", ipnet.IP, b.Port())
		if resp, err := client.Get(target); err == nil {
			resp.Body.Close()
			t.Errorf("桥口暴露到了 %s，应只绑 127.0.0.1", ipnet.IP)
		}
	}
}

func postJSON(t *testing.T, url string, payload map[string]interface{}) {
	t.Helper()
	resp := postJSONRaw(t, url, payload)
	defer resp.Body.Close()
	if resp.StatusCode != http.StatusOK {
		b, _ := io.ReadAll(resp.Body)
		t.Errorf("POST %s 状态 = %d, body=%s", url, resp.StatusCode, string(b))
	}
}

func postJSONRaw(t *testing.T, url string, payload map[string]interface{}) *http.Response {
	t.Helper()
	raw, err := json.Marshal(payload)
	if err != nil {
		t.Fatal(err)
	}
	req, err := http.NewRequest("POST", url, bytes.NewReader(raw))
	if err != nil {
		t.Fatal(err)
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-HD4K-Token", BridgeToken)
	resp, err := http.DefaultClient.Do(req)
	if err != nil {
		t.Fatal(err)
	}
	return resp
}

// ---------------------------------------------------------------- 杂项

func TestBridgeTokenIsNotEmpty(t *testing.T) {
	if BridgeToken == "" {
		t.Fatal("桥口 token 不能为空")
	}
	if AppName != "4KHD Downloader" {
		t.Errorf("AppName = %q", AppName)
	}
}

// 默认桥口端口不该和参考项目（EH 的 37777）撞车 —— 两个工具可能同时开着
func TestDefaultBridgePortDiffersFromReferenceProject(t *testing.T) {
	port := DefaultSettings().BridgePort
	if port == 37777 {
		t.Error("默认端口与 EH Downloader 相同，两个工具会互相抢占桥口")
	}
	if port < 1024 || port > 65535 {
		t.Errorf("默认端口不合法: %d", port)
	}
}

func TestManifestFileName(t *testing.T) {
	if ManifestFileName != ".hd4k.json" {
		t.Errorf("清单文件名 = %q, 想要 .hd4k.json", ManifestFileName)
	}
	// 必须是点开头 —— 扫描时要靠这个跳过隐藏目录
	if !strings.HasPrefix(ManifestFileName, ".") {
		t.Error("清单文件名应以点开头")
	}
}

func TestVersionConstants(t *testing.T) {
	if AppVersion == "" {
		t.Error("AppVersion 不能为空")
	}
}

// 清理小图：只删小的图片文件，清单和 .part 临时文件不能碰
func TestSweepSmallImages(t *testing.T) {
	root := t.TempDir()
	gallery := filepath.Join(root, "画廊A")
	if err := os.MkdirAll(gallery, 0o755); err != nil {
		t.Fatal(err)
	}

	// 正常图（大于门槛）—— 必须留着
	big := filepath.Join(gallery, "ok.webp")
	if err := os.WriteFile(big, make([]byte, 20*1024), 0o644); err != nil {
		t.Fatal(err)
	}
	// 小图 —— 应被删
	tiny := filepath.Join(gallery, "junk.webp")
	if err := os.WriteFile(tiny, make([]byte, 3*1024), 0o644); err != nil {
		t.Fatal(err)
	}
	// 清单 —— 绝不能删（要留着续传）
	man := filepath.Join(gallery, ManifestFileName)
	if err := os.WriteFile(man, []byte(`{"tiny":"json"}`), 0o644); err != nil {
		t.Fatal(err)
	}
	// .part 未完成的下载 —— 不该当图片删（可能正好小于门槛但没下完）
	part := filepath.Join(gallery, "incomplete.webp.part")
	if err := os.WriteFile(part, make([]byte, 1024), 0o644); err != nil {
		t.Fatal(err)
	}
	// 非图片的小文件 —— 不该删
	other := filepath.Join(gallery, "note.txt")
	if err := os.WriteFile(other, []byte("hi"), 0o644); err != nil {
		t.Fatal(err)
	}

	removed, freed, err := SweepSmallImages(root, MinImageBytes)
	if err != nil {
		t.Fatal(err)
	}
	if removed != 1 {
		t.Errorf("删了 %d 个文件，想要 1", removed)
	}
	if freed != 3*1024 {
		t.Errorf("释放 %d 字节，想要 %d", freed, 3*1024)
	}

	for _, keep := range []string{big, man, part, other} {
		if _, err := os.Stat(keep); err != nil {
			t.Errorf("不该被删的文件没了: %s", keep)
		}
	}
	if _, err := os.Stat(tiny); !os.IsNotExist(err) {
		t.Error("小图没被删掉")
	}
}

// 根目录不存在时不该报错炸掉
func TestSweepSmallImagesMissingRoot(t *testing.T) {
	removed, _, err := SweepSmallImages(filepath.Join(t.TempDir(), "nope"), MinImageBytes)
	if err != nil {
		t.Errorf("根目录不存在不该报错: %v", err)
	}
	if removed != 0 {
		t.Errorf("删了 %d 个，想要 0", removed)
	}
}