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
|
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/testhelper"
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
"google.golang.org/grpc/codes"
)
func TestSuccessfulCalculateChecksum(t *testing.T) {
t.Parallel()
ctx := testhelper.Context(t)
cfg, repo, repoPath, client := setupRepositoryService(ctx, t)
// Force the refs database of testRepo into a known state
require.NoError(t, os.RemoveAll(filepath.Join(repoPath, "refs")))
for _, d := range []string{"refs/heads", "refs/tags", "refs/notes"} {
require.NoError(t, os.MkdirAll(filepath.Join(repoPath, d), 0o755))
}
testhelper.CopyFile(t, "testdata/checksum-test-packed-refs", filepath.Join(repoPath, "packed-refs"))
gittest.Exec(t, cfg, "-C", repoPath, "symbolic-ref", "HEAD", "refs/heads/feature")
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
response, err := client.CalculateChecksum(ctx, request)
require.NoError(t, err)
require.Equal(t, "0c500d7c8a9dbf65e4cf5e58914bec45bfb6e9ab", response.Checksum)
}
func TestEmptyRepositoryCalculateChecksum(t *testing.T) {
t.Parallel()
cfg, client := setupRepositoryServiceWithoutRepo(t)
testCtx := testhelper.Context(t)
repo, _ := gittest.CreateRepository(testCtx, t, cfg)
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
response, err := client.CalculateChecksum(testCtx, request)
require.NoError(t, err)
require.Equal(t, git.ZeroOID.String(), response.Checksum)
}
func TestBrokenRepositoryCalculateChecksum(t *testing.T) {
t.Parallel()
cfg, client := setupRepositoryServiceWithoutRepo(t)
testCtx := testhelper.Context(t)
repo, repoPath := gittest.CreateRepository(testCtx, t, cfg)
// Force an empty HEAD file
require.NoError(t, os.Truncate(filepath.Join(repoPath, "HEAD"), 0))
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
_, err := client.CalculateChecksum(testCtx, request)
testhelper.RequireGrpcCode(t, err, codes.DataLoss)
}
func TestFailedCalculateChecksum(t *testing.T) {
t.Parallel()
_, client := setupRepositoryServiceWithoutRepo(t)
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 := testhelper.Context(t)
_, err := client.CalculateChecksum(testCtx, testCase.request)
testhelper.RequireGrpcCode(t, err, testCase.code)
}
}
func TestInvalidRefsCalculateChecksum(t *testing.T) {
t.Parallel()
ctx := testhelper.Context(t)
_, repo, repoPath, client := setupRepositoryService(ctx, t)
// Force the refs database of testRepo into a known state
require.NoError(t, os.RemoveAll(filepath.Join(repoPath, "refs")))
for _, d := range []string{"refs/heads", "refs/tags", "refs/notes"} {
require.NoError(t, os.MkdirAll(filepath.Join(repoPath, d), 0o755))
}
require.NoError(t, exec.Command("cp", "testdata/checksum-test-invalid-refs", filepath.Join(repoPath, "packed-refs")).Run())
request := &gitalypb.CalculateChecksumRequest{Repository: repo}
response, err := client.CalculateChecksum(ctx, request)
require.NoError(t, err)
require.Equal(t, git.ZeroOID.String(), response.Checksum)
}
|
