99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package vault
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func trashAt(t *testing.T, v *Vault, path string, when time.Time) TrashItem {
|
|
t.Helper()
|
|
if _, err := v.Write(path, []byte("content"), Precondition{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
item, err := v.Delete(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
meta, err := v.readTrashMetadata(item.ID)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
meta.DeletedAt = when.UnixMilli()
|
|
data, err := json.Marshal(meta)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := os.WriteFile(filepath.Join(v.trash, item.ID, "meta.json"), data, 0600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
return meta.TrashItem
|
|
}
|
|
|
|
func TestTrashRetentionAndInvalidMetadata(t *testing.T) {
|
|
v := newVault(t)
|
|
now := time.Now()
|
|
old := trashAt(t, v, "old.md", now.Add(-31*24*time.Hour))
|
|
boundary := trashAt(t, v, "boundary.md", now.Add(-30*24*time.Hour))
|
|
fresh := trashAt(t, v, "fresh.md", now.Add(-29*24*time.Hour))
|
|
future := trashAt(t, v, "future.md", now.Add(time.Hour))
|
|
broken := trashAt(t, v, "broken.md", now.Add(-40*24*time.Hour))
|
|
if err := os.WriteFile(filepath.Join(v.trash, broken.ID, "meta.json"), []byte(`{"deletedAt":0}`), 0600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
result, err := v.PurgeTrash(30, now)
|
|
if err != nil || len(result.Deleted) != 2 || len(result.Failed) != 1 || result.Failed[0].ID != broken.ID {
|
|
t.Fatalf("结果异常: %+v, %v", result, err)
|
|
}
|
|
for _, item := range []TrashItem{old, boundary} {
|
|
if _, err := os.Stat(filepath.Join(v.trash, item.ID)); !errors.Is(err, os.ErrNotExist) {
|
|
t.Fatalf("过期条目未删除: %v", err)
|
|
}
|
|
}
|
|
for _, item := range []TrashItem{fresh, future, broken} {
|
|
if _, err := os.Stat(filepath.Join(v.trash, item.ID)); err != nil {
|
|
t.Fatalf("不应删除未过期/无效条目: %v", err)
|
|
}
|
|
}
|
|
// 极大天数不应溢出成过期期限。
|
|
result, err = v.PurgeTrash(1<<53-1, now)
|
|
if err != nil || len(result.Deleted) != 0 || len(result.Failed) != 1 {
|
|
t.Fatalf("极大期限: %+v, %v", result, err)
|
|
}
|
|
result, err = v.PurgeTrash(0, now)
|
|
if err != nil || len(result.Deleted) != 2 || len(result.Failed) != 1 {
|
|
t.Fatalf("清空全部应独立报告损坏条目: %+v, %v", result, err)
|
|
}
|
|
}
|
|
|
|
func TestTrashRestoreAndPurgeSerialize(t *testing.T) {
|
|
for range 20 {
|
|
v := newVault(t)
|
|
item := trashAt(t, v, "note.md", time.Now().Add(-40*24*time.Hour))
|
|
var restoreErr, purgeErr error
|
|
var result TrashPurgeResult
|
|
var wg sync.WaitGroup
|
|
wg.Add(2)
|
|
go func() { defer wg.Done(); _, restoreErr = v.Restore(item.ID) }()
|
|
go func() { defer wg.Done(); result, purgeErr = v.PurgeTrash(30, time.Now()) }()
|
|
wg.Wait()
|
|
if purgeErr != nil || len(result.Failed) != 0 {
|
|
t.Fatalf("并发清理失败: %+v %v", result, purgeErr)
|
|
}
|
|
if restoreErr == nil {
|
|
if len(result.Deleted) != 0 {
|
|
t.Fatal("已恢复的内容不可再次清理")
|
|
}
|
|
if body, err := os.ReadFile(filepath.Join(v.Root(), item.Path)); err != nil || string(body) != "content" {
|
|
t.Fatalf("恢复内容丢失: %v", err)
|
|
}
|
|
} else if !errors.Is(restoreErr, ErrNotFound) || len(result.Deleted) != 1 {
|
|
t.Fatalf("清理先完成时应明确不存在: %v %+v", restoreErr, result)
|
|
}
|
|
}
|
|
}
|