108 lines
2.9 KiB
Go
108 lines
2.9 KiB
Go
package fileutil
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
)
|
|
|
|
func TestWriteFileAtomicCreatesParentDirectory(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "nested", "artifact.txt")
|
|
|
|
if err := WriteFileAtomic(path, []byte("artifact")); err != nil {
|
|
t.Fatalf("WriteFileAtomic() error = %v", err)
|
|
}
|
|
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile() error = %v", err)
|
|
}
|
|
if string(data) != "artifact" {
|
|
t.Fatalf("data = %q, want artifact", data)
|
|
}
|
|
}
|
|
|
|
func TestWriteFileAtomicOverwritesTarget(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "artifact.txt")
|
|
if err := WriteFileAtomic(path, []byte("old")); err != nil {
|
|
t.Fatalf("WriteFileAtomic() initial error = %v", err)
|
|
}
|
|
|
|
if err := WriteFileAtomic(path, []byte("new")); err != nil {
|
|
t.Fatalf("WriteFileAtomic() overwrite error = %v", err)
|
|
}
|
|
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile() error = %v", err)
|
|
}
|
|
if string(data) != "new" {
|
|
t.Fatalf("data = %q, want new", data)
|
|
}
|
|
}
|
|
|
|
func TestWriteFileAtomicSupportsLongestFileName(t *testing.T) {
|
|
directory := t.TempDir()
|
|
name := strings.Repeat("a", maxFileNameBytes-len(".md")) + ".md"
|
|
path := filepath.Join(directory, name)
|
|
if err := os.WriteFile(path, []byte("old"), 0o600); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := WriteFileAtomic(path, []byte("new")); err != nil {
|
|
t.Fatalf("WriteFileAtomic() error = %v", err)
|
|
}
|
|
data, err := os.ReadFile(path)
|
|
if err != nil || string(data) != "new" {
|
|
t.Fatalf("output/error = %q/%v", data, err)
|
|
}
|
|
info, err := os.Stat(path)
|
|
if err != nil || info.Mode().Perm() != 0o600 {
|
|
t.Fatalf("output mode/error = %o/%v", info.Mode().Perm(), err)
|
|
}
|
|
matches, err := filepath.Glob(filepath.Join(directory, ".weatherreporter-*.tmp"))
|
|
if err != nil || len(matches) != 0 {
|
|
t.Fatalf("temporary files/error = %v/%v", matches, err)
|
|
}
|
|
}
|
|
|
|
func TestWriteFileAtomicCleansTemporaryFileAfterRenameError(t *testing.T) {
|
|
dir := t.TempDir()
|
|
target := filepath.Join(dir, "target")
|
|
if err := os.Mkdir(target, 0o755); err != nil {
|
|
t.Fatalf("Mkdir() error = %v", err)
|
|
}
|
|
|
|
err := WriteFileAtomic(target, []byte("data"))
|
|
if err == nil {
|
|
t.Fatal("WriteFileAtomic() error = nil, want rename error")
|
|
}
|
|
if !strings.Contains(err.Error(), "save") {
|
|
t.Fatalf("error = %q, want save context", err.Error())
|
|
}
|
|
matches, err := filepath.Glob(filepath.Join(dir, ".weatherreporter-*.tmp"))
|
|
if err != nil {
|
|
t.Fatalf("Glob() error = %v", err)
|
|
}
|
|
if len(matches) != 0 {
|
|
t.Fatalf("temporary files = %v, want none", matches)
|
|
}
|
|
}
|
|
|
|
func TestWriteJSONAtomic(t *testing.T) {
|
|
path := filepath.Join(t.TempDir(), "artifact.json")
|
|
|
|
if err := WriteJSONAtomic(path, map[string]string{"status": "ok"}); err != nil {
|
|
t.Fatalf("WriteJSONAtomic() error = %v", err)
|
|
}
|
|
|
|
data, err := os.ReadFile(path)
|
|
if err != nil {
|
|
t.Fatalf("ReadFile() error = %v", err)
|
|
}
|
|
if string(data) != "{\n \"status\": \"ok\"\n}" {
|
|
t.Fatalf("json = %q, want indented object", data)
|
|
}
|
|
}
|