hardbox/internal/modules/util/atomicwrite_test.go at 8a70f6e8fb772385406cf3921aac0620f9e191bf · jackby03/hardbox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
package util_test

import (
	"os"
	"path/filepath"
	"testing"

	"github.com/hardbox-io/hardbox/internal/modules/util"
)

func TestAtomicWrite(t *testing.T) {
	tempDir := t.TempDir()

	t.Run("success_new_file", func(t *testing.T) {
		path := filepath.Join(tempDir, "new_file.txt")
		data := []byte("hello world")
		mode := os.FileMode(0644)

		err := util.AtomicWrite(path, data, mode)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}

		info, err := os.Stat(path)
		if err != nil {
			t.Fatalf("failed to stat file: %v", err)
		}

		if info.Mode() != mode {
			t.Errorf("expected mode %v, got %v", mode, info.Mode())
		}

		content, err := os.ReadFile(path)
		if err != nil {
			t.Fatalf("failed to read file: %v", err)
		}

		if string(content) != string(data) {
			t.Errorf("expected content %q, got %q", string(data), string(content))
		}
	})

	t.Run("success_overwrite_file", func(t *testing.T) {
		path := filepath.Join(tempDir, "overwrite_file.txt")
		initialData := []byte("initial")
		mode := os.FileMode(0644)

		if err := os.WriteFile(path, initialData, mode); err != nil {
			t.Fatalf("failed to write initial file: %v", err)
		}

		newData := []byte("new data")
		newMode := os.FileMode(0600)

		err := util.AtomicWrite(path, newData, newMode)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}

		info, err := os.Stat(path)
		if err != nil {
			t.Fatalf("failed to stat file: %v", err)
		}

		if info.Mode() != newMode {
			t.Errorf("expected mode %v, got %v", newMode, info.Mode())
		}

		content, err := os.ReadFile(path)
		if err != nil {
			t.Fatalf("failed to read file: %v", err)
		}

		if string(content) != string(newData) {
			t.Errorf("expected content %q, got %q", string(newData), string(content))
		}
	})

	t.Run("success_creates_parent_directories", func(t *testing.T) {
		path := filepath.Join(tempDir, "nested", "dir", "file.txt")
		data := []byte("nested")
		mode := os.FileMode(0644)

		err := util.AtomicWrite(path, data, mode)
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}

		content, err := os.ReadFile(path)
		if err != nil {
			t.Fatalf("failed to read file: %v", err)
		}

		if string(content) != string(data) {
			t.Errorf("expected content %q, got %q", string(data), string(content))
		}
	})

	t.Run("error_invalid_directory", func(t *testing.T) {
		filePath := filepath.Join(tempDir, "is_a_file")
		if err := os.WriteFile(filePath, []byte("test"), 0644); err != nil {
			t.Fatalf("failed to write file: %v", err)
		}

		path := filepath.Join(filePath, "file.txt")
		data := []byte("test")
		mode := os.FileMode(0644)

		err := util.AtomicWrite(path, data, mode)
		if err == nil {
			t.Fatal("expected error, got nil")
		}
	})
}