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
115
116
117
118
119
120
121
122
123
124
125
126
|
package streamcache
import (
"fmt"
"os"
"path/filepath"
"regexp"
"testing"
"time"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v16/internal/helper/perm"
"gitlab.com/gitlab-org/gitaly/v16/internal/testhelper"
)
func TestFilestoreCreate(t *testing.T) {
tmp := testhelper.TempDir(t)
fs := newFilestore(tmp, 0, time.After, testhelper.SharedLogger(t))
defer fs.Stop()
f, err := fs.Create()
require.NoError(t, err)
_, err = f.Write([]byte{0})
require.NoError(t, err, "file is writeable")
require.NoError(t, f.Close())
relpath, err := filepath.Rel(fs.dir, f.Name())
require.NoError(t, err)
require.Regexp(
t,
regexp.MustCompile(`^[0-9a-f]{2}/[^/]+$`),
relpath,
"path should be one directory deep and use aa/foobar hex scheme",
)
}
func TestFilestoreCreate_concurrency(t *testing.T) {
tmp := testhelper.TempDir(t)
fs := newFilestore(tmp, time.Hour, time.After, testhelper.SharedLogger(t))
defer fs.Stop()
const N = 100
errors := make(chan error, N)
start := make(chan struct{})
for i := 0; i < N; i++ {
go func(i int) {
<-start
errors <- func() error {
f, err := fs.Create()
if err != nil {
return err
}
if err := f.Close(); err != nil {
return err
}
return os.Remove(f.Name())
}()
}(i)
}
close(start)
for i := 0; i < N; i++ {
require.NoError(t, <-errors)
}
}
func TestFilestoreCreate_uniqueness(t *testing.T) {
tmp := testhelper.TempDir(t)
const (
M = 10
N = 10
)
filenames := make(map[string]struct{})
for j := 0; j < M; j++ {
fs := newFilestore(tmp, time.Hour, time.After, testhelper.SharedLogger(t))
defer fs.Stop()
for i := 0; i < N; i++ {
t.Run(fmt.Sprintf("create file %d/%d", i, j), func(t *testing.T) {
f, err := fs.Create()
require.NoError(t, err)
require.NoError(t, f.Close())
filenames[f.Name()] = struct{}{}
})
}
}
require.Len(t, filenames, M*N, "all filenames must be unique")
}
func TestFilestoreCleanwalk(t *testing.T) {
tmp := testhelper.TempDir(t)
fs := newFilestore(tmp, time.Hour, time.After, testhelper.SharedLogger(t))
defer fs.Stop()
dir1 := filepath.Join(tmp, "dir1")
dir2 := filepath.Join(tmp, "dir2")
file := filepath.Join(dir2, "file")
require.NoError(t, os.Mkdir(dir1, perm.SharedDir))
require.NoError(t, os.Mkdir(dir2, perm.SharedDir))
require.NoError(t, os.WriteFile(file, nil, perm.SharedFile))
require.NoError(t, os.Chmod(dir2, 0), "create dir with pathological permissions")
require.NoError(t, fs.cleanWalk(time.Now().Add(time.Hour)))
for _, d := range []string{dir1, dir2} {
fi, err := os.Stat(d)
require.NoError(t, err, "directories do not get deleted")
const mask = perm.PrivateDir
require.True(t, fi.Mode()&mask >= mask, "unexpected file mode %o", fi.Mode())
}
require.NoFileExists(t, file)
}
|
