72 lines
1.7 KiB
Go
72 lines
1.7 KiB
Go
|
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)
|
||
|
|
}
|
||
|
|
}
|