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
|
package remote_test
import (
"context"
"net"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/gitaly-proto/go/gitalypb"
serverPkg "gitlab.com/gitlab-org/gitaly/internal/server"
"gitlab.com/gitlab-org/gitaly/internal/service/remote"
"gitlab.com/gitlab-org/gitaly/internal/testhelper"
"google.golang.org/grpc"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/metadata"
)
func TestSuccessfulFetchInternalRemote(t *testing.T) {
server, serverSocketPath := runFullServer(t)
defer server.Stop()
client, conn := remote.NewRemoteClient(t, serverSocketPath)
defer conn.Close()
remoteRepo, remoteRepoPath, remoteCleanupFn := testhelper.NewTestRepo(t)
defer remoteCleanupFn()
repo, repoPath, cleanupFn := testhelper.InitBareRepo(t)
defer cleanupFn()
ctxOuter, cancel := testhelper.Context()
defer cancel()
md := testhelper.GitalyServersMetadata(t, serverSocketPath)
ctx := metadata.NewOutgoingContext(ctxOuter, md)
request := &gitalypb.FetchInternalRemoteRequest{
Repository: repo,
RemoteRepository: remoteRepo,
}
c, err := client.FetchInternalRemote(ctx, request)
require.NoError(t, err)
require.True(t, c.GetResult())
remoteRefs := testhelper.GetRepositoryRefs(t, remoteRepoPath)
refs := testhelper.GetRepositoryRefs(t, repoPath)
require.Equal(t, remoteRefs, refs)
}
func TestFailedFetchInternalRemote(t *testing.T) {
server, serverSocketPath := runFullServer(t)
defer server.Stop()
client, conn := remote.NewRemoteClient(t, serverSocketPath)
defer conn.Close()
repo, _, cleanupFn := testhelper.InitBareRepo(t)
defer cleanupFn()
ctxOuter, cancel := testhelper.Context()
defer cancel()
md := testhelper.GitalyServersMetadata(t, serverSocketPath)
ctx := metadata.NewOutgoingContext(ctxOuter, md)
// Non-existing remote repo
remoteRepo := &gitalypb.Repository{StorageName: "default", RelativePath: "fake.git"}
request := &gitalypb.FetchInternalRemoteRequest{
Repository: repo,
RemoteRepository: remoteRepo,
}
c, err := client.FetchInternalRemote(ctx, request)
require.NoError(t, err)
require.False(t, c.GetResult())
}
func TestFailedFetchInternalRemoteDueToValidations(t *testing.T) {
server, serverSocketPath := runFullServer(t)
defer server.Stop()
client, conn := remote.NewRemoteClient(t, serverSocketPath)
defer conn.Close()
ctx, cancel := context.WithCancel(context.Background())
defer cancel()
repo := &gitalypb.Repository{StorageName: "default", RelativePath: "repo.git"}
testCases := []struct {
desc string
request *gitalypb.FetchInternalRemoteRequest
}{
{
desc: "empty Repository",
request: &gitalypb.FetchInternalRemoteRequest{RemoteRepository: repo},
},
{
desc: "empty Remote Repository",
request: &gitalypb.FetchInternalRemoteRequest{Repository: repo},
},
}
for _, tc := range testCases {
t.Run(tc.desc, func(t *testing.T) {
_, err := client.FetchInternalRemote(ctx, tc.request)
testhelper.RequireGrpcError(t, err, codes.InvalidArgument)
require.Contains(t, err.Error(), tc.desc)
})
}
}
func runFullServer(t *testing.T) (*grpc.Server, string) {
server := serverPkg.New(remote.RubyServer)
serverSocketPath := testhelper.GetTemporaryGitalySocketFileName()
listener, err := net.Listen("unix", serverSocketPath)
if err != nil {
t.Fatal(err)
}
go server.Serve(listener)
return server, "unix://" + serverSocketPath
}
|