1package filesystem
2
3import (
4 "bytes"
5 "context"
6 "io"
7 "path/filepath"
8 "testing"
9 "time"
10
11 "github.com/bnema/gordon/internal/domain"
12 "github.com/stretchr/testify/assert"
13 "github.com/stretchr/testify/require"
14)
15
16func TestBackupStorage_StoreAndGet(t *testing.T) {
17 storage, err := NewBackupStorage(t.TempDir(), testLogger())
18 require.NoError(t, err)
19
20 now := time.Date(2026, 2, 7, 11, 0, 0, 0, time.UTC)
21 payload := []byte("backup-content")
22 path, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, now, bytes.NewReader(payload))
23 require.NoError(t, err)
24
25 rc, err := storage.Get(context.Background(), path)
26 require.NoError(t, err)
27 defer rc.Close()
28
29 data, err := io.ReadAll(rc)
30 require.NoError(t, err)
31 assert.Equal(t, payload, data)
32}
33
34func TestBackupStorage_StoreSameSecondBackupsUseDistinctFinalAndTempPaths(t *testing.T) {
35 rootDir := t.TempDir()
36 storage, err := NewBackupStorage(rootDir, testLogger())
37 require.NoError(t, err)
38
39 now := time.Date(2026, 2, 7, 11, 0, 0, 0, time.UTC)
40 firstPayload := []byte("first-backup")
41 secondPayload := []byte("second-backup")
42
43 firstPath, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, now, bytes.NewReader(firstPayload))
44 require.NoError(t, err)
45 secondPath, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, now, bytes.NewReader(secondPayload))
46 require.NoError(t, err)
47
48 assert.NotEqual(t, firstPath, secondPath)
49 assert.NoFileExists(t, firstPath+".tmp")
50 assert.NoFileExists(t, secondPath+".tmp")
51
52 firstRead, err := storage.Get(context.Background(), firstPath)
53 require.NoError(t, err)
54 defer firstRead.Close()
55 firstData, err := io.ReadAll(firstRead)
56 require.NoError(t, err)
57 assert.Equal(t, firstPayload, firstData)
58
59 secondRead, err := storage.Get(context.Background(), secondPath)
60 require.NoError(t, err)
61 defer secondRead.Close()
62 secondData, err := io.ReadAll(secondRead)
63 require.NoError(t, err)
64 assert.Equal(t, secondPayload, secondData)
65}
66
67func TestBackupStorage_ListWithScheduleFilter(t *testing.T) {
68 storage, err := NewBackupStorage(t.TempDir(), testLogger())
69 require.NoError(t, err)
70
71 base := time.Date(2026, 2, 7, 10, 0, 0, 0, time.UTC)
72 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base, bytes.NewReader([]byte("d1")))
73 require.NoError(t, err)
74 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base.Add(time.Hour), bytes.NewReader([]byte("d2")))
75 require.NoError(t, err)
76 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleWeekly, base.Add(2*time.Hour), bytes.NewReader([]byte("w1")))
77 require.NoError(t, err)
78
79 schedule := domain.ScheduleDaily
80 jobs, err := storage.List(context.Background(), "app.example.com", &schedule)
81 require.NoError(t, err)
82 require.Len(t, jobs, 2)
83 assert.Equal(t, domain.ScheduleDaily, jobs[0].Schedule)
84 assert.Equal(t, domain.ScheduleDaily, jobs[1].Schedule)
85 assert.True(t, jobs[0].StartedAt.After(jobs[1].StartedAt))
86}
87
88func TestBackupStorage_Delete(t *testing.T) {
89 storage, err := NewBackupStorage(t.TempDir(), testLogger())
90 require.NoError(t, err)
91
92 path, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, time.Now().UTC(), bytes.NewReader([]byte("data")))
93 require.NoError(t, err)
94
95 err = storage.Delete(context.Background(), path)
96 require.NoError(t, err)
97
98 _, err = storage.Get(context.Background(), path)
99 assert.Error(t, err)
100}
101
102func TestBackupStorage_ApplyRetention(t *testing.T) {
103 storage, err := NewBackupStorage(t.TempDir(), testLogger())
104 require.NoError(t, err)
105
106 base := time.Date(2026, 2, 7, 6, 0, 0, 0, time.UTC)
107 for i := range 4 {
108 _, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base.Add(time.Duration(i)*time.Hour), bytes.NewReader([]byte("data")))
109 require.NoError(t, err)
110 }
111
112 deleted, err := storage.ApplyRetention(context.Background(), "app.example.com", domain.RetentionPolicy{Daily: 2})
113 require.NoError(t, err)
114 assert.Equal(t, 2, deleted)
115
116 schedule := domain.ScheduleDaily
117 jobs, err := storage.List(context.Background(), "app.example.com", &schedule)
118 require.NoError(t, err)
119 assert.Len(t, jobs, 2)
120}
121
122func TestBackupStorage_StoreSanitizesDotOnlyPathComponents(t *testing.T) {
123 storage, err := NewBackupStorage(t.TempDir(), testLogger())
124 require.NoError(t, err)
125
126 path, err := storage.Store(context.Background(), "..", "...", domain.ScheduleDaily, time.Now().UTC(), bytes.NewReader([]byte("data")))
127 require.NoError(t, err)
128 assert.NotContains(t, path, "..")
129}
130
131func TestBackupStorage_StoreSanitizesSchedulePathComponent(t *testing.T) {
132 rootDir := t.TempDir()
133 storage, err := NewBackupStorage(rootDir, testLogger())
134 require.NoError(t, err)
135
136 path, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.BackupSchedule("../../escape"), time.Now().UTC(), bytes.NewReader([]byte("data")))
137 require.NoError(t, err)
138
139 rel, err := filepath.Rel(rootDir, path)
140 require.NoError(t, err)
141 assert.NotEqual(t, "..", rel)
142 assert.NotContains(t, rel, "../")
143 assert.NotContains(t, rel, "..\\")
144}