mirror of
https://gitcode.com/JianFeeeee/HomeAgent.git
synced 2026-09-21 09:28:14 +00:00
bc9ef15eb0
- L0: files 插件写受保护系统路径(/etc 等)前自动留档，AbstractBeforeWrite 到 data/file_baseline - L1: failback 受限 worker 执行恢复梯子 probe→还原DNS/proxy→还原LLM配置+ReloadFromConfig→probe，N轮有界 - L2: tracker changeset 持久化原文 blob，guard 离线 RollbackFromDisk 回滚 agentfs；SystemSnapshot 支撑 - guard 父守护: 心跳 IPC(PING/ACK unix socket, 文件心跳回退)、失败计数、退出码协议(42/43/44)、最后手段 - 发行版路径适配: system.protected_paths/network_paths 可注入，默认面向主流 Linux - Windows 兼容: guard.go/failback.go 加 //go:build linux, guard_windows.go 提供 no-op 桩 - 修复: guard.yaml last_resort 键冲突、changeset Content 不落盘导致离线回滚丢原文 Build 全绿, vet 干净, system/recovery/ipc/tracker 单元测试全过
93 lines
2.2 KiB
Go
93 lines
2.2 KiB
Go
package tracker
|
||
|
||
import (
|
||
"os"
|
||
"path/filepath"
|
||
"testing"
|
||
)
|
||
|
||
// 构造一条写入磁盘的 changeset（模拟 worker 运行期间的变更），随后用
|
||
// RollbackFromDisk 离线还原。
|
||
func TestRollbackFromDisk(t *testing.T) {
|
||
dataDir := t.TempDir()
|
||
workDir := t.TempDir()
|
||
|
||
trk := NewTracker(dataDir, workDir)
|
||
if err := trk.Init(); err != nil {
|
||
t.Fatalf("Init: %v", err)
|
||
}
|
||
upper := trk.upperDir
|
||
os.MkdirAll(upper, 0755)
|
||
|
||
// 1. 先创建一个文件（对应"新建"）
|
||
newFile := filepath.Join(upper, "new.txt")
|
||
os.WriteFile(newFile, []byte("brand new"), 0644)
|
||
|
||
// 2. 修改一个文件（对应"修改"，带原文）
|
||
modFile := filepath.Join(upper, "mod.txt")
|
||
os.WriteFile(modFile, []byte("after"), 0644)
|
||
beforeContent := []byte("before")
|
||
csMod := &ChangeSet{
|
||
ID: "cs_mod",
|
||
Files: []FileChange{{
|
||
Path: "mod.txt",
|
||
Type: ChangeFileModified,
|
||
Content: beforeContent,
|
||
}},
|
||
}
|
||
trk.saveChangeSet(csMod)
|
||
csNew := &ChangeSet{
|
||
ID: "cs_new",
|
||
Files: []FileChange{{
|
||
Path: "new.txt",
|
||
Type: ChangeFileCreated,
|
||
}},
|
||
}
|
||
trk.saveChangeSet(csNew)
|
||
|
||
if trk.ChangesetsOnDisk() != 2 {
|
||
t.Fatalf("ChangesetsOnDisk = %d, want 2", trk.ChangesetsOnDisk())
|
||
}
|
||
|
||
// 离线回滚（模拟 guard 在 worker 崩溃后调用）
|
||
n, err := trk.RollbackFromDisk()
|
||
if err != nil {
|
||
t.Fatalf("RollbackFromDisk: %v", err)
|
||
}
|
||
if n != 2 {
|
||
t.Fatalf("rolled back %d, want 2", n)
|
||
}
|
||
if trk.ChangesetsOnDisk() != 0 {
|
||
t.Fatalf("ChangesetsOnDisk after rollback = %d, want 0", trk.ChangesetsOnDisk())
|
||
}
|
||
|
||
// 新建文件被删除
|
||
if _, err := os.Stat(newFile); !os.IsNotExist(err) {
|
||
t.Fatal("created file should be removed after rollback")
|
||
}
|
||
// 修改文件还原原文
|
||
data, err := os.ReadFile(modFile)
|
||
if err != nil {
|
||
t.Fatalf("read modFile: %v", err)
|
||
}
|
||
if string(data) != "before" {
|
||
t.Fatalf("modFile restored to %q, want %q", data, beforeContent)
|
||
}
|
||
}
|
||
|
||
func TestRollbackFromDiskEmpty(t *testing.T) {
|
||
dataDir := t.TempDir()
|
||
workDir := t.TempDir()
|
||
trk := NewTracker(dataDir, workDir)
|
||
if err := trk.Init(); err != nil {
|
||
t.Fatalf("Init: %v", err)
|
||
}
|
||
n, err := trk.RollbackFromDisk()
|
||
if err != nil {
|
||
t.Fatalf("RollbackFromDisk: %v", err)
|
||
}
|
||
if n != 0 {
|
||
t.Fatalf("rolled back %d, want 0", n)
|
||
}
|
||
}
|