Sakurasan
d5dac5a6fc
llm（三协议 Provider）/ session（JSONL 持久化）/ tools（Registry、围栏、 builtin、computer）/ cdp（browser）/ skills / mcp / agent（循环、两级压缩、 编排、审批门控）+ 底座 jsonl/textx/execx/diag/scratch/spill/imagecache。 产品装配代码（config 桥接、schedule 工具、memory/plan/retro/prompt） 留在 AutoGo，宿主接入方式见 MIGRATION.md。
99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
package spill
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSaveAndReuse(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path, err := Save(dir, "call_abc-123", "hello world")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if filepath.Dir(path) != dir {
|
|
t.Errorf("unexpected dir: %s", path)
|
|
}
|
|
got, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if string(got) != "hello world" {
|
|
t.Errorf("content = %q", got)
|
|
}
|
|
|
|
// 幂等：同 callID 再存一次，路径相同且目录只多出 1 个文件。
|
|
path2, err := Save(dir, "call_abc-123", "hello world")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if path2 != path {
|
|
t.Errorf("path changed on reuse: %s vs %s", path, path2)
|
|
}
|
|
entries, _ := os.ReadDir(dir)
|
|
if len(entries) != 1 {
|
|
t.Errorf("want 1 file, got %d", len(entries))
|
|
}
|
|
}
|
|
|
|
func TestSaveSanitizesCallID(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path, err := Save(dir, "../../etc/passwd", "x")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// 必须落在 dir 内（无路径穿越）。
|
|
if filepath.Dir(path) != dir {
|
|
t.Fatalf("path escaped dir: %s", path)
|
|
}
|
|
name := filepath.Base(path)
|
|
if strings.ContainsAny(name, "/\\") || strings.Contains(name, "..") {
|
|
t.Errorf("unsafe filename: %q", name)
|
|
}
|
|
}
|
|
|
|
func TestSaveHashFallbackAndEmptyDir(t *testing.T) {
|
|
dir := t.TempDir()
|
|
path, err := Save(dir, "", strings.Repeat("z", 10))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.HasPrefix(filepath.Base(path), "sha-") {
|
|
t.Errorf("empty callID should fall back to hash name, got %q", filepath.Base(path))
|
|
}
|
|
if _, err := Save("", "id", "x"); err == nil {
|
|
t.Error("empty dir must error")
|
|
}
|
|
}
|
|
|
|
func TestPruneRemovesOldFiles(t *testing.T) {
|
|
dir := t.TempDir()
|
|
old := filepath.Join(dir, "old.txt")
|
|
if err := os.WriteFile(old, []byte("x"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// 把 mtime 拨到 8 天前。
|
|
past := time.Now().Add(-8 * 24 * time.Hour)
|
|
if err := os.Chtimes(old, past, past); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
fresh := filepath.Join(dir, "fresh.txt")
|
|
if err := os.WriteFile(fresh, []byte("y"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
Prune(dir)
|
|
|
|
if _, err := os.Stat(old); !os.IsNotExist(err) {
|
|
t.Error("old file should have been pruned")
|
|
}
|
|
if _, err := os.Stat(fresh); err != nil {
|
|
t.Error("fresh file should survive")
|
|
}
|
|
// 目录不存在时不应 panic。
|
|
Prune(filepath.Join(dir, "nope"))
|
|
}
|