fix(web,agent): P1 正确性与鉴权加固,P2 抽 fsutil 并补测试(v1.2.8)

This commit is contained in:
2026-09-24 19:03:05 +08:00
parent 6abedbbe42
commit 8f75e99339
42 changed files with 1190 additions and 215 deletions
+71
View File
@@ -0,0 +1,71 @@
package fsutil
import (
"os"
"path/filepath"
"strings"
"testing"
)
func TestWriteFileAtomicReplacesContent(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "frps.toml")
if err := os.WriteFile(path, []byte("old"), 0o600); err != nil {
t.Fatal(err)
}
if err := WriteFileAtomic(path, []byte("new"), 0o640); err != nil {
t.Fatal(err)
}
data, err := os.ReadFile(path)
if err != nil {
t.Fatal(err)
}
if string(data) != "new" {
t.Fatalf("内容未替换:%q", data)
}
}
// TestWriteFileAtomicLeavesNoTempFiles:临时文件必须被 rename 或清理,不能在配置目录里留垃圾。
func TestWriteFileAtomicLeavesNoTempFiles(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "nginx.conf")
if err := WriteFileAtomic(path, []byte("worker_processes 1;"), 0o600); err != nil {
t.Fatal(err)
}
entries, err := os.ReadDir(dir)
if err != nil {
t.Fatal(err)
}
for _, entry := range entries {
if strings.Contains(entry.Name(), ".tmp-") {
t.Fatalf("残留临时文件:%s", entry.Name())
}
}
if len(entries) != 1 {
t.Fatalf("目录里应只剩目标文件,实际 %d 个", len(entries))
}
}
func TestBackupSkipsMissingFile(t *testing.T) {
if err := Backup(filepath.Join(t.TempDir(), "nope.conf")); err != nil {
t.Fatalf("文件不存在时应视为无需备份,实际 %v", err)
}
}
func TestBackupCopiesContent(t *testing.T) {
dir := t.TempDir()
path := filepath.Join(dir, "frpc.toml")
if err := os.WriteFile(path, []byte("token = \"x\""), 0o600); err != nil {
t.Fatal(err)
}
if err := Backup(path); err != nil {
t.Fatal(err)
}
data, err := os.ReadFile(path + ".bak")
if err != nil {
t.Fatal(err)
}
if string(data) != "token = \"x\"" {
t.Fatalf("备份内容不符:%q", data)
}
}