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
|
package gitaly_test
import (
"context"
"io"
"testing"
"github.com/stretchr/testify/require"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/gitaly"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/gitaly/vendored/gitalypb"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/matcher"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/mock_gitaly"
"gitlab.com/gitlab-org/cluster-integration/gitlab-agent/v16/internal/tool/testing/mock_internalgitaly"
"go.uber.org/mock/gomock"
)
const (
revision = "507ebc6de9bcac25628aa7afd52802a91a0685d8"
manifestRevision = "7afd52802a91a0685d8507ebc6de9bcac25628aa"
repoPath = "dir"
)
func TestPathVisitor_HappyPath(t *testing.T) {
ctrl := gomock.NewController(t)
r := repo()
treeEntriesReq := &gitalypb.GetTreeEntriesRequest{
Repository: r,
Revision: []byte(revision),
Path: []byte(repoPath),
Recursive: false,
}
commitClient := mock_gitaly.NewMockCommitServiceClient(ctrl)
expectedEntry := &gitalypb.TreeEntry{
Path: []byte("manifest.yaml"),
Type: gitalypb.TreeEntry_BLOB,
CommitOid: manifestRevision,
}
features := map[string]string{
"f1": "true",
}
mockGetTreeEntries(t, ctrl, matcher.GRPCOutgoingCtx(features), commitClient, treeEntriesReq, []*gitalypb.TreeEntry{expectedEntry})
mockVisitor := mock_internalgitaly.NewMockPathEntryVisitor(ctrl)
mockVisitor.EXPECT().
Entry(matcher.ProtoEq(t, expectedEntry))
v := gitaly.PathVisitor{
Client: commitClient,
Features: features,
}
err := v.Visit(context.Background(), r, []byte(revision), []byte(repoPath), false, mockVisitor)
require.NoError(t, err)
}
func mockGetTreeEntries(t *testing.T, ctrl *gomock.Controller, ctx gomock.Matcher, commitClient *mock_gitaly.MockCommitServiceClient, req *gitalypb.GetTreeEntriesRequest, entries []*gitalypb.TreeEntry) {
treeEntriesClient := mock_gitaly.NewMockCommitService_GetTreeEntriesClient(ctrl)
gomock.InOrder(
commitClient.EXPECT().
GetTreeEntries(ctx, matcher.ProtoEq(t, req), gomock.Any()).
Return(treeEntriesClient, nil),
treeEntriesClient.EXPECT().
Recv().
Return(&gitalypb.GetTreeEntriesResponse{Entries: entries}, nil),
treeEntriesClient.EXPECT().
Recv().
Return(nil, io.EOF),
)
}
func repo() *gitalypb.Repository {
return &gitalypb.Repository{
StorageName: "StorageName1",
RelativePath: "RelativePath1",
GitObjectDirectory: "GitObjectDirectory1",
GlRepository: "GlRepository1",
GlProjectPath: "GlProjectPath1",
}
}
|
