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
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
|
package repository
import (
"os"
"os/exec"
"path/filepath"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/git"
"gitlab.com/gitlab-org/gitaly/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/config"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/proto/go/gitalypb"
"google.golang.org/grpc/codes"
)
func TestSuccessfulCalculateChecksum(t *testing.T) {
locator := config.NewLocator(config.Config)
serverSocketPath, stop := runRepoServer(t, locator)
defer stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
testRepo, testRepoPath, cleanupFn := gittest.CloneRepo(t)
defer cleanupFn()
// Force the refs database of testRepo into a known state
require.NoError(t, os.RemoveAll(filepath.Join(testRepoPath, "refs")))
for _, d := range []string{"refs/heads", "refs/tags", "refs/notes"} {
require.NoError(t, os.MkdirAll(filepath.Join(testRepoPath, d), 0755))
}
require.NoError(t, exec.Command("cp", "testdata/checksum-test-packed-refs", filepath.Join(testRepoPath, "packed-refs")).Run())
require.NoError(t, exec.Command("git", "-C", testRepoPath, "symbolic-ref", "HEAD", "refs/heads/feature").Run())
request := &gitalypb.CalculateChecksumRequest{Repository: testRepo}
testCtx, cancelCtx := testhelper.Context()
defer cancelCtx()
response, err := client.CalculateChecksum(testCtx, request)
require.NoError(t, err)
require.Equal(t, "0c500d7c8a9dbf65e4cf5e58914bec45bfb6e9ab", response.Checksum)
}
func TestRefWhitelist(t *testing.T) {
testCases := []struct {
ref string
match bool
}{
{ref: "1450cd639e0bc6721eb02800169e464f212cde06 foo/bar", match: false},
{ref: "259a6fba859cc91c54cd86a2cbd4c2f720e3a19d foo/bar:baz", match: false},
{ref: "d0a293c0ac821fadfdc086fe528f79423004229d refs/foo/bar:baz", match: false},
{ref: "21751bf5cb2b556543a11018c1f13b35e44a99d7 tags/v0.0.1", match: false},
{ref: "498214de67004b1da3d820901307bed2a68a8ef6 keep-around/498214de67004b1da3d820901307bed2a68a8ef6", match: false},
{ref: "38008cb17ce1466d8fec2dfa6f6ab8dcfe5cf49e merge-requests/11/head", match: false},
{ref: "c347ca2e140aa667b968e51ed0ffe055501fe4f4 environments/3/head", match: false},
{ref: "4ed78158b5b018c43005cec917129861541e25bc notes/commits", match: false},
{ref: "9298d46305ee0d3f4ce288370beaa02637584ff2 HEAD", match: true},
{ref: "0ed8c6c6752e8c6ea63e7b92a517bf5ac1209c80 refs/heads/markdown", match: true},
{ref: "f4e6814c3e4e7a0de82a9e7cd20c626cc963a2f8 refs/tags/v1.0.0", match: true},
{ref: "78a30867c755d774340108cdad5f11254818fb0c refs/keep-around/78a30867c755d774340108cdad5f11254818fb0c", match: true},
{ref: "c347ca2e140aa667b968e51ed0ffe055501fe4f4 refs/merge-requests/10/head", match: true},
{ref: "c347ca2e140aa667b968e51ed0ffe055501fe4f4 refs/environments/2/head", match: true},
{ref: "4ed78158b5b018c43005cec917129861541e25bc refs/notes/commits", match: true},
}
for _, tc := range testCases {
t.Run(tc.ref, func(t *testing.T) {
match := refWhitelist.Match([]byte(tc.ref))
require.Equal(t, match, tc.match)
})
}
}
func TestEmptyRepositoryCalculateChecksum(t *testing.T) {
locator := config.NewLocator(config.Config)
serverSocketPath, stop := runRepoServer(t, locator)
defer stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
repo, _, cleanupFn := gittest.InitBareRepo(t)
defer cleanupFn()
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
testCtx, cancelCtx := testhelper.Context()
defer cancelCtx()
response, err := client.CalculateChecksum(testCtx, request)
require.NoError(t, err)
require.Equal(t, git.ZeroOID.String(), response.Checksum)
}
func TestBrokenRepositoryCalculateChecksum(t *testing.T) {
locator := config.NewLocator(config.Config)
serverSocketPath, stop := runRepoServer(t, locator)
defer stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
repo, testRepoPath, cleanupFn := gittest.InitBareRepo(t)
defer cleanupFn()
// Force an empty HEAD file
require.NoError(t, os.Truncate(filepath.Join(testRepoPath, "HEAD"), 0))
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
testCtx, cancelCtx := testhelper.Context()
defer cancelCtx()
_, err := client.CalculateChecksum(testCtx, request)
testhelper.RequireGrpcError(t, err, codes.DataLoss)
}
func TestFailedCalculateChecksum(t *testing.T) {
locator := config.NewLocator(config.Config)
serverSocketPath, stop := runRepoServer(t, locator)
defer stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
invalidRepo := &gitalypb.Repository{StorageName: "fake", RelativePath: "path"}
testCases := []struct {
desc string
request *gitalypb.CalculateChecksumRequest
code codes.Code
}{
{
desc: "Invalid repository",
request: &gitalypb.CalculateChecksumRequest{Repository: invalidRepo},
code: codes.InvalidArgument,
},
{
desc: "Repository is nil",
request: &gitalypb.CalculateChecksumRequest{},
code: codes.InvalidArgument,
},
}
for _, testCase := range testCases {
testCtx, cancelCtx := testhelper.Context()
defer cancelCtx()
_, err := client.CalculateChecksum(testCtx, testCase.request)
testhelper.RequireGrpcError(t, err, testCase.code)
}
}
func TestInvalidRefsCalculateChecksum(t *testing.T) {
locator := config.NewLocator(config.Config)
serverSocketPath, stop := runRepoServer(t, locator)
defer stop()
client, conn := newRepositoryClient(t, serverSocketPath)
defer conn.Close()
testRepo, testRepoPath, cleanupFn := gittest.CloneRepo(t)
defer cleanupFn()
// Force the refs database of testRepo into a known state
require.NoError(t, os.RemoveAll(filepath.Join(testRepoPath, "refs")))
for _, d := range []string{"refs/heads", "refs/tags", "refs/notes"} {
require.NoError(t, os.MkdirAll(filepath.Join(testRepoPath, d), 0755))
}
require.NoError(t, exec.Command("cp", "testdata/checksum-test-invalid-refs", filepath.Join(testRepoPath, "packed-refs")).Run())
request := &gitalypb.CalculateChecksumRequest{Repository: testRepo}
testCtx, cancelCtx := testhelper.Context()
defer cancelCtx()
response, err := client.CalculateChecksum(testCtx, request)
require.NoError(t, err)
require.Equal(t, git.ZeroOID.String(), response.Checksum)
}
|
