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_ListWithScheduleFilter(t *testing.T) {
35 storage, err := NewBackupStorage(t.TempDir(), testLogger())
36 require.NoError(t, err)
37
38 base := time.Date(2026, 2, 7, 10, 0, 0, 0, time.UTC)
39 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base, bytes.NewReader([]byte("d1")))
40 require.NoError(t, err)
41 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base.Add(time.Hour), bytes.NewReader([]byte("d2")))
42 require.NoError(t, err)
43 _, err = storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleWeekly, base.Add(2*time.Hour), bytes.NewReader([]byte("w1")))
44 require.NoError(t, err)
45
46 schedule := domain.ScheduleDaily
47 jobs, err := storage.List(context.Background(), "app.example.com", &schedule)
48 require.NoError(t, err)
49 require.Len(t, jobs, 2)
50 assert.Equal(t, domain.ScheduleDaily, jobs[0].Schedule)
51 assert.Equal(t, domain.ScheduleDaily, jobs[1].Schedule)
52 assert.True(t, jobs[0].StartedAt.After(jobs[1].StartedAt))
53}
54
55func TestBackupStorage_Delete(t *testing.T) {
56 storage, err := NewBackupStorage(t.TempDir(), testLogger())
57 require.NoError(t, err)
58
59 path, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, time.Now().UTC(), bytes.NewReader([]byte("data")))
60 require.NoError(t, err)
61
62 err = storage.Delete(context.Background(), path)
63 require.NoError(t, err)
64
65 _, err = storage.Get(context.Background(), path)
66 assert.Error(t, err)
67}
68
69func TestBackupStorage_ApplyRetention(t *testing.T) {
70 storage, err := NewBackupStorage(t.TempDir(), testLogger())
71 require.NoError(t, err)
72
73 base := time.Date(2026, 2, 7, 6, 0, 0, 0, time.UTC)
74 for i := 0; i < 4; i++ {
75 _, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.ScheduleDaily, base.Add(time.Duration(i)*time.Hour), bytes.NewReader([]byte("data")))
76 require.NoError(t, err)
77 }
78
79 deleted, err := storage.ApplyRetention(context.Background(), "app.example.com", domain.RetentionPolicy{Daily: 2})
80 require.NoError(t, err)
81 assert.Equal(t, 2, deleted)
82
83 schedule := domain.ScheduleDaily
84 jobs, err := storage.List(context.Background(), "app.example.com", &schedule)
85 require.NoError(t, err)
86 assert.Len(t, jobs, 2)
87}
88
89func TestBackupStorage_StoreSanitizesDotOnlyPathComponents(t *testing.T) {
90 storage, err := NewBackupStorage(t.TempDir(), testLogger())
91 require.NoError(t, err)
92
93 path, err := storage.Store(context.Background(), "..", "...", domain.ScheduleDaily, time.Now().UTC(), bytes.NewReader([]byte("data")))
94 require.NoError(t, err)
95 assert.NotContains(t, path, "..")
96}
97
98func TestBackupStorage_StoreSanitizesSchedulePathComponent(t *testing.T) {
99 rootDir := t.TempDir()
100 storage, err := NewBackupStorage(rootDir, testLogger())
101 require.NoError(t, err)
102
103 path, err := storage.Store(context.Background(), "app.example.com", "postgres", domain.BackupSchedule("../../escape"), time.Now().UTC(), bytes.NewReader([]byte("data")))
104 require.NoError(t, err)
105
106 rel, err := filepath.Rel(rootDir, path)
107 require.NoError(t, err)
108 assert.NotEqual(t, "..", rel)
109 assert.NotContains(t, rel, "../")
110 assert.NotContains(t, rel, "..\\")
111}