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
|
package ref
import (
"bytes"
"context"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/v14/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/config"
"gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/service"
hookservice "gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/service/hook"
"gitlab.com/gitlab-org/gitaly/v14/internal/gitaly/service/repository"
"gitlab.com/gitlab-org/gitaly/v14/internal/helper/lines"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper/testcfg"
"gitlab.com/gitlab-org/gitaly/v14/internal/testhelper/testserver"
"gitlab.com/gitlab-org/gitaly/v14/proto/go/gitalypb"
"google.golang.org/grpc"
)
var localBranches = map[string]*gitalypb.GitCommit{
"refs/heads/100%branch": testhelper.GitLabTestCommit("1b12f15a11fc6e62177bef08f47bc7b5ce50b141"),
"refs/heads/improve/awesome": testhelper.GitLabTestCommit("5937ac0a7beb003549fc5fd26fc247adbce4a52e"),
"refs/heads/'test'": testhelper.GitLabTestCommit("e56497bb5f03a90a51293fc6d516788730953899"),
}
func TestMain(m *testing.M) {
testhelper.Run(m, testhelper.WithSetup(func() error {
// Force small messages to test that fragmenting the
// ref list works correctly
lines.ItemsPerMessage = 3
return nil
}))
}
func setupRefService(ctx context.Context, t testing.TB) (config.Cfg, *gitalypb.Repository, string, gitalypb.RefServiceClient) {
cfg, client := setupRefServiceWithoutRepo(t)
repo, repoPath := gittest.CreateRepository(ctx, t, cfg, gittest.CreateRepositoryConfig{
Seed: gittest.SeedGitLabTest,
})
return cfg, repo, repoPath, client
}
func setupRefServiceWithoutRepo(t testing.TB) (config.Cfg, gitalypb.RefServiceClient) {
cfg := testcfg.Build(t)
testcfg.BuildGitalyHooks(t, cfg)
serverSocketPath := runRefServiceServer(t, cfg)
cfg.SocketPath = serverSocketPath
client, conn := newRefServiceClient(t, serverSocketPath)
t.Cleanup(func() { conn.Close() })
return cfg, client
}
func runRefServiceServer(t testing.TB, cfg config.Cfg) string {
return testserver.RunGitalyServer(t, cfg, nil, func(srv *grpc.Server, deps *service.Dependencies) {
gitalypb.RegisterRefServiceServer(srv, NewServer(
deps.GetLocator(),
deps.GetGitCmdFactory(),
deps.GetTxManager(),
deps.GetCatfileCache(),
))
gitalypb.RegisterHookServiceServer(srv, hookservice.NewServer(deps.GetHookManager(), deps.GetGitCmdFactory(), deps.GetPackObjectsCache()))
gitalypb.RegisterRepositoryServiceServer(srv, repository.NewServer(
deps.GetCfg(),
deps.GetRubyServer(),
deps.GetLocator(),
deps.GetTxManager(),
deps.GetGitCmdFactory(),
deps.GetCatfileCache(),
deps.GetConnsPool(),
deps.GetGit2goExecutor(),
deps.GetHousekeepingManager(),
))
})
}
func newRefServiceClient(t testing.TB, serverSocketPath string) (gitalypb.RefServiceClient, *grpc.ClientConn) {
connOpts := []grpc.DialOption{
grpc.WithInsecure(),
}
conn, err := grpc.Dial(serverSocketPath, connOpts...)
require.NoError(t, err)
return gitalypb.NewRefServiceClient(conn), conn
}
func assertContainsLocalBranch(t *testing.T, branches []*gitalypb.FindLocalBranchResponse, branch *gitalypb.FindLocalBranchResponse) {
t.Helper()
for _, b := range branches {
if bytes.Equal(branch.Name, b.Name) {
if !findLocalBranchResponsesEqual(branch, b) {
t.Errorf("Expected branch\n%v\ngot\n%v", branch, b)
}
testhelper.ProtoEqual(t, branch.Commit, b.Commit)
return // Found the branch and it maches. Success!
}
}
t.Errorf("Expected to find branch %q in local branches", branch.Name)
}
func findLocalBranchCommitAuthorsEqual(a *gitalypb.FindLocalBranchCommitAuthor, b *gitalypb.FindLocalBranchCommitAuthor) bool {
return bytes.Equal(a.Name, b.Name) &&
bytes.Equal(a.Email, b.Email) &&
a.Date.Seconds == b.Date.Seconds
}
func findLocalBranchResponsesEqual(a *gitalypb.FindLocalBranchResponse, b *gitalypb.FindLocalBranchResponse) bool {
return a.CommitId == b.CommitId &&
bytes.Equal(a.CommitSubject, b.CommitSubject) &&
findLocalBranchCommitAuthorsEqual(a.CommitAuthor, b.CommitAuthor) &&
findLocalBranchCommitAuthorsEqual(a.CommitCommitter, b.CommitCommitter)
}
func assertContainsBranch(t *testing.T, branches []*gitalypb.FindAllBranchesResponse_Branch, branch *gitalypb.FindAllBranchesResponse_Branch) {
t.Helper()
var branchNames [][]byte
for _, b := range branches {
if bytes.Equal(branch.Name, b.Name) {
testhelper.ProtoEqual(t, b.Target, branch.Target)
return // Found the branch and it maches. Success!
}
branchNames = append(branchNames, b.Name)
}
t.Errorf("Expected to find branch %q in branches %s", branch.Name, branchNames)
}
func gitalyOrPraefect(gitalyMsg, praefectMsg string) string {
if testhelper.IsPraefectEnabled() {
return praefectMsg
}
return gitalyMsg
}
|
