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
|
package ref
import (
"bytes"
"os"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly/internal/git/gittest"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/config"
"gitlab.com/gitlab-org/gitaly/internal/gitaly/service"
hookservice "gitlab.com/gitlab-org/gitaly/internal/gitaly/service/hook"
"gitlab.com/gitlab-org/gitaly/internal/helper/lines"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"gitlab.com/gitlab-org/gitaly/internal/testhelper/testcfg"
"gitlab.com/gitlab-org/gitaly/internal/testhelper/testserver"
"gitlab.com/gitlab-org/gitaly/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) {
os.Exit(testMain(m))
}
func testMain(m *testing.M) int {
defer testhelper.MustHaveNoChildProcess()
cleanup := testhelper.Configure()
defer cleanup()
// Force small messages to test that fragmenting the
// ref list works correctly
lines.ItemsPerMessage = 3
return m.Run()
}
func setupRefService(t testing.TB) (config.Cfg, *gitalypb.Repository, string, gitalypb.RefServiceClient) {
cfg, client := setupRefServiceWithoutRepo(t)
repo, repoPath, cleanup := gittest.CloneRepoAtStorage(t, cfg, cfg.Storages[0], t.Name())
t.Cleanup(cleanup)
testhelper.ConfigureGitalyHooksBin(t, cfg)
return cfg, repo, repoPath, client
}
func setupRefServiceWithoutRepo(t testing.TB) (config.Cfg, gitalypb.RefServiceClient) {
cfg := testcfg.Build(t)
testhelper.ConfigureGitalyHooksBin(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.GetCfg(),
deps.GetLocator(),
deps.GetGitCmdFactory(),
deps.GetTxManager(),
deps.GetCatfileCache(),
))
gitalypb.RegisterHookServiceServer(srv, hookservice.NewServer(deps.GetCfg(), deps.GetHookManager(), deps.GetGitCmdFactory()))
})
}
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)
}
|